File: object_storage_quota_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 (80 lines) | stat: -rw-r--r-- 2,597 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
package unit

import (
	"context"
	"testing"

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

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

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

	base.MockGet("object-storage/quotas/obj-objects-us-ord-1", fixtureData)

	quota, err := base.Client.GetObjectStorageQuota(context.Background(), "obj-objects-us-ord-1")
	if err != nil {
		t.Fatalf("Error getting object-storage quota: %v", err)
	}

	assert.Equal(t, "obj-objects-us-ord-1", quota.QuotaID)
	assert.Equal(t, "Object Storage Maximum Objects", quota.QuotaName)
	assert.Equal(t, "Maximum number of Objects this customer is allowed to have on this endpoint.", quota.Description)
	assert.Equal(t, "E1", quota.EndpointType)
	assert.Equal(t, "us-iad-1.linodeobjects.com", quota.S3Endpoint)
	assert.Equal(t, 50, quota.QuotaLimit)
	assert.Equal(t, "object", quota.ResourceMetric)
}

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

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

	base.MockGet("object-storage/quotas", fixtureData)

	quotas, err := base.Client.ListObjectStorageQuotas(context.Background(), &linodego.ListOptions{})
	if err != nil {
		t.Fatalf("Error listing object-storage quotas: %v", err)
	}

	if len(quotas) < 1 {
		t.Fatalf("Expected to get a list of object-storage quotas but failed.")
	}

	assert.Equal(t, "obj-objects-us-ord-1", quotas[0].QuotaID)
	assert.Equal(t, "Object Storage Maximum Objects", quotas[0].QuotaName)
	assert.Equal(t, "Maximum number of Objects this customer is allowed to have on this endpoint.", quotas[0].Description)
	assert.Equal(t, "E1", quotas[0].EndpointType)
	assert.Equal(t, "us-iad-1.linodeobjects.com", quotas[0].S3Endpoint)
	assert.Equal(t, 50, quotas[0].QuotaLimit)
	assert.Equal(t, "object", quotas[0].ResourceMetric)
}

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

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

	base.MockGet("object-storage/quotas/obj-bucket-us-ord-1/usage", fixtureData)

	quotaUsage, err := base.Client.GetObjectStorageQuotaUsage(context.Background(), "obj-bucket-us-ord-1")
	if err != nil {
		t.Fatalf("Error getting object storage quota usage: %v", err)
	}

	assert.Equal(t, 100, quotaUsage.QuotaLimit)
	assert.Equal(t, 10, *quotaUsage.Usage)
}