1 files changed, 53 insertions, 0 deletions
diff --git a/internal/handlers/atoms_test.go b/internal/handlers/atoms_test.go
index 521b147..7cb94a7 100644
--- a/internal/handlers/atoms_test.go
+++ b/internal/handlers/atoms_test.go
@@ -44,6 +44,59 @@ func TestBuildUnifiedAtomList_WithNativeTasks(t *testing.T) {
 	}
 }
 
+func TestBuildUnifiedAtomList_ExcludesLockedChainTasks(t *testing.T) {
+	s, err := store.New(":memory:", "../../migrations")
+	if err != nil {
+		t.Fatalf("failed to create in-memory store: %v", err)
+	}
+	defer s.Close()
+
+	if _, err := s.CreateChain("Track", []models.ChainTaskInput{
+		{Content: "Step 1"}, {Content: "Step 2"}, {Content: "Step 3"},
+	}); err != nil {
+		t.Fatal(err)
+	}
+
+	atoms, _, err := BuildUnifiedAtomList(s, nil)
+	if err != nil {
+		t.Fatalf("unexpected error: %v", err)
+	}
+	for _, a := range atoms {
+		if a.Title == "Step 1" || a.Title == "Step 2" || a.Title == "Step 3" {
+			t.Errorf("chain task %q should not appear in the flat atom list (chains get their own Chains section)", a.Title)
+		}
+	}
+}
+
+func TestBuildUnifiedAtomList_ExcludesDormantBucketItems(t *testing.T) {
+	s, err := store.New(":memory:", "../../migrations")
+	if err != nil {
+		t.Fatalf("failed to create in-memory store: %v", err)
+	}
+	defer s.Close()
+
+	bucket, err := s.CreateBucket("Gutters", 30, 1)
+	if err != nil {
+		t.Fatal(err)
+	}
+	if err := s.CreateNativeTask(models.Task{ID: "pool-item", Content: "Clean gutters", Priority: 1}); err != nil {
+		t.Fatal(err)
+	}
+	if err := s.AddBucketItem(bucket.ID, "pool-item"); err != nil {
+		t.Fatal(err)
+	}
+
+	atoms, _, err := BuildUnifiedAtomList(s, nil)
+	if err != nil {
+		t.Fatalf("unexpected error: %v", err)
+	}
+	for _, a := range atoms {
+		if a.Title == "Clean gutters" {
+			t.Error("dormant bucket-pool item should not appear in the flat atom list")
+		}
+	}
+}
+
 func TestBuildUnifiedAtomList_WithClaudomator(t *testing.T) {
 	s, err := store.New(":memory:", "../../migrations")
 	if err != nil {