Skip to content

Commit 36f766d

Browse files
AlancereAlancere
authored
add sdk/resourcemanager/cosmos/armcosmos live test (#20705)
* add sdk/resourcemanager/cosmos/armcosmos live test * update assets.json * update assets.json * update assets.json * update assets.json
1 parent c005ed6 commit 36f766d

12 files changed

+2220
-18
lines changed

‎sdk/resourcemanager/cosmos/armcosmos/assets.json

+1-1
Original file line numberDiff line numberDiff line change
@@ -2,5 +2,5 @@
22
"AssetsRepo": "Azure/azure-sdk-assets",
33
"AssetsRepoPrefixPath": "go",
44
"TagPrefix": "go/resourcemanager/cosmos/armcosmos",
5-
"Tag": "go/resourcemanager/cosmos/armcosmos_f300ab67c7"
5+
"Tag": "go/resourcemanager/cosmos/armcosmos_b256b80d8d"
66
}

‎sdk/resourcemanager/cosmos/armcosmos/cassandraresources_live_test.go

+266
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,266 @@
1+
//go:build go1.18
2+
// +build go1.18
3+
4+
// Copyright (c) Microsoft Corporation. All rights reserved.
5+
// Licensed under the MIT License. See License.txt in the project root for license information.
6+
7+
package armcosmos_test
8+
9+
import (
10+
"context"
11+
"fmt"
12+
"testing"
13+
14+
"github.com/Azure/azure-sdk-for-go/sdk/azcore"
15+
"github.com/Azure/azure-sdk-for-go/sdk/azcore/arm"
16+
"github.com/Azure/azure-sdk-for-go/sdk/azcore/to"
17+
"github.com/Azure/azure-sdk-for-go/sdk/internal/recording"
18+
"github.com/Azure/azure-sdk-for-go/sdk/resourcemanager/cosmos/armcosmos/v2"
19+
"github.com/Azure/azure-sdk-for-go/sdk/resourcemanager/internal/testutil"
20+
"github.com/stretchr/testify/suite"
21+
)
22+
23+
type CassandraResourcesTestSuite struct {
24+
suite.Suite
25+
26+
ctx context.Context
27+
cred azcore.TokenCredential
28+
options *arm.ClientOptions
29+
accountName string
30+
keyspaceName string
31+
tableName string
32+
location string
33+
resourceGroupName string
34+
subscriptionId string
35+
}
36+
37+
func (testsuite *CassandraResourcesTestSuite) SetupSuite() {
38+
testutil.StartRecording(testsuite.T(), "sdk/resourcemanager/cosmos/armcosmos/testdata")
39+
40+
testsuite.ctx = context.Background()
41+
testsuite.cred, testsuite.options = testutil.GetCredAndClientOptions(testsuite.T())
42+
testsuite.accountName, _ = recording.GenerateAlphaNumericID(testsuite.T(), "accountn", 14, true)
43+
testsuite.keyspaceName, _ = recording.GenerateAlphaNumericID(testsuite.T(), "keyspace", 14, false)
44+
testsuite.tableName, _ = recording.GenerateAlphaNumericID(testsuite.T(), "tablenam", 14, false)
45+
testsuite.location = testutil.GetEnv("LOCATION", "westus")
46+
testsuite.resourceGroupName = testutil.GetEnv("RESOURCE_GROUP_NAME", "scenarioTestTempGroup")
47+
testsuite.subscriptionId = testutil.GetEnv("AZURE_SUBSCRIPTION_ID", "00000000-0000-0000-0000-000000000000")
48+
resourceGroup, _, err := testutil.CreateResourceGroup(testsuite.ctx, testsuite.subscriptionId, testsuite.cred, testsuite.options, testsuite.location)
49+
testsuite.Require().NoError(err)
50+
testsuite.resourceGroupName = *resourceGroup.Name
51+
testsuite.Prepare()
52+
}
53+
54+
func (testsuite *CassandraResourcesTestSuite) TearDownSuite() {
55+
testsuite.Cleanup()
56+
_, err := testutil.DeleteResourceGroup(testsuite.ctx, testsuite.subscriptionId, testsuite.cred, testsuite.options, testsuite.resourceGroupName)
57+
testsuite.Require().NoError(err)
58+
testutil.StopRecording(testsuite.T())
59+
}
60+
61+
func TestCassandraResourcesTestSuite(t *testing.T) {
62+
suite.Run(t, new(CassandraResourcesTestSuite))
63+
}
64+
65+
func (testsuite *CassandraResourcesTestSuite) Prepare() {
66+
var err error
67+
// From step DatabaseAccounts_CreateOrUpdate
68+
fmt.Println("Call operation: DatabaseAccounts_CreateOrUpdate")
69+
databaseAccountsClient, err := armcosmos.NewDatabaseAccountsClient(testsuite.subscriptionId, testsuite.cred, testsuite.options)
70+
testsuite.Require().NoError(err)
71+
databaseAccountsClientCreateOrUpdateResponsePoller, err := databaseAccountsClient.BeginCreateOrUpdate(testsuite.ctx, testsuite.resourceGroupName, testsuite.accountName, armcosmos.DatabaseAccountCreateUpdateParameters{
72+
Location: to.Ptr(testsuite.location),
73+
Properties: &armcosmos.DatabaseAccountCreateUpdateProperties{
74+
Capabilities: []*armcosmos.Capability{
75+
{
76+
Name: to.Ptr("EnableCassandra"),
77+
}},
78+
CreateMode: to.Ptr(armcosmos.CreateModeDefault),
79+
DatabaseAccountOfferType: to.Ptr("Standard"),
80+
Locations: []*armcosmos.Location{
81+
{
82+
FailoverPriority: to.Ptr[int32](0),
83+
IsZoneRedundant: to.Ptr(false),
84+
LocationName: to.Ptr(testsuite.location),
85+
}},
86+
},
87+
}, nil)
88+
testsuite.Require().NoError(err)
89+
_, err = testutil.PollForTest(testsuite.ctx, databaseAccountsClientCreateOrUpdateResponsePoller)
90+
testsuite.Require().NoError(err)
91+
}
92+
93+
// Microsoft.DocumentDB/databaseAccounts/{accountName}/cassandraKeyspaces/{keyspaceName}
94+
func (testsuite *CassandraResourcesTestSuite) TestCassandraKeyspace() {
95+
var err error
96+
// From step CassandraResources_CreateUpdateCassandraKeyspace
97+
fmt.Println("Call operation: CassandraResources_CreateUpdateCassandraKeyspace")
98+
cassandraResourcesClient, err := armcosmos.NewCassandraResourcesClient(testsuite.subscriptionId, testsuite.cred, testsuite.options)
99+
testsuite.Require().NoError(err)
100+
cassandraResourcesClientCreateUpdateCassandraKeyspaceResponsePoller, err := cassandraResourcesClient.BeginCreateUpdateCassandraKeyspace(testsuite.ctx, testsuite.resourceGroupName, testsuite.accountName, testsuite.keyspaceName, armcosmos.CassandraKeyspaceCreateUpdateParameters{
101+
Location: to.Ptr(testsuite.location),
102+
Tags: map[string]*string{},
103+
Properties: &armcosmos.CassandraKeyspaceCreateUpdateProperties{
104+
Options: &armcosmos.CreateUpdateOptions{
105+
Throughput: to.Ptr[int32](2000),
106+
},
107+
Resource: &armcosmos.CassandraKeyspaceResource{
108+
ID: to.Ptr(testsuite.keyspaceName),
109+
},
110+
},
111+
}, nil)
112+
testsuite.Require().NoError(err)
113+
_, err = testutil.PollForTest(testsuite.ctx, cassandraResourcesClientCreateUpdateCassandraKeyspaceResponsePoller)
114+
testsuite.Require().NoError(err)
115+
116+
// From step CassandraResources_ListCassandraKeyspaces
117+
fmt.Println("Call operation: CassandraResources_ListCassandraKeyspaces")
118+
cassandraResourcesClientNewListCassandraKeyspacesPager := cassandraResourcesClient.NewListCassandraKeyspacesPager(testsuite.resourceGroupName, testsuite.accountName, nil)
119+
for cassandraResourcesClientNewListCassandraKeyspacesPager.More() {
120+
_, err := cassandraResourcesClientNewListCassandraKeyspacesPager.NextPage(testsuite.ctx)
121+
testsuite.Require().NoError(err)
122+
break
123+
}
124+
125+
// From step CassandraResources_GetCassandraKeyspaceThroughput
126+
fmt.Println("Call operation: CassandraResources_GetCassandraKeyspaceThroughput")
127+
_, err = cassandraResourcesClient.GetCassandraKeyspaceThroughput(testsuite.ctx, testsuite.resourceGroupName, testsuite.accountName, testsuite.keyspaceName, nil)
128+
testsuite.Require().NoError(err)
129+
130+
// From step CassandraResources_GetCassandraKeyspace
131+
fmt.Println("Call operation: CassandraResources_GetCassandraKeyspace")
132+
_, err = cassandraResourcesClient.GetCassandraKeyspace(testsuite.ctx, testsuite.resourceGroupName, testsuite.accountName, testsuite.keyspaceName, nil)
133+
testsuite.Require().NoError(err)
134+
135+
// From step CassandraResources_MigrateCassandraKeyspaceToAutoscale
136+
fmt.Println("Call operation: CassandraResources_MigrateCassandraKeyspaceToAutoscale")
137+
cassandraResourcesClientMigrateCassandraKeyspaceToAutoscaleResponsePoller, err := cassandraResourcesClient.BeginMigrateCassandraKeyspaceToAutoscale(testsuite.ctx, testsuite.resourceGroupName, testsuite.accountName, testsuite.keyspaceName, nil)
138+
testsuite.Require().NoError(err)
139+
_, err = testutil.PollForTest(testsuite.ctx, cassandraResourcesClientMigrateCassandraKeyspaceToAutoscaleResponsePoller)
140+
testsuite.Require().NoError(err)
141+
142+
// From step CassandraResources_MigrateCassandraKeyspaceToManualThroughput
143+
fmt.Println("Call operation: CassandraResources_MigrateCassandraKeyspaceToManualThroughput")
144+
cassandraResourcesClientMigrateCassandraKeyspaceToManualThroughputResponsePoller, err := cassandraResourcesClient.BeginMigrateCassandraKeyspaceToManualThroughput(testsuite.ctx, testsuite.resourceGroupName, testsuite.accountName, testsuite.keyspaceName, nil)
145+
testsuite.Require().NoError(err)
146+
_, err = testutil.PollForTest(testsuite.ctx, cassandraResourcesClientMigrateCassandraKeyspaceToManualThroughputResponsePoller)
147+
testsuite.Require().NoError(err)
148+
149+
// From step CassandraResources_UpdateCassandraKeyspaceThroughput
150+
fmt.Println("Call operation: CassandraResources_UpdateCassandraKeyspaceThroughput")
151+
cassandraResourcesClientUpdateCassandraKeyspaceThroughputResponsePoller, err := cassandraResourcesClient.BeginUpdateCassandraKeyspaceThroughput(testsuite.ctx, testsuite.resourceGroupName, testsuite.accountName, testsuite.keyspaceName, armcosmos.ThroughputSettingsUpdateParameters{
152+
Location: to.Ptr(testsuite.location),
153+
Tags: map[string]*string{},
154+
Properties: &armcosmos.ThroughputSettingsUpdateProperties{
155+
Resource: &armcosmos.ThroughputSettingsResource{
156+
Throughput: to.Ptr[int32](400),
157+
},
158+
},
159+
}, nil)
160+
testsuite.Require().NoError(err)
161+
_, err = testutil.PollForTest(testsuite.ctx, cassandraResourcesClientUpdateCassandraKeyspaceThroughputResponsePoller)
162+
testsuite.Require().NoError(err)
163+
}
164+
165+
// Microsoft.DocumentDB/databaseAccounts/{accountName}/cassandraKeyspaces/{keyspaceName}/tables/{tableName}
166+
func (testsuite *CassandraResourcesTestSuite) TestCassandraTable() {
167+
var err error
168+
// From step CassandraResources_CreateUpdateCassandraTable
169+
fmt.Println("Call operation: CassandraResources_CreateUpdateCassandraTable")
170+
cassandraResourcesClient, err := armcosmos.NewCassandraResourcesClient(testsuite.subscriptionId, testsuite.cred, testsuite.options)
171+
testsuite.Require().NoError(err)
172+
cassandraResourcesClientCreateUpdateCassandraTableResponsePoller, err := cassandraResourcesClient.BeginCreateUpdateCassandraTable(testsuite.ctx, testsuite.resourceGroupName, testsuite.accountName, testsuite.keyspaceName, testsuite.tableName, armcosmos.CassandraTableCreateUpdateParameters{
173+
Location: to.Ptr(testsuite.location),
174+
Tags: map[string]*string{},
175+
Properties: &armcosmos.CassandraTableCreateUpdateProperties{
176+
Options: &armcosmos.CreateUpdateOptions{
177+
Throughput: to.Ptr[int32](2000),
178+
},
179+
Resource: &armcosmos.CassandraTableResource{
180+
Schema: &armcosmos.CassandraSchema{
181+
Columns: []*armcosmos.Column{
182+
{
183+
Name: to.Ptr("columnA"),
184+
Type: to.Ptr("Ascii"),
185+
}},
186+
PartitionKeys: []*armcosmos.CassandraPartitionKey{
187+
{
188+
Name: to.Ptr("columnA"),
189+
}},
190+
},
191+
DefaultTTL: to.Ptr[int32](100),
192+
ID: to.Ptr(testsuite.tableName),
193+
},
194+
},
195+
}, nil)
196+
testsuite.Require().NoError(err)
197+
_, err = testutil.PollForTest(testsuite.ctx, cassandraResourcesClientCreateUpdateCassandraTableResponsePoller)
198+
testsuite.Require().NoError(err)
199+
200+
// From step CassandraResources_GetCassandraTable
201+
fmt.Println("Call operation: CassandraResources_GetCassandraTable")
202+
_, err = cassandraResourcesClient.GetCassandraTable(testsuite.ctx, testsuite.resourceGroupName, testsuite.accountName, testsuite.keyspaceName, testsuite.tableName, nil)
203+
testsuite.Require().NoError(err)
204+
205+
// From step CassandraResources_ListCassandraTables
206+
fmt.Println("Call operation: CassandraResources_ListCassandraTables")
207+
cassandraResourcesClientNewListCassandraTablesPager := cassandraResourcesClient.NewListCassandraTablesPager(testsuite.resourceGroupName, testsuite.accountName, testsuite.keyspaceName, nil)
208+
for cassandraResourcesClientNewListCassandraTablesPager.More() {
209+
_, err := cassandraResourcesClientNewListCassandraTablesPager.NextPage(testsuite.ctx)
210+
testsuite.Require().NoError(err)
211+
break
212+
}
213+
214+
// From step CassandraResources_GetCassandraTableThroughput
215+
fmt.Println("Call operation: CassandraResources_GetCassandraTableThroughput")
216+
_, err = cassandraResourcesClient.GetCassandraTableThroughput(testsuite.ctx, testsuite.resourceGroupName, testsuite.accountName, testsuite.keyspaceName, testsuite.tableName, nil)
217+
testsuite.Require().NoError(err)
218+
219+
// From step CassandraResources_MigrateCassandraTableToAutoscale
220+
fmt.Println("Call operation: CassandraResources_MigrateCassandraTableToAutoscale")
221+
cassandraResourcesClientMigrateCassandraTableToAutoscaleResponsePoller, err := cassandraResourcesClient.BeginMigrateCassandraTableToAutoscale(testsuite.ctx, testsuite.resourceGroupName, testsuite.accountName, testsuite.keyspaceName, testsuite.tableName, nil)
222+
testsuite.Require().NoError(err)
223+
_, err = testutil.PollForTest(testsuite.ctx, cassandraResourcesClientMigrateCassandraTableToAutoscaleResponsePoller)
224+
testsuite.Require().NoError(err)
225+
226+
// From step CassandraResources_MigrateCassandraTableToManualThroughput
227+
fmt.Println("Call operation: CassandraResources_MigrateCassandraTableToManualThroughput")
228+
cassandraResourcesClientMigrateCassandraTableToManualThroughputResponsePoller, err := cassandraResourcesClient.BeginMigrateCassandraTableToManualThroughput(testsuite.ctx, testsuite.resourceGroupName, testsuite.accountName, testsuite.keyspaceName, testsuite.tableName, nil)
229+
testsuite.Require().NoError(err)
230+
_, err = testutil.PollForTest(testsuite.ctx, cassandraResourcesClientMigrateCassandraTableToManualThroughputResponsePoller)
231+
testsuite.Require().NoError(err)
232+
233+
// From step CassandraResources_UpdateCassandraTableThroughput
234+
fmt.Println("Call operation: CassandraResources_UpdateCassandraTableThroughput")
235+
cassandraResourcesClientUpdateCassandraTableThroughputResponsePoller, err := cassandraResourcesClient.BeginUpdateCassandraTableThroughput(testsuite.ctx, testsuite.resourceGroupName, testsuite.accountName, testsuite.keyspaceName, testsuite.tableName, armcosmos.ThroughputSettingsUpdateParameters{
236+
Location: to.Ptr(testsuite.location),
237+
Tags: map[string]*string{},
238+
Properties: &armcosmos.ThroughputSettingsUpdateProperties{
239+
Resource: &armcosmos.ThroughputSettingsResource{
240+
Throughput: to.Ptr[int32](400),
241+
},
242+
},
243+
}, nil)
244+
testsuite.Require().NoError(err)
245+
_, err = testutil.PollForTest(testsuite.ctx, cassandraResourcesClientUpdateCassandraTableThroughputResponsePoller)
246+
testsuite.Require().NoError(err)
247+
248+
// From step CassandraResources_DeleteCassandraTable
249+
fmt.Println("Call operation: CassandraResources_DeleteCassandraTable")
250+
cassandraResourcesClientDeleteCassandraTableResponsePoller, err := cassandraResourcesClient.BeginDeleteCassandraTable(testsuite.ctx, testsuite.resourceGroupName, testsuite.accountName, testsuite.keyspaceName, testsuite.tableName, nil)
251+
testsuite.Require().NoError(err)
252+
_, err = testutil.PollForTest(testsuite.ctx, cassandraResourcesClientDeleteCassandraTableResponsePoller)
253+
testsuite.Require().NoError(err)
254+
}
255+
256+
func (testsuite *CassandraResourcesTestSuite) Cleanup() {
257+
var err error
258+
// From step CassandraResources_DeleteCassandraKeyspace
259+
fmt.Println("Call operation: CassandraResources_DeleteCassandraKeyspace")
260+
cassandraResourcesClient, err := armcosmos.NewCassandraResourcesClient(testsuite.subscriptionId, testsuite.cred, testsuite.options)
261+
testsuite.Require().NoError(err)
262+
cassandraResourcesClientDeleteCassandraKeyspaceResponsePoller, err := cassandraResourcesClient.BeginDeleteCassandraKeyspace(testsuite.ctx, testsuite.resourceGroupName, testsuite.accountName, testsuite.keyspaceName, nil)
263+
testsuite.Require().NoError(err)
264+
_, err = testutil.PollForTest(testsuite.ctx, cassandraResourcesClientDeleteCassandraKeyspaceResponsePoller)
265+
testsuite.Require().NoError(err)
266+
}

0 commit comments

Comments
 (0)