|
3 | 3 | import static org.junit.jupiter.api.Assertions.assertEquals;
|
4 | 4 | import static org.junit.jupiter.api.Assertions.assertNotNull;
|
5 | 5 | import static org.junit.jupiter.api.Assertions.assertTrue;
|
| 6 | +import static org.mockito.ArgumentMatchers.argThat; |
6 | 7 | import static org.mockito.Mockito.eq;
|
7 | 8 | import static org.mockito.Mockito.mock;
|
8 | 9 | import static org.mockito.Mockito.when;
|
9 | 10 | import static org.ovirt.engine.api.restapi.test.util.TestHelper.eqParams;
|
10 | 11 |
|
11 | 12 | import java.util.ArrayList;
|
| 13 | +import java.util.Date; |
12 | 14 | import java.util.HashMap;
|
13 | 15 | import java.util.List;
|
14 | 16 | import java.util.Locale;
|
|
22 | 24 | import org.junit.jupiter.api.AfterEach;
|
23 | 25 | import org.junit.jupiter.api.BeforeEach;
|
24 | 26 | import org.junit.jupiter.api.Test;
|
| 27 | +import org.mockito.ArgumentMatcher; |
25 | 28 | import org.ovirt.engine.api.model.Api;
|
26 | 29 | import org.ovirt.engine.api.model.Link;
|
27 | 30 | import org.ovirt.engine.api.model.SpecialObjects;
|
28 | 31 | import org.ovirt.engine.api.restapi.invocation.Current;
|
29 | 32 | import org.ovirt.engine.api.restapi.invocation.CurrentManager;
|
30 | 33 | import org.ovirt.engine.api.restapi.invocation.VersionSource;
|
31 | 34 | import org.ovirt.engine.api.restapi.logging.MessageBundle;
|
| 35 | +import org.ovirt.engine.core.common.businessentities.EngineBackupLog; |
| 36 | +import org.ovirt.engine.core.common.businessentities.EngineBackupScope; |
32 | 37 | import org.ovirt.engine.core.common.businessentities.aaa.DbUser;
|
33 | 38 | import org.ovirt.engine.core.common.config.ConfigValues;
|
34 | 39 | import org.ovirt.engine.core.common.interfaces.BackendLocal;
|
35 | 40 | import org.ovirt.engine.core.common.mode.ApplicationMode;
|
36 | 41 | import org.ovirt.engine.core.common.queries.GetConfigurationValueParameters;
|
| 42 | +import org.ovirt.engine.core.common.queries.GetLastEngineBackupParameters; |
37 | 43 | import org.ovirt.engine.core.common.queries.GetSystemStatisticsQueryParameters;
|
38 | 44 | import org.ovirt.engine.core.common.queries.QueryParametersBase;
|
39 | 45 | import org.ovirt.engine.core.common.queries.QueryReturnValue;
|
@@ -306,6 +312,7 @@ private void setupExpectations(ApplicationMode appMode) {
|
306 | 312 | setUpGetInstanceIdExpectations();
|
307 | 313 | setUpGetUserBySessionExpectations();
|
308 | 314 | setUpGetSystemStatisticsExpectations();
|
| 315 | + setUpGetLastDbBackendExpectations(); |
309 | 316 | }
|
310 | 317 |
|
311 | 318 | protected void doTestGlusterOnlyGet() {
|
@@ -465,6 +472,22 @@ private QueryParametersBase getProductVersionParams() {
|
465 | 472 | return eqParams(QueryParametersBase.class, new String[0], new Object[0]);
|
466 | 473 | }
|
467 | 474 |
|
| 475 | + protected void setUpGetLastDbBackendExpectations() { |
| 476 | + QueryReturnValue queryResult = new QueryReturnValue(); |
| 477 | + queryResult.setSucceeded(true); |
| 478 | + EngineBackupLog emptyReturn = new EngineBackupLog(); |
| 479 | + emptyReturn.setDoneAt(new Date()); |
| 480 | + emptyReturn.setScope(EngineBackupScope.DB.getName()); |
| 481 | + queryResult.setReturnValue(emptyReturn); |
| 482 | + when(backend.runQuery(eq(QueryType.GetLastEngineBackup), argThat(new ArgumentMatcher<GetLastEngineBackupParameters>() { |
| 483 | + @Override |
| 484 | + public boolean matches(GetLastEngineBackupParameters argument) { |
| 485 | + return argument != null && argument.getEngineBackupScope() != null; // Match any non-null EngineBackupScope |
| 486 | + } |
| 487 | + } |
| 488 | + ))).thenReturn(queryResult); |
| 489 | + } |
| 490 | + |
468 | 491 | protected void setUpGetSystemStatisticsExpectations() {
|
469 | 492 | QueryReturnValue queryResult = new QueryReturnValue();
|
470 | 493 |
|
|
0 commit comments