more cleanup

Cloud-Foundations · Nov 6, 2024 · b26a8e9 · b26a8e9
1 parent b77096d
commit b26a8e9
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/cmd/keymasterd/2fa_totp_test.go b/cmd/keymasterd/2fa_totp_test.go
@@ -127,6 +127,7 @@ func TestGenerateNewTOTPSuccess(t *testing.T) {
 	if valid {
 		t.Fatal("should NOT have been valid")
 	}
+	state.dbDone <- struct{}{}
 
 }
 
@@ -213,6 +214,7 @@ func TestVerifyTOTPHandlerSuccess(t *testing.T) {
 	if err != nil {
 		t.Fatal(err)
 	}
+	state.dbDone <- struct{}{}
 }
 
 func TestAuthTOTPHandlerSuccess(t *testing.T) {
@@ -259,6 +261,7 @@ func TestAuthTOTPHandlerSuccess(t *testing.T) {
 	if err != nil {
 		t.Fatal(err)
 	}
+	state.dbDone <- struct{}{}
 }
 
 func TestTOTPTokenManagerHandlerUpdateSuccess(t *testing.T) {
@@ -327,4 +330,5 @@ func TestTOTPTokenManagerHandlerUpdateSuccess(t *testing.T) {
 	if profile.TOTPAuthData[0].Name != newName {
 		t.Fatal("update not successul")
 	}
+	state.dbDone <- struct{}{}
 }
diff --git a/cmd/keymasterd/2fa_webauthn_test.go b/cmd/keymasterd/2fa_webauthn_test.go
@@ -6,7 +6,6 @@ import (
 	"net/http"
 	"os"
 	"testing"
-	"time"
 
 	"github.com/Cloud-Foundations/keymaster/lib/webapi/v0/proto"
 
@@ -94,5 +93,5 @@ func TestWebAuthnRegistrationBegin(t *testing.T) {
 			"{\"id\":\"_N2M7t9Qe2rwS4asNZ15I4Thd-nkXow6_lyDT6CURM3gD1sAq0FyMnf8NDOARMWMjjNgPfeHpPWP0Q8nkx-v7pNRuR0IwRHkvZeZxaV3Ql3HFigByVOhuB3OCq2em8Ve\",\"rawId\":\"_N2M7t9Qe2rwS4asNZ15I4Thd-nkXow6_lyDT6CURM3gD1sAq0FyMnf8NDOARMWMjjNgPfeHpPWP0Q8nkx-v7pNRuR0IwRHkvZeZxaV3Ql3HFigByVOhuB3OCq2em8Ve\",\"type\":\"public-key\",\"response\":{\"attestationObject\":\"o2NmbXRkbm9uZWdhdHRTdG10oGhhdXRoRGF0YVjkSZYN5YgOjGh0NBcPZHZgW4_krrmihjLHmVzzuoMdl2NBAAADlwAAAAAAAAAAAAAAAAAAAAAAYPzdjO7fUHtq8EuGrDWdeSOE4Xfp5F6MOv5cg0-glETN4A9bAKtBcjJ3_DQzgETFjI4zYD33h6T1j9EPJ5Mfr-6TUbkdCMER5L2XmcWld0JdxxYoAclTobgdzgqtnpvFXqUBAgMmIAEhWCBwm_S46LuncSKubWLGS7236xBQyY-Ptg0dTKpOmddRMCJYIG02ZJischNpyUqMXRdiJfBW2kDmG3TROzKzHHBHmLlp\",\"clientDataJSON\":\"eyJjaGFsbGVuZ2UiOiJlTW1Ca0gxQ05KZzFsbGRQb3ZXQUN6R0pMZUpYRHZndmViUXIycDRxdWNVIiwib3JpZ2luIjoiaHR0cHM6Ly9sb2NhbGhvc3Q6MzM0NDMiLCJ0eXBlIjoid2ViYXV0aG4uY3JlYXRlIn0\"}}": ""
 		}
 	*/
-	time.Sleep(time.Second)
+	state.dbDone <- struct{}{}
 }