[tor-commits] [collector/master] Added tests for file-storage using PersistenceUtils.

[karsten at torproject.org]

commit f2079ed4408fce241d0f774d8600bc89bc910058
Author: iwakeh <iwakeh at torproject.org>
Date:   Tue Oct 25 17:45:08 2016 +0200

    Added tests for file-storage using PersistenceUtils.
---
 .../collector/persist/PersistUtilsTest.java        | 101 +++++++++++++++++++++
 1 file changed, 101 insertions(+)

diff --git a/src/test/java/org/torproject/collector/persist/PersistUtilsTest.java b/src/test/java/org/torproject/collector/persist/PersistUtilsTest.java
new file mode 100644
index 0000000..0fcfe14
--- /dev/null
+++ b/src/test/java/org/torproject/collector/persist/PersistUtilsTest.java
@@ -0,0 +1,101 @@
+/* Copyright 2016 The Tor Project
+ * See LICENSE for licensing information */
+
+package org.torproject.collector.persist;
+
+import static org.junit.Assert.assertEquals;
+import static org.junit.Assert.assertFalse;
+import static org.junit.Assert.assertTrue;
+
+import org.junit.Rule;
+import org.junit.Test;
+import org.junit.rules.TemporaryFolder;
+
+import java.nio.file.Files;
+import java.nio.file.Path;
+import java.nio.file.Paths;
+import java.nio.file.StandardOpenOption;
+import java.util.List;
+
+public class PersistUtilsTest {
+
+  private static final String ANNO1 = "1 at annotation\n";
+  private static final String ANNO2 = "2 at annotation\n";
+
+  @Rule
+  public TemporaryFolder tmpf = new TemporaryFolder();
+
+  @Test()
+  public void testCreateNew() throws Exception {
+    Path out = tmpf.newFile().toPath();
+    assertTrue("File should exist, but doesn't: " + out.toString(),
+        Files.exists(out));
+    assertFalse("Files shouldn't be created.",
+        PersistenceUtils.storeToFileSystem(ANNO1.getBytes(),
+        "some text".getBytes(), out, StandardOpenOption.CREATE_NEW));
+    assertFalse("Files shouldn't be created.",
+        PersistenceUtils.storeToFileSystem(ANNO1.getBytes(),
+        "some text".getBytes(), out, StandardOpenOption.CREATE_NEW, true));
+    List<String> text = Files.readAllLines(out);
+    assertTrue("List should be empty: " + text, text.isEmpty());
+  }
+
+  @Test()
+  public void testCreate() throws Exception {
+    Path out = tmpf.newFolder().toPath();
+    Path pathToCreate = Paths.get(out.toString(), "very-new-file");
+    String theText = "some text";
+    assertTrue("Files should be created.",
+        PersistenceUtils.storeToFileSystem(ANNO1.getBytes(),
+        (theText + "\n").getBytes(), pathToCreate, StandardOpenOption.CREATE));
+    List<String> text = Files.readAllLines(pathToCreate);
+    assertEquals("File contained: " + text, 2, text.size());
+    assertEquals("File contained: " + text, theText, text.get(1));
+  }
+
+  @Test()
+  public void testTruncateExisting() throws Exception {
+    Path out = tmpf.newFolder().toPath();
+    Path pathToCreate = Paths.get(out.toString(), "very-new-file");
+    String theText = "some text";
+    assertTrue("Files should be created.",
+        PersistenceUtils.storeToFileSystem(ANNO1.getBytes(),
+        (theText + "\n").getBytes(), pathToCreate, StandardOpenOption.CREATE));
+    List<String> text = Files.readAllLines(pathToCreate);
+    assertEquals("File contained: " + text, 2, text.size());
+    assertEquals("File contained: " + text, theText, text.get(1));
+    String theText2 = "other symbols";
+    assertTrue("Files should be written.",
+        PersistenceUtils.storeToFileSystem((ANNO2).getBytes(),
+        (theText2 + "\n").getBytes(), pathToCreate,
+        StandardOpenOption.TRUNCATE_EXISTING));
+    text = Files.readAllLines(pathToCreate);
+    assertEquals("File contained: " + text, 2, text.size());
+    assertEquals("File contained: " + text, "2 at annotation", text.get(0));
+    assertEquals("File contained: " + text, theText2, text.get(1));
+  }
+
+  @Test()
+  public void testAppend() throws Exception {
+    Path out = tmpf.newFolder().toPath();
+    Path pathToCreate = Paths.get(out.toString(), "very-new-file");
+    String theText = "some text";
+    assertTrue("Files should be created.",
+        PersistenceUtils.storeToFileSystem(ANNO1.getBytes(),
+        (theText + "\n").getBytes(), pathToCreate, StandardOpenOption.CREATE));
+    List<String> text = Files.readAllLines(pathToCreate);
+    assertEquals("File contained: " + text, 2, text.size());
+    assertEquals("File contained: " + text, theText, text.get(1));
+    String theText2 = "other symbols";
+    assertTrue("Files should be created.",
+        PersistenceUtils.storeToFileSystem((ANNO2).getBytes(),
+        (theText2 + "\n").getBytes(), pathToCreate, StandardOpenOption.APPEND));
+    text = Files.readAllLines(pathToCreate);
+    assertEquals("File contained: " + text, 4, text.size());
+    assertEquals("File contained: " + text, "1 at annotation", text.get(0));
+    assertEquals("File contained: " + text, theText, text.get(1));
+    assertEquals("File contained: " + text, "2 at annotation", text.get(2));
+    assertEquals("File contained: " + text, theText2, text.get(3));
+  }
+
+}