Jomon/internal/model/account_manager_test.go at 0a6925f771455c8a4734d29cefa8b38ed3c5bba3 · traPtitech/Jomon · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
package model

import (
	"testing"

	"github.com/google/uuid"
	"github.com/stretchr/testify/require"
	"github.com/traPtitech/Jomon/internal/service"
	"github.com/traPtitech/Jomon/internal/testutil"
	"github.com/traPtitech/Jomon/internal/testutil/random"
)

func TestEntRepository_GetAccountManagers(t *testing.T) {
	ctx := testutil.NewContext(t)
	client, err := setup(t, ctx, "get_accountManagers")
	require.NoError(t, err)
	repo := NewEntRepository(client)
	client2, err := setup(t, ctx, "get_accountManagers2")
	require.NoError(t, err)
	repo2 := NewEntRepository(client2)

	t.Run("Success", func(t *testing.T) {
		t.Parallel()
		user1, err := repo.CreateUser(
			ctx,
			random.AlphaNumeric(t, 20),
			random.AlphaNumeric(t, 30),
			true)
		require.NoError(t, err)
		user2, err := repo.CreateUser(
			ctx,
			random.AlphaNumeric(t, 20),
			random.AlphaNumeric(t, 30),
			true)
		require.NoError(t, err)

		got, err := repo.GetAccountManagers(ctx)
		require.NoError(t, err)
		exp := []*service.AccountManager{
			{ID: user1.ID},
			{ID: user2.ID},
		}
		require.ElementsMatch(t, exp, got)
	})

	t.Run("Success2", func(t *testing.T) {
		t.Parallel()

		got, err := repo2.GetAccountManagers(ctx)
		require.NoError(t, err)
		require.Empty(t, got)
	})
}

func TestEntRepository_AddAccountManagers(t *testing.T) {
	ctx := testutil.NewContext(t)
	client, err := setup(t, ctx, "add_accountManagers")
	require.NoError(t, err)
	repo := NewEntRepository(client)

	t.Run("Success", func(t *testing.T) {
		t.Parallel()
		user, err := repo.CreateUser(
			ctx,
			random.AlphaNumeric(t, 20),
			random.AlphaNumeric(t, 30),
			false)
		require.NoError(t, err)

		err = repo.AddAccountManagers(ctx, []uuid.UUID{user.ID})
		require.NoError(t, err)

		u, err := repo.GetUserByID(ctx, user.ID)
		require.NoError(t, err)
		require.True(t, u.AccountManager)
	})
}

func TestEntRepository_DeleteAccountManagers(t *testing.T) {
	ctx := testutil.NewContext(t)
	client, err := setup(t, ctx, "delete_accountManagers")
	require.NoError(t, err)
	repo := NewEntRepository(client)

	t.Run("Success", func(t *testing.T) {
		t.Parallel()
		user, err := repo.CreateUser(
			ctx,
			random.AlphaNumeric(t, 20),
			random.AlphaNumeric(t, 30),
			true)
		require.NoError(t, err)

		err = repo.DeleteAccountManagers(ctx, []uuid.UUID{user.ID})
		require.NoError(t, err)

		u, err := repo.GetUserByID(ctx, user.ID)
		require.NoError(t, err)
		require.False(t, u.AccountManager)
	})
}