torproject/collector
1
0
Fork 0
mirror of https://github.com/torproject/collector.git synced 2024-11-23 09:29:46 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Added tests for file-storage using PersistenceUtils.

Browse Source
This commit is contained in:
iwakeh 2016-10-25 17:45:08 +02:00
parent f4cb959d1e
commit f2079ed440
1 changed files with 101 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

101
src/test/java/org/torproject/collector/persist/PersistUtilsTest.java Normal file
View File

@ -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));
}
}