Init.

Author: Joshua Reed

pkg/cloud/client.go

@@ -37,6 +37,7 @@ type Client interface {
 	ResolveLoadBalancerRuleDetails(*infrav1.CloudStackCluster) error
 	GetOrCreateLoadBalancerRule(*infrav1.CloudStackCluster) error
 	AffinityGroupIFace
+	TagIFace
 }
 
 type client struct {

pkg/cloud/tags.go

@@ -0,0 +1,28 @@
+/*
+Copyright 2022.
+
+Licensed under the Apache License, Version 2.0 (the "License");
+you may not use this file except in compliance with the License.
+You may obtain a copy of the License at
+
+    http://www.apache.org/licenses/LICENSE-2.0
+
+Unless required by applicable law or agreed to in writing, software
+distributed under the License is distributed on an "AS IS" BASIS,
+WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+See the License for the specific language governing permissions and
+limitations under the License.
+*/
+
+package cloud
+
+type TagIFace interface {
+	TagNetwork(string, map[string]string) error
+}
+
+// TagNetwork adds tags to a network by network id.
+func (c *client) TagNetwork(networkId string, tag map[string]string) error {
+	p := c.cs.Resourcetags.NewCreateTagsParams([]string{"someid"}, "network", map[string]string{"some": "tag"})
+	_, err := c.cs.Resourcetags.CreateTags(p)
+	return err
+}

pkg/cloud/tags_test.go

@@ -0,0 +1,115 @@
+/*
+Copyright 2022.
+
+Licensed under the Apache License, Version 2.0 (the "License");
+you may not use this file except in compliance with the License.
+You may obtain a copy of the License at
+
+    http://www.apache.org/licenses/LICENSE-2.0
+
+Unless required by applicable law or agreed to in writing, software
+distributed under the License is distributed on an "AS IS" BASIS,
+WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+See the License for the specific language governing permissions and
+limitations under the License.
+*/
+
+package cloud_test
+
+import (
+	infrav1 "github.com/aws/cluster-api-provider-cloudstack/api/v1beta1"
+	"github.com/aws/cluster-api-provider-cloudstack/pkg/cloud"
+	"github.com/golang/mock/gomock"
+	. "github.com/onsi/ginkgo"
+	. "github.com/onsi/gomega"
+)
+
+var _ = Describe("AffinityGroup Unit Tests", func() {
+	var ( // Declare shared vars.
+		mockCtrl *gomock.Controller
+		// 	mockClient  *cloudstack.CloudStackClient
+		// 	ags         *cloudstack.MockAffinityGroupServiceIface
+		// 	fakeAG      *cloud.AffinityGroup
+		cluster *infrav1.CloudStackCluster
+		// machine     *infrav1.CloudStackMachine
+		// capiMachine *capiv1.Machine
+	// 	client      cloud.Client
+	)
+
+	BeforeEach(func() {
+		// 	// Setup new mock services.
+		// 	mockCtrl = gomock.NewController(GinkgoT())
+		// 	mockClient = cloudstack.NewMockClient(mockCtrl)
+		// 	ags = mockClient.AffinityGroup.(*cloudstack.MockAffinityGroupServiceIface)
+		// 	client = cloud.NewClientFromCSAPIClient(mockClient)
+		// 	fakeAG = &cloud.AffinityGroup{
+		// 		Name: "FakeAffinityGroup",
+		// 		Type: cloud.AffinityGroupType}
+		cluster = &infrav1.CloudStackCluster{Spec: infrav1.CloudStackClusterSpec{
+			Zone: "Zone1", Network: "SharedGuestNet1"}}
+		// machine = &infrav1.CloudStackMachine{Spec: infrav1.CloudStackMachineSpec{
+		// 	Offering: "Medium Instance", Template: "Ubuntu20"}}
+		// 	machine.ObjectMeta.SetName("rejoshed-affinity-group-test-vm")
+		// capiMachine = &capiv1.Machine{}
+	})
+
+	AfterEach(func() {
+		mockCtrl.Finish()
+	})
+
+	// It("fetches an affinity group", func() {
+	// 	ags.EXPECT().GetAffinityGroupByName(fakeAG.Name).Return(&cloudstack.AffinityGroup{}, 1, nil)
+
+	// 	Ω(client.GetOrCreateAffinityGroup(cluster, fakeAG)).Should(Succeed())
+	// })
+	// It("creates an affinity group", func() {
+	// 	fakeAG.Id = "FakeID"
+	// 	cluster.Spec.Account = "FakeAccount"
+	// 	cluster.Status.DomainID = "FakeDomainId"
+	// 	ags.EXPECT().GetAffinityGroupByID(fakeAG.Id).Return(nil, -1, errors.New("FakeError"))
+	// 	ags.EXPECT().NewCreateAffinityGroupParams(fakeAG.Name, fakeAG.Type).
+	// 		Return(&cloudstack.CreateAffinityGroupParams{})
+	// 	ags.EXPECT().CreateAffinityGroup(ParamMatch(And(AccountEquals("FakeAccount"), DomainIdEquals("FakeDomainId")))).
+	// 		Return(&cloudstack.CreateAffinityGroupResponse{}, nil)
+
+	// 	Ω(client.GetOrCreateAffinityGroup(cluster, fakeAG)).Should(Succeed())
+	// })
+
+	Context("Tag Integ Tests", func() {
+		client, connectionErr := cloud.NewClient("../../cloud-config")
+
+		var ( // Declare shared vars.
+			arbitraryTag *map[string]string
+			networkId    string
+		)
+
+		BeforeEach(func() {
+			if connectionErr != nil { // Only do these tests if an actual ACS instance is available via cloud-config.
+				Skip("Could not connect to ACS instance.")
+			}
+			arbitraryTag = &map[string]string{"Arbitrary": "Tag"}
+			client.GetOrCreateNetwork(cluster)
+			networkId = cluster.Status.NetworkID
+		})
+
+		AfterEach(func() {
+			mockCtrl.Finish()
+		})
+
+		PIt("Tags a network with an arbitrary tag.", func() {
+			// https://cloudstack.apache.org/api/apidocs-4.16/apis/createTags.html
+			Ω(client.TagNetwork(networkId, *arbitraryTag)).Should(Succeed())
+		})
+		PIt("Fethes said tag.", func() {
+			// It's hard to say what exactly the best method here is. I assume there are many ways to fetch a tag.
+			// Maybe something like GetNetworkTags, GetLBTags, etc...
+			// https://cloudstack.apache.org/api/apidocs-4.16/apis/listTags.html
+			// Ω(client.FetchTag(*arbitraryTag)).Should(Succeed())
+		})
+		PIt("Deletes said tag.", func() {
+			// Same, need some design through around how to delete tags.
+			// https://cloudstack.apache.org/api/apidocs-4.16/apis/deleteTags.html
+			// Ω(client.DeleteTags(networkId, *arbitraryTag)).Should(Succeed())
+		})
+	})
+})