commit f2079ed4408fce241d0f774d8600bc89bc910058 Author: iwakeh <iwakeh@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@annotation\n"; + private static final String ANNO2 = "2@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@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@annotation", text.get(0)); + assertEquals("File contained: " + text, theText, text.get(1)); + assertEquals("File contained: " + text, "2@annotation", text.get(2)); + assertEquals("File contained: " + text, theText2, text.get(3)); + } + +}