sweep: can you add tests for ui.go

[harperreed]

---

Add unit tests for UI functionality in ui.go

Description:

Create a new test file to cover the UI functionality implemented in ui.go. These tests will ensure the proper behavior of the model, command handling, and view rendering.

Tasks:

• Create a new file ui_test.go in the root directory
• Implement unit tests for the model struct and its methods
• Add tests for the initialModel function
• Create tests for the Update method, covering different message types
• Implement tests for the handleCommand method
• Add tests for the View method

Test:

• Test initialModel function
  • Verify correct initialization of model struct
• Test Update method
  • Test behavior for different key presses (Enter, Ctrl+C, Esc)
  • Test window size message handling
• Test handleCommand method
  • Verify correct handling of "r", "reset", "c", "clear", "q", "quit" commands
  • Test handling of unknown commands
• Test View method
  • Verify correct rendering of UI elements
  • Test quitting state view
• Test startUI function
  • Verify proper initialization and running of the tea program

Additional tasks:

• Update main_test.go to include new UI tests if necessary
• Ensure all new tests use Go's built-in testing package

[sweep-ai[bot]]

❌ Error

Traceback (most recent call last):
  File "/app/sweepai/handlers/on_ticket.py", line 141, in on_ticket
    results = await fix_issue(
  File "/app/sweepai/backend/api.py", line 1175, in fix_issue
    _message, snippets, new_messages = wrapped_file_searcher(
  File "/app/sweepai/utils/streamable_functions.py", line 82, in __call__
    return last(self.stream(*args, **kwargs))
  File "/app/sweepai/utils/streamable_functions.py", line 26, in last
    result = next(generator)
  File "/app/sweepai/utils/streamable_functions.py", line 71, in stream
    item = next(stream)
  File "/app/sweepai/backend/api.py", line 535, in wrapped_file_searcher
    raise e
  File "/app/sweepai/backend/api.py", line 509, in wrapped_file_searcher
    for message, snippets, messages in file_searcher.stream( # type: ignore
  File "/app/sweepai/utils/streamable_functions.py", line 71, in stream
    item = next(stream)
  File "/app/sweepai/search/agent/search_agent.py", line 1332, in file_searcher
    for thinking, function_calls_response, function_calls in get_multi_function_calls.stream(
  File "/app/sweepai/utils/streamable_functions.py", line 71, in stream
    item = next(stream)
  File "/app/sweepai/search/agent/agent_utils.py", line 348, in get_multi_function_calls
    for function_calls_response in handle_multiple_function_calls.stream(
  File "/app/sweepai/utils/streamable_functions.py", line 71, in stream
    item = next(stream)
  File "/app/sweepai/search/agent/agent_utils.py", line 212, in handle_multiple_function_calls
    for result in handle_function_call.stream(function_call, tools, **kwargs):
  File "/app/sweepai/utils/streamable_functions.py", line 71, in stream
    item = next(stream)
  File "/app/sweepai/search/agent/agent_utils.py", line 191, in handle_function_call
    for result in tool.function._stream(**function_kwargs):
  File "/app/sweepai/search/agent/search_agent.py", line 616, in vector_search
    for message, retrieved_snippets in search_codebase.stream(
  File "/app/sweepai/utils/streamable_functions.py", line 71, in stream
    item = next(stream)
  File "/app/sweepai/search/agent/agent_utils.py", line 784, in search_codebase
    for message, snippets in prep_snippets.stream(
  File "/app/sweepai/utils/streamable_functions.py", line 71, in stream
    item = next(stream)
  File "/app/sweepai/search/query/ticket_utils.py", line 456, in prep_snippets
    for message, snippets in multi_prep_snippets.stream(
  File "/app/sweepai/utils/streamable_functions.py", line 71, in stream
    item = next(stream)
  File "/app/sweepai/search/query/ticket_utils.py", line 351, in multi_prep_snippets
    for message, ranked_snippets, snippets, content_to_lexical_score in get_top_k_snippets.stream(
  File "/app/sweepai/utils/streamable_functions.py", line 71, in stream
    item = next(stream)
  File "/app/sweepai/search/query/ticket_utils.py", line 225, in get_top_k_snippets
    for message, ranked_snippets_list, snippets, content_to_lexical_score_list in multi_get_top_k_snippets.stream(
  File "/app/sweepai/utils/streamable_functions.py", line 71, in stream
    item = next(stream)
  File "/app/sweepai/search/query/ticket_utils.py", line 174, in multi_get_top_k_snippets
    content_to_lexical_score_list = [search_index(query, lexical_index) for query in queries]
  File "/app/sweepai/search/query/ticket_utils.py", line 174, in <listcomp>
    content_to_lexical_score_list = [search_index(query, lexical_index) for query in queries]
  File "/app/sweepai/search/query/lexical_search.py", line 179, in search_index
    results_with_metadata = index.search_index(query)
AttributeError: 'object' object has no attribute 'search_index'

'object' object has no attribute 'search_index'

Sweep has encountered a runtime error unrelated to your request. Please let us know via this link or at support@sweep.dev directly.

:book: For more information on how to use Sweep, please read our documentation.

^{Tracking ID: bc6ca792c6}