fix: server tests

charmbracelet · May 2, 2023 · 443e713 · 443e713
1 parent 0ff2f2d
commit 443e713
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 6 deletions.
diff --git a/server/config/config_test.go b/server/config/config_test.go
@@ -9,8 +9,8 @@ import (
 
 func TestParseMultipleKeys(t *testing.T) {
 	is := is.New(t)
-	is.NoErr(os.Setenv("SOFT_SERVE_INITIAL_ADMIN_KEY", "testdata/k1.pub\ntestdata/k2.pub"))
-	t.Cleanup(func() { is.NoErr(os.Unsetenv("SOFT_SERVE_INITIAL_ADMIN_KEY")) })
+	is.NoErr(os.Setenv("SOFT_SERVE_INITIAL_ADMIN_KEYS", "testdata/k1.pub\ntestdata/k2.pub"))
+	t.Cleanup(func() { is.NoErr(os.Unsetenv("SOFT_SERVE_INITIAL_ADMIN_KEYS")) })
 	cfg := DefaultConfig()
 	is.Equal(cfg.InitialAdminKeys, []string{
 		"testdata/k1.pub",

diff --git a/server/server_test.go b/server/server_test.go
@@ -46,11 +46,10 @@ func createKeyPair(tb testing.TB) (ssh.PublicKey, string) {
 	tb.Helper()
 	is := is.New(tb)
 	keyDir := tb.TempDir()
-	kp, err := keygen.NewWithWrite(filepath.Join(keyDir, "id"), nil, keygen.Ed25519)
+	fp := filepath.Join(keyDir, "id_ed25519")
+	kp, err := keygen.New(fp, keygen.WithKeyType(keygen.Ed25519), keygen.WithWrite())
 	is.NoErr(err)
-	pubkey, _, _, _, err := ssh.ParseAuthorizedKey(kp.PublicKey())
-	is.NoErr(err)
-	return pubkey, filepath.Join(keyDir, "id_ed25519")
+	return kp.PublicKey(), fp
 }
 
 func authorizedKey(pk ssh.PublicKey) string {