usergroups_test.go 1.1 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243
  1. package pro
  2. import (
  3. "testing"
  4. "github.com/gravitl/netmaker/database"
  5. "github.com/gravitl/netmaker/models/promodels"
  6. "github.com/stretchr/testify/assert"
  7. )
  8. func TestUserGroupLogic(t *testing.T) {
  9. database.InitializeDatabase()
  10. t.Run("User Groups initialized successfully", func(t *testing.T) {
  11. err := InitializeGroups()
  12. assert.Nil(t, err)
  13. })
  14. t.Run("Check for default group", func(t *testing.T) {
  15. groups, err := GetUserGroups()
  16. assert.Nil(t, err)
  17. var hasdefault bool
  18. for k := range groups {
  19. if string(k) == DEFAULT_ALLOWED_GROUPS {
  20. hasdefault = true
  21. }
  22. }
  23. assert.True(t, hasdefault)
  24. })
  25. t.Run("User Groups created successfully", func(t *testing.T) {
  26. err := InsertUserGroup(promodels.UserGroupName("group1"))
  27. assert.Nil(t, err)
  28. err = InsertUserGroup(promodels.UserGroupName("group2"))
  29. assert.Nil(t, err)
  30. })
  31. t.Run("User Groups deleted successfully", func(t *testing.T) {
  32. err := DeleteUserGroup(promodels.UserGroupName("group1"))
  33. assert.Nil(t, err)
  34. assert.False(t, DoesUserGroupExist(promodels.UserGroupName("group1")))
  35. })
  36. }