Fix tests

Author: vladhanzha

cloudconnexa/cloudconnexa_test.go

@@ -79,10 +79,11 @@ func TestDoRequest(t *testing.T) {
 	defer server.Close()
 
 	client := &Client{
-		client:      server.Client(),
-		BaseURL:     server.URL,
-		Token:       "mock-access-token",
-		RateLimiter: rate.NewLimiter(rate.Every(1), 5),
+		client:            server.Client(),
+		BaseURL:           server.URL,
+		Token:             "mock-access-token",
+		ReadRateLimiter:   rate.NewLimiter(rate.Every(1), 5),
+		UpdateRateLimiter: rate.NewLimiter(rate.Every(1), 5),
 	}
 
 	tests := []struct {

cloudconnexa/devices_test.go

@@ -13,10 +13,11 @@ import (
 // createTestClient creates a test client with the given server
 func createTestClient(server *httptest.Server) *Client {
 	client := &Client{
-		client:      server.Client(),
-		BaseURL:     server.URL,
-		Token:       "test-token",
-		RateLimiter: rate.NewLimiter(rate.Every(1*time.Second), 5),
+		client:            server.Client(),
+		BaseURL:           server.URL,
+		Token:             "test-token",
+		ReadRateLimiter:   rate.NewLimiter(rate.Every(1), 5),
+		UpdateRateLimiter: rate.NewLimiter(rate.Every(1), 5),
 	}
 	client.Devices = (*DevicesService)(&service{client: client})
 	client.Sessions = (*SessionsService)(&service{client: client})
@@ -280,7 +281,8 @@ func TestDevicesService_ListByUserID(t *testing.T) {
 
 func TestDevicesService_List_InvalidSize(t *testing.T) {
 	client := &Client{
-		RateLimiter: rate.NewLimiter(rate.Every(1*time.Second), 5),
+		ReadRateLimiter:   rate.NewLimiter(rate.Every(1), 5),
+		UpdateRateLimiter: rate.NewLimiter(rate.Every(1), 5),
 	}
 	client.Devices = (*DevicesService)(&service{client: client})
 

cloudconnexa/dns_records_direct_test.go

@@ -2,21 +2,20 @@ package cloudconnexa
 
 import (
 	"encoding/json"
+	"golang.org/x/time/rate"
 	"net/http"
 	"net/http/httptest"
 	"testing"
-	"time"
-
-	"golang.org/x/time/rate"
 )
 
 // createTestDNSClient creates a test client with the given server for DNS testing
 func createTestDNSClient(server *httptest.Server) *Client {
 	client := &Client{
-		client:      server.Client(),
-		BaseURL:     server.URL,
-		Token:       "test-token",
-		RateLimiter: rate.NewLimiter(rate.Every(1*time.Second), 5),
+		client:            server.Client(),
+		BaseURL:           server.URL,
+		Token:             "test-token",
+		ReadRateLimiter:   rate.NewLimiter(rate.Every(1), 5),
+		UpdateRateLimiter: rate.NewLimiter(rate.Every(1), 5),
 	}
 	client.DNSRecords = (*DNSRecordsService)(&service{client: client})
 	return client

cloudconnexa/host_ip_services_dto_test.go

@@ -2,21 +2,20 @@ package cloudconnexa
 
 import (
 	"encoding/json"
+	"golang.org/x/time/rate"
 	"net/http"
 	"net/http/httptest"
 	"testing"
-	"time"
-
-	"golang.org/x/time/rate"
 )
 
 // createTestHostIPServicesClient creates a test client with the given server for host IP services testing
 func createTestHostIPServicesClient(server *httptest.Server) *Client {
 	client := &Client{
-		client:      server.Client(),
-		BaseURL:     server.URL,
-		Token:       "test-token",
-		RateLimiter: rate.NewLimiter(rate.Every(1*time.Second), 5),
+		client:            server.Client(),
+		BaseURL:           server.URL,
+		Token:             "test-token",
+		ReadRateLimiter:   rate.NewLimiter(rate.Every(1), 5),
+		UpdateRateLimiter: rate.NewLimiter(rate.Every(1), 5),
 	}
 	client.HostIPServices = (*HostIPServicesService)(&service{client: client})
 	return client

cloudconnexa/network_connectors_ipsec_test.go

@@ -2,21 +2,20 @@ package cloudconnexa
 
 import (
 	"encoding/json"
+	"golang.org/x/time/rate"
 	"net/http"
 	"net/http/httptest"
 	"testing"
-	"time"
-
-	"golang.org/x/time/rate"
 )
 
 // createTestIPsecClient creates a test client with the given server for IPsec testing
 func createTestIPsecClient(server *httptest.Server) *Client {
 	client := &Client{
-		client:      server.Client(),
-		BaseURL:     server.URL,
-		Token:       "test-token",
-		RateLimiter: rate.NewLimiter(rate.Every(1*time.Second), 5),
+		client:            server.Client(),
+		BaseURL:           server.URL,
+		Token:             "test-token",
+		ReadRateLimiter:   rate.NewLimiter(rate.Every(1), 5),
+		UpdateRateLimiter: rate.NewLimiter(rate.Every(1), 5),
 	}
 	client.NetworkConnectors = (*NetworkConnectorsService)(&service{client: client})
 	return client

cloudconnexa/sessions_test.go

@@ -13,10 +13,11 @@ import (
 // createTestSessionsClient creates a test client with the given server for sessions testing
 func createTestSessionsClient(server *httptest.Server) *Client {
 	client := &Client{
-		client:      server.Client(),
-		BaseURL:     server.URL,
-		Token:       "test-token",
-		RateLimiter: rate.NewLimiter(rate.Every(1*time.Second), 5),
+		client:            server.Client(),
+		BaseURL:           server.URL,
+		Token:             "test-token",
+		ReadRateLimiter:   rate.NewLimiter(rate.Every(1), 5),
+		UpdateRateLimiter: rate.NewLimiter(rate.Every(1), 5),
 	}
 	client.Sessions = (*SessionsService)(&service{client: client})
 	return client
@@ -176,7 +177,8 @@ func TestSessionsService_ListByDateRange(t *testing.T) {
 
 func TestSessionsService_List_InvalidSize(t *testing.T) {
 	client := &Client{
-		RateLimiter: rate.NewLimiter(rate.Every(1*time.Second), 5),
+		ReadRateLimiter:   rate.NewLimiter(rate.Every(1), 5),
+		UpdateRateLimiter: rate.NewLimiter(rate.Every(1), 5),
 	}
 	client.Sessions = (*SessionsService)(&service{client: client})
 

cloudconnexa/user_groups_direct_test.go

@@ -2,21 +2,20 @@ package cloudconnexa
 
 import (
 	"encoding/json"
+	"golang.org/x/time/rate"
 	"net/http"
 	"net/http/httptest"
 	"testing"
-	"time"
-
-	"golang.org/x/time/rate"
 )
 
 // createTestUserGroupsClient creates a test client with the given server for user groups testing
 func createTestUserGroupsClient(server *httptest.Server) *Client {
 	client := &Client{
-		client:      server.Client(),
-		BaseURL:     server.URL,
-		Token:       "test-token",
-		RateLimiter: rate.NewLimiter(rate.Every(1*time.Second), 5),
+		client:            server.Client(),
+		BaseURL:           server.URL,
+		Token:             "test-token",
+		ReadRateLimiter:   rate.NewLimiter(rate.Every(1), 5),
+		UpdateRateLimiter: rate.NewLimiter(rate.Every(1), 5),
 	}
 	client.UserGroups = (*UserGroupsService)(&service{client: client})
 	return client