test validator file suffix

rhysd · Dec 19, 2019 · b5118a9 · b5118a9
1 parent a4d48e7
commit b5118a9
Showing 1 changed file with 22 additions and 0 deletions.
diff --git a/selfupdate/validate_test.go b/selfupdate/validate_test.go
@@ -112,3 +112,25 @@ func TestECDSAValidatorFail(t *testing.T) {
 		t.Fatal(err)
 	}
 }
+
+func TestValidatorSuffix(t *testing.T) {
+	for _, test := range []struct {
+		v      Validator
+		suffix string
+	}{
+		{
+			v:      &SHA2Validator{},
+			suffix: ".sha256",
+		},
+		{
+			v:      &ECDSAValidator{},
+			suffix: ".sig",
+		},
+	} {
+		want := test.suffix
+		got := test.v.Suffix()
+		if want != got {
+			t.Errorf("Wanted %q but got %q", want, got)
+		}
+	}
+}