github.com/Azure/azure-sdk-for-go/sdk/resourcemanager/network/armnetwork/v6@v6.2.0/serviceendpointpolicy_live_test.go (about)

     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 armnetwork_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/internal/v3/testutil"
    19  	"github.com/Azure/azure-sdk-for-go/sdk/resourcemanager/network/armnetwork/v6"
    20  	"github.com/stretchr/testify/suite"
    21  )
    22  
    23  type ServiceEndpointPolicyTestSuite struct {
    24  	suite.Suite
    25  
    26  	ctx                                 context.Context
    27  	cred                                azcore.TokenCredential
    28  	options                             *arm.ClientOptions
    29  	serviceEndpointPolicyName           string
    30  	serviceEndpointPolicyDefinitionName string
    31  	location                            string
    32  	resourceGroupName                   string
    33  	subscriptionId                      string
    34  }
    35  
    36  func (testsuite *ServiceEndpointPolicyTestSuite) SetupSuite() {
    37  	testutil.StartRecording(testsuite.T(), pathToPackage)
    38  
    39  	testsuite.ctx = context.Background()
    40  	testsuite.cred, testsuite.options = testutil.GetCredAndClientOptions(testsuite.T())
    41  	testsuite.serviceEndpointPolicyName, _ = recording.GenerateAlphaNumericID(testsuite.T(), "serviceend", 16, false)
    42  	testsuite.serviceEndpointPolicyDefinitionName, _ = recording.GenerateAlphaNumericID(testsuite.T(), "serviceenddef", 19, false)
    43  	testsuite.location = recording.GetEnvVariable("LOCATION", "westus")
    44  	testsuite.resourceGroupName = recording.GetEnvVariable("RESOURCE_GROUP_NAME", "scenarioTestTempGroup")
    45  	testsuite.subscriptionId = recording.GetEnvVariable("AZURE_SUBSCRIPTION_ID", "00000000-0000-0000-0000-000000000000")
    46  	resourceGroup, _, err := testutil.CreateResourceGroup(testsuite.ctx, testsuite.subscriptionId, testsuite.cred, testsuite.options, testsuite.location)
    47  	testsuite.Require().NoError(err)
    48  	testsuite.resourceGroupName = *resourceGroup.Name
    49  	testsuite.Prepare()
    50  }
    51  
    52  func (testsuite *ServiceEndpointPolicyTestSuite) TearDownSuite() {
    53  	testsuite.Cleanup()
    54  	_, err := testutil.DeleteResourceGroup(testsuite.ctx, testsuite.subscriptionId, testsuite.cred, testsuite.options, testsuite.resourceGroupName)
    55  	testsuite.Require().NoError(err)
    56  	testutil.StopRecording(testsuite.T())
    57  }
    58  
    59  func TestServiceEndpointPolicyTestSuite(t *testing.T) {
    60  	suite.Run(t, new(ServiceEndpointPolicyTestSuite))
    61  }
    62  
    63  func (testsuite *ServiceEndpointPolicyTestSuite) Prepare() {
    64  	var err error
    65  	// From step ServiceEndpointPolicies_CreateOrUpdate
    66  	fmt.Println("Call operation: ServiceEndpointPolicies_CreateOrUpdate")
    67  	serviceEndpointPoliciesClient, err := armnetwork.NewServiceEndpointPoliciesClient(testsuite.subscriptionId, testsuite.cred, testsuite.options)
    68  	testsuite.Require().NoError(err)
    69  	serviceEndpointPoliciesClientCreateOrUpdateResponsePoller, err := serviceEndpointPoliciesClient.BeginCreateOrUpdate(testsuite.ctx, testsuite.resourceGroupName, testsuite.serviceEndpointPolicyName, armnetwork.ServiceEndpointPolicy{
    70  		Location: to.Ptr(testsuite.location),
    71  	}, nil)
    72  	testsuite.Require().NoError(err)
    73  	_, err = testutil.PollForTest(testsuite.ctx, serviceEndpointPoliciesClientCreateOrUpdateResponsePoller)
    74  	testsuite.Require().NoError(err)
    75  }
    76  
    77  // Microsoft.Network/serviceEndpointPolicies/{serviceEndpointPolicyName}
    78  func (testsuite *ServiceEndpointPolicyTestSuite) TestServiceEndpointPolicies() {
    79  	var err error
    80  	// From step ServiceEndpointPolicies_List
    81  	fmt.Println("Call operation: ServiceEndpointPolicies_List")
    82  	serviceEndpointPoliciesClient, err := armnetwork.NewServiceEndpointPoliciesClient(testsuite.subscriptionId, testsuite.cred, testsuite.options)
    83  	testsuite.Require().NoError(err)
    84  	serviceEndpointPoliciesClientNewListPager := serviceEndpointPoliciesClient.NewListPager(nil)
    85  	for serviceEndpointPoliciesClientNewListPager.More() {
    86  		_, err := serviceEndpointPoliciesClientNewListPager.NextPage(testsuite.ctx)
    87  		testsuite.Require().NoError(err)
    88  		break
    89  	}
    90  
    91  	// From step ServiceEndpointPolicies_ListByResourceGroup
    92  	fmt.Println("Call operation: ServiceEndpointPolicies_ListByResourceGroup")
    93  	serviceEndpointPoliciesClientNewListByResourceGroupPager := serviceEndpointPoliciesClient.NewListByResourceGroupPager(testsuite.resourceGroupName, nil)
    94  	for serviceEndpointPoliciesClientNewListByResourceGroupPager.More() {
    95  		_, err := serviceEndpointPoliciesClientNewListByResourceGroupPager.NextPage(testsuite.ctx)
    96  		testsuite.Require().NoError(err)
    97  		break
    98  	}
    99  
   100  	// From step ServiceEndpointPolicies_Get
   101  	fmt.Println("Call operation: ServiceEndpointPolicies_Get")
   102  	_, err = serviceEndpointPoliciesClient.Get(testsuite.ctx, testsuite.resourceGroupName, testsuite.serviceEndpointPolicyName, &armnetwork.ServiceEndpointPoliciesClientGetOptions{Expand: nil})
   103  	testsuite.Require().NoError(err)
   104  
   105  	// From step ServiceEndpointPolicies_UpdateTags
   106  	fmt.Println("Call operation: ServiceEndpointPolicies_UpdateTags")
   107  	_, err = serviceEndpointPoliciesClient.UpdateTags(testsuite.ctx, testsuite.resourceGroupName, testsuite.serviceEndpointPolicyName, armnetwork.TagsObject{
   108  		Tags: map[string]*string{
   109  			"tag1": to.Ptr("value1"),
   110  			"tag2": to.Ptr("value2"),
   111  		},
   112  	}, nil)
   113  	testsuite.Require().NoError(err)
   114  }
   115  
   116  // Microsoft.Network/serviceEndpointPolicies/{serviceEndpointPolicyName}/serviceEndpointPolicyDefinitions/{serviceEndpointPolicyDefinitionName}
   117  func (testsuite *ServiceEndpointPolicyTestSuite) TestServiceEndpointPolicyDefinitions() {
   118  	var err error
   119  	// From step ServiceEndpointPolicyDefinitions_CreateOrUpdate
   120  	fmt.Println("Call operation: ServiceEndpointPolicyDefinitions_CreateOrUpdate")
   121  	serviceEndpointPolicyDefinitionsClient, err := armnetwork.NewServiceEndpointPolicyDefinitionsClient(testsuite.subscriptionId, testsuite.cred, testsuite.options)
   122  	testsuite.Require().NoError(err)
   123  	serviceEndpointPolicyDefinitionsClientCreateOrUpdateResponsePoller, err := serviceEndpointPolicyDefinitionsClient.BeginCreateOrUpdate(testsuite.ctx, testsuite.resourceGroupName, testsuite.serviceEndpointPolicyName, testsuite.serviceEndpointPolicyDefinitionName, armnetwork.ServiceEndpointPolicyDefinition{
   124  		Properties: &armnetwork.ServiceEndpointPolicyDefinitionPropertiesFormat{
   125  			Description: to.Ptr("Storage Service EndpointPolicy Definition"),
   126  			Service:     to.Ptr("Microsoft.Storage"),
   127  			ServiceResources: []*string{
   128  				to.Ptr("/subscriptions/" + testsuite.subscriptionId)},
   129  		},
   130  	}, nil)
   131  	testsuite.Require().NoError(err)
   132  	_, err = testutil.PollForTest(testsuite.ctx, serviceEndpointPolicyDefinitionsClientCreateOrUpdateResponsePoller)
   133  	testsuite.Require().NoError(err)
   134  
   135  	// From step ServiceEndpointPolicyDefinitions_ListByResourceGroup
   136  	fmt.Println("Call operation: ServiceEndpointPolicyDefinitions_ListByResourceGroup")
   137  	serviceEndpointPolicyDefinitionsClientNewListByResourceGroupPager := serviceEndpointPolicyDefinitionsClient.NewListByResourceGroupPager(testsuite.resourceGroupName, testsuite.serviceEndpointPolicyName, nil)
   138  	for serviceEndpointPolicyDefinitionsClientNewListByResourceGroupPager.More() {
   139  		_, err := serviceEndpointPolicyDefinitionsClientNewListByResourceGroupPager.NextPage(testsuite.ctx)
   140  		testsuite.Require().NoError(err)
   141  		break
   142  	}
   143  
   144  	// From step ServiceEndpointPolicyDefinitions_Get
   145  	fmt.Println("Call operation: ServiceEndpointPolicyDefinitions_Get")
   146  	_, err = serviceEndpointPolicyDefinitionsClient.Get(testsuite.ctx, testsuite.resourceGroupName, testsuite.serviceEndpointPolicyName, testsuite.serviceEndpointPolicyDefinitionName, nil)
   147  	testsuite.Require().NoError(err)
   148  
   149  	// From step ServiceEndpointPolicyDefinitions_Delete
   150  	fmt.Println("Call operation: ServiceEndpointPolicyDefinitions_Delete")
   151  	serviceEndpointPolicyDefinitionsClientDeleteResponsePoller, err := serviceEndpointPolicyDefinitionsClient.BeginDelete(testsuite.ctx, testsuite.resourceGroupName, testsuite.serviceEndpointPolicyName, testsuite.serviceEndpointPolicyDefinitionName, nil)
   152  	testsuite.Require().NoError(err)
   153  	_, err = testutil.PollForTest(testsuite.ctx, serviceEndpointPolicyDefinitionsClientDeleteResponsePoller)
   154  	testsuite.Require().NoError(err)
   155  }
   156  
   157  func (testsuite *ServiceEndpointPolicyTestSuite) Cleanup() {
   158  	var err error
   159  	// From step ServiceEndpointPolicies_Delete
   160  	fmt.Println("Call operation: ServiceEndpointPolicies_Delete")
   161  	serviceEndpointPoliciesClient, err := armnetwork.NewServiceEndpointPoliciesClient(testsuite.subscriptionId, testsuite.cred, testsuite.options)
   162  	testsuite.Require().NoError(err)
   163  	serviceEndpointPoliciesClientDeleteResponsePoller, err := serviceEndpointPoliciesClient.BeginDelete(testsuite.ctx, testsuite.resourceGroupName, testsuite.serviceEndpointPolicyName, nil)
   164  	testsuite.Require().NoError(err)
   165  	_, err = testutil.PollForTest(testsuite.ctx, serviceEndpointPoliciesClientDeleteResponsePoller)
   166  	testsuite.Require().NoError(err)
   167  }