update tests

Author: TheTeaCat

src/main_test.go

@@ -16,7 +16,7 @@ func TestLoadEnvVars(t *testing.T) {
 	loadEnvVars()
 
 	assert.Equal(t, firetailApiUrl, MockFiretailApiUrl)
-	assert.Equal(t, firetailApiToken, MockFiretailApiToken)
+	assert.Equal(t, firetailToken, MockFiretailApiToken)
 }
 
 func TestLoadEnvVarsApiUrlUnset(t *testing.T) {
@@ -27,5 +27,5 @@ func TestLoadEnvVarsApiUrlUnset(t *testing.T) {
 	loadEnvVars()
 
 	assert.Equal(t, firetailApiUrl, DefaultFiretailApiUrl)
-	assert.Equal(t, firetailApiToken, MockFiretailApiToken)
-}
+	assert.Equal(t, firetailToken, MockFiretailApiToken)
+}

src/send_to_firetail_test.go

@@ -28,7 +28,7 @@ func TestSendToFiretail(t *testing.T) {
 			Query:     &testQuery,
 			RequestID: "TEST_ID",
 		},
-	}, testServer.URL, "TEST_KEY")
+	}, testServer.URL, "x-ft-api-key", "TEST_KEY")
 	require.Nil(t, err)
 
 	wg.Wait()
@@ -51,7 +51,7 @@ func TestSendToFiretailBadServer(t *testing.T) {
 			Query:     &testQuery,
 			RequestID: "TEST_ID",
 		},
-	}, testServer.URL, "TEST_KEY")
+	}, testServer.URL, "x-ft-api-key", "TEST_KEY")
 	require.NotNil(t, err)
 	assert.Equal(t, "got err response from firetail api: map[message:fail]", err.Error())
 
@@ -65,7 +65,7 @@ func TestSendToFiretailNoServer(t *testing.T) {
 			Query:     &testQuery,
 			RequestID: "TEST_ID",
 		},
-	}, "http://127.0.0.1:0", "TEST_KEY")
+	}, "http://127.0.0.1:0", "x-ft-api-key", "TEST_KEY")
 	require.NotNil(t, err)
 	assert.Contains(t, err.Error(), "Post \"http://127.0.0.1:0\": dial tcp 127.0.0.1:0")
 }
@@ -77,7 +77,7 @@ func TestSendToFiretailBadUrl(t *testing.T) {
 			Query:     &testQuery,
 			RequestID: "TEST_ID",
 		},
-	}, "\n", "TEST_KEY")
+	}, "\n", "x-ft-api-key", "TEST_KEY")
 	require.NotNil(t, err)
 	assert.Equal(t, "parse \"\\n\": net/url: invalid control character in URL", err.Error())
 }