Add attachment replace workflow UI

appstate/state_test.go

@@ -1056,6 +1056,76 @@ func TestAddAttachmentToSelectedEntryPersistsAndMarksDirty(t *testing.T) {
 	}
 }
 
+func TestAddAttachmentToSelectedEntryRejectsDuplicateNames(t *testing.T) {
+	t.Parallel()
+
+	model := testVaultModel()
+	model.Entries[0].Attachments = map[string][]byte{"token.txt": []byte("secret")}
+	sess := &mutableStubSession{model: model}
+	state := State{
+		Session:         sess,
+		CurrentPath:     []string{"Root", "Internet"},
+		SelectedEntryID: "bellagio",
+	}
+
+	err := state.AddAttachmentToSelectedEntry("token.txt", []byte("replacement"))
+	if !errors.Is(err, ErrAttachmentAlreadyExists) {
+		t.Fatalf("AddAttachmentToSelectedEntry() error = %v, want ErrAttachmentAlreadyExists", err)
+	}
+
+	got, currentErr := sess.Current()
+	if currentErr != nil {
+		t.Fatalf("Current() error = %v", currentErr)
+	}
+	if string(got.Entries[0].Attachments["token.txt"]) != "secret" {
+		t.Fatalf("attachment content = %q, want %q", got.Entries[0].Attachments["token.txt"], "secret")
+	}
+}
+
+func TestReplaceAttachmentOnSelectedEntryPersistsAndMarksDirty(t *testing.T) {
+	t.Parallel()
+
+	model := testVaultModel()
+	model.Entries[0].Attachments = map[string][]byte{"token.txt": []byte("secret")}
+	sess := &mutableStubSession{model: model}
+	state := State{
+		Session:         sess,
+		CurrentPath:     []string{"Root", "Internet"},
+		SelectedEntryID: "bellagio",
+	}
+
+	if err := state.ReplaceAttachmentOnSelectedEntry("token.txt", []byte("replacement")); err != nil {
+		t.Fatalf("ReplaceAttachmentOnSelectedEntry() error = %v", err)
+	}
+
+	got, err := state.VisibleEntries()
+	if err != nil {
+		t.Fatalf("VisibleEntries() error = %v", err)
+	}
+	if string(got[0].Attachments["token.txt"]) != "replacement" {
+		t.Fatalf("attachment content = %q, want %q", got[0].Attachments["token.txt"], "replacement")
+	}
+	if !state.Dirty {
+		t.Fatal("Dirty = false, want true after ReplaceAttachmentOnSelectedEntry")
+	}
+}
+
+func TestReplaceAttachmentOnSelectedEntryRequiresExistingAttachment(t *testing.T) {
+	t.Parallel()
+
+	sess := &mutableStubSession{model: testVaultModel()}
+	state := State{
+		Session:         sess,
+		CurrentPath:     []string{"Root", "Internet"},
+		SelectedEntryID: "bellagio",
+	}
+
+	err := state.ReplaceAttachmentOnSelectedEntry("token.txt", []byte("replacement"))
+	if !errors.Is(err, ErrAttachmentNotFound) {
+		t.Fatalf("ReplaceAttachmentOnSelectedEntry() error = %v, want ErrAttachmentNotFound", err)
+	}
+}
+
 func TestDeleteAttachmentFromSelectedEntryPersistsAndMarksDirty(t *testing.T) {
 	t.Parallel()
 
@@ -1084,6 +1154,22 @@ func TestDeleteAttachmentFromSelectedEntryPersistsAndMarksDirty(t *testing.T) {
 	}
 }
 
+func TestDeleteAttachmentFromSelectedEntryRequiresExistingAttachment(t *testing.T) {
+	t.Parallel()
+
+	sess := &mutableStubSession{model: testVaultModel()}
+	state := State{
+		Session:         sess,
+		CurrentPath:     []string{"Root", "Internet"},
+		SelectedEntryID: "bellagio",
+	}
+
+	err := state.DeleteAttachmentFromSelectedEntry("token.txt")
+	if !errors.Is(err, ErrAttachmentNotFound) {
+		t.Fatalf("DeleteAttachmentFromSelectedEntry() error = %v, want ErrAttachmentNotFound", err)
+	}
+}
+
 type mutableStubSession struct {
 	model vault.Model
 	err   error