|
@@ -4,6 +4,7 @@ import (
|
|
"testing"
|
|
"testing"
|
|
"time"
|
|
"time"
|
|
|
|
|
|
|
|
+ "github.com/google/uuid"
|
|
"github.com/gravitl/netmaker/database"
|
|
"github.com/gravitl/netmaker/database"
|
|
"github.com/gravitl/netmaker/models"
|
|
"github.com/gravitl/netmaker/models"
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/assert"
|
|
@@ -13,35 +14,35 @@ func TestCreateEnrollmentKey(t *testing.T) {
|
|
database.InitializeDatabase()
|
|
database.InitializeDatabase()
|
|
defer database.CloseDB()
|
|
defer database.CloseDB()
|
|
t.Run("Can_Not_Create_Key", func(t *testing.T) {
|
|
t.Run("Can_Not_Create_Key", func(t *testing.T) {
|
|
- newKey, err := CreateEnrollmentKey(0, time.Time{}, nil, nil, false)
|
|
|
|
|
|
+ newKey, err := CreateEnrollmentKey(0, time.Time{}, nil, nil, false, uuid.Nil)
|
|
assert.Nil(t, newKey)
|
|
assert.Nil(t, newKey)
|
|
assert.NotNil(t, err)
|
|
assert.NotNil(t, err)
|
|
assert.Equal(t, err, EnrollmentErrors.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, uuid.Nil)
|
|
assert.Nil(t, err)
|
|
assert.Nil(t, err)
|
|
assert.Equal(t, 1, newKey.UsesRemaining)
|
|
assert.Equal(t, 1, newKey.UsesRemaining)
|
|
assert.True(t, newKey.IsValid())
|
|
assert.True(t, newKey.IsValid())
|
|
})
|
|
})
|
|
t.Run("Can_Create_Key_Time", func(t *testing.T) {
|
|
t.Run("Can_Create_Key_Time", func(t *testing.T) {
|
|
- newKey, err := CreateEnrollmentKey(0, time.Now().Add(time.Minute), nil, nil, false)
|
|
|
|
|
|
+ newKey, err := CreateEnrollmentKey(0, time.Now().Add(time.Minute), nil, nil, false, uuid.Nil)
|
|
assert.Nil(t, err)
|
|
assert.Nil(t, err)
|
|
assert.True(t, newKey.IsValid())
|
|
assert.True(t, newKey.IsValid())
|
|
})
|
|
})
|
|
t.Run("Can_Create_Key_Unlimited", func(t *testing.T) {
|
|
t.Run("Can_Create_Key_Unlimited", func(t *testing.T) {
|
|
- newKey, err := CreateEnrollmentKey(0, time.Time{}, nil, nil, true)
|
|
|
|
|
|
+ newKey, err := CreateEnrollmentKey(0, time.Time{}, nil, nil, true, uuid.Nil)
|
|
assert.Nil(t, err)
|
|
assert.Nil(t, err)
|
|
assert.True(t, newKey.IsValid())
|
|
assert.True(t, newKey.IsValid())
|
|
})
|
|
})
|
|
t.Run("Can_Create_Key_WithNetworks", func(t *testing.T) {
|
|
t.Run("Can_Create_Key_WithNetworks", func(t *testing.T) {
|
|
- newKey, err := CreateEnrollmentKey(0, time.Time{}, []string{"mynet", "skynet"}, nil, true)
|
|
|
|
|
|
+ newKey, err := CreateEnrollmentKey(0, time.Time{}, []string{"mynet", "skynet"}, nil, true, uuid.Nil)
|
|
assert.Nil(t, err)
|
|
assert.Nil(t, err)
|
|
assert.True(t, newKey.IsValid())
|
|
assert.True(t, newKey.IsValid())
|
|
assert.True(t, len(newKey.Networks) == 2)
|
|
assert.True(t, len(newKey.Networks) == 2)
|
|
})
|
|
})
|
|
t.Run("Can_Create_Key_WithTags", func(t *testing.T) {
|
|
t.Run("Can_Create_Key_WithTags", func(t *testing.T) {
|
|
- newKey, err := CreateEnrollmentKey(0, time.Time{}, nil, []string{"tag1", "tag2"}, true)
|
|
|
|
|
|
+ newKey, err := CreateEnrollmentKey(0, time.Time{}, nil, []string{"tag1", "tag2"}, true, uuid.Nil)
|
|
assert.Nil(t, err)
|
|
assert.Nil(t, err)
|
|
assert.True(t, newKey.IsValid())
|
|
assert.True(t, newKey.IsValid())
|
|
assert.True(t, len(newKey.Tags) == 2)
|
|
assert.True(t, len(newKey.Tags) == 2)
|
|
@@ -61,7 +62,7 @@ func TestCreateEnrollmentKey(t *testing.T) {
|
|
func TestDelete_EnrollmentKey(t *testing.T) {
|
|
func TestDelete_EnrollmentKey(t *testing.T) {
|
|
database.InitializeDatabase()
|
|
database.InitializeDatabase()
|
|
defer database.CloseDB()
|
|
defer database.CloseDB()
|
|
- newKey, _ := CreateEnrollmentKey(0, time.Time{}, []string{"mynet", "skynet"}, nil, true)
|
|
|
|
|
|
+ newKey, _ := CreateEnrollmentKey(0, time.Time{}, []string{"mynet", "skynet"}, nil, true, uuid.Nil)
|
|
t.Run("Can_Delete_Key", func(t *testing.T) {
|
|
t.Run("Can_Delete_Key", func(t *testing.T) {
|
|
assert.True(t, newKey.IsValid())
|
|
assert.True(t, newKey.IsValid())
|
|
err := DeleteEnrollmentKey(newKey.Value)
|
|
err := DeleteEnrollmentKey(newKey.Value)
|
|
@@ -82,7 +83,7 @@ func TestDelete_EnrollmentKey(t *testing.T) {
|
|
func TestDecrement_EnrollmentKey(t *testing.T) {
|
|
func TestDecrement_EnrollmentKey(t *testing.T) {
|
|
database.InitializeDatabase()
|
|
database.InitializeDatabase()
|
|
defer database.CloseDB()
|
|
defer database.CloseDB()
|
|
- newKey, _ := CreateEnrollmentKey(1, time.Time{}, nil, nil, false)
|
|
|
|
|
|
+ newKey, _ := CreateEnrollmentKey(1, time.Time{}, nil, nil, false, uuid.Nil)
|
|
t.Run("Check_initial_uses", func(t *testing.T) {
|
|
t.Run("Check_initial_uses", func(t *testing.T) {
|
|
assert.True(t, newKey.IsValid())
|
|
assert.True(t, newKey.IsValid())
|
|
assert.Equal(t, newKey.UsesRemaining, 1)
|
|
assert.Equal(t, newKey.UsesRemaining, 1)
|
|
@@ -106,9 +107,9 @@ func TestDecrement_EnrollmentKey(t *testing.T) {
|
|
func TestUsability_EnrollmentKey(t *testing.T) {
|
|
func TestUsability_EnrollmentKey(t *testing.T) {
|
|
database.InitializeDatabase()
|
|
database.InitializeDatabase()
|
|
defer database.CloseDB()
|
|
defer database.CloseDB()
|
|
- key1, _ := CreateEnrollmentKey(1, time.Time{}, nil, nil, false)
|
|
|
|
- key2, _ := CreateEnrollmentKey(0, time.Now().Add(time.Minute<<4), nil, nil, false)
|
|
|
|
- key3, _ := CreateEnrollmentKey(0, time.Time{}, nil, nil, true)
|
|
|
|
|
|
+ key1, _ := CreateEnrollmentKey(1, time.Time{}, nil, nil, false, uuid.Nil)
|
|
|
|
+ key2, _ := CreateEnrollmentKey(0, time.Now().Add(time.Minute<<4), nil, nil, false, uuid.Nil)
|
|
|
|
+ key3, _ := CreateEnrollmentKey(0, time.Time{}, nil, nil, true, uuid.Nil)
|
|
t.Run("Check if valid use key can be used", func(t *testing.T) {
|
|
t.Run("Check if valid use key can be used", func(t *testing.T) {
|
|
assert.Equal(t, key1.UsesRemaining, 1)
|
|
assert.Equal(t, key1.UsesRemaining, 1)
|
|
ok := TryToUseEnrollmentKey(key1)
|
|
ok := TryToUseEnrollmentKey(key1)
|
|
@@ -144,7 +145,7 @@ func removeAllEnrollments() {
|
|
func TestTokenize_EnrollmentKeys(t *testing.T) {
|
|
func TestTokenize_EnrollmentKeys(t *testing.T) {
|
|
database.InitializeDatabase()
|
|
database.InitializeDatabase()
|
|
defer database.CloseDB()
|
|
defer database.CloseDB()
|
|
- newKey, _ := CreateEnrollmentKey(0, time.Time{}, []string{"mynet", "skynet"}, nil, true)
|
|
|
|
|
|
+ newKey, _ := CreateEnrollmentKey(0, time.Time{}, []string{"mynet", "skynet"}, nil, true, uuid.Nil)
|
|
const defaultValue = "MwE5MwE5MwE5MwE5MwE5MwE5MwE5MwE5"
|
|
const defaultValue = "MwE5MwE5MwE5MwE5MwE5MwE5MwE5MwE5"
|
|
const b64value = "eyJzZXJ2ZXIiOiJhcGkubXlzZXJ2ZXIuY29tIiwidmFsdWUiOiJNd0U1TXdFNU13RTVNd0U1TXdFNU13RTVNd0U1TXdFNSJ9"
|
|
const b64value = "eyJzZXJ2ZXIiOiJhcGkubXlzZXJ2ZXIuY29tIiwidmFsdWUiOiJNd0U1TXdFNU13RTVNd0U1TXdFNU13RTVNd0U1TXdFNSJ9"
|
|
const serverAddr = "api.myserver.com"
|
|
const serverAddr = "api.myserver.com"
|
|
@@ -177,7 +178,7 @@ func TestTokenize_EnrollmentKeys(t *testing.T) {
|
|
func TestDeTokenize_EnrollmentKeys(t *testing.T) {
|
|
func TestDeTokenize_EnrollmentKeys(t *testing.T) {
|
|
database.InitializeDatabase()
|
|
database.InitializeDatabase()
|
|
defer database.CloseDB()
|
|
defer database.CloseDB()
|
|
- newKey, _ := CreateEnrollmentKey(0, time.Time{}, []string{"mynet", "skynet"}, nil, true)
|
|
|
|
|
|
+ newKey, _ := CreateEnrollmentKey(0, time.Time{}, []string{"mynet", "skynet"}, nil, true, uuid.Nil)
|
|
const b64Value = "eyJzZXJ2ZXIiOiJhcGkubXlzZXJ2ZXIuY29tIiwidmFsdWUiOiJNd0U1TXdFNU13RTVNd0U1TXdFNU13RTVNd0U1TXdFNSJ9"
|
|
const b64Value = "eyJzZXJ2ZXIiOiJhcGkubXlzZXJ2ZXIuY29tIiwidmFsdWUiOiJNd0U1TXdFNU13RTVNd0U1TXdFNU13RTVNd0U1TXdFNSJ9"
|
|
const serverAddr = "api.myserver.com"
|
|
const serverAddr = "api.myserver.com"
|
|
|
|
|