diff options
Diffstat (limited to 'internal/server/handlers/authkeycommand_test.go')
| -rw-r--r-- | internal/server/handlers/authkeycommand_test.go | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/internal/server/handlers/authkeycommand_test.go b/internal/server/handlers/authkeycommand_test.go index bb9488b..f510038 100644 --- a/internal/server/handlers/authkeycommand_test.go +++ b/internal/server/handlers/authkeycommand_test.go @@ -33,11 +33,11 @@ func TestHandleAuthKeyCommandSuccess(t *testing.T) { if message := readServerMessage(t, handler.serverMessages); message != "AUTHKEY OK\n" { t.Fatalf("Unexpected response: %q", message) } - if !sshserver.ServerAuthKeyStore().Has(handler.user.Name, key) { + if !sshserver.AuthKeys().Has(handler.user.Name, key) { t.Fatalf("Expected key to be stored for user") } - sshserver.ServerAuthKeyStore().Remove(handler.user.Name, key) + sshserver.AuthKeys().Remove(handler.user.Name, key) } func TestHandleAuthKeyCommandFeatureDisabled(t *testing.T) { @@ -51,7 +51,7 @@ func TestHandleAuthKeyCommandFeatureDisabled(t *testing.T) { if message := readServerMessage(t, handler.serverMessages); message != "AUTHKEY ERR feature disabled\n" { t.Fatalf("Unexpected response: %q", message) } - if sshserver.ServerAuthKeyStore().Has(handler.user.Name, key) { + if sshserver.AuthKeys().Has(handler.user.Name, key) { t.Fatalf("Expected no key to be stored while feature is disabled") } } |