mirror of
https://github.com/gravitl/netmaker.git
synced 2025-10-06 17:29:15 +08:00
shortened name
This commit is contained in:
@@ -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 {
|
||||||
|
@@ -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)
|
||||||
|
Reference in New Issue
Block a user