コード例 #1
0
  @Test
  public void testCountUpStream() {
    ObjectId streamId = new ObjectId();
    counter.countUpStream(streamId, 100);
    counter.countUpStream(streamId, 150);

    int res = counter.getStreamCounts().get(streamId.toString());
    assertEquals(250, res);
  }
コード例 #2
0
  @Test
  public void testIncrementStream() {
    ObjectId streamId = new ObjectId();
    counter.countUpStream(streamId, 100);
    counter.incrementStream(streamId);

    int res = counter.getStreamCounts().get(streamId.toString());
    assertEquals(101, res);
  }
コード例 #3
0
  @Test
  public void testGetStreamCounts() {
    ObjectId stream1 = new ObjectId();
    ObjectId stream2 = new ObjectId();
    ObjectId stream3 = new ObjectId();

    Map expected = new HashMap<String, Integer>();
    expected.put(stream1.toString(), 1);
    expected.put(stream2.toString(), 5);
    expected.put(stream3.toString(), 2);

    counter.countUpStream(stream1, 1);
    counter.countUpStream(stream2, 3);
    counter.countUpStream(stream2, 2);
    counter.countUpStream(stream3, 1);
    counter.incrementStream(stream3);

    assertEquals(expected, counter.getStreamCounts());
  }
コード例 #4
0
  @Test
  public void testResetAllCounts() {
    counter.countUpTotal(100);
    counter.countUpHost("foo.example.org", 9001);
    counter.countUpStream(new ObjectId(), 5);

    assertEquals(100, counter.getTotalCount()); // Just to make sure.

    counter.resetAllCounts();

    assertEquals(0, counter.getTotalCount());
    assertEquals(0, counter.getHostCounts().size());
    assertEquals(0, counter.getStreamCounts().size());
  }
コード例 #5
0
  @Test
  public void testGetHostCounts() {
    String host1 = "example.org";
    String host2 = "foo.example.org";
    String host3 = "example.com";

    Map expected = new HashMap<String, Integer>();
    expected.put(Tools.encodeBase64(host1), 5);
    expected.put(Tools.encodeBase64(host2), 1);
    expected.put(Tools.encodeBase64(host3), 3);

    counter.countUpStream(new ObjectId(), 5); // Add a stream count for complexity.
    counter.countUpHost(host1, 4);
    counter.countUpHost(host1, 1);
    counter.incrementHost(host2);
    counter.countUpHost(host3, 3);

    assertEquals(expected, counter.getHostCounts());
  }
コード例 #6
0
 @Test
 public void testResetStreamCounts() {
   counter.countUpStream(new ObjectId(), 100);
   counter.resetStreamCounts();
   assertEquals(new HashMap<ObjectId, Integer>(), counter.getStreamCounts());
 }