1 files changed, 65 insertions, 0 deletions
diff --git a/internal/executor/container_test.go b/internal/executor/container_test.go
new file mode 100644
index 0000000..b1513ea
--- /dev/null
+++ b/internal/executor/container_test.go
@@ -0,0 +1,65 @@
+package executor
+
+import (
+	"strings"
+	"testing"
+
+	"github.com/thepeterstone/claudomator/internal/task"
+)
+
+func TestContainerRunner_BuildDockerArgs(t *testing.T) {
+	runner := &ContainerRunner{
+		APIURL:   "http://localhost:8484",
+		DropsDir: "/data/drops",
+	}
+	workspace := "/tmp/ws"
+	taskID := "task-123"
+
+	args := runner.buildDockerArgs(workspace, taskID)
+	
+	expected := []string{
+		"run", "--rm",
+		"-v", "/tmp/ws:/workspace",
+		"-w", "/workspace",
+		"--env-file", "/workspace/.claudomator-env",
+		"-e", "CLAUDOMATOR_API_URL=http://localhost:8484",
+		"-e", "CLAUDOMATOR_TASK_ID=task-123",
+		"-e", "CLAUDOMATOR_DROP_DIR=/data/drops",
+	}
+
+	if len(args) != len(expected) {
+		t.Fatalf("expected %d args, got %d", len(expected), len(args))
+	}
+	for i, v := range args {
+		if v != expected[i] {
+			t.Errorf("arg %d: expected %q, got %q", i, expected[i], v)
+		}
+	}
+}
+
+func TestContainerRunner_BuildInnerCmd(t *testing.T) {
+	runner := &ContainerRunner{}
+	
+	t.Run("claude", func(t *testing.T) {
+		tk := &task.Task{Agent: task.AgentConfig{Type: "claude"}}
+		cmd := runner.buildInnerCmd(tk, "exec-456")
+		
+		cmdStr := strings.Join(cmd, " ")
+		if !strings.Contains(cmdStr, "--resume exec-456") {
+			t.Errorf("expected --resume flag, got %q", cmdStr)
+		}
+		if !strings.Contains(cmdStr, "-p /workspace/.claudomator-instructions.txt") {
+			t.Errorf("expected instructions file path, got %q", cmdStr)
+		}
+	})
+
+	t.Run("gemini", func(t *testing.T) {
+		tk := &task.Task{Agent: task.AgentConfig{Type: "gemini"}}
+		cmd := runner.buildInnerCmd(tk, "exec-456")
+		
+		cmdStr := strings.Join(cmd, " ")
+		if !strings.HasPrefix(cmdStr, "gemini") {
+			t.Errorf("expected gemini command, got %q", cmdStr)
+		}
+	})
+}