-
- WatchdogDistributionStatusDb wdds = Mockito.mock(WatchdogDistributionStatusDb.class);
- Mockito.when(wdds.getWatchdogDistributionIdStatus("ff305d54-75b4-431b-adb2-eb6b9e5ff001")).thenReturn("ff3514e3-5a33-55df-13ab-12abad84e7ff");
- String actual = wdds.getWatchdogDistributionIdStatus("ff305d54-75b4-431b-adb2-eb6b9e5ff001");
- assertEquals(actual, distributionId);
- verify(wdds, times(1)).getWatchdogDistributionIdStatus(any(String.class));
+ WatchdogDistributionStatusDb wds = new WatchdogDistributionStatusDb(this.sessionFactoryRequest);
+ Query mockQuery = mock(Query.class);
+ when(session.createQuery(any(String.class))).thenReturn(mockQuery);
+ when(mockQuery.uniqueResult()).thenReturn("myValue");
+ when(session.isOpen()).thenReturn(true);
+ assertEquals("myValue", wds.getWatchdogDistributionIdStatus("test"));
+ }
+
+ @Test
+ public void testGetWatchdogDistributionIdStatusNotFound() {
+ WatchdogDistributionStatusDb wds = new WatchdogDistributionStatusDb(this.sessionFactoryRequest);
+ Query mockQuery = mock(Query.class);
+ when(session.createQuery(any(String.class))).thenReturn(mockQuery);
+ when(mockQuery.uniqueResult()).thenReturn(null);
+ when(session.isOpen()).thenReturn(true);
+ assertEquals(null, wds.getWatchdogDistributionIdStatus("test"));