@Test public void testGroupEntitiesByService() throws EagleServiceClientException { List<TaggedLogAPIEntity> entities = new ArrayList<>(); TaggedLogAPIEntity testEntity1 = new TestTimeSeriesAPIEntity(); entities.add(testEntity1); EagleServiceBaseClient eagleServiceBaseClient = new EagleServiceClientImpl("localhost", 9090, "admin", "secret"); Map<String,List<TaggedLogAPIEntity>> serviceEntityMap = eagleServiceBaseClient.groupEntitiesByService(entities); System.out.println(serviceEntityMap); Set<String> keySet = serviceEntityMap.keySet(); for(Map.Entry<String, List<TaggedLogAPIEntity>> entry : serviceEntityMap.entrySet()) { Assert.assertEquals("TestTimeSeriesAPIEntity", entry.getKey()); Assert.assertEquals(1, entry.getValue().size()); } }
InternalLog internalLog = new InternalLog(); Map<String, byte[]> map = new HashMap<String, byte[]>(); TestTimeSeriesAPIEntity apiEntity = new TestTimeSeriesAPIEntity(); EntityDefinition ed = null; try {
@Test public void testGenericEntityServiceResourceupdateDatabase() throws IllegalDataStorageTypeException, IOException, IllegalAccessException, InstantiationException { TestTimeSeriesAPIEntity e = new TestTimeSeriesAPIEntity(); List<TestTimeSeriesAPIEntity> entities = new ArrayList<>(); entities.add(e); EntityDefinition ed = EntityDefinitionManager.getEntityByServiceName("TestTimeSeriesAPIEntity"); ModifyResult<String> modifyResult = new ModifyResult<>(); List<String> identifiers = new ArrayList<>(); identifiers.add("test"); modifyResult.setSuccess(true); modifyResult.setIdentifiers(identifiers); when(dataStorage.update(entities, ed)).thenReturn(modifyResult); GenericServiceAPIResponseEntity responseEntity = genericEntityServiceResource.updateDatabase(new UpdateStatement(entities, "TestTimeSeriesAPIEntity")); Assert.assertTrue(responseEntity.isSuccess()); Assert.assertTrue(responseEntity.getMeta().toString().startsWith("{totalResults=1")); Assert.assertEquals(String.class, responseEntity.getType()); Assert.assertEquals("test", responseEntity.getObj().get(0)); Assert.assertEquals(null, responseEntity.getException()); verify(dataStorage).update(entities, ed); }
private TestTimeSeriesAPIEntity newInstance(){ TestTimeSeriesAPIEntity instance = new TestTimeSeriesAPIEntity(); instance.setField1(123); instance.setField2(234); instance.setField3(1231312312l); instance.setField4(12312312312l); instance.setField5(123123.12312); instance.setField6(-12312312.012); instance.setField7(UUID.randomUUID().toString()); instance.setTags(new HashMap<String, String>() {{ put("cluster", "c4ut"); put("datacenter", "d4ut"); put("random",UUID.randomUUID().toString()); }}); instance.setTimestamp(System.currentTimeMillis()); return instance; } }
@Test public void testUnmarshalEntitiesByServie() throws NoSuchMethodException, InvocationTargetException, JsonProcessingException, IllegalAccessException, InstantiationException { TestTimeSeriesAPIEntity timeSeriesAPIEntity = new TestTimeSeriesAPIEntity(); timeSeriesAPIEntity.setTimestamp(System.currentTimeMillis()); timeSeriesAPIEntity.setField1(1);
private TestTimeSeriesAPIEntity newInstance() { TestTimeSeriesAPIEntity instance = new TestTimeSeriesAPIEntity(); instance.setField1(123); instance.setField2(234); instance.setField3(1231312312L); instance.setField4(12312312312L); instance.setField5(123123.12312); instance.setField6(-12312312.012); instance.setField7(UUID.randomUUID().toString()); instance.setTags(new HashMap<String, String>() { { put("cluster", "c4ut"); put("datacenter", "d4ut"); put("random", UUID.randomUUID().toString()); } }); instance.setTimestamp(System.currentTimeMillis()); return instance; }
private TestTimeSeriesAPIEntity newInstance() { TestTimeSeriesAPIEntity instance = new TestTimeSeriesAPIEntity(); instance.setField1(123); instance.setField2(234); instance.setField3(1231312312L); instance.setField4(12312312312L); instance.setField5(123123.12312); instance.setField6(-12312312.012); instance.setField7(UUID.randomUUID().toString()); instance.setTags(new HashMap<String, String>() { { put("cluster", "c4ut"); put("datacenter", "d4ut"); put("random", UUID.randomUUID().toString()); } }); instance.setTimestamp(System.currentTimeMillis()); return instance; }
private List<String> prepareData(int count) throws Exception { List<TestTimeSeriesAPIEntity> list = new ArrayList<TestTimeSeriesAPIEntity>(); if (ed == null) { EntityDefinitionManager.registerEntity(TestTimeSeriesAPIEntity.class); ed = EntityDefinitionManager.getEntityDefinitionByEntityClass(TestTimeSeriesAPIEntity.class); } for (int i = 0; i < count; i++) { TestTimeSeriesAPIEntity e = new TestTimeSeriesAPIEntity(); e.setTimestamp(System.currentTimeMillis()); e.setField1(1); e.setField2(2); e.setField3(3); e.setField4(4L); e.setField5(5.0); e.setField6(5.0); e.setField7("7"); e.setTags(new HashMap<String, String>()); e.getTags().put("cluster", "test4UT"); e.getTags().put("datacenter", "dc1"); e.getTags().put("index", "" + i); e.getTags().put("jobId", "job_" + System.currentTimeMillis()); list.add(e); } GenericEntityWriter writer = new GenericEntityWriter(ed.getService()); List<String> result = writer.write(list); return result; }
@Test public void testGenericEntityServiceResourceupdateDatabase1() throws IllegalDataStorageTypeException, IOException, IllegalAccessException, InstantiationException { TestTimeSeriesAPIEntity e = new TestTimeSeriesAPIEntity(); List<TestTimeSeriesAPIEntity> entities = new ArrayList<>(); entities.add(e); EntityDefinition ed = EntityDefinitionManager.getEntityByServiceName("TestTimeSeriesAPIEntity"); ModifyResult<String> modifyResult = new ModifyResult<>(); List<String> identifiers = null; modifyResult.setSuccess(true); modifyResult.setIdentifiers(identifiers); when(dataStorage.update(entities, ed)).thenReturn(modifyResult); GenericServiceAPIResponseEntity responseEntity = genericEntityServiceResource.updateDatabase(new UpdateStatement(entities, "TestTimeSeriesAPIEntity")); Assert.assertTrue(responseEntity.isSuccess()); Assert.assertTrue(responseEntity.getMeta().toString().startsWith("{totalResults=0")); Assert.assertEquals(null, responseEntity.getType()); Assert.assertEquals(null, responseEntity.getObj()); Assert.assertEquals(null, responseEntity.getException()); verify(dataStorage).update(entities, ed); }
@SuppressWarnings("unused") private List<String> prepareTestEntity(int count) throws Exception { List<TestTimeSeriesAPIEntity> list = new ArrayList<TestTimeSeriesAPIEntity>(); EntityDefinition ed = EntityDefinitionManager.getEntityDefinitionByEntityClass(TestTimeSeriesAPIEntity.class); if (ed == null) { EntityDefinitionManager.registerEntity(TestTimeSeriesAPIEntity.class); ed = EntityDefinitionManager.getEntityDefinitionByEntityClass(TestTimeSeriesAPIEntity.class); } for(int i=0;i<count;i++){ TestTimeSeriesAPIEntity e = new TestTimeSeriesAPIEntity(); e.setTimestamp(System.currentTimeMillis()); e.setField1(1); e.setField2(2); e.setField3(3); e.setField4(4L); e.setField5(5.0); e.setField6(5.0); e.setField7("7"); e.setTags(new HashMap<String, String>()); e.getTags().put("cluster", "test4UT"); e.getTags().put("datacenter", "dc1"); e.getTags().put("index", ""+i); e.getTags().put("jobId", "job_"+System.currentTimeMillis()); list.add(e); } GenericEntityWriter writer = new GenericEntityWriter(ed.getService()); List<String> result = writer.write(list); return result; }
@Test public void testGenericEntityServiceResourceSearch() throws IllegalAccessException, InstantiationException, IOException { List<TestTimeSeriesAPIEntity> result = new ArrayList<>(); TestTimeSeriesAPIEntity e = new TestTimeSeriesAPIEntity(); long timestamp = System.currentTimeMillis(); e.setTimestamp(timestamp);
@Test public void testGenericEntityServiceResourceupdateDatabaseFalse() throws IllegalDataStorageTypeException, IOException, IllegalAccessException, InstantiationException { TestTimeSeriesAPIEntity e = new TestTimeSeriesAPIEntity(); List<TestTimeSeriesAPIEntity> entities = new ArrayList<>(); entities.add(e); EntityDefinition ed = EntityDefinitionManager.getEntityByServiceName("TestTimeSeriesAPIEntity"); ModifyResult<String> modifyResult = new ModifyResult<>(); modifyResult.setSuccess(false); when(dataStorage.update(entities, ed)).thenReturn(modifyResult); GenericServiceAPIResponseEntity responseEntity = genericEntityServiceResource.updateDatabase(new UpdateStatement(entities, "TestTimeSeriesAPIEntity")); Assert.assertFalse(responseEntity.isSuccess()); Assert.assertEquals(null, responseEntity.getMeta()); Assert.assertEquals(null, responseEntity.getType()); Assert.assertEquals(null, responseEntity.getObj()); Assert.assertEquals(null, responseEntity.getException()); verify(dataStorage).update(entities, ed); }
private TestTimeSeriesAPIEntity newEntity(){ TestTimeSeriesAPIEntity entity = new TestTimeSeriesAPIEntity(); entity.setTimestamp(System.currentTimeMillis()); entity.setTags(new HashMap<String, String>() {{ put("cluster", "cluster4ut"); put("datacenter", "datacenter4ut"); }}); entity.setField1(1); entity.setField2(1); entity.setField3(1); entity.setField4(1l); entity.setField5(1l); entity.setField5(1.2); entity.setField6(-1.2); entity.setField7("test unit string attribute"); return entity; }
@Test public void testCreate() throws IllegalDataStorageTypeException, IOException { final List<TestTimeSeriesAPIEntity> entities = new ArrayList<>(); TestTimeSeriesAPIEntity entity = new TestTimeSeriesAPIEntity(); entity.setField1(1); entity.setField2(2); entity.setField3(10000000L); entity.setField4(10000000L); entity.setField5(0.12345678); entity.setTags(new HashMap<String, String>() { { put("cluster", "test"); put("datacenter", "test"); put("name", "unit.test.name"); } }); entities.add(entity); CreateStatement createStatement = new CreateStatement(entities, "TestTimeSeriesAPIEntity"); ModifyResult resultSet = createStatement.execute(DataStorageManager.newDataStorage("hbase")); Assert.assertEquals(1, resultSet.getIdentifiers().size()); createStatement = new CreateStatement(entities, "TestTimeSeriesAPIEntity"); resultSet = createStatement.execute(DataStorageManager.newDataStorage("hbase")); Assert.assertEquals(1, resultSet.getIdentifiers().size()); }
TestTimeSeriesAPIEntity entity = new TestTimeSeriesAPIEntity(); entity.setTimestamp(System.currentTimeMillis()); entity.setTags(new HashMap<String, String>() {{
List<TestTimeSeriesAPIEntity> entities = new ArrayList<TestTimeSeriesAPIEntity>(); for(int i=0;i<100;i++){ TestTimeSeriesAPIEntity entity = new TestTimeSeriesAPIEntity(); entity.setTimestamp(System.currentTimeMillis()); entity.setTags(new HashMap<String, String>() {{
List<TestTimeSeriesAPIEntity> entities = new ArrayList<TestTimeSeriesAPIEntity>(); for(int i=0;i<100;i++){ TestTimeSeriesAPIEntity entity = new TestTimeSeriesAPIEntity(); entity.setTimestamp(System.currentTimeMillis()); entity.setTags(new HashMap<String, String>() {{