Add method for validation of the secret

This commit is contained in:
9seconds
2021-03-12 14:40:07 +03:00
parent 1218f18af9
commit 4daf927a72
3 changed files with 20 additions and 5 deletions
+2 -2
View File
@@ -55,8 +55,8 @@ type Config struct {
}
func (c *Config) Validate() error {
if len(c.Secret.Key) == 0 || c.Secret.Host == "" {
return fmt.Errorf("incorrect secret %s", c.Secret.String())
if !c.Secret.Valid() {
return fmt.Errorf("invalid secret %s", c.Secret.String())
}
return nil
+7 -3
View File
@@ -22,11 +22,11 @@ type Secret struct {
}
func (s Secret) MarshalText() ([]byte, error) {
if s.Key == secretEmptyKey {
return nil, nil
if s.Valid() {
return []byte(s.String()), nil
}
return []byte(s.String()), nil
return nil, nil
}
func (s *Secret) UnmarshalText(data []byte) error {
@@ -72,6 +72,10 @@ func (s *Secret) UnmarshalText(data []byte) error {
return nil
}
func (s Secret) Valid() bool {
return s.Key != secretEmptyKey && s.Host != ""
}
func (s Secret) String() string {
return s.Base64()
}
+11
View File
@@ -99,6 +99,17 @@ func (suite *SecretTestSuite) TestInvariant() {
suite.Equal("google.com", parsed.Host)
}
func (suite *SecretTestSuite) TestValid() {
s := mtglib.Secret{}
suite.False(s.Valid())
s.Key[0] = 1
suite.False(s.Valid())
s.Host = "11"
suite.True(s.Valid())
}
func TestSecret(t *testing.T) {
t.Parallel()
suite.Run(t, &SecretTestSuite{})