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)
}
|