Add new option tests for New for the TPMKMS

hslatman · hslatman · commit a9d2714473ab · 2023-07-24T17:04:25.000+02:00
diff --git a/kms/tpmkms/tpmkms_test.go b/kms/tpmkms/tpmkms_test.go
@@ -18,9 +18,10 @@ func TestNew(t *testing.T) {
 		want    *TPMKMS
 		wantErr bool
 	}{
-		{"ok/defaults", args{apiv1.Options{Type: "tpmkms"}}, &TPMKMS{}, false},
-		{"ok/uri", args{apiv1.Options{Type: "tpmkms", URI: "tpmkms:device=/dev/tpm0;storage-directory=/tmp/tpmstorage"}}, &TPMKMS{}, false},
+		{"ok/defaults", args{apiv1.Options{Type: "tpmkms"}}, &TPMKMS{identityEarlyRenewalEnabled: true, identityRenewalPeriodPercentage: 60}, false},
+		{"ok/uri", args{apiv1.Options{Type: "tpmkms", URI: "tpmkms:device=/dev/tpm0;storage-directory=/tmp/tpmstorage;renewal-percentage=70"}}, &TPMKMS{identityEarlyRenewalEnabled: true, identityRenewalPeriodPercentage: 70}, false},
 		{"fail/uri-scheme", args{apiv1.Options{Type: "tpmkms", URI: "tpmkmz://device=/dev/tpm0"}}, &TPMKMS{}, true},
+		{"fail/renewal-percentage-too-low", args{apiv1.Options{Type: "tpmkms", URI: "tpmkms:renewal-percentage=0"}}, &TPMKMS{}, true},
 	}
 	for _, tt := range tests {
 		t.Run(tt.name, func(t *testing.T) {
@@ -32,6 +33,8 @@ func TestNew(t *testing.T) {
 
 			if assert.NotNil(t, got) {
 				assert.NotNil(t, got.tpm)
+				assert.Equal(t, tt.want.identityEarlyRenewalEnabled, got.identityEarlyRenewalEnabled)
+				assert.Equal(t, tt.want.identityRenewalPeriodPercentage, got.identityRenewalPeriodPercentage)
 			}
 		})
 	}
