File: profile_sshkey_test.go

package info (click to toggle)
golang-github-linode-linodego 1.55.0-1
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid
  • size: 13,112 kB
  • sloc: makefile: 96; sh: 52; python: 24
file content (112 lines) | stat: -rw-r--r-- 2,564 bytes parent folder | download
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
package unit

import (
	"context"
	"testing"
	"time"

	"github.com/linode/linodego"
	"github.com/stretchr/testify/assert"
)

func TestProfileSSHKey_Get(t *testing.T) {
	fixtureData, err := fixtures.GetFixture("profile_sshkey_get")
	assert.NoError(t, err)

	var base ClientBaseCase
	base.SetUp(t)
	defer base.TearDown(t)

	base.MockGet("profile/sshkeys/123", fixtureData)

	key, err := base.Client.GetSSHKey(context.Background(), 123)
	assert.NoError(t, err)
	assert.NotNil(t, key)

	expectedTime, _ := time.Parse(time.RFC3339, "2024-03-10T12:00:00Z")

	if assert.NotNil(t, key.Created) {
		assert.Equal(t, expectedTime, *key.Created)
	}
}

func TestProfileSSHKeys_List(t *testing.T) {
	fixtureData, err := fixtures.GetFixture("profile_sshkeys_list")
	assert.NoError(t, err)

	var base ClientBaseCase
	base.SetUp(t)
	defer base.TearDown(t)

	base.MockGet("profile/sshkeys", fixtureData)

	keys, err := base.Client.ListSSHKeys(context.Background(), nil)
	assert.NoError(t, err)
	assert.NotNil(t, keys)
	assert.Len(t, keys, 2)

	expectedTimes := []string{
		"2024-03-10T12:00:00Z",
		"2024-03-11T15:45:00Z",
	}

	for i, key := range keys {
		if assert.NotNil(t, key.Created) {
			expectedTime, _ := time.Parse(time.RFC3339, expectedTimes[i])
			assert.Equal(t, expectedTime, *key.Created)
		}
	}
}

func TestProfileSSHKey_Create(t *testing.T) {
	fixtureData, err := fixtures.GetFixture("profile_sshkey_create")
	assert.NoError(t, err)

	var base ClientBaseCase
	base.SetUp(t)
	defer base.TearDown(t)

	base.MockPost("profile/sshkeys", fixtureData)

	opts := linodego.SSHKeyCreateOptions{
		Label:  "Test Key",
		SSHKey: "ssh-rsa AAAAB3...",
	}

	key, err := base.Client.CreateSSHKey(context.Background(), opts)
	assert.NoError(t, err)
	assert.NotNil(t, key)
	assert.Equal(t, opts.Label, key.Label)
	assert.Equal(t, opts.SSHKey, key.SSHKey)
}

func TestProfileSSHKey_Update(t *testing.T) {
	fixtureData, err := fixtures.GetFixture("profile_sshkey_update")
	assert.NoError(t, err)

	var base ClientBaseCase
	base.SetUp(t)
	defer base.TearDown(t)

	base.MockPut("profile/sshkeys/123", fixtureData)

	opts := linodego.SSHKeyUpdateOptions{
		Label: "Updated Key",
	}

	key, err := base.Client.UpdateSSHKey(context.Background(), 123, opts)
	assert.NoError(t, err)
	assert.NotNil(t, key)
	assert.Equal(t, opts.Label, key.Label)
}

func TestProfileSSHKey_Delete(t *testing.T) {
	var base ClientBaseCase
	base.SetUp(t)
	defer base.TearDown(t)

	base.MockDelete("profile/sshkeys/123", nil)

	err := base.Client.DeleteSSHKey(context.Background(), 123)
	assert.NoError(t, err)
}