task: scope auth key dependencies to server instances (task 375)

1 files changed, 4 insertions, 4 deletions

diff --git a/internal/ssh/server/publickeycallback_test.go b/internal/ssh/server/publickeycallback_test.go
index 7ded4f3..97baa72 100644
--- a/internal/ssh/server/publickeycallback_test.go
+++ b/internal/ssh/server/publickeycallback_test.go
@@ -16,13 +16,13 @@ func TestAuthKeyStorePermissions(t *testing.T) {
 
 	key := testPublicKey(t, 21)
 
-	if permissions := authKeyStorePermissions("alice", key); permissions != nil {
+	if permissions := authKeyStorePermissions(authKeyStore, "alice", key); permissions != nil {
 		t.Fatalf("Expected nil permissions when no key is cached")
 	}
 
 	authKeyStore.Add("alice", key)
 
-	permissions := authKeyStorePermissions("alice", key)
+	permissions := authKeyStorePermissions(authKeyStore, "alice", key)
 	if permissions == nil {
 		t.Fatalf("Expected permissions when key is cached")
 	}
@@ -30,12 +30,12 @@ func TestAuthKeyStorePermissions(t *testing.T) {
 		t.Fatalf("Unexpected fingerprint: %s", fingerprint)
 	}
 
-	if permissions := authKeyStorePermissions("bob", key); permissions != nil {
+	if permissions := authKeyStorePermissions(authKeyStore, "bob", key); permissions != nil {
 		t.Fatalf("Expected nil permissions for different user")
 	}
 
 	unknownKey := testPublicKey(t, 22)
-	if permissions := authKeyStorePermissions("alice", unknownKey); permissions != nil {
+	if permissions := authKeyStorePermissions(authKeyStore, "alice", unknownKey); permissions != nil {
 		t.Fatalf("Expected nil permissions for unknown key")
 	}
 }