Function testTui_ai

client/integration_test.go:2313–2344

(t *testing.T)

}

func testTui_ai(t *testing.T) {
	// Setup
	defer testutils.BackupAndRestore(t)()
	defer testutils.BackupAndRestoreEnv("OPENAI_API_KEY")()
	os.Setenv("OPENAI_API_KEY", "")
	tester, _, _ := setupTestTui(t, Online)
	req, err := json.Marshal(
		ai.TestOnlyOverrideAiSuggestionRequest{Query: "myQuery", Suggestions: []string{"result 1", "result 2", "longer result 3"}},
	)
	require.NoError(t, err)
	_, err = lib.ApiPost(hctx.MakeContext(), "/api/v1/ai-suggest-override", "application/json", req)
	require.NoError(t, err)

	// Test running an AI query
	require.Equal(t, "true", strings.TrimSpace(tester.RunInteractiveShell(t, `hishtory config-get ai-completion`)))
	out := captureTerminalOutputWithComplexCommands(t, tester, []TmuxCommand{
		{Keys: "hishtory SPACE tquery ENTER"},
		// ExtraDelay since AI queries are debounced and thus slower
		{Keys: "'?myQuery'", ExtraDelay: 1.0},
	})
	out = stripTuiCommandPrefix(t, out)
	testutils.CompareGoldens(t, out, "TestTui-AiQuery")

	// Test that when it is disabled, no AI queries are run
	tester.RunInteractiveShell(t, `hishtory config-set ai-completion false`)
	out = captureTerminalOutput(t, tester, []string{
		"hishtory SPACE tquery ENTER",
		"'?myQuery'",
	})
	out = stripTuiCommandPrefix(t, out)
	testutils.CompareGoldens(t, out, "TestTui-AiQuery-Disabled")
}

func testControlR(t *testing.T, tester shellTester, shellName string, onlineStatus OnlineStatus) {
	// Setup