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(expected = EagleServiceClientException.class) public void testGroupEntitiesByServiceException() throws EagleServiceClientException { List<TaggedLogAPIEntity> entities = new ArrayList<>(); TaggedLogAPIEntity testEntity1 = null; TaggedLogAPIEntity testEntity2 = new TestEntity(); entities.add(testEntity1); entities.add(testEntity2); EagleServiceBaseClient eagleServiceBaseClient = new EagleServiceClientImpl("localhost", 9090, "admin", "secret"); eagleServiceBaseClient.groupEntitiesByService(entities); }
TestLogAPIEntity e = new TestLogAPIEntity(); 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);
@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()); }
@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()); } }
@Test public void testConvert() throws Exception { Map<String, Object> objectMap = new HashMap<>(); objectMap.put("prefix", "eagle"); objectMap.put("timestamp", 1479264382L); objectMap.put("encodedRowkey", "rowkey"); Map<String, String> tags = new HashMap<>(); tags.put("field1", "value1"); objectMap.put("tags", tags); objectMap.put("remediationID", "ID"); objectMap.put("remediationStatus", "status"); objectMap.put("count", 123456789L); objectMap.put("numHosts", 9); objectMap.put("numClusters", 123456789L); RowkeyQueryAPIResponseEntity rowkeyQueryAPIResponseEntity = new RowkeyQueryAPIResponseEntity(); rowkeyQueryAPIResponseEntity.setObj(objectMap); RowkeyQueryAPIResponseConvertHelper rowkeyQueryAPIResponseConvertHelper = new RowkeyQueryAPIResponseConvertHelper(); rowkeyQueryAPIResponseEntity = rowkeyQueryAPIResponseConvertHelper.convert(TestEntity.class, rowkeyQueryAPIResponseEntity); TestEntity entity = (TestEntity)rowkeyQueryAPIResponseEntity.getObj(); Assert.assertEquals("eagle", entity.getPrefix()); Assert.assertEquals(1479264382L, entity.getTimestamp()); Assert.assertEquals("rowkey", entity.getEncodedRowkey()); Assert.assertEquals("ID", entity.getRemediationID()); Assert.assertEquals("status", entity.getRemediationStatus()); Assert.assertEquals(1, entity.getTags().size()); } }
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; } }
TestLogAPIEntity e = new TestLogAPIEntity(); 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()); 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"); list.add(e);
@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); }
@Test public void testConvert() throws Exception { Map<String, Object> objectMap = new HashMap<>(); objectMap.put("prefix", "eagle"); objectMap.put("timestamp", 1479264382L); objectMap.put("encodedRowkey", "rowkey"); Map<String, String> tags = new HashMap<>(); tags.put("field1", "value1"); objectMap.put("tags", tags); objectMap.put("remediationID", "ID"); objectMap.put("remediationStatus", "status"); objectMap.put("count", 123456789L); objectMap.put("numHosts", 9); objectMap.put("numClusters", 123456789L); List<Map<String, Object>> mapCollection = new ArrayList<Map<String, Object>>(); mapCollection.add(objectMap); ListQueryAPIResponseEntity listQueryAPIResponseEntity = new ListQueryAPIResponseEntity(); listQueryAPIResponseEntity.setObj(mapCollection); ListQueryAPIResponseConvertHelper listQueryAPIResponseConvertHelper = new ListQueryAPIResponseConvertHelper(); listQueryAPIResponseEntity = listQueryAPIResponseConvertHelper.convert(TestEntity.class, listQueryAPIResponseEntity); List<TestEntity> objs = (List<TestEntity>) listQueryAPIResponseEntity.getObj(); TestEntity entity = objs.get(0); Assert.assertEquals("eagle", entity.getPrefix()); Assert.assertEquals(1479264382L, entity.getTimestamp()); Assert.assertEquals("rowkey", entity.getEncodedRowkey()); Assert.assertEquals("ID", entity.getRemediationID()); Assert.assertEquals("status", entity.getRemediationStatus()); Assert.assertEquals(1, entity.getTags().size()); } }
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 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); }
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 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 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; }
@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; }
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>() {{ 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("updated"); entities.add(entity);
TestTimeSeriesAPIEntity entity = new TestTimeSeriesAPIEntity(); entity.setTimestamp(System.currentTimeMillis()); entity.setTags(new HashMap<String, String>() {{ put("cluster", "cluster4ut"); put("datacenter", "datacenter4ut"); put("timestampStr",System.currentTimeMillis()+""); }}); 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"); entities.add(entity);
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>() {{ 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(" unit test oriented string attribute"); entities.add(entity);