github
/
k3s
mirror of https://github.com/k3s-io/k3s
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

Reduce LOC in security context tests

pull/6/head
Paul Morie 2016-04-28 00:03:18 -04:00
parent 492762d394
commit 6f940a1a78
1 changed files with 15 additions and 43 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

58
pkg/securitycontext/util_test.go
View File

@ -84,6 +84,10 @@ func compareContexts(name string, ex, ac *api.SELinuxOptions, t *testing.T) {
} }
} }
func containerWithUser(ptr *int64) *api.Container {
return &api.Container{SecurityContext: &api.SecurityContext{RunAsUser: ptr}}
}
func TestHaRootUID(t *testing.T) { func TestHaRootUID(t *testing.T) {
var nonRoot int64 = 1 var nonRoot int64 = 1
var root int64 = 0 var root int64 = 0
@ -96,26 +100,14 @@ func TestHaRootUID(t *testing.T) {
container: &api.Container{SecurityContext: nil}, container: &api.Container{SecurityContext: nil},
}, },
"nil runAsuser": { "nil runAsuser": {
container: &api.Container{ container: containerWithUser(nil),
SecurityContext: &api.SecurityContext{
RunAsUser: nil,
},
},
}, },
"runAsUser non-root": { "runAsUser non-root": {
container: &api.Container{ container: containerWithUser(&nonRoot),
SecurityContext: &api.SecurityContext{
RunAsUser: &nonRoot,
},
},
}, },
"runAsUser root": { "runAsUser root": {
container: &api.Container{ container: containerWithUser(&root),
SecurityContext: &api.SecurityContext{ expect: true,
RunAsUser: &root,
},
},
expect: true,
}, },
} }
@ -138,19 +130,11 @@ func TestHasRunAsUser(t *testing.T) {
container: &api.Container{SecurityContext: nil}, container: &api.Container{SecurityContext: nil},
}, },
"nil runAsUser": { "nil runAsUser": {
container: &api.Container{ container: containerWithUser(nil),
SecurityContext: &api.SecurityContext{
RunAsUser: nil,
},
},
}, },
"valid runAsUser": { "valid runAsUser": {
container: &api.Container{ container: containerWithUser(&runAsUser),
SecurityContext: &api.SecurityContext{ expect: true,
RunAsUser: &runAsUser,
},
},
expect: true,
}, },
} }
@ -174,26 +158,14 @@ func TestHasRootRunAsUser(t *testing.T) {
container: &api.Container{SecurityContext: nil}, container: &api.Container{SecurityContext: nil},
}, },
"nil runAsuser": { "nil runAsuser": {
container: &api.Container{ container: containerWithUser(nil),
SecurityContext: &api.SecurityContext{
RunAsUser: nil,
},
},
}, },
"runAsUser non-root": { "runAsUser non-root": {
container: &api.Container{ container: containerWithUser(&nonRoot),
SecurityContext: &api.SecurityContext{
RunAsUser: &nonRoot,
},
},
}, },
"runAsUser root": { "runAsUser root": {
container: &api.Container{ container: containerWithUser(&root),
SecurityContext: &api.SecurityContext{ expect: true,
RunAsUser: &root,
},
},
expect: true,
}, },
} }