adyxax/tfstated
1
0
Fork 0
Code Issues Pull requests Releases Packages Activity

chore(tfstated): fix testing race condition

Browse source
This commit is contained in:
Julien Dessaux 2024-12-01 12:51:18 +01:00
parent 6cf7872b22
commit 6cf2ca82de
Signed by: adyxax
GPG key ID: F92E51B86E07177E
1 changed files with 6 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
cmd/tfstated/main_test.go
View file

@ -7,6 +7,7 @@ import (
"net/http" "net/http"
"net/url" "net/url"
"os" "os"
"sync"
"testing" "testing"
"time" "time"
@ -20,6 +21,7 @@ var baseURI = url.URL{
} }
var db *database.DB var db *database.DB
var adminPassword string var adminPassword string
var adminPasswordMutex sync.Mutex
func TestMain(m *testing.M) { func TestMain(m *testing.M) {
ctx := context.Background() ctx := context.Background()
@ -47,6 +49,8 @@ func TestMain(m *testing.M) {
} }
database.AdvertiseAdminPassword = func(password string) { database.AdvertiseAdminPassword = func(password string) {
adminPasswordMutex.Lock()
defer adminPasswordMutex.Unlock()
adminPassword = password adminPassword = password
} }
go run( go run(
@ -84,6 +88,8 @@ func runHTTPRequest(method string, auth bool, uriRef *url.URL, body io.Reader, t
return return
} }
if auth { if auth {
adminPasswordMutex.Lock()
defer adminPasswordMutex.Unlock()
req.SetBasicAuth("admin", adminPassword) req.SetBasicAuth("admin", adminPassword)
} }
resp, err := client.Do(req) resp, err := client.Do(req)