@Test public void testNullGetTheDataSource() { EasyMock.expect(server.getDataSource("none")).andReturn(null).atLeastOnce(); EasyMock.expect(inventoryView.getInventory()).andReturn(ImmutableList.of(server)).atLeastOnce(); EasyMock.replay(inventoryView, server); DataSourcesResource DataSourcesResource = new DataSourcesResource(inventoryView, null, null, null, new AuthConfig(), null); Assert.assertEquals(204, DataSourcesResource.getTheDataSource("none", null).getStatus()); EasyMock.verify(inventoryView, server); }
private static ShardSpec shardSpec(String dimension, boolean contained) { ShardSpec shard = EasyMock.createMock(ShardSpec.class); EasyMock.expect(shard.getDomainDimensions()) .andReturn(ImmutableList.of(dimension)) .anyTimes(); EasyMock.expect(shard.possibleInDomain(EasyMock.anyObject())) .andReturn(contained) .anyTimes(); return shard; } }
@Test public void shouldPrintServerAddressWhenEmptyStringArg() throws Exception { expect(restClient.makeRootRequest()).andReturn(RestResponse.successful(SERVER_INFO)); expect(restClient.getServerAddress()).andReturn(new URI(INITIAL_SERVER_ADDRESS)); restClient.setServerAddress(anyString()); expectLastCall().andThrow(new AssertionError("should not set address")); replay(restClient); command.execute(ImmutableList.of(), terminal); assertThat(out.toString(), equalTo(INITIAL_SERVER_ADDRESS + "\n")); }
private void expectAuthorizationTokenCheck() { AuthenticationResult authenticationResult = new AuthenticationResult("druid", "druid", null, null); EasyMock.expect(req.getAttribute(AuthConfig.DRUID_ALLOW_UNSECURED_PATH)).andReturn(null).anyTimes(); EasyMock.expect(req.getAttribute(AuthConfig.DRUID_AUTHORIZATION_CHECKED)).andReturn(null).atLeastOnce(); EasyMock.expect(req.getAttribute(AuthConfig.DRUID_AUTHENTICATION_RESULT)) .andReturn(authenticationResult) .atLeastOnce(); req.setAttribute(AuthConfig.DRUID_AUTHORIZATION_CHECKED, false); EasyMock.expectLastCall().anyTimes(); req.setAttribute(AuthConfig.DRUID_AUTHORIZATION_CHECKED, true); EasyMock.expectLastCall().anyTimes(); }
private void mockEmptyPeon() { EasyMock.expect(mockPeon.getSegmentsToLoad()).andReturn(new HashSet<>()).anyTimes(); EasyMock.expect(mockPeon.getSegmentsMarkedToDrop()).andReturn(new HashSet<>()).anyTimes(); EasyMock.expect(mockPeon.getLoadQueueSize()).andReturn(0L).atLeastOnce(); EasyMock.expect(mockPeon.getNumberOfSegmentsInQueue()).andReturn(0).anyTimes(); EasyMock.replay(mockPeon); }
@Test public void testPlayback() throws IOException { Scene scene = EasyMock.createMock(Scene.class); EasyMock.expect(scene.isReadable()).andReturn(true).anyTimes(); EasyMock.expect(scene.isSequential()).andReturn(false).anyTimes(); SceneWriter sceneWriter = EasyMock.createStrictMock(SceneWriter.class); RecordedHttpExchange recordedHttpExchange1 = EasyMock.createStrictMock(RecordedHttpExchange.class); RecordedHttpExchange recordedHttpExchange2 = EasyMock.createStrictMock(RecordedHttpExchange.class); RecordedHttpRequest recordedHttpRequest1 = EasyMock.createStrictMock(RecordedHttpRequest.class); RecordedHttpRequest recordedHttpRequest2 = EasyMock.createStrictMock(RecordedHttpRequest.class); RecordedHttpResponse recordedHttpResponse1 = EasyMock.createStrictMock(RecordedHttpResponse.class); RecordedHttpResponse recordedHttpResponse2 = EasyMock.createStrictMock(RecordedHttpResponse.class); EasyMock.expect(recordedHttpExchange1.getRecordedHttpRequest()).andReturn(recordedHttpRequest1); EasyMock.expect(recordedHttpExchange2.getRecordedHttpRequest()).andReturn(recordedHttpRequest2); EasyMock.expect(recordedHttpExchange1.getRecordedHttpResponse()).andReturn(recordedHttpResponse1); EasyMock.expect(recordedHttpExchange2.getRecordedHttpResponse()).andReturn(recordedHttpResponse2); RecordedHttpRequest incomingHttpRequest = EasyMock.createStrictMock(RecordedHttpRequest.class); ArrayList<RecordedHttpExchange> recordedHttpExchangeArrayList = new ArrayList<>(); recordedHttpExchangeArrayList.add(recordedHttpExchange1); recordedHttpExchangeArrayList.add(recordedHttpExchange2); EasyMock.expect(scene.getRecordedHttpExchangeList()).andReturn(recordedHttpExchangeArrayList).times(2); MatchRule matchRule = EasyMock.createStrictMock(MatchRule.class); EasyMock.expect(matchRule.test(incomingHttpRequest, recordedHttpRequest1)).andReturn(true); EasyMock.replay(scene, recordedHttpExchange1, recordedHttpRequest1, matchRule); SceneAccessLayer sceneAccessLayer = new SceneAccessLayer(scene, sceneWriter, matchRule); Assert.assertEquals(sceneAccessLayer.playback(incomingHttpRequest), recordedHttpResponse1); EasyMock.verify(scene, recordedHttpExchange1, recordedHttpRequest1, matchRule); }
List<String> tasksIds = ImmutableList.of("id_1", "id_2"); EasyMock.<Collection<? extends TaskRunnerWorkItem>>expect(taskRunner.getRunningTasks()).andReturn( ImmutableList.of( new MockTaskRunnerWorkItem(tasksIds.get(0), null), new MockTaskRunnerWorkItem(tasksIds.get(1), null) EasyMock.expect(taskStorageQueryAdapter.getActiveTaskInfo(null)).andStubReturn( ImmutableList.of( new TaskInfo( "id_1", EasyMock.replay(taskRunner, taskMaster, taskStorageQueryAdapter, indexerMetadataStorageAdapter, req);
@Test(expectedExceptions = IllegalStateException.class, expectedExceptionsMessageRegExp = SceneAccessLayer.THE_SCENE_IS_NOT_READABLE) public void testPlaybackNotReadable() throws IOException { Scene scene = EasyMock.createStrictMock(Scene.class); EasyMock.expect(scene.isReadable()).andReturn(false); SceneWriter sceneWriter = EasyMock.createStrictMock(SceneWriter.class); MatchRule matchRule = EasyMock.createStrictMock(MatchRule.class); RecordedHttpRequest incomingHttpRequest = EasyMock.createStrictMock(RecordedHttpRequest.class); EasyMock.replay(scene); SceneAccessLayer sceneAccessLayer = new SceneAccessLayer(scene, sceneWriter, matchRule); sceneAccessLayer.playback(incomingHttpRequest); EasyMock.verify(scene); }
@Test public void testGetSpecificIntervals() { EasyMock.expect(inventoryView.getInventory()).andReturn( ImmutableList.of(server) ).atLeastOnce(); EasyMock.expect(request.getAttribute(AuthConfig.DRUID_ALLOW_UNSECURED_PATH)).andReturn(null).once(); EasyMock.expect(request.getAttribute(AuthConfig.DRUID_AUTHORIZATION_CHECKED)).andReturn(null).once(); EasyMock.expect(request.getAttribute(AuthConfig.DRUID_AUTHENTICATION_RESULT)).andReturn( new AuthenticationResult("druid", "druid", null, null) ).once(); request.setAttribute(AuthConfig.DRUID_AUTHORIZATION_CHECKED, true); EasyMock.expectLastCall().times(1); EasyMock.replay(inventoryView, request); IntervalsResource intervalsResource = new IntervalsResource( inventoryView, new AuthConfig(), AuthTestUtils.TEST_AUTHORIZER_MAPPER ); Response response = intervalsResource.getSpecificIntervals("2010-01-01T00:00:00.000Z/P1D", null, null, request); Map<String, Object> actualIntervals = (Map) response.getEntity(); Assert.assertEquals(2, actualIntervals.size()); Assert.assertEquals(25L, actualIntervals.get("size")); Assert.assertEquals(2, actualIntervals.get("count")); }
@Test public void testListEmptyImagesForProject() { compute = options.getService(); ImmutableList<com.google.api.services.compute.model.Image> images = ImmutableList.of(); Tuple<String, Iterable<com.google.api.services.compute.model.Image>> result = Tuple.<String, Iterable<com.google.api.services.compute.model.Image>>of(null, images); EasyMock.expect(computeRpcMock.listImages("otherProject", EMPTY_RPC_OPTIONS)).andReturn(result); EasyMock.replay(computeRpcMock); Page<Image> page = compute.listImages("otherProject"); assertNull(page.getNextPageToken()); assertArrayEquals(images.toArray(), Iterables.toArray(page.getValues(), Image.class)); }
@Test public void testCreateSceneExistInSequentialRecordMode() throws Exception { Scene expectedScene = EasyMock.createStrictMock(Scene.class); List<RecordedHttpExchange> recordedHttpExchanges = new ArrayList<>(); Date now = new Date(); recordedHttpExchanges.add(new RecordedHttpExchange(null, null, now)); EasyMock.expect(expectedScene.getRecordedHttpExchangeList()).andReturn(recordedHttpExchanges); EasyMock.replay(expectedScene); Scene result = runTestGetResult(SceneMode.SEQUENTIAL_RECORD, expectedScene); Assert.assertNotNull(result); Assert.assertEquals(result.getName(), NAME); Assert.assertEquals(result.getSceneRoot(), ROOT); Assert.assertFalse(result.isReadable()); // Creating a Scene in sequential record mode should clear the list of exchanges to allow for re-recording Assert.assertEquals(result.getRecordedHttpExchangeList().size(), 0); }
protected DatastoreBatchWriter() { super("test"); datastore = EasyMock.createMock(Datastore.class); IncompleteKey[] expected = {INCOMPLETE_KEY, INCOMPLETE_KEY}; List<Key> result = ImmutableList.of(KEY2, KEY3); expect(datastore.allocateId(expected)).andReturn(result).times(0, 1); replay(datastore); }
TaskQueue mockQueue = EasyMock.createMock(TaskQueue.class); EasyMock.expect(taskMaster.isLeader()).andReturn(true).anyTimes(); EasyMock.expect(taskMaster.getTaskRunner()).andReturn( Optional.of(taskRunner) ).anyTimes(); EasyMock.expect(taskMaster.getTaskQueue()).andReturn( Optional.of(mockQueue) ).anyTimes(); EasyMock.expect(taskStorageQueryAdapter.getActiveTaskInfo("datasource")).andStubReturn(ImmutableList.of( new TaskInfo( "id_1",
@Test public void testListEmptyNetworks() { compute = options.getService(); ImmutableList<com.google.api.services.compute.model.Network> networks = ImmutableList.of(); Tuple<String, Iterable<com.google.api.services.compute.model.Network>> result = Tuple.<String, Iterable<com.google.api.services.compute.model.Network>>of(null, networks); EasyMock.expect(computeRpcMock.listNetworks(EMPTY_RPC_OPTIONS)).andReturn(result); EasyMock.replay(computeRpcMock); Page<Network> page = compute.listNetworks(); assertNull(page.getNextPageToken()); assertArrayEquals(networks.toArray(), Iterables.toArray(page.getValues(), Network.class)); }
@Test public void testListEmptyRegions() { ImmutableList<com.google.api.services.compute.model.Region> regions = ImmutableList.of(); Tuple<String, Iterable<com.google.api.services.compute.model.Region>> result = Tuple.<String, Iterable<com.google.api.services.compute.model.Region>>of(null, regions); EasyMock.expect(computeRpcMock.listRegions(EMPTY_RPC_OPTIONS)).andReturn(result); EasyMock.replay(computeRpcMock); compute = options.getService(); Page<Region> page = compute.listRegions(); assertNull(page.getNextPageToken()); assertArrayEquals(regions.toArray(), Iterables.toArray(page.getValues(), Region.class)); }
@Test public void testAggregatedListEmptyAddresses() { ImmutableList<com.google.api.services.compute.model.Address> addresses = ImmutableList.of(); Tuple<String, Iterable<com.google.api.services.compute.model.Address>> result = Tuple.<String, Iterable<com.google.api.services.compute.model.Address>>of(null, addresses); EasyMock.expect(computeRpcMock.listAddresses(EMPTY_RPC_OPTIONS)).andReturn(result); EasyMock.replay(computeRpcMock); compute = options.getService(); Page<Address> page = compute.listAddresses(); assertNull(page.getNextPageToken()); assertArrayEquals(addresses.toArray(), Iterables.toArray(page.getValues(), Address.class)); }
@Test public void testListEmptySnapshots() { compute = options.getService(); ImmutableList<com.google.api.services.compute.model.Snapshot> snapshots = ImmutableList.of(); Tuple<String, Iterable<com.google.api.services.compute.model.Snapshot>> result = Tuple.<String, Iterable<com.google.api.services.compute.model.Snapshot>>of(null, snapshots); EasyMock.expect(computeRpcMock.listSnapshots(EMPTY_RPC_OPTIONS)).andReturn(result); EasyMock.replay(computeRpcMock); Page<Snapshot> page = compute.listSnapshots(); assertNull(page.getNextPageToken()); assertArrayEquals(snapshots.toArray(), Iterables.toArray(page.getValues(), Snapshot.class)); }
@Test public void testListEmptyGlobalAddresses() { ImmutableList<com.google.api.services.compute.model.Address> addresses = ImmutableList.of(); Tuple<String, Iterable<com.google.api.services.compute.model.Address>> result = Tuple.<String, Iterable<com.google.api.services.compute.model.Address>>of(null, addresses); EasyMock.expect(computeRpcMock.listGlobalAddresses(EMPTY_RPC_OPTIONS)).andReturn(result); EasyMock.replay(computeRpcMock); compute = options.getService(); Page<Address> page = compute.listGlobalAddresses(); assertNull(page.getNextPageToken()); assertArrayEquals(addresses.toArray(), Iterables.toArray(page.getValues(), Address.class)); }
@Test public void testListEmptyZones() { ImmutableList<com.google.api.services.compute.model.Zone> zones = ImmutableList.of(); Tuple<String, Iterable<com.google.api.services.compute.model.Zone>> result = Tuple.<String, Iterable<com.google.api.services.compute.model.Zone>>of(null, zones); EasyMock.expect(computeRpcMock.listZones(EMPTY_RPC_OPTIONS)).andReturn(result); EasyMock.replay(computeRpcMock); compute = options.getService(); Page<Zone> page = compute.listZones(); assertNull(page.getNextPageToken()); assertArrayEquals(zones.toArray(), Iterables.toArray(page.getValues(), Zone.class)); }
@Test public void testAggregatedListEmptyDisks() { compute = options.getService(); ImmutableList<com.google.api.services.compute.model.Disk> diskList = ImmutableList.of(); Tuple<String, Iterable<com.google.api.services.compute.model.Disk>> result = Tuple.<String, Iterable<com.google.api.services.compute.model.Disk>>of(null, diskList); EasyMock.expect(computeRpcMock.listDisks(EMPTY_RPC_OPTIONS)).andReturn(result); EasyMock.replay(computeRpcMock); Page<Disk> page = compute.listDisks(); assertNull(page.getNextPageToken()); assertArrayEquals(diskList.toArray(), Iterables.toArray(page.getValues(), Disk.class)); }