read: simplify TestSelectTotpCode()

vmiklos · Apr 4, 2024 · eeaeb4e · eeaeb4e
1 parent 300f8ea
commit eeaeb4e
Showing 1 changed file with 2 additions and 5 deletions.
diff --git a/commands/read_test.go b/commands/read_test.go
@@ -82,12 +82,9 @@ func TestSelectTotpCode(t *testing.T) {
 	expectedMachine := "mymachine"
 	expectedService := "myservice"
 	expectedUser := "myuser"
-	expectedPassword := "totppassword"
-	var expectedType PasswordType = "totp"
-	secure := false
-	_, err := createPassword(&ctx, expectedMachine, expectedService, expectedUser, expectedPassword, expectedType, secure)
+	_, err := ctx.Database.Exec(`insert into passwords (machine, service, user, password, type) values('mymachine', 'myservice', 'myuser', 'totppassword', 'totp');`)
 	if err != nil {
-		t.Fatalf("createPassword() = %q, want nil", err)
+		t.Fatalf("db.Exec() = %q, want nil", err)
 	}
 	os.Args = []string{"", "search", "--noid", "--totp", "-m", expectedMachine, "-s", expectedService, "-u", expectedUser}
 	inBuf := new(bytes.Buffer)