shortened name

This commit is contained in:
0xdcarns
2023-02-16 18:56:45 -05:00
parent bae3a5a028
commit 596de6b9e3
2 changed files with 17 additions and 17 deletions

View File

@@ -12,8 +12,8 @@ import (
"github.com/gravitl/netmaker/netclient/ncutils" "github.com/gravitl/netmaker/netclient/ncutils"
) )
// EnrollmentKeyErrors - struct for holding EnrollmentKey error messages // EnrollmentErrors - struct for holding EnrollmentKey error messages
var EnrollmentKeyErrors = struct { var EnrollmentErrors = struct {
InvalidCreate error InvalidCreate error
NoKeyFound error NoKeyFound error
InvalidKey error InvalidKey error
@@ -56,7 +56,7 @@ func CreateEnrollmentKey(uses int, expiration time.Time, networks, tags []string
k.Tags = tags k.Tags = tags
} }
if ok := k.Validate(); !ok { if ok := k.Validate(); !ok {
return nil, EnrollmentKeyErrors.InvalidCreate return nil, EnrollmentErrors.InvalidCreate
} }
if err = upsertEnrollmentKey(k); err != nil { if err = upsertEnrollmentKey(k); err != nil {
return nil, err return nil, err
@@ -87,7 +87,7 @@ func GetEnrollmentKey(value string) (*models.EnrollmentKey, error) {
if key, ok := currentKeys[value]; ok { if key, ok := currentKeys[value]; ok {
return key, nil return key, nil
} }
return nil, EnrollmentKeyErrors.NoKeyFound return nil, EnrollmentErrors.NoKeyFound
} }
// DeleteEnrollmentKey - delete's a given enrollment key by value // DeleteEnrollmentKey - delete's a given enrollment key by value
@@ -104,7 +104,7 @@ func DeleteEnrollmentKey(value string) error {
func TryToUseEnrollmentKey(k *models.EnrollmentKey) bool { func TryToUseEnrollmentKey(k *models.EnrollmentKey) bool {
key, err := decrementEnrollmentKey(k.Value) key, err := decrementEnrollmentKey(k.Value)
if err != nil { if err != nil {
if errors.Is(err, EnrollmentKeyErrors.NoUsesRemaining) { if errors.Is(err, EnrollmentErrors.NoUsesRemaining) {
return k.IsValid() return k.IsValid()
} }
} else { } else {
@@ -118,7 +118,7 @@ func TryToUseEnrollmentKey(k *models.EnrollmentKey) bool {
// and attaches it to the Token field on the struct // and attaches it to the Token field on the struct
func Tokenize(k *models.EnrollmentKey, serverAddr string) error { func Tokenize(k *models.EnrollmentKey, serverAddr string) error {
if len(serverAddr) == 0 || k == nil { if len(serverAddr) == 0 || k == nil {
return EnrollmentKeyErrors.FailedToTokenize return EnrollmentErrors.FailedToTokenize
} }
newToken := models.EnrollmentToken{ newToken := models.EnrollmentToken{
Server: serverAddr, Server: serverAddr,
@@ -136,7 +136,7 @@ func Tokenize(k *models.EnrollmentKey, serverAddr string) error {
// and finds the associated enrollment key // and finds the associated enrollment key
func DeTokenize(b64Token string) (*models.EnrollmentKey, error) { func DeTokenize(b64Token string) (*models.EnrollmentKey, error) {
if len(b64Token) == 0 { if len(b64Token) == 0 {
return nil, EnrollmentKeyErrors.FailedToDeTokenize return nil, EnrollmentErrors.FailedToDeTokenize
} }
tokenData, err := b64.StdEncoding.DecodeString(b64Token) tokenData, err := b64.StdEncoding.DecodeString(b64Token)
if err != nil { if err != nil {
@@ -164,7 +164,7 @@ func decrementEnrollmentKey(value string) (*models.EnrollmentKey, error) {
return nil, err return nil, err
} }
if k.UsesRemaining == 0 { if k.UsesRemaining == 0 {
return nil, EnrollmentKeyErrors.NoUsesRemaining return nil, EnrollmentErrors.NoUsesRemaining
} }
k.UsesRemaining = k.UsesRemaining - 1 k.UsesRemaining = k.UsesRemaining - 1
if err = upsertEnrollmentKey(k); err != nil { if err = upsertEnrollmentKey(k); err != nil {
@@ -176,7 +176,7 @@ func decrementEnrollmentKey(value string) (*models.EnrollmentKey, error) {
func upsertEnrollmentKey(k *models.EnrollmentKey) error { func upsertEnrollmentKey(k *models.EnrollmentKey) error {
if k == nil { if k == nil {
return EnrollmentKeyErrors.InvalidKey return EnrollmentErrors.InvalidKey
} }
data, err := json.Marshal(k) data, err := json.Marshal(k)
if err != nil { if err != nil {

View File

@@ -16,7 +16,7 @@ func TestCreateEnrollmentKey(t *testing.T) {
newKey, err := CreateEnrollmentKey(0, time.Time{}, nil, nil, false) newKey, err := CreateEnrollmentKey(0, time.Time{}, nil, nil, false)
assert.Nil(t, newKey) assert.Nil(t, newKey)
assert.NotNil(t, err) assert.NotNil(t, err)
assert.Equal(t, err, EnrollmentKeyErrors.InvalidCreate) assert.Equal(t, err, EnrollmentErrors.InvalidCreate)
}) })
t.Run("Can_Create_Key_Uses", func(t *testing.T) { t.Run("Can_Create_Key_Uses", func(t *testing.T) {
newKey, err := CreateEnrollmentKey(1, time.Time{}, nil, nil, false) newKey, err := CreateEnrollmentKey(1, time.Time{}, nil, nil, false)
@@ -69,12 +69,12 @@ func TestDelete_EnrollmentKey(t *testing.T) {
oldKey, err := GetEnrollmentKey(newKey.Value) oldKey, err := GetEnrollmentKey(newKey.Value)
assert.Nil(t, oldKey) assert.Nil(t, oldKey)
assert.NotNil(t, err) assert.NotNil(t, err)
assert.Equal(t, err, EnrollmentKeyErrors.NoKeyFound) assert.Equal(t, err, EnrollmentErrors.NoKeyFound)
}) })
t.Run("Can_Not_Delete_Invalid_Key", func(t *testing.T) { t.Run("Can_Not_Delete_Invalid_Key", func(t *testing.T) {
err := DeleteEnrollmentKey("notakey") err := DeleteEnrollmentKey("notakey")
assert.NotNil(t, err) assert.NotNil(t, err)
assert.Equal(t, err, EnrollmentKeyErrors.NoKeyFound) assert.Equal(t, err, EnrollmentErrors.NoKeyFound)
}) })
removeAllEnrollments() removeAllEnrollments()
} }
@@ -97,7 +97,7 @@ func TestDecrement_EnrollmentKey(t *testing.T) {
assert.Equal(t, newKey.UsesRemaining, 0) assert.Equal(t, newKey.UsesRemaining, 0)
_, err := decrementEnrollmentKey(newKey.Value) _, err := decrementEnrollmentKey(newKey.Value)
assert.NotNil(t, err) assert.NotNil(t, err)
assert.Equal(t, err, EnrollmentKeyErrors.NoUsesRemaining) assert.Equal(t, err, EnrollmentErrors.NoUsesRemaining)
}) })
removeAllEnrollments() removeAllEnrollments()
@@ -151,12 +151,12 @@ func TestTokenize_EnrollmentKeys(t *testing.T) {
t.Run("Can_Not_Tokenize_Nil_Key", func(t *testing.T) { t.Run("Can_Not_Tokenize_Nil_Key", func(t *testing.T) {
err := Tokenize(nil, "ServerAddress") err := Tokenize(nil, "ServerAddress")
assert.NotNil(t, err) assert.NotNil(t, err)
assert.Equal(t, err, EnrollmentKeyErrors.FailedToTokenize) assert.Equal(t, err, EnrollmentErrors.FailedToTokenize)
}) })
t.Run("Can_Not_Tokenize_Empty_Server_Address", func(t *testing.T) { t.Run("Can_Not_Tokenize_Empty_Server_Address", func(t *testing.T) {
err := Tokenize(newKey, "") err := Tokenize(newKey, "")
assert.NotNil(t, err) assert.NotNil(t, err)
assert.Equal(t, err, EnrollmentKeyErrors.FailedToTokenize) assert.Equal(t, err, EnrollmentErrors.FailedToTokenize)
}) })
t.Run("Can_Tokenize", func(t *testing.T) { t.Run("Can_Tokenize", func(t *testing.T) {
@@ -185,13 +185,13 @@ func TestDeTokenize_EnrollmentKeys(t *testing.T) {
value, err := DeTokenize("") value, err := DeTokenize("")
assert.Nil(t, value) assert.Nil(t, value)
assert.NotNil(t, err) assert.NotNil(t, err)
assert.Equal(t, err, EnrollmentKeyErrors.FailedToDeTokenize) assert.Equal(t, err, EnrollmentErrors.FailedToDeTokenize)
}) })
t.Run("Can_Not_Find_Key", func(t *testing.T) { t.Run("Can_Not_Find_Key", func(t *testing.T) {
value, err := DeTokenize(b64Value) value, err := DeTokenize(b64Value)
assert.Nil(t, value) assert.Nil(t, value)
assert.NotNil(t, err) assert.NotNil(t, err)
assert.Equal(t, err, EnrollmentKeyErrors.NoKeyFound) assert.Equal(t, err, EnrollmentErrors.NoKeyFound)
}) })
t.Run("Can_DeTokenize", func(t *testing.T) { t.Run("Can_DeTokenize", func(t *testing.T) {
err := Tokenize(newKey, serverAddr) err := Tokenize(newKey, serverAddr)