Examples with SourceState com.airbnb.spinaltap.common.source.SourceState used on opensource projects

Example 21 with SourceState

use of com.airbnb.spinaltap.common.source.SourceState in project SpinalTap by airbnb.

the class AbstractMysqlSourceTest method testSaveState.

@Test
public void testSaveState() throws Exception {
    TestSource source = new TestSource();
    SourceState savedState = mock(SourceState.class);
    SourceState newState = mock(SourceState.class);
    when(stateRepository.read()).thenReturn(savedState);
    source.saveState(newState);
    verify(stateRepository, times(1)).save(newState);
    assertEquals(newState, source.getLastSavedState().get());
}