Fix tests

Signed-off-by: Karanjot Singh <[email protected]>
atlanhq · May 7, 2024 · cb32fb5 · cb32fb5
1 parent 3af39d1
commit cb32fb5
Show file tree

Hide file tree

Showing 6 changed files with 7 additions and 12 deletions.
diff --git a/atlan/client/atlan_tag_cache_test.go b/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)
 

diff --git a/atlan/client/client.go b/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{} {

diff --git a/atlan/client/client_test.go b/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{}),
 	}
 

diff --git a/atlan/client/fluent_search_test.go b/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

diff --git a/atlan/client/glossary_client_test.go b/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)

diff --git a/main.go b/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().