piv: add test that Certificate() returns ErrNotFound

go-piv · Apr 22, 2020 · 04943c8 · 04943c8
1 parent 1d325bc
commit 04943c8
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/piv/key_test.go b/piv/key_test.go
@@ -80,6 +80,12 @@ func TestYubiKeySignECDSA(t *testing.T) {
 }
 
 func TestSlots(t *testing.T) {
+	yk, close := newTestYubiKey(t)
+	if err := yk.Reset(); err != nil {
+		t.Fatalf("resetting yubikey: %v", err)
+	}
+	close()
+
 	tests := []struct {
 		name string
 		slot Slot
@@ -125,6 +131,10 @@ func TestSlots(t *testing.T) {
 			if err != nil {
 				t.Fatalf("parse certificate: %v", err)
 			}
+
+			if _, err := yk.Certificate(test.slot); err == nil || !errors.Is(err, ErrNotFound) {
+				t.Errorf("get certificate, got err=%v, want=ErrNotFound", err)
+			}
 			if err := yk.SetCertificate(DefaultManagementKey, test.slot, cert); err != nil {
 				t.Fatalf("set certificate: %v", err)
 			}