profile_tests (#698)

vshanthe · web-flow · commit 5308e5af448c · 2025-03-20T11:56:39.000+05:30
diff --git a/test/unit/fixtures/profile_login_get.json b/test/unit/fixtures/profile_login_get.json
@@ -0,0 +1,9 @@
+{
+    "id": 456,
+    "ip": "192.0.2.10",
+    "restricted": false,
+    "status": "successful",
+    "username": "testuser",
+    "datetime": "2024-01-15T14:30:00"
+}
+  
diff --git a/test/unit/fixtures/profile_logins_list.json b/test/unit/fixtures/profile_logins_list.json
@@ -0,0 +1,24 @@
+{
+    "data": [
+      {
+        "id": 123,
+        "ip": "203.0.113.5",
+        "restricted": true,
+        "status": "failed",
+        "username": "user1",
+        "datetime": "2024-02-01T10:15:30"
+      },
+      {
+        "id": 124,
+        "ip": "198.51.100.25",
+        "restricted": false,
+        "status": "successful",
+        "username": "user2",
+        "datetime": "2024-02-02T18:45:00"
+      }
+    ],
+    "page": 1,
+    "pages": 1,
+    "results": 2
+}
+  
diff --git a/test/unit/fixtures/profile_sshkey_create.json b/test/unit/fixtures/profile_sshkey_create.json
@@ -0,0 +1,6 @@
+{
+	"id": 125,
+	"label": "Test Key",
+	"ssh_key": "ssh-rsa AAAAB3...",
+	"created": "2024-03-12T08:30:00"
+}
diff --git a/test/unit/fixtures/profile_sshkey_get.json b/test/unit/fixtures/profile_sshkey_get.json
@@ -0,0 +1,6 @@
+{
+	"id": 123,
+	"label": "My SSH Key",
+	"ssh_key": "ssh-rsa AAAAB3...",
+	"created": "2024-03-10T12:00:00"
+}
diff --git a/test/unit/fixtures/profile_sshkey_update.json b/test/unit/fixtures/profile_sshkey_update.json
@@ -0,0 +1,6 @@
+{
+	"id": 123,
+	"label": "Updated Key",
+	"ssh_key": "ssh-rsa AAAAB3...",
+	"created": "2024-03-10T12:00:00"
+}
diff --git a/test/unit/fixtures/profile_sshkeys_list.json b/test/unit/fixtures/profile_sshkeys_list.json
@@ -0,0 +1,18 @@
+{
+    "data": [
+        {
+            "id": 123,
+            "label": "My SSH Key",
+            "ssh_key": "ssh-rsa AAAAB3...",
+            "created": "2024-03-10T12:00:00"
+        },
+        {
+            "id": 124,
+            "label": "Another SSH Key",
+            "ssh_key": "ssh-rsa AAAAB3...",
+            "created": "2024-03-11T15:45:00" 
+        }
+    ],
+    "pages": 1,
+    "results": 2
+}
diff --git a/test/unit/fixtures/profile_token_create.json b/test/unit/fixtures/profile_token_create.json
@@ -0,0 +1,8 @@
+{
+    "id": 125,
+    "label": "New API Token",
+    "scopes": "read_write",
+    "token": "newtoken123",
+    "created": "2025-01-01T08:00:00",
+    "expiry": "2025-06-01T08:00:00"
+}
diff --git a/test/unit/fixtures/profile_token_get.json b/test/unit/fixtures/profile_token_get.json
@@ -0,0 +1,8 @@
+{
+    "id": 123,
+    "label": "My API Token",
+    "scopes": "read_write",
+    "token": "abcd1234",
+    "created": "2024-03-10T12:00:00",
+    "expiry": "2024-09-10T12:00:00"
+}
diff --git a/test/unit/fixtures/profile_token_update.json b/test/unit/fixtures/profile_token_update.json
@@ -0,0 +1,8 @@
+{
+    "id": 123,
+    "label": "Updated API Token",
+    "scopes": "read_write",
+    "token": "abcd1234",
+    "created": "2024-03-10T12:00:00",
+    "expiry": "2024-09-10T12:00:00"
+}
diff --git a/test/unit/fixtures/profile_tokens_list.json b/test/unit/fixtures/profile_tokens_list.json
@@ -0,0 +1,22 @@
+{
+    "data": [
+        {
+            "id": 123,
+            "label": "My API Token",
+            "scopes": "read_write",
+            "token": "abcd1234",
+            "created": "2024-03-10T12:00:00",
+            "expiry": "2024-09-10T12:00:00"
+        },
+        {
+            "id": 124,
+            "label": "Another API Token",
+            "scopes": "read_only",
+            "token": "xyz9876",
+            "created": "2024-03-11T15:45:00",
+            "expiry": "2024-10-11T15:45:00"
+        }
+    ],
+    "pages": 1,
+    "results": 2
+}
diff --git a/test/unit/profile_login_test.go b/test/unit/profile_login_test.go
@@ -0,0 +1,58 @@
+package unit
+
+import (
+	"context"
+	"testing"
+	"time"
+
+	"github.com/stretchr/testify/assert"
+)
+
+func TestProfileLogin_Get(t *testing.T) {
+	fixtureData, err := fixtures.GetFixture("profile_login_get")
+	assert.NoError(t, err)
+
+	var base ClientBaseCase
+	base.SetUp(t)
+	defer base.TearDown(t)
+
+	base.MockGet("profile/logins/456", fixtureData)
+
+	login, err := base.Client.GetProfileLogin(context.Background(), 456)
+	assert.NoError(t, err)
+	assert.NotNil(t, login)
+
+	expectedTime, _ := time.Parse(time.RFC3339, "2024-01-15T14:30:00Z")
+
+	if assert.NotNil(t, login.Datetime) {
+		assert.Equal(t, expectedTime, *login.Datetime)
+	}
+}
+
+func TestProfileLogins_List(t *testing.T) {
+	fixtureData, err := fixtures.GetFixture("profile_logins_list")
+	assert.NoError(t, err)
+
+	var base ClientBaseCase
+	base.SetUp(t)
+	defer base.TearDown(t)
+
+	base.MockGet("profile/logins", fixtureData)
+
+	logins, err := base.Client.ListProfileLogins(context.Background(), nil)
+	assert.NoError(t, err)
+	assert.NotNil(t, logins)
+	assert.Len(t, logins, 2)
+
+	expectedTimes := []string{
+		"2024-02-01T10:15:30Z",
+		"2024-02-02T18:45:00Z",
+	}
+
+	for i, login := range logins {
+		if assert.NotNil(t, login.Datetime) {
+			expectedTime, _ := time.Parse(time.RFC3339, expectedTimes[i])
+			assert.Equal(t, expectedTime, *login.Datetime)
+		}
+	}
+}
diff --git a/test/unit/profile_sshkey_test.go b/test/unit/profile_sshkey_test.go
@@ -0,0 +1,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)
+}
diff --git a/test/unit/profile_token_test.go b/test/unit/profile_token_test.go
