From f0ad28ab62adb71d2442ae327f3c55c77c5c0686 Mon Sep 17 00:00:00 2001
From: Pieter Noordhuis <pieter.noordhuis@databricks.com>
Date: Wed, 26 Jul 2023 11:37:18 +0200
Subject: [PATCH] Fix tests under ./cmd/configure if DATABRICKS_TOKEN is set
 (#605)

## Changes

The assertions would fail because `DATABRICKS_TOKEN` overrides a token
set in the profile.

## Tests

Tests now pass if `DATABRICKS_TOKEN` is set.
---
 cmd/configure/configure_test.go | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/cmd/configure/configure_test.go b/cmd/configure/configure_test.go
index 0dbf114d..7b627ba9 100644
--- a/cmd/configure/configure_test.go
+++ b/cmd/configure/configure_test.go
@@ -14,8 +14,9 @@ import (
 
 func assertKeyValueInSection(t *testing.T, section *ini.Section, keyName, expectedValue string) {
 	key, err := section.GetKey(keyName)
-	assert.NoError(t, err)
-	assert.Equal(t, key.Value(), expectedValue)
+	if assert.NoError(t, err) {
+		assert.Equal(t, expectedValue, key.Value())
+	}
 }
 
 func setup(t *testing.T) string {
@@ -26,6 +27,7 @@ func setup(t *testing.T) string {
 	}
 	t.Setenv(homeEnvVar, tempHomeDir)
 	t.Setenv("DATABRICKS_CONFIG_FILE", "")
+	t.Setenv("DATABRICKS_TOKEN", "")
 	return tempHomeDir
 }