Browse Source

add missing GetUserInternal and GetUsers tests

Matthew R Kasun 3 years ago
parent
commit
2f24acbaf1
2 changed files with 43 additions and 27 deletions
  1. 43 0
      controllers/userHttpController_test.go
  2. 0 27
      test/testutilities_test.go

+ 43 - 0
controllers/userHttpController_test.go

@@ -153,6 +153,49 @@ func TestGetUser(t *testing.T) {
 	})
 }
 
+func TestGetUserInternal(t *testing.T) {
+	database.InitializeDatabase()
+	deleteAllUsers()
+	t.Run("NonExistantUser", func(t *testing.T) {
+		admin, err := GetUserInternal("admin")
+		assert.EqualError(t, err, "could not find any records")
+		assert.Equal(t, "", admin.UserName)
+	})
+	t.Run("UserExisits", func(t *testing.T) {
+		user := models.User{"admin", "password", nil, true}
+		CreateUser(user)
+		admin, err := GetUserInternal("admin")
+		assert.Nil(t, err)
+		assert.Equal(t, user.UserName, admin.UserName)
+	})
+}
+
+func TestGetUsers(t *testing.T) {
+	database.InitializeDatabase()
+	deleteAllUsers()
+	t.Run("NonExistantUser", func(t *testing.T) {
+		admin, err := GetUsers()
+		assert.EqualError(t, err, "could not find any records")
+		assert.Equal(t, []models.ReturnUser(nil), admin)
+	})
+	t.Run("UserExisits", func(t *testing.T) {
+		user := models.User{"admin", "password", nil, true}
+		CreateUser(user)
+		admins, err := GetUsers()
+		assert.Nil(t, err)
+		assert.Equal(t, user.UserName, admins[0].UserName)
+	})
+	t.Run("MulipleUsers", func(t *testing.T) {
+		user := models.User{"user", "password", nil, true}
+		CreateUser(user)
+		admins, err := GetUsers()
+		assert.Nil(t, err)
+		assert.Equal(t, "admin", admins[0].UserName)
+		assert.Equal(t, user.UserName, admins[1].UserName)
+	})
+
+}
+
 func TestUpdateUser(t *testing.T) {
 	database.InitializeDatabase()
 	deleteAllUsers()

+ 0 - 27
test/testutilities_test.go

@@ -1,27 +0,0 @@
-package tests
-
-import "github.com/gravitl/netmaker/models"
-
-func DeleteAllNetworks() {
-	deleteAllNodes()
-	nets, _ := models.GetNetworks()
-	for _, net := range nets {
-		DeleteNetwork(net.NetID)
-	}
-}
-
-func CreateNet() {
-	var network models.Network
-	network.NetID = "skynet"
-	network.AddressRange = "10.0.0.1/24"
-	network.DisplayName = "mynetwork"
-	_, err := GetNetwork("skynet")
-	if err != nil {
-		CreateNetwork(network)
-	}
-}
-
-func GetNet() models.Network {
-	network, _ := GetNetwork("skynet")
-	return network
-}