113 lines
2.8 KiB
Go
113 lines
2.8 KiB
Go
package cmd
|
|
|
|
import (
|
|
"os"
|
|
"path/filepath"
|
|
"testing"
|
|
|
|
"filippo.io/age"
|
|
)
|
|
|
|
func setupEditTest(t *testing.T) (*age.X25519Identity, string) {
|
|
t.Helper()
|
|
dataDir := t.TempDir()
|
|
configDir := t.TempDir()
|
|
t.Setenv("PDA_DATA", dataDir)
|
|
t.Setenv("PDA_CONFIG", configDir)
|
|
|
|
id, err := age.GenerateX25519Identity()
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if err := os.WriteFile(filepath.Join(dataDir, "identity.txt"), []byte(id.String()+"\n"), 0o600); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
// Reset global config to defaults with test env vars active
|
|
config, _, _, _ = loadConfig()
|
|
|
|
return id, dataDir
|
|
}
|
|
|
|
func TestEditCreatesNewKey(t *testing.T) {
|
|
id, _ := setupEditTest(t)
|
|
|
|
// Create editor script that writes "hello"
|
|
script := filepath.Join(t.TempDir(), "editor.sh")
|
|
if err := os.WriteFile(script, []byte("#!/bin/sh\necho hello > \"$1\"\n"), 0o755); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
t.Setenv("EDITOR", script)
|
|
|
|
// Run edit for a new key
|
|
rootCmd.SetArgs([]string{"edit", "newkey@testedit"})
|
|
if err := rootCmd.Execute(); err != nil {
|
|
t.Fatalf("edit failed: %v", err)
|
|
}
|
|
|
|
// Verify key was created
|
|
store := &Store{}
|
|
p, _ := store.storePath("testedit")
|
|
entries, _ := readStoreFile(p, id)
|
|
idx := findEntry(entries, "newkey")
|
|
if idx < 0 {
|
|
t.Fatal("key was not created")
|
|
}
|
|
if string(entries[idx].Value) != "hello" {
|
|
t.Fatalf("unexpected value: %q", entries[idx].Value)
|
|
}
|
|
}
|
|
|
|
func TestEditModifiesExistingKey(t *testing.T) {
|
|
id, _ := setupEditTest(t)
|
|
|
|
// Create an existing key
|
|
store := &Store{}
|
|
p, _ := store.storePath("testedit2")
|
|
entries := []Entry{{Key: "existing", Value: []byte("original")}}
|
|
if err := writeStoreFile(p, entries, nil); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
// Editor that replaces content
|
|
script := filepath.Join(t.TempDir(), "editor.sh")
|
|
if err := os.WriteFile(script, []byte("#!/bin/sh\necho modified > \"$1\"\n"), 0o755); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
t.Setenv("EDITOR", script)
|
|
|
|
rootCmd.SetArgs([]string{"edit", "existing@testedit2"})
|
|
if err := rootCmd.Execute(); err != nil {
|
|
t.Fatalf("edit failed: %v", err)
|
|
}
|
|
|
|
// Verify
|
|
entries, _ = readStoreFile(p, id)
|
|
idx := findEntry(entries, "existing")
|
|
if idx < 0 {
|
|
t.Fatal("key disappeared")
|
|
}
|
|
if string(entries[idx].Value) != "modified" {
|
|
t.Fatalf("unexpected value: %q", entries[idx].Value)
|
|
}
|
|
}
|
|
|
|
func TestEditNoChangeSkipsWrite(t *testing.T) {
|
|
setupEditTest(t)
|
|
|
|
store := &Store{}
|
|
p, _ := store.storePath("testedit3")
|
|
entries := []Entry{{Key: "unchanged", Value: []byte("same")}}
|
|
if err := writeStoreFile(p, entries, nil); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
// "true" command does nothing — file stays the same
|
|
t.Setenv("EDITOR", "true")
|
|
|
|
rootCmd.SetArgs([]string{"edit", "unchanged@testedit3"})
|
|
if err := rootCmd.Execute(); err != nil {
|
|
t.Fatalf("edit failed: %v", err)
|
|
}
|
|
// Should print "no changes" — we just verify it didn't error
|
|
}
|