Fix tests

Signed-off-by: Karanjot Singh <[email protected]>

Author: 0xquark

atlan/client/atlan_tag_cache_test.go

@@ -10,7 +10,6 @@ func TestIntegrationAtlanTagCache_RefreshCache(t *testing.T) {
 	if testing.Short() {
 		t.Skip("skipping integration test")
 	}
-	LoggingEnabled = false
 	client := NewContext()
 	cache := NewAtlanTagCache(client)
 
@@ -27,7 +26,6 @@ func TestIntegrationAtlanTagCache_GetIDForName(t *testing.T) {
 	if testing.Short() {
 		t.Skip("skipping integration test")
 	}
-	LoggingEnabled = false
 
 	client := NewContext()
 	cache := NewAtlanTagCache(client)
@@ -54,7 +52,6 @@ func TestIntegrationAtlanTagCache_GetNameForID(t *testing.T) {
 	if testing.Short() {
 		t.Skip("skipping integration test")
 	}
-	LoggingEnabled = false
 	client := NewContext()
 	cache := NewAtlanTagCache(client)
 

atlan/client/client.go

@@ -303,7 +303,7 @@ func (ac *AtlanClient) logHTTPStatus(response *http.Response) {
 }
 
 func (ac *AtlanClient) logResponse(responseJSON []byte) {
-	ac.logger.Infof("<== __call_api", string(responseJSON))
+	ac.logger.Infof("<== __call_api %s", string(responseJSON))
 }
 
 func deepCopy(original map[string]interface{}) map[string]interface{} {

atlan/client/client_test.go

@@ -40,11 +40,10 @@ func TestCallAPI(t *testing.T) {
 
 	// Create a new AtlanClient instance
 	atlanClient := &AtlanClient{
-		Session:        http.DefaultClient,
-		host:           mockServer.URL,
-		ApiKey:         "mock_api_key",
-		loggingEnabled: false,
-		//	logger:         logger,
+		Session: http.DefaultClient,
+		host:    mockServer.URL,
+		ApiKey:  "mock_api_key",
+		//logger:        logger,
 		requestParams: make(map[string]interface{}),
 	}
 

atlan/client/fluent_search_test.go

@@ -12,7 +12,6 @@ func TestIntegrationFluentSearch(t *testing.T) {
 	if testing.Short() {
 		t.Skip("skipping integration test")
 	}
-	LoggingEnabled = false
 	ctx := NewContext()
 
 	// Create a glossary

atlan/client/glossary_client_test.go

@@ -14,7 +14,6 @@ func TestIntegrationGlossary(t *testing.T) {
 		t.Skip("skipping integration test")
 	}
 
-	LoggingEnabled = false
 	NewContext()
 
 	glossaryGUID := testCreateGlossary(t)

main.go

@@ -8,9 +8,10 @@ import (
 
 func main() {
 
-	client.LoggingEnabled = true
 	ctx := client.NewContext()
 
+	ctx.SetLogger(true, "info")
+
 	qualifiedName := "default/snowflake/1714501359/RAW/WIDEWORLDIMPORTERS_SALESFORCE/WAITLIST_WORK_TYPE_HISTORY"
 
 	TableResponse, _ := client.NewFluentSearch().