upstream-corteza/server/store/tests/roles_test.go

package tests

import (
	"context"
	"strings"
	"testing"
	"time"

	"github.com/cortezaproject/corteza/server/pkg/filter"
	"github.com/cortezaproject/corteza/server/pkg/id"
	"github.com/cortezaproject/corteza/server/pkg/rand"
	"github.com/cortezaproject/corteza/server/store"
	"github.com/cortezaproject/corteza/server/system/types"
	_ "github.com/joho/godotenv/autoload"
	"github.com/stretchr/testify/require"
)

func testRoles(t *testing.T, s store.Roles) {
	var (
		ctx = context.Background()

		makeNew = func(nn ...string) *types.Role {
			name := strings.Join(nn, "")
			return &types.Role{
				ID:        id.Next(),
				CreatedAt: time.Now(),
				Name:      "RoleCRUD+" + name,
				Handle:    "rolecrud+" + name,
			}
		}

		truncAndCreate = func(t *testing.T) (*require.Assertions, *types.Role) {
			req := require.New(t)
			req.NoError(s.TruncateRoles(ctx))
			role := makeNew()
			req.NoError(s.CreateRole(ctx, role))
			return req, role
		}

		truncAndFill = func(t *testing.T, l int) (*require.Assertions, types.RoleSet) {
			req := require.New(t)
			req.NoError(s.TruncateRoles(ctx))

			set := make([]*types.Role, l)

			for i := 0; i < l; i++ {
				set[i] = makeNew(string(rand.Bytes(10)))
			}

			req.NoError(s.CreateRole(ctx, set...))
			return req, set
		}
	)

	t.Run("create", func(t *testing.T) {
		req := require.New(t)
		req.NoError(s.CreateRole(ctx, makeNew()))
	})

	t.Run("lookup by ID", func(t *testing.T) {
		req, role := truncAndCreate(t)
		fetched, err := s.LookupRoleByID(ctx, role.ID)
		req.NoError(err)
		req.Equal(role.Name, fetched.Name)
		req.Equal(role.Handle, fetched.Handle)
		req.Equal(role.ID, fetched.ID)
		req.NotNil(fetched.CreatedAt)
		req.Nil(fetched.UpdatedAt)
		req.Nil(fetched.DeletedAt)
	})

	t.Run("update", func(t *testing.T) {
		req, role := truncAndCreate(t)
		req.NoError(s.UpdateRole(ctx, role))
	})

	t.Run("create and update deleted with existing handle", func(t *testing.T) {
		req, role := truncAndCreate(t)
		deletedRole := makeNew("copy")
		deletedRole.DeletedAt = now()
		deletedRole.Handle = role.Handle
		req.NoError(store.CreateRole(ctx, s, deletedRole))
		req.NoError(store.UpdateRole(ctx, s, deletedRole))
	})

	t.Run("lookup by handle", func(t *testing.T) {
		req, role := truncAndCreate(t)
		fetched, err := s.LookupRoleByHandle(ctx, role.Handle)
		req.NoError(err)
		req.Equal(role.ID, fetched.ID)
	})

	t.Run("search", func(t *testing.T) {
		t.Run("by ID", func(t *testing.T) {
			req, prefill := truncAndFill(t, 5)

			set, f, err := s.SearchRoles(ctx, types.RoleFilter{RoleID: id.Strings(prefill[0].ID)})
			req.NoError(err)
			req.Equal(id.Strings(prefill[0].ID), f.RoleID)
			req.Len(set, 1)
		})

		t.Run("by query", func(t *testing.T) {
			req, prefill := truncAndFill(t, 5)

			set, _, err := s.SearchRoles(ctx, types.RoleFilter{Query: prefill[0].Handle})
			req.NoError(err)
			req.Len(set, 1)
		})

		t.Run("by state", func(t *testing.T) {
			t.Run("deleted", func(t *testing.T) {
				req, prefill := truncAndFill(t, 5)

				prefill[0].DeletedAt = &(prefill[0].CreatedAt)
				s.UpdateRole(ctx, prefill[0])

				set, _, err := s.SearchRoles(ctx, types.RoleFilter{Deleted: filter.StateExcluded})
				req.NoError(err)
				req.Len(set, 4)

				set, _, err = s.SearchRoles(ctx, types.RoleFilter{Deleted: filter.StateInclusive})
				req.NoError(err)
				req.Len(set, 5)

				set, _, err = s.SearchRoles(ctx, types.RoleFilter{Deleted: filter.StateExclusive})
				req.NoError(err)
				req.Len(set, 1)
			})

			t.Run("archived", func(t *testing.T) {
				req, prefill := truncAndFill(t, 5)

				prefill[0].ArchivedAt = &(prefill[0].CreatedAt)
				s.UpdateRole(ctx, prefill[0])

				set, _, err := s.SearchRoles(ctx, types.RoleFilter{Archived: filter.StateExcluded})
				req.NoError(err)
				req.Len(set, 4)

				set, _, err = s.SearchRoles(ctx, types.RoleFilter{Archived: filter.StateInclusive})
				req.NoError(err)
				req.Len(set, 5)

				set, _, err = s.SearchRoles(ctx, types.RoleFilter{Archived: filter.StateExclusive})
				req.NoError(err)
				req.Len(set, 1)
			})
		})

		t.Run("with check", func(t *testing.T) {
			req, prefill := truncAndFill(t, 5)

			set, _, err := s.SearchRoles(ctx, types.RoleFilter{
				Check: func(role *types.Role) (bool, error) {
					return role.ID == prefill[0].ID, nil
				},
			})
			req.NoError(err)
			req.Len(set, 1)
			req.Equal(prefill[0].ID, set[0].ID)
		})
	})

	t.Run("ordered search", func(t *testing.T) {
		t.Skip("not implemented")
	})

	t.Run("metrics", func(t *testing.T) {
		var (
			req = require.New(t)

			oct, _ = time.Parse(time.RFC3339, "2020-10-02T10:01:10Z")
			nov, _ = time.Parse(time.RFC3339, "2020-11-02T20:02:10Z")

			octu = uint(oct.Truncate(time.Hour * 24).Unix())
			novu = uint(nov.Truncate(time.Hour * 24).Unix())

			e = &types.RoleMetrics{
				Total:         8,
				Valid:         2,
				Deleted:       3,
				Archived:      3,
				DailyCreated:  []uint{octu, 7, novu, 1},
				DailyUpdated:  []uint{octu, 2},
				DailyArchived: []uint{octu, 2, novu, 1},
				DailyDeleted:  []uint{novu, 3},
			}
		)

		req.NoError(s.TruncateRoles(ctx))
		req.NoError(s.CreateRole(ctx, &types.Role{ID: id.Next(), CreatedAt: oct, UpdatedAt: &oct}))
		req.NoError(s.CreateRole(ctx, &types.Role{ID: id.Next(), CreatedAt: oct, UpdatedAt: &oct}))
		req.NoError(s.CreateRole(ctx, &types.Role{ID: id.Next(), CreatedAt: oct, ArchivedAt: &oct}))
		req.NoError(s.CreateRole(ctx, &types.Role{ID: id.Next(), CreatedAt: oct, ArchivedAt: &oct}))
		req.NoError(s.CreateRole(ctx, &types.Role{ID: id.Next(), CreatedAt: oct, ArchivedAt: &nov}))
		req.NoError(s.CreateRole(ctx, &types.Role{ID: id.Next(), CreatedAt: oct, DeletedAt: &nov}))
		req.NoError(s.CreateRole(ctx, &types.Role{ID: id.Next(), CreatedAt: oct, DeletedAt: &nov}))
		req.NoError(s.CreateRole(ctx, &types.Role{ID: id.Next(), CreatedAt: nov, DeletedAt: &nov}))

		m, err := store.RoleMetrics(ctx, s)
		req.NoError(err)
		req.Equal(e, m)
	})
}