@@ -2,7 +2,6 @@ package fakeserver
22
33import (
44 "bytes"
5- "io/ioutil"
65 "log"
76 "net/http"
87 "net/http/httptest"
@@ -41,7 +40,7 @@ something_something_enabled: "true"
4140func TestMain (m * testing.M ) {
4241 current , exists := os .LookupEnv ("TESTTRACK_FAKE_SERVER_CONFIG_DIR" )
4342
44- dir , err := ioutil . TempDir ("" , "testtrack-cli" )
43+ dir , err := os . MkdirTemp ("" , "testtrack-cli" )
4544 if err != nil {
4645 log .Fatal (err )
4746 }
@@ -53,12 +52,12 @@ func TestMain(m *testing.M) {
5352 }
5453
5554 schemaContent := []byte (testSchema )
56- if err := ioutil .WriteFile (filepath .Join (schemasDir , "test.yml" ), schemaContent , 0644 ); err != nil {
55+ if err := os .WriteFile (filepath .Join (schemasDir , "test.yml" ), schemaContent , 0644 ); err != nil {
5756 log .Fatal (err )
5857 }
5958
6059 assignmentsContent := []byte (testAssignments )
61- if err := ioutil .WriteFile (filepath .Join (dir , "assignments.yml" ), assignmentsContent , 0644 ); err != nil {
60+ if err := os .WriteFile (filepath .Join (dir , "assignments.yml" ), assignmentsContent , 0644 ); err != nil {
6261 log .Fatal (err )
6362 }
6463
@@ -230,7 +229,7 @@ func TestCors(t *testing.T) {
230229 h .ServeHTTP (w , request )
231230
232231 require .Equal (t , http .StatusOK , w .Code )
233- require .Equal (t , "" , w .HeaderMap .Get ("Access-Control-Allow-Origin" ))
232+ require .Equal (t , "" , w .Result (). Header .Get ("Access-Control-Allow-Origin" ))
234233 })
235234
236235 t .Run ("it passes cors with an allowed origin" , func (t * testing.T ) {
@@ -243,7 +242,7 @@ func TestCors(t *testing.T) {
243242 h .ServeHTTP (w , request )
244243
245244 require .Equal (t , http .StatusOK , w .Code )
246- require .Equal (t , "http://www.allowed.com" , w .HeaderMap .Get ("Access-Control-Allow-Origin" ))
245+ require .Equal (t , "http://www.allowed.com" , w .Result (). Header .Get ("Access-Control-Allow-Origin" ))
247246 })
248247
249248 t .Run ("it passes cors from localhost" , func (t * testing.T ) {
@@ -256,7 +255,7 @@ func TestCors(t *testing.T) {
256255 h .ServeHTTP (w , request )
257256
258257 require .Equal (t , http .StatusOK , w .Code )
259- require .Equal (t , "http://localhost:3000" , w .HeaderMap .Get ("Access-Control-Allow-Origin" ))
258+ require .Equal (t , "http://localhost:3000" , w .Result (). Header .Get ("Access-Control-Allow-Origin" ))
260259 })
261260
262261 t .Run ("it passes cors from loopback ip" , func (t * testing.T ) {
@@ -269,7 +268,7 @@ func TestCors(t *testing.T) {
269268 h .ServeHTTP (w , request )
270269
271270 require .Equal (t , http .StatusOK , w .Code )
272- require .Equal (t , "http://127.0.0.1:3000" , w .HeaderMap .Get ("Access-Control-Allow-Origin" ))
271+ require .Equal (t , "http://127.0.0.1:3000" , w .Result (). Header .Get ("Access-Control-Allow-Origin" ))
273272 })
274273
275274 os .Unsetenv ("TESTTRACK_ALLOWED_ORIGINS" )
@@ -305,11 +304,11 @@ func TestPersistAssignmentV2(t *testing.T) {
305304
306305 overrides := v2AssignmentOverrideRequestBody {
307306 Assignments : []v1Assignment {
308- v1Assignment {
307+ {
309308 SplitName : "test.test_experiment" ,
310309 Variant : "control" ,
311310 },
312- v1Assignment {
311+ {
313312 SplitName : "test.test2_experiment" ,
314313 Variant : "treatment" ,
315314 },
0 commit comments