From b5118a9ebf6d8c94bc7345b1f497d4a55210de46 Mon Sep 17 00:00:00 2001
From: rhysd <lin90162@yahoo.co.jp>
Date: Thu, 19 Dec 2019 11:40:54 +0900
Subject: [PATCH] test validator file suffix

---
 selfupdate/validate_test.go | 22 ++++++++++++++++++++++
 1 file changed, 22 insertions(+)

diff --git a/selfupdate/validate_test.go b/selfupdate/validate_test.go
index f52f2a3..3c6ed6e 100644
--- 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)
+		}
+	}
+}