private Subscription sub(int i) { return new Subscription() .withId("" + i) .withExpression("name," + i + ",:eq,:sum") .withFrequency(60000); }
@Test public void removeSub() { List<Subscription> subs = new ArrayList<>(); subs.add(new Subscription().withId("sum").withExpression(":true,:sum")); subs.add(new Subscription().withId("max").withExpression(":true,:max")); Evaluator evaluator = new Evaluator().addGroupSubscriptions("local", subs); evaluator.removeGroupSubscriptions("local"); EvalPayload payload = evaluator.eval("test", 0L, Collections.emptyList()); EvalPayload expected = new EvalPayload(0L, Collections.emptyList()); Assertions.assertEquals(expected, payload); }
@Test public void dataExprInvalid() { Assertions.assertThrows(IllegalArgumentException.class, () -> new Subscription().withExpression(":true")); } }
@Test public void sumAndMaxForGroup() { List<Subscription> subs = new ArrayList<>(); subs.add(new Subscription().withId("sum").withExpression(":true,:sum")); subs.add(new Subscription().withId("max").withExpression(":true,:max")); Evaluator evaluator = new Evaluator().addGroupSubscriptions("local", subs); EvalPayload payload = evaluator.eval("local", 0L, data("foo", 1.0, 2.0, 3.0)); List<EvalPayload.Metric> metrics = new ArrayList<>(); metrics.add(new EvalPayload.Metric("sum", Collections.emptyMap(), 6.0)); metrics.add(new EvalPayload.Metric("max", Collections.emptyMap(), 3.0)); EvalPayload expected = new EvalPayload(0L, metrics); Assertions.assertEquals(expected, payload); }
private Subscription newSub(String id, String expr, long freq) { return new Subscription().withId(id).withExpression(expr).withFrequency(freq); }
@Test public void updateSub() { // Eval with sum List<Subscription> sumSub = new ArrayList<>(); sumSub.add(new Subscription().withId("sum").withExpression(":true,:sum")); Evaluator evaluator = new Evaluator().addGroupSubscriptions("local", sumSub); EvalPayload payload = evaluator.eval("local", 0L, data("foo", 1.0, 2.0, 3.0)); List<EvalPayload.Metric> metrics = new ArrayList<>(); metrics.add(new EvalPayload.Metric("sum", Collections.emptyMap(), 6.0)); EvalPayload expected = new EvalPayload(0L, metrics); Assertions.assertEquals(expected, payload); // Update to use max instead List<Subscription> maxSub = new ArrayList<>(); maxSub.add(new Subscription().withId("sum").withExpression(":true,:max")); evaluator.addGroupSubscriptions("local", maxSub); payload = evaluator.eval("local", 0L, data("foo", 1.0, 2.0, 3.0)); metrics = new ArrayList<>(); metrics.add(new EvalPayload.Metric("sum", Collections.emptyMap(), 3.0)); expected = new EvalPayload(0L, metrics); Assertions.assertEquals(expected, payload); } }
@Test public void dataExpr() { Subscription sub = new Subscription().withExpression(":true,:sum"); Assertions.assertEquals(new DataExpr.Sum(Query.TRUE), sub.dataExpr()); }