Skip to content

Commit 2d6d817

Browse files
authored
fix creating workspace test (#193)
1 parent 93ff0ee commit 2d6d817

File tree

1 file changed

+12
-6
lines changed
  • crates/nebula-backbone/src/application/workspace

1 file changed

+12
-6
lines changed

crates/nebula-backbone/src/application/workspace/mod.rs

Lines changed: 12 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -155,7 +155,9 @@ mod test {
155155
#[tokio::test]
156156
async fn when_creating_workspace_use_case_should_delegate_to_service() {
157157
const WORKSPACE_NAME: &str = "testworkspace";
158-
let mock_database = Arc::new(MockDatabase::new(DatabaseBackend::Postgres).into_connection());
158+
let mock_database = MockDatabase::new(DatabaseBackend::Postgres)
159+
.append_exec_results([MockExecResult { last_insert_id: 0, rows_affected: 1 }]);
160+
let mock_connection = Arc::new(mock_database.into_connection());
159161
let mut workspace_service_mock = MockWorkspaceService::new();
160162

161163
workspace_service_mock
@@ -175,7 +177,7 @@ mod test {
175177
.returning(move |_| Ok(Parameter { version: 1, value: GlobalParams::<Bn462Curve>::new(&mut rng) }));
176178

177179
let workspace_use_case = WorkspaceUseCaseImpl::new(
178-
mock_database,
180+
mock_connection,
179181
Arc::new(workspace_service_mock),
180182
Arc::new(secret_service_mock),
181183
Arc::new(parameter_service_mock),
@@ -189,7 +191,9 @@ mod test {
189191
#[tokio::test]
190192
async fn when_creating_workspace_failed_with_anyhow_use_case_should_returns_anyhow() {
191193
const WORKSPACE_NAME: &str = "testworkspace";
192-
let mock_database = Arc::new(MockDatabase::new(DatabaseBackend::Postgres).into_connection());
194+
let mock_database = MockDatabase::new(DatabaseBackend::Postgres)
195+
.append_exec_results([MockExecResult { last_insert_id: 0, rows_affected: 1 }]);
196+
let mock_connection = Arc::new(mock_database.into_connection());
193197
let mut workspace_service_mock = MockWorkspaceService::new();
194198

195199
workspace_service_mock
@@ -201,7 +205,7 @@ mod test {
201205
let parameter_service_mock = MockParameterService::new();
202206

203207
let workspace_use_case = WorkspaceUseCaseImpl::new(
204-
mock_database,
208+
mock_connection,
205209
Arc::new(workspace_service_mock),
206210
Arc::new(secret_service_mock),
207211
Arc::new(parameter_service_mock),
@@ -216,7 +220,9 @@ mod test {
216220
async fn when_creating_workspace_failed_with_workspace_name_conflicted_use_case_should_returns_workspace_name_conflicted_err(
217221
) {
218222
const WORKSPACE_NAME: &str = "testworkspace";
219-
let mock_database = Arc::new(MockDatabase::new(DatabaseBackend::Postgres).into_connection());
223+
let mock_database = MockDatabase::new(DatabaseBackend::Postgres)
224+
.append_exec_results([MockExecResult { last_insert_id: 0, rows_affected: 1 }]);
225+
let mock_connection = Arc::new(mock_database.into_connection());
220226
let mut workspace_service_mock = MockWorkspaceService::new();
221227

222228
workspace_service_mock
@@ -228,7 +234,7 @@ mod test {
228234
let parameter_service_mock = MockParameterService::new();
229235

230236
let workspace_use_case = WorkspaceUseCaseImpl::new(
231-
mock_database,
237+
mock_connection,
232238
Arc::new(workspace_service_mock),
233239
Arc::new(secret_service_mock),
234240
Arc::new(parameter_service_mock),

0 commit comments

Comments
 (0)