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(); }
@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 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 testFullGetTheDataSource() { DruidDataSource dataSource1 = new DruidDataSource("datasource1", new HashMap<>()); EasyMock.expect(server.getDataSource("datasource1")).andReturn(dataSource1).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); Response response = DataSourcesResource.getTheDataSource("datasource1", "full"); ImmutableDruidDataSource result = (ImmutableDruidDataSource) response.getEntity(); Assert.assertEquals(200, response.getStatus()); Assert.assertEquals(dataSource1.toImmutableDruidDataSource(), result); EasyMock.verify(inventoryView, server); }
/** * Test method for {@link SystemTime#asDate()}. */ public void testAsDate() { expect(mockTimeSource.timeInMillis()).andReturn(1100L).atLeastOnce(); replay(mockTimeSource); SystemTime.setGlobalTimeSource(mockTimeSource); assertEquals(1100L, SystemTime.asDate().getTime()); verify(); }
/** * Test method for {@link SystemTime#asCalendar()}. */ public void testAsCalendar() { expect(mockTimeSource.timeInMillis()).andReturn(1200L).atLeastOnce(); replay(mockTimeSource); SystemTime.setGlobalTimeSource(mockTimeSource); assertEquals(1200L, SystemTime.asCalendar().getTimeInMillis()); verify(); }
@Test public void testGetSimpleQueryableDataSources() EasyMock.expect(server.getDataSources()).andReturn(listDataSources).atLeastOnce(); EasyMock.expect(server.getDataSource("datasource1")).andReturn(listDataSources.get(0)).atLeastOnce(); EasyMock.expect(server.getTier()).andReturn(null).atLeastOnce(); EasyMock.expect(server.getDataSource("datasource2")).andReturn(listDataSources.get(1)).atLeastOnce(); 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) ).atLeastOnce(); request.setAttribute(AuthConfig.DRUID_AUTHORIZATION_CHECKED, true); EasyMock.expectLastCall().times(1); EasyMock.replay(inventoryView, server, request); DataSourcesResource DataSourcesResource = new DataSourcesResource( inventoryView, index++; EasyMock.verify(inventoryView, server);
@Test public void testSimpleGetTheDataSourceManyTiers() EasyMock.expect(server.getDataSource("datasource1")).andReturn( listDataSources.get(0) ).atLeastOnce(); EasyMock.expect(server.getTier()).andReturn("cold").atLeastOnce(); EasyMock.expect(server2.getDataSource("datasource1")).andReturn( listDataSources.get(1) ).atLeastOnce(); EasyMock.expect(server2.getTier()).andReturn("hot").atLeastOnce(); EasyMock.expect(server3.getDataSource("datasource1")).andReturn( listDataSources.get(1) ).atLeastOnce(); EasyMock.expect(server3.getTier()).andReturn("cold").atLeastOnce(); EasyMock.expect(inventoryView.getInventory()).andReturn( ImmutableList.of(server, server2, server3) ).atLeastOnce(); EasyMock.replay(inventoryView, server, server2, server3); DataSourcesResource DataSourcesResource = new DataSourcesResource(inventoryView, null, null, null, new AuthConfig(), null); Assert.assertEquals(2, result.get("segments").get("count")); Assert.assertEquals(30L, result.get("segments").get("size")); EasyMock.verify(inventoryView, server, server2, server3);
@Test public void testSimpleGetTheDataSource() { DruidDataSource dataSource1 = new DruidDataSource("datasource1", new HashMap<>()); dataSource1.addSegment( new DataSegment("datasegment1", Intervals.of("2010-01-01/P1D"), "", null, null, null, null, 0x9, 10) ); EasyMock.expect(server.getDataSource("datasource1")).andReturn(dataSource1).atLeastOnce(); EasyMock.expect(server.getTier()).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); Response response = DataSourcesResource.getTheDataSource("datasource1", null); Assert.assertEquals(200, response.getStatus()); Map<String, Map<String, Object>> result = (Map<String, Map<String, Object>>) response.getEntity(); Assert.assertEquals(1, ((Map) (result.get("tiers").get(null))).get("segmentCount")); Assert.assertEquals(10L, ((Map) (result.get("tiers").get(null))).get("size")); Assert.assertNotNull(result.get("segments")); Assert.assertEquals("2010-01-01T00:00:00.000Z", result.get("segments").get("minTime").toString()); Assert.assertEquals("2010-01-02T00:00:00.000Z", result.get("segments").get("maxTime").toString()); Assert.assertEquals(1, result.get("segments").get("count")); Assert.assertEquals(10L, result.get("segments").get("size")); EasyMock.verify(inventoryView, server); }
/** * Test method for {@link SystemTime#setGlobalTimeSource(TimeSource)}. */ public void testSetGlobalTimeSource() { expect(mockTimeSource.timeInMillis()).andReturn(100L).atLeastOnce(); replay(mockTimeSource); SystemTime.setGlobalTimeSource(mockTimeSource); assertEquals(100L, SystemTime.asMillis()); verify(); }
/** * Test method for {@link SystemTime#asMillis()}. */ public void testAsMillis() { expect(mockTimeSource.timeInMillis()).andReturn(1000L).atLeastOnce(); replay(mockTimeSource); SystemTime.setGlobalTimeSource(mockTimeSource); assertEquals(1000L, SystemTime.asMillis()); verify(); }
/** * Test method for {@link SystemTime#setLocalTimeSource(TimeSource)}. */ public void testSetLocalTimeSource() { expect(mockTimeSource.timeInMillis()).andReturn(300L).atLeastOnce(); replay(mockTimeSource); SystemTime.setLocalTimeSource(mockTimeSource); assertEquals(300L, SystemTime.asMillis()); verify(); }
/** * Test method for {@link SystemTime#asDate(boolean)}. */ public void testAsDateBoolean() { Calendar cal = new GregorianCalendar(2010, 1, 2, 3, 4, 5); cal.set(Calendar.MILLISECOND, 3); // Add a few milliseconds for good measure expect(mockTimeSource.timeInMillis()).andReturn(cal.getTimeInMillis()).atLeastOnce(); replay(mockTimeSource); SystemTime.setGlobalTimeSource(mockTimeSource); Calendar calMidnight = new GregorianCalendar(2010, 1, 2, 0, 0, 0); calMidnight.set(Calendar.MILLISECOND, 0); assertEquals(calMidnight.getTimeInMillis(), SystemTime.asDate(false).getTime()); assertEquals(cal.getTimeInMillis(), SystemTime.asDate(true).getTime()); verify(); } }
/** * Test method for {@link SystemTime#asMillis(boolean)}. */ public void testAsMillisBoolean() { Calendar cal = new GregorianCalendar(2010, 1, 2, 3, 4, 5); long timeInMillis = cal.getTimeInMillis() + 3; // Add a few milliseconds for good measure expect(mockTimeSource.timeInMillis()).andReturn(timeInMillis).atLeastOnce(); replay(mockTimeSource); SystemTime.setGlobalTimeSource(mockTimeSource); Calendar calMidnight = new GregorianCalendar(2010, 1, 2, 0, 0, 0); calMidnight.set(Calendar.MILLISECOND, 0); assertEquals(calMidnight.getTimeInMillis(), SystemTime.asMillis(false)); assertEquals(timeInMillis, SystemTime.asMillis(true)); verify(); }
/** * Test method for {@link SystemTime#asCalendar(boolean)}. */ public void testAsCalendarBoolean() { Calendar cal = new GregorianCalendar(2010, 1, 2, 3, 4, 5); cal.set(Calendar.MILLISECOND, 3); // Add a few milliseconds for good measure expect(mockTimeSource.timeInMillis()).andReturn(cal.getTimeInMillis()).atLeastOnce(); replay(mockTimeSource); SystemTime.setGlobalTimeSource(mockTimeSource); Calendar calMidnight = new GregorianCalendar(2010, 1, 2, 0, 0, 0); calMidnight.set(Calendar.MILLISECOND, 0); assertEquals(calMidnight, SystemTime.asCalendar(false)); assertEquals(cal, SystemTime.asCalendar(true)); verify(); }
@Test public void testFullGetSpecificIntervals() { 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); List<Interval> expectedIntervals = new ArrayList<>(); expectedIntervals.add(Intervals.of("2010-01-01T00:00:00.000Z/2010-01-02T00:00:00.000Z")); IntervalsResource intervalsResource = new IntervalsResource( inventoryView, new AuthConfig(), AuthTestUtils.TEST_AUTHORIZER_MAPPER ); Response response = intervalsResource.getSpecificIntervals("2010-01-01T00:00:00.000Z/P1D", null, "full", request); TreeMap<Interval, Map<String, Map<String, Object>>> actualIntervals = (TreeMap) response.getEntity(); Assert.assertEquals(1, actualIntervals.size()); Assert.assertEquals(expectedIntervals.get(0), actualIntervals.firstKey()); Assert.assertEquals(20L, actualIntervals.get(expectedIntervals.get(0)).get("datasource1").get("size")); Assert.assertEquals(1, actualIntervals.get(expectedIntervals.get(0)).get("datasource1").get("count")); Assert.assertEquals(5L, actualIntervals.get(expectedIntervals.get(0)).get("datasource2").get("size")); Assert.assertEquals(1, actualIntervals.get(expectedIntervals.get(0)).get("datasource2").get("count")); }
private void mockDruidServer( ImmutableDruidServer druidServer, String name, String tier, long currentSize, long maxSize, List<DataSegment> segments ) { EasyMock.expect(druidServer.getName()).andReturn(name).anyTimes(); EasyMock.expect(druidServer.getTier()).andReturn(tier).anyTimes(); EasyMock.expect(druidServer.getCurrSize()).andReturn(currentSize).atLeastOnce(); EasyMock.expect(druidServer.getMaxSize()).andReturn(maxSize).atLeastOnce(); EasyMock.expect(druidServer.getSegments()).andReturn(segments).anyTimes(); if (!segments.isEmpty()) { segments.forEach( s -> EasyMock.expect(druidServer.getSegment(s.getId())).andReturn(s).anyTimes() ); } else { EasyMock.expect(druidServer.getSegment(EasyMock.anyObject())).andReturn(null).anyTimes(); } EasyMock.replay(druidServer); }
EasyMock.expect(taskMaster.getSupervisorManager()).andReturn(Optional.of(supervisorManager)); EasyMock.expect(supervisorManager.getSupervisorIds()).andReturn(supervisorIds).atLeastOnce(); EasyMock.expect(supervisorManager.getSupervisorSpec("id1")).andReturn(Optional.of(spec1)).times(2); EasyMock.expect(supervisorManager.getSupervisorSpec("id2")).andReturn(Optional.of(spec2)).times(2); EasyMock.expect(request.getAttribute(AuthConfig.DRUID_ALLOW_UNSECURED_PATH)).andReturn(null).atLeastOnce(); EasyMock.expect(request.getAttribute(AuthConfig.DRUID_AUTHORIZATION_CHECKED)).andReturn(null).atLeastOnce(); EasyMock.expect(request.getAttribute(AuthConfig.DRUID_AUTHENTICATION_RESULT)).andReturn( new AuthenticationResult("druid", "druid", null, null) ).atLeastOnce(); request.setAttribute(AuthConfig.DRUID_AUTHORIZATION_CHECKED, true); EasyMock.expectLastCall().anyTimes();
@Test public void testScale() RunInstancesResult runInstancesResult = EasyMock.createMock(RunInstancesResult.class); ); EasyMock.expect(runInstancesResult.getReservation()).andReturn(reservation).atLeastOnce(); EasyMock.replay(runInstancesResult); EasyMock.expect(describeInstancesResult.getReservations()) .andReturn(Collections.singletonList(reservation)) .atLeastOnce(); EasyMock.replay(describeInstancesResult); EasyMock.expect(reservation.getInstances()).andReturn(Collections.singletonList(instance)).atLeastOnce(); EasyMock.replay(reservation); Assert.assertEquals(INSTANCE_ID, deleted.getNodeIds().get(0)); EasyMock.verify(runInstancesResult);
@Test public void testGetIntervals() 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) request.setAttribute(AuthConfig.DRUID_AUTHORIZATION_CHECKED, true); EasyMock.expectLastCall().times(1); EasyMock.replay(inventoryView, request);