private CreateStreamAsSelect mockCSAS(final String name) { final CreateStreamAsSelect mockStatement = mock(CreateStreamAsSelect.class); expect(mockStatement.getName()).andStubReturn(QualifiedName.of(name)); expect(mockStatement.getQuery()).andStubReturn(mockCSASQuery(name)); expect(mockStatement.getProperties()).andStubReturn(Collections.emptyMap()); expect(mockStatement.getPartitionByColumn()).andStubReturn(Optional.empty()); return mockStatement; }
private Query mockCSASQuery(final String name) { final Query mockQuery = mock(Query.class); final QuerySpecification mockQuerySpec = mock(QuerySpecification.class); final Table mockRelation = mock(Table.class); expect(mockQuery.getQueryBody()).andStubReturn(mockQuerySpec); expect(mockQuery.getLimit()).andStubReturn(Optional.empty()); expect(mockQuerySpec.getInto()).andStubReturn(mockRelation); expect(mockRelation.getName()).andStubReturn(QualifiedName.of(name)); return mockQuery; }
private Statement mockTerminate(final QueryId queryId) { final TerminateQuery mockStatement = mock(TerminateQuery.class); expect(mockStatement.getQueryId()).andStubReturn(queryId); return mockStatement; }
private static void createStubs(Vector v, Vector realV) { expect(v.getLookupCost()) .andStubReturn(realV instanceof SequentialAccessSparseVector ? Math.round(Math.log(1000)) : realV.getLookupCost()); expect(v.getIteratorAdvanceCost()) .andStubReturn(realV.getIteratorAdvanceCost()); expect(v.isAddConstantTime()) .andStubReturn(realV.isAddConstantTime()); expect(v.isSequentialAccess()) .andStubReturn(realV.isSequentialAccess()); expect(v.isDense()) .andStubReturn(realV.isDense()); expect(v.getNumNondefaultElements()) .andStubReturn(realV.isDense() ? realV.size() : 1000); expect(v.size()) .andStubReturn(realV.size()); }
private DropStream mockDropStream(final String name) { final DropStream mockDropStream = mock(DropStream.class); expect(mockDropStream.getName()).andStubReturn(QualifiedName.of(name)); expect(mockDropStream.getStreamName()).andStubReturn(QualifiedName.of(name)); expect(mockParser.parseSingleStatement("DROP")) .andReturn(new PreparedStatement<>("DROP", mockDropStream)); return mockDropStream; }
private static void createStubs(Vector v, Vector realV) { expect(v.getLookupCost()) .andStubReturn(realV instanceof SequentialAccessSparseVector ? Math.round(Math.log(1000)) : realV.getLookupCost()); expect(v.getIteratorAdvanceCost()) .andStubReturn(realV.getIteratorAdvanceCost()); expect(v.isAddConstantTime()) .andStubReturn(realV.isAddConstantTime()); expect(v.isSequentialAccess()) .andStubReturn(realV.isSequentialAccess()); expect(v.isDense()) .andStubReturn(realV.isDense()); expect(v.getNumNondefaultElements()) .andStubReturn(realV.isDense() ? realV.size() : 1000); expect(v.size()) .andStubReturn(realV.size()); }
private PersistentQueryMetadata mockReplayCSAS( final String statement, final String name, final QueryId queryId) { final CreateStreamAsSelect mockCSAS = mockCSAS(name); final PersistentQueryMetadata mockQuery = mock(PersistentQueryMetadata.class); expect(mockQuery.getQueryId()).andStubReturn(queryId); final PreparedStatement<Statement> csas = new PreparedStatement<>("CSAS", mockCSAS); expect(mockParser.parseSingleStatement(statement)) .andReturn(csas); expect(mockMetaStore.getSource(name)).andStubReturn(null); expect(mockEngine.numberOfPersistentQueries()).andReturn(0L); expect(mockEngine.execute(eq(csas), anyObject(), anyObject())) .andReturn(Optional.of(mockQuery)); return mockQuery; }
EasyMock.expect(taskStorageQueryAdapter.getActiveTaskInfo(null)).andStubReturn( ImmutableList.of( new TaskInfo(
@Test public void shouldCascade4Dot1DropStreamCommand() { // Given: final DropStream mockDropStream = mockDropStream("foo"); expect(mockMetaStore.getSource("foo")) .andStubReturn(mock(StructuredDataSource.class)); expect(mockMetaStore.getQueriesWithSink("foo")) .andStubReturn(ImmutableSet.of("query-id")); expect(mockEngine.getMetaStore()).andStubReturn(mockMetaStore); expect(mockEngine.getPersistentQuery(new QueryId("query-id"))).andReturn(Optional.of(mockQueryMetadata)); mockQueryMetadata.close(); expectLastCall(); expect(mockEngine.executeDdlStatement("DROP", mockDropStream, Collections.emptyMap())) .andReturn(new DdlCommandResult(true, "SUCCESS")); replayAll(); // When: statementExecutorWithMocks.handleRestore( new QueuedCommand( new CommandId(Type.STREAM, "foo", Action.DROP), new Command("DROP", Collections.emptyMap(), PRE_VERSION_5_NULL_ORIGNAL_PROPS) ) ); // Then: verify(mockParser, mockEngine, mockMetaStore); }
Optional.of(mockQueue) ).anyTimes(); EasyMock.expect(taskStorageQueryAdapter.getActiveTaskInfo("datasource")).andStubReturn(ImmutableList.of( new TaskInfo( "id_1",
new MockTaskRunnerWorkItem(tasksIds.get(2), null))); EasyMock.expect(taskStorageQueryAdapter.getCompletedTaskInfoByCreatedTimeDuration(null, null, null)).andStubReturn( ImmutableList.of( new TaskInfo(
List<String> tasksIds = ImmutableList.of("id_1", "id_2", "id_3"); Duration duration = new Period("PT86400S").toStandardDuration(); EasyMock.expect(taskStorageQueryAdapter.getCompletedTaskInfoByCreatedTimeDuration(null, duration, null)).andStubReturn( ImmutableList.of( new TaskInfo(
EasyMock.expect(taskStorageQueryAdapter.getActiveTaskInfo("allow")).andStubReturn( ImmutableList.of( new TaskInfo(
EasyMock.expect(taskStorageQueryAdapter.getCompletedTaskInfoByCreatedTimeDuration(null, null, null)).andStubReturn( ImmutableList.of( new TaskInfo(
EasyMock.expect(taskStorageQueryAdapter.getCompletedTaskInfoByCreatedTimeDuration(null, null, null)).andStubReturn( ImmutableList.of( new TaskInfo(
EasyMock.expect(taskStorageQueryAdapter.getActiveTaskInfo(null)).andStubReturn( ImmutableList.of( new TaskInfo(
EasyMock.expect(taskStorageQueryAdapter.getActiveTaskInfo(null)).andStubReturn( ImmutableList.of( new TaskInfo(
EasyMock.expect(taskStorageQueryAdapter.getActiveTaskInfo(null)).andStubReturn( ImmutableList.of( new TaskInfo(
EasyMock.expect(taskStorageQueryAdapter.getCompletedTaskInfoByCreatedTimeDuration(null, null, null)).andStubReturn( ImmutableList.of( new TaskInfo( EasyMock.expect(taskStorageQueryAdapter.getActiveTaskInfo(null)).andStubReturn( ImmutableList.of( new TaskInfo(
EasyMock.expect(taskStorageQueryAdapter.getCompletedTaskInfoByCreatedTimeDuration(null, null, "allow")).andStubReturn( ImmutableList.of( new TaskInfo( EasyMock.expect(taskStorageQueryAdapter.getActiveTaskInfo("allow")).andStubReturn( ImmutableList.of( new TaskInfo(