Skip to content

Commit c6be84f

Browse files
committed
Register singleton for tests
1 parent 0822435 commit c6be84f

4 files changed

Lines changed: 8 additions & 0 deletions

File tree

app/src/pane_group/mod_tests.rs

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -16,6 +16,7 @@ use crate::{
1616
},
1717
document::ai_document_model::AIDocumentModel,
1818
execution_profiles::profiles::AIExecutionProfilesModel,
19+
harness_availability::HarnessAvailabilityModel,
1920
llms::LLMPreferences,
2021
mcp::{
2122
templatable_manager::TemplatableMCPServerManager, FileBasedMCPManager, FileMCPWatcher,
@@ -154,6 +155,7 @@ fn initialize_app(app: &mut App) {
154155
});
155156
app.add_singleton_model(SessionPermissionsManager::new);
156157
app.add_singleton_model(LLMPreferences::new);
158+
app.add_singleton_model(HarnessAvailabilityModel::new);
157159
#[cfg(feature = "voice_input")]
158160
app.add_singleton_model(voice_input::VoiceInput::new);
159161
#[cfg(feature = "local_fs")]

app/src/terminal/input_test.rs

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -5,6 +5,7 @@ use crate::ai::active_agent_views_model::ActiveAgentViewsModel;
55
use crate::ai::agent_conversations_model::AgentConversationsModel;
66
use crate::ai::blocklist::{AIQueryHistory, BlocklistAIPermissions};
77
use crate::ai::execution_profiles::profiles::AIExecutionProfilesModel;
8+
use crate::ai::harness_availability::HarnessAvailabilityModel;
89
use crate::ai::llms::LLMPreferences;
910
use crate::ai::mcp::gallery::MCPGalleryManager;
1011
use crate::ai::mcp::templatable_manager::TemplatableMCPServerManager;
@@ -143,6 +144,7 @@ pub fn initialize_app(app: &mut App) {
143144
app.add_singleton_model(AppTelemetryContextProvider::new_context_provider);
144145
app.add_singleton_model(AuthManager::new_for_test);
145146
app.add_singleton_model(LLMPreferences::new);
147+
app.add_singleton_model(HarnessAvailabilityModel::new);
146148
app.add_singleton_model(SessionPermissionsManager::new);
147149
app.add_singleton_model(DirectoryWatcher::new);
148150
app.add_singleton_model(|_| DetectedRepositories::default());

app/src/test_util/terminal.rs

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -27,6 +27,7 @@ use crate::ai::blocklist::task_status_sync_model::TaskStatusSyncModel;
2727
use crate::ai::blocklist::BlocklistAIPermissions;
2828
use crate::ai::blocklist::SerializedBlockListItem;
2929
use crate::ai::execution_profiles::profiles::AIExecutionProfilesModel;
30+
use crate::ai::harness_availability::HarnessAvailabilityModel;
3031
use crate::ai::llms::LLMPreferences;
3132
use crate::ai::outline::RepoOutlines;
3233
use crate::ai::restored_conversations::RestoredAgentConversations;
@@ -116,6 +117,7 @@ pub fn initialize_app_for_terminal_view(app: &mut App) {
116117
app.add_singleton_model(AppTelemetryContextProvider::new_context_provider);
117118
app.add_singleton_model(AuthManager::new_for_test);
118119
app.add_singleton_model(LLMPreferences::new);
120+
app.add_singleton_model(HarnessAvailabilityModel::new);
119121
app.add_singleton_model(SessionPermissionsManager::new);
120122
app.add_singleton_model(DirectoryWatcher::new);
121123
app.add_singleton_model(|_| DetectedRepositories::default());

app/src/workspace/view_test.rs

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -3,6 +3,7 @@ use crate::ai::blocklist::{BlocklistAIHistoryModel, BlocklistAIPermissions};
33
use crate::ai::document::ai_document_model::AIDocumentModel;
44
use crate::ai::execution_profiles::profiles::AIExecutionProfilesModel;
55
use crate::ai::facts::manager::AIFactManager;
6+
use crate::ai::harness_availability::HarnessAvailabilityModel;
67
use crate::ai::llms::LLMPreferences;
78
use crate::ai::outline::RepoOutlines;
89
use crate::ai::persisted_workspace::PersistedWorkspace;
@@ -140,6 +141,7 @@ fn initialize_app(app: &mut App) {
140141
app.add_singleton_model(AgentConversationsModel::new);
141142
app.add_singleton_model(SessionPermissionsManager::new);
142143
app.add_singleton_model(LLMPreferences::new);
144+
app.add_singleton_model(HarnessAvailabilityModel::new);
143145
app.add_singleton_model(|_| SettingsPaneManager::new());
144146
app.add_singleton_model(|_| AIFactManager::new());
145147

0 commit comments

Comments
 (0)