go-proxmox/access_tfa_test.go at main · luthermonson/go-proxmox · 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
package proxmox

import (
	"context"
	"testing"

	"github.com/luthermonson/go-proxmox/tests/mocks"
	"github.com/stretchr/testify/assert"
)

func TestClient_TFAUsers(t *testing.T) {
	mocks.On(mockConfig)
	defer mocks.Off()
	users, err := mockClient().TFAUsers(context.Background())
	assert.Nil(t, err)
	assert.Len(t, users, 1)
	assert.Equal(t, "alice@pve", users[0].UserID)
	assert.True(t, users[0].TOTP)
}

func TestClient_TFAEntries(t *testing.T) {
	mocks.On(mockConfig)
	defer mocks.Off()
	entries, err := mockClient().TFAEntries(context.Background(), "alice@pve")
	assert.Nil(t, err)
	assert.Len(t, entries, 1)
	assert.Equal(t, "totp-1", entries[0].ID)

	_, err = mockClient().TFAEntries(context.Background(), "")
	assert.NotNil(t, err)
}

func TestClient_TFAEntry(t *testing.T) {
	mocks.On(mockConfig)
	defer mocks.Off()
	entry, err := mockClient().TFAEntry(context.Background(), "alice@pve", "totp-1")
	assert.Nil(t, err)
	assert.Equal(t, "phone", entry.Description)

	_, err = mockClient().TFAEntry(context.Background(), "", "totp-1")
	assert.NotNil(t, err)
}

func TestClient_NewTFAEntry(t *testing.T) {
	mocks.On(mockConfig)
	defer mocks.Off()
	id, err := mockClient().NewTFAEntry(context.Background(), "alice@pve", &TFAEntryOptions{
		Type:        "totp",
		Description: "phone",
		TOTP:        "otpauth://totp/Proxmox:alice?secret=ABCD&issuer=Proxmox",
		Value:       "123456",
	})
	assert.Nil(t, err)
	assert.Equal(t, "totp-2", id)

	_, err = mockClient().NewTFAEntry(context.Background(), "alice@pve", &TFAEntryOptions{})
	assert.NotNil(t, err)

	_, err = mockClient().NewTFAEntry(context.Background(), "", nil)
	assert.NotNil(t, err)
}

func TestClient_UpdateTFAEntry(t *testing.T) {
	mocks.On(mockConfig)
	defer mocks.Off()
	enable := false
	err := mockClient().UpdateTFAEntry(context.Background(), "alice@pve", "totp-1", &TFAEntryUpdateOptions{Enable: &enable})
	assert.Nil(t, err)

	err = mockClient().UpdateTFAEntry(context.Background(), "", "totp-1", nil)
	assert.NotNil(t, err)
}

func TestClient_DeleteTFAEntry(t *testing.T) {
	mocks.On(mockConfig)
	defer mocks.Off()
	err := mockClient().DeleteTFAEntry(context.Background(), "alice@pve", "totp-1", "currentpw")
	assert.Nil(t, err)

	err = mockClient().DeleteTFAEntry(context.Background(), "alice@pve", "totp-1", "")
	assert.Nil(t, err)
}

func TestClient_UnlockUserTFA(t *testing.T) {
	mocks.On(mockConfig)
	defer mocks.Off()
	err := mockClient().UnlockUserTFA(context.Background(), "alice@pve")
	assert.Nil(t, err)

	err = mockClient().UnlockUserTFA(context.Background(), "")
	assert.NotNil(t, err)
}