timeSeriesAPIEntity.setField7("7"); timeSeriesAPIEntity.setTags(new HashMap<>()); timeSeriesAPIEntity.getTags().put("cluster", "test4UT"); timeSeriesAPIEntity.getTags().put("datacenter", "dc1"); timeSeriesAPIEntity.getTags().put("index", "" + 1); timeSeriesAPIEntity.getTags().put("jobId", "job_" + timeSeriesAPIEntity.getTimestamp()); Assert.assertEquals(timeSeriesAPIEntityList.get(0).getTags().get("cluster"), result.get(0).getTags().get("cluster")); Assert.assertEquals(timeSeriesAPIEntityList.get(0).getTags().get("datacenter"), result.get(0).getTags().get("datacenter")); Assert.assertEquals(timeSeriesAPIEntityList.get(0).getTags().get("index"), result.get(0).getTags().get("index")); Assert.assertEquals(timeSeriesAPIEntityList.get(0).getTags().get("jobId"), result.get(0).getTags().get("jobId"));
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; }
Assert.assertTrue(entity instanceof TestTimeSeriesAPIEntity); TestTimeSeriesAPIEntity tsentity = (TestTimeSeriesAPIEntity)entity; Assert.assertEquals("cluster4ut", tsentity.getTags().get("cluster")); Assert.assertEquals("datacenter4ut", tsentity.getTags().get("datacenter")); Assert.assertEquals(12, tsentity.getField1()); Assert.assertEquals(123432432l, tsentity.getField3());