Skip to content

Commit 25445dc

Browse files
committed
Register singleton for tests
1 parent 9574c98 commit 25445dc

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
@@ -7,6 +7,7 @@ use crate::{
77
blocklist::BlocklistAIHistoryModel,
88
document::ai_document_model::AIDocumentModel,
99
execution_profiles::profiles::AIExecutionProfilesModel,
10+
harness_availability::HarnessAvailabilityModel,
1011
llms::LLMPreferences,
1112
mcp::{
1213
templatable_manager::TemplatableMCPServerManager, FileBasedMCPManager, FileMCPWatcher,
@@ -134,6 +135,7 @@ fn initialize_app(app: &mut App) {
134135
});
135136
app.add_singleton_model(SessionPermissionsManager::new);
136137
app.add_singleton_model(LLMPreferences::new);
138+
app.add_singleton_model(HarnessAvailabilityModel::new);
137139
#[cfg(feature = "voice_input")]
138140
app.add_singleton_model(voice_input::VoiceInput::new);
139141
#[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;
@@ -140,6 +141,7 @@ pub fn initialize_app(app: &mut App) {
140141
app.add_singleton_model(AppTelemetryContextProvider::new_context_provider);
141142
app.add_singleton_model(AuthManager::new_for_test);
142143
app.add_singleton_model(LLMPreferences::new);
144+
app.add_singleton_model(HarnessAvailabilityModel::new);
143145
app.add_singleton_model(SessionPermissionsManager::new);
144146
app.add_singleton_model(DirectoryWatcher::new);
145147
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
@@ -24,6 +24,7 @@ use super::settings::initialize_settings_for_tests;
2424
use crate::ai::blocklist::BlocklistAIPermissions;
2525
use crate::ai::blocklist::SerializedBlockListItem;
2626
use crate::ai::execution_profiles::profiles::AIExecutionProfilesModel;
27+
use crate::ai::harness_availability::HarnessAvailabilityModel;
2728
use crate::ai::llms::LLMPreferences;
2829
use crate::ai::outline::RepoOutlines;
2930
use crate::ai::restored_conversations::RestoredAgentConversations;
@@ -107,6 +108,7 @@ pub fn initialize_app_for_terminal_view(app: &mut App) {
107108
app.add_singleton_model(AppTelemetryContextProvider::new_context_provider);
108109
app.add_singleton_model(AuthManager::new_for_test);
109110
app.add_singleton_model(LLMPreferences::new);
111+
app.add_singleton_model(HarnessAvailabilityModel::new);
110112
app.add_singleton_model(SessionPermissionsManager::new);
111113
app.add_singleton_model(DirectoryWatcher::new);
112114
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)