Function TestGeminiEngine

pkg/workflow/gemini_engine_test.go:13–72

(t *testing.T)

)

func TestGeminiEngine(t *testing.T) {
	engine := NewGeminiEngine()

	t.Run("engine identity", func(t *testing.T) {
		assert.Equal(t, "gemini", engine.GetID(), "Engine ID should be 'gemini'")
		assert.Equal(t, "Google Gemini CLI", engine.GetDisplayName(), "Display name should be 'Google Gemini CLI'")
		assert.NotEmpty(t, engine.GetDescription(), "Description should not be empty")
		assert.False(t, engine.IsExperimental(), "Gemini engine should not be experimental")
	})

	t.Run("capabilities", func(t *testing.T) {
		capabilities := engine.GetCapabilities()
		assert.True(t, capabilities.ToolsAllowlist, "Should support tools allowlist")
		assert.True(t, capabilities.MaxTurns, "Should support max turns")
		assert.False(t, capabilities.WebSearch, "Should not support built-in web search")
	})

	t.Run("required secrets", func(t *testing.T) {
		workflowData := &WorkflowData{
			Name:        "test",
			ParsedTools: &ToolsConfig{},
			Tools:       map[string]any{},
		}
		secrets := engine.GetRequiredSecretNames(workflowData)
		assert.Contains(t, secrets, "GEMINI_API_KEY", "Should require GEMINI_API_KEY")
	})

	t.Run("required secrets with MCP servers", func(t *testing.T) {
		workflowData := &WorkflowData{
			Name: "test",
			ParsedTools: &ToolsConfig{
				GitHub: &GitHubToolConfig{},
			},
			Tools: map[string]any{
				"github": map[string]any{},
			},
		}
		secrets := engine.GetRequiredSecretNames(workflowData)
		assert.Contains(t, secrets, "GEMINI_API_KEY", "Should require GEMINI_API_KEY")
		assert.Contains(t, secrets, "MCP_GATEWAY_API_KEY", "Should require MCP_GATEWAY_API_KEY when MCP servers present")
		assert.Contains(t, secrets, "GITHUB_MCP_SERVER_TOKEN", "Should require GITHUB_MCP_SERVER_TOKEN for GitHub tool")
	})

	t.Run("declared output files", func(t *testing.T) {
		outputFiles := engine.GetDeclaredOutputFiles()
		require.Len(t, outputFiles, 1, "Should declare one output file path")
		assert.Equal(t, "/tmp/gh-aw/gemini-client-error-*.json", outputFiles[0], "Should declare Gemini error log wildcard path under /tmp/gh-aw/")
	})

	t.Run("pre-bundle steps move files to /tmp/gh-aw/", func(t *testing.T) {
		workflowData := &WorkflowData{Name: "test-workflow"}
		steps := engine.GetPreBundleSteps(workflowData)
		require.Len(t, steps, 1, "Should return exactly one pre-bundle step")

		stepContent := strings.Join(steps[0], "\n")
		assert.Contains(t, stepContent, "Move Gemini error files", "Step name should describe move operation")
		assert.Contains(t, stepContent, "mv /tmp/gemini-client-error-*.json /tmp/gh-aw/", "Step should move files to /tmp/gh-aw/")
		assert.Contains(t, stepContent, "if: always()", "Step should run always so files are captured on failure")