Refactor encrypted opts and added unit test

Signed-off-by: Brandon Lum <lumjjb@gmail.com>
This commit is contained in:
Brandon Lum
2020-01-08 10:13:23 +00:00
parent ac8ec18813
commit c43a7588f6
2 changed files with 35 additions and 5 deletions

View File

@@ -288,3 +288,25 @@ func TestDefaultScheme(t *testing.T) {
assert.Equal(t, test.expected, got)
}
}
func TestEncryptedImagePullOpts(t *testing.T) {
for desc, test := range map[string]struct {
keyModel string
expectedOpts int
}{
"node key model should return one unpack opt": {
keyModel: criconfig.EncryptionKeyModelNode,
expectedOpts: 1,
},
"no key model selected should not add any opts": {
keyModel: "",
expectedOpts: 0,
},
} {
t.Logf("TestCase %q", desc)
c := newTestCRIService()
c.config.EncryptedImagesConfig.KeyModel = test.keyModel
got := len(c.encryptedImagesPullOpts())
assert.Equal(t, test.expectedOpts, got)
}
}