client-go/user_test.go at main · DependencyTrack/client-go · 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
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
package dtrack

import (
	"context"
	"github.com/stretchr/testify/require"
	"testing"
)

func TestCreateManagedUser(t *testing.T) {
	client := setUpContainer(t, testContainerOptions{
		APIPermissions: []string{
			PermissionAccessManagement,
		},
	})

	user, err := client.User.CreateManaged(context.Background(), ManagedUser{
		Username:        "test-managed",
		Fullname:        "test-managed-full-name",
		Email:           "test-managed-email@localhost",
		NewPassword:     "test-managed-password",
		ConfirmPassword: "test-managed-password",
	})
	require.NoError(t, err)

	require.Equal(t, user.Username, "test-managed")
	require.NotNil(t, user.LastPasswordChange)
	require.Equal(t, user.Fullname, "test-managed-full-name")
	require.Equal(t, user.Email, "test-managed-email@localhost")
	require.Equal(t, user.Teams, []Team{})
	require.Equal(t, user.Permissions, []Permission{})
	require.Equal(t, user.Suspended, false)
	require.Equal(t, user.ForcePasswordChange, false)
	require.Equal(t, user.NonExpiryPassword, false)
}

func TestUpdateManagedUser(t *testing.T) {
	client := setUpContainer(t, testContainerOptions{
		APIPermissions: []string{
			PermissionAccessManagement,
		},
	})

	user, err := client.User.CreateManaged(context.Background(), ManagedUser{
		Username:        "test-managed",
		Fullname:        "test-managed-full-name",
		Email:           "test-managed-email@localhost",
		NewPassword:     "test-managed-password",
		ConfirmPassword: "test-managed-password",
	})
	require.NoError(t, err)
	require.Equal(t, user.Username, "test-managed")
	require.Equal(t, user.Fullname, "test-managed-full-name")
	require.Equal(t, user.Email, "test-managed-email@localhost")

	updated, err := client.User.UpdateManaged(context.Background(), ManagedUser{
		Username: user.Username,
		Fullname: "test-managed-full-name-updated",
		Email:    user.Email,
	})

	require.NoError(t, err)
	require.Equal(t, updated.Username, user.Username)
	require.Equal(t, updated.Fullname, "test-managed-full-name-updated")
	require.Equal(t, updated.Email, user.Email)
}

func TestGetAllManagedUsers(t *testing.T) {
	client := setUpContainer(t, testContainerOptions{
		APIPermissions: []string{
			PermissionAccessManagement,
		},
	})

	user, err := client.User.CreateManaged(context.Background(), ManagedUser{
		Username:        "test-managed",
		Fullname:        "test-managed-full-name",
		Email:           "test-managed-email@localhost",
		NewPassword:     "test-managed-password",
		ConfirmPassword: "test-managed-password",
	})
	require.NoError(t, err)
	require.Equal(t, user.Username, "test-managed")
	require.Equal(t, user.Fullname, "test-managed-full-name")
	require.Equal(t, user.Email, "test-managed-email@localhost")

	users, err := FetchAll(func(po PageOptions) (Page[ManagedUser], error) {
		return client.User.GetAllManaged(context.Background(), po)
	})

	require.NoError(t, err)
	require.Equal(t, len(users), 2)
	require.Equal(t, users[0].Username, "admin")
	require.Equal(t, len(users[0].Teams), 1)
	require.Equal(t, users[0].Teams[0].Name, "Administrators")
	require.Equal(t, len(users[0].Permissions), 14)

	require.Equal(t, users[1].Username, "test-managed")
}

func TestDeleteManagedUser(t *testing.T) {
	// Create User
	// Validate creation of user
	// Delete user
	// Validate deletion of user
	client := setUpContainer(t, testContainerOptions{
		APIPermissions: []string{
			PermissionAccessManagement,
		},
	})

	user, err := client.User.CreateManaged(context.Background(), ManagedUser{
		Username:        "test-managed",
		Fullname:        "test-managed-full-name",
		Email:           "test-managed-email@localhost",
		NewPassword:     "test-managed-password",
		ConfirmPassword: "test-managed-password",
	})
	require.NoError(t, err)
	require.Equal(t, user.Username, "test-managed")
	require.Equal(t, user.Fullname, "test-managed-full-name")
	require.Equal(t, user.Email, "test-managed-email@localhost")

	users, err := FetchAll(func(po PageOptions) (Page[ManagedUser], error) {
		return client.User.GetAllManaged(context.Background(), po)
	})

	require.NoError(t, err)
	require.Equal(t, len(users), 2)

	err = client.User.DeleteManaged(context.Background(), ManagedUser{
		Username: user.Username,
	})

	require.NoError(t, err)

	users, err = FetchAll(func(po PageOptions) (Page[ManagedUser], error) {
		return client.User.GetAllManaged(context.Background(), po)
	})

	require.NoError(t, err)
	require.Equal(t, len(users), 1)
}