[build] add At method for loadbalancer

vngcloud · Jun 22, 2024 · 0cde8bb · 0cde8bb
1 parent 54e3b23
commit 0cde8bb
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 1 deletion.
diff --git a/test/lb_test.go b/test/lb_test.go
@@ -221,7 +221,7 @@ func TestCreatePoolWithoutMembersSuccess(t *ltesting.T) {
 func TestCreatePoolWithMembersSuccess(t *ltesting.T) {
 	vngcloud := validSdkConfig()
 	opt := lslbv2.NewCreatePoolRequest("cuongdm3-test-pool-3", lslbv2.PoolProtocolTCP).
-		WithLoadBalancerId("lb-96138e73-0864-4016-8db1-bae387002378").
+		WithLoadBalancerId("lb-8bd4ea07-ab40-483d-8387-124ed2f2cecb").
 		WithMembers(lslbv2.NewMember("cuongdm3-member-1", "10.84.0.32", 80, 80)).
 		WithHealthMonitor(lslbv2.NewHealthMonitor(lslbv2.HealthCheckProtocolTCP))
 

diff --git a/vngcloud/entity/loadbalancer.go b/vngcloud/entity/loadbalancer.go
@@ -40,3 +40,11 @@ func (s *ListLoadBalancers) Empty() bool {
 func (s *ListLoadBalancers) Add(item *LoadBalancer) {
 	s.Items = append(s.Items, item)
 }
+
+func (s *ListLoadBalancers) At(pidx int) *LoadBalancer {
+	if pidx < 0 || pidx >= s.Len() {
+		return nil
+	}
+
+	return s.Items[pidx]
+}