[tor-commits] [onionoo/master] Add unit tests for BandwidthStatus.

[karsten@xxxxxxxxxxxxxx]

commit 9248ccd903069fad8cd48ab6a538b09eadc77144
Author: Karsten Loesing <karsten.loesing@xxxxxxx>
Date:   Fri Aug 15 18:46:04 2014 +0200

    Add unit tests for BandwidthStatus.
---
 .../onionoo/docs/BandwidthStatusTest.java          |  146 ++++++++++++++++++++
 1 file changed, 146 insertions(+)

diff --git a/src/test/java/org/torproject/onionoo/docs/BandwidthStatusTest.java b/src/test/java/org/torproject/onionoo/docs/BandwidthStatusTest.java
new file mode 100644
index 0000000..91d086d
--- /dev/null
+++ b/src/test/java/org/torproject/onionoo/docs/BandwidthStatusTest.java
@@ -0,0 +1,146 @@
+/* Copyright 2014 The Tor Project
+ * See LICENSE for licensing information */
+package org.torproject.onionoo.docs;
+
+import static org.junit.Assert.assertEquals;
+import static org.junit.Assert.assertFalse;
+import static org.junit.Assert.assertTrue;
+
+import java.util.SortedMap;
+import java.util.TreeMap;
+
+import org.junit.Before;
+import org.junit.Test;
+import org.torproject.descriptor.BandwidthHistory;
+import org.torproject.onionoo.DummyTime;
+import org.torproject.onionoo.util.TimeFactory;
+
+public class BandwidthStatusTest {
+
+  private long currentTimeMillis = DateTimeHelper.parse(
+      "2014-08-01 02:22:22");
+
+  @Before
+  public void createDummyTime() {
+    TimeFactory.setTime(new DummyTime(this.currentTimeMillis));
+  }
+
+  @Test()
+  public void testEmptyStatusNotDirty() {
+    BandwidthStatus bandwidthStatus = new BandwidthStatus();
+    assertFalse("Newly created bandwidth status should not be dirty.",
+        bandwidthStatus.isDirty());
+  }
+
+  private static class DummyBandwidthHistory implements BandwidthHistory {
+    DummyBandwidthHistory(String line) {
+      this.line = line;
+      String[] parts = line.split(" ");
+      this.historyEndMillis = DateTimeHelper.parse(parts[1] + " "
+          + parts[2]);
+      this.intervalLength = Long.parseLong(parts[3].substring(1));
+      long intervalEndMillis = this.historyEndMillis,
+          intervalLengthMillis = this.intervalLength * 1000L;
+      String[] valueStrings = parts[5].split(",");
+      for (int i = valueStrings.length - 1; i >= 0; i--) {
+        this.bandwidthValues.put(intervalEndMillis,
+            Long.parseLong(valueStrings[i]));
+        intervalEndMillis -= intervalLengthMillis;
+      }
+    }
+    private String line;
+    public String getLine() {
+      return this.line;
+    }
+    private long historyEndMillis;
+    public long getHistoryEndMillis() {
+      return this.historyEndMillis;
+    }
+    private long intervalLength;
+    public long getIntervalLength() {
+      return this.intervalLength;
+    }
+    private SortedMap<Long, Long> bandwidthValues =
+        new TreeMap<Long, Long>();
+    public SortedMap<Long, Long> getBandwidthValues() {
+      return this.bandwidthValues;
+    }
+  }
+
+  @Test()
+  public void testNewStatusWithSingleInterval() {
+    BandwidthStatus bandwidthStatus = new BandwidthStatus();
+    bandwidthStatus.addToWriteHistory(new DummyBandwidthHistory(
+        "write-history 2014-08-01 00:22:22 (900 s) 30720"));
+    assertTrue("Updated bandwidth status should be marked as dirty.",
+        bandwidthStatus.isDirty());
+    assertEquals("Formatted document string not as expected.",
+        "w 2014-08-01 00:07:22 2014-08-01 00:22:22 30720\n",
+        bandwidthStatus.toDocumentString());
+    bandwidthStatus.clearDirty();
+    assertFalse("Dirty flag should be cleared.",
+        bandwidthStatus.isDirty());
+  }
+
+  @Test()
+  public void testNewStatusWithTwoIntervals() {
+    BandwidthStatus bandwidthStatus = new BandwidthStatus();
+    bandwidthStatus.addToWriteHistory(new DummyBandwidthHistory(
+        "write-history 2014-08-01 00:22:22 (900 s) 4096,30720"));
+    assertEquals("Formatted document string not as expected.",
+        "w 2014-07-31 23:52:22 2014-08-01 00:07:22 4096\n"
+        + "w 2014-08-01 00:07:22 2014-08-01 00:22:22 30720\n",
+        bandwidthStatus.toDocumentString());
+  }
+
+  @Test()
+  public void testExistingStatusWithNewIntervals() {
+    BandwidthStatus bandwidthStatus = new BandwidthStatus();
+    String existingLines =
+        "w 2014-07-31 23:52:22 2014-08-01 00:07:22 4096\n"
+        + "w 2014-08-01 00:07:22 2014-08-01 00:22:22 30720\n";
+    bandwidthStatus.fromDocumentString(existingLines);
+    bandwidthStatus.addToWriteHistory(new DummyBandwidthHistory(
+        "write-history 2014-08-01 00:37:22 (900 s) 0"));
+    assertEquals("New interval should be appended.",
+        existingLines + "w 2014-08-01 00:22:22 2014-08-01 00:37:22 0\n",
+        bandwidthStatus.toDocumentString());
+  }
+
+  @Test()
+  public void testCompressRecentIntervals() {
+    BandwidthStatus bandwidthStatus = new BandwidthStatus();
+    String existingLines =
+        "w 2014-08-01 00:07:22 2014-08-01 00:22:22 30720\n"
+        + "w 2014-08-01 00:22:22 2014-08-01 00:37:22 4096\n";
+    bandwidthStatus.fromDocumentString(existingLines);
+    bandwidthStatus.compressHistory();
+    assertEquals("Two recent intervals should not be compressed.",
+        existingLines, bandwidthStatus.toDocumentString());
+  }
+
+  @Test()
+  public void testCompressOldIntervals() {
+    BandwidthStatus bandwidthStatus = new BandwidthStatus();
+    bandwidthStatus.fromDocumentString(
+        "w 2013-08-01 00:07:22 2013-08-01 00:22:22 30720\n"
+        + "w 2013-08-01 00:22:22 2013-08-01 00:37:22 4096\n");
+    bandwidthStatus.compressHistory();
+    assertEquals("Two old intervals should be compressed into one.",
+        "w 2013-08-01 00:07:22 2013-08-01 00:37:22 34816\n",
+        bandwidthStatus.toDocumentString());
+  }
+
+  @Test()
+  public void testCompressOldIntervalsOverMonthEnd() {
+    BandwidthStatus bandwidthStatus = new BandwidthStatus();
+    String statusLines =
+        "w 2013-07-31 23:52:22 2013-08-01 00:07:22 4096\n"
+        + "w 2013-08-01 00:07:22 2013-08-01 00:22:22 30720\n";
+    bandwidthStatus.fromDocumentString(statusLines);
+    bandwidthStatus.compressHistory();
+    assertEquals("Two old intervals should not be merged over month end.",
+        statusLines, bandwidthStatus.toDocumentString());
+  }
+}
+

_______________________________________________
tor-commits mailing list
tor-commits@xxxxxxxxxxxxxxxxxxxx
https://lists.torproject.org/cgi-bin/mailman/listinfo/tor-commits