diff options
| -rw-r--r-- | datarouter-prov/src/test/java/org/onap/dmaap/datarouter/provisioning/utils/LogfileLoaderTest.java | 37 |
1 files changed, 14 insertions, 23 deletions
diff --git a/datarouter-prov/src/test/java/org/onap/dmaap/datarouter/provisioning/utils/LogfileLoaderTest.java b/datarouter-prov/src/test/java/org/onap/dmaap/datarouter/provisioning/utils/LogfileLoaderTest.java index ce71698e..7f6d7de6 100644 --- a/datarouter-prov/src/test/java/org/onap/dmaap/datarouter/provisioning/utils/LogfileLoaderTest.java +++ b/datarouter-prov/src/test/java/org/onap/dmaap/datarouter/provisioning/utils/LogfileLoaderTest.java @@ -20,10 +20,11 @@ package org.onap.dmaap.datarouter.provisioning.utils; -import org.junit.After; import org.junit.AfterClass; import org.junit.Assert; import org.junit.BeforeClass; +import org.junit.Rule; +import org.junit.rules.TemporaryFolder; import org.junit.runner.RunWith; import org.onap.dmaap.datarouter.provisioning.InternalServlet; @@ -38,14 +39,11 @@ import javax.persistence.Persistence; import java.io.File; import java.io.FileWriter; import java.io.IOException; -import java.util.Map; - -import static org.junit.Assert.assertEquals; import static org.junit.Assert.assertFalse; import org.junit.Test; -import java.util.HashMap; + @RunWith(PowerMockRunner.class) @@ -56,6 +54,9 @@ public class LogfileLoaderTest { private static EntityManager em; private LogfileLoader lfl = LogfileLoader.getLoader(); + @Rule + public TemporaryFolder folder = new TemporaryFolder(); + @BeforeClass public static void init() { @@ -76,46 +77,36 @@ public class LogfileLoaderTest { } - @After - public void tearDown() { - try { - new FileWriter("unit-test-logs/testFile1.txt").close(); - }catch (IOException ioe){ - System.out.println(ioe.getMessage()); - } - } - - @Test - public void Verify_File_Processing_when_Req_Type_LOG() { + public void Verify_File_Processing_when_Req_Type_LOG() throws IOException { String fileContent = "2018-08-29-10-10-10-543.|LOG|1|1|url/file123|method|1|1|type|1|128.0.0.9|user123|2|1|1|1|other|1"; - int[] actual = lfl.process(prepFile(fileContent)); + int[] actual = lfl.process(prepFile(fileContent, "file1")); int[] expect = {0, 1}; Assert.assertArrayEquals(expect, actual); } @Test - public void Verify_File_Processing_when_Req_Type_EXP() { + public void Verify_File_Processing_when_Req_Type_EXP() throws IOException{ String fileContent = "2018-08-29-10-10-10-543.|EXP|1|1|1|'url/file123'|method|ctype|3|other|4"; - int[] actual = lfl.process(prepFile(fileContent)); + int[] actual = lfl.process(prepFile(fileContent, "file2")); int[] expect = {0, 1}; Assert.assertArrayEquals(expect, actual); } @Test - public void Verify_Records_Prune_When_Record_Count_Is_Less_Then_Threshold() { + public void Verify_Records_Prune_When_Record_Count_Is_Less_Then_Threshold() throws IOException{ String fileContent = "2018-08-29-10-10-10-543.|PUB|1|1|https://dmaap-dr-prov:8443/publish/1/file123/|POST|application/vnd.att-dr.feed|2|128.0.0.9|user123|200"; - lfl.process(prepFile(fileContent)); + lfl.process(prepFile(fileContent, "file3")); PowerMockito.mockStatic(Parameters.class); PowerMockito.when(Parameters.getParameter(Parameters.PROV_LOG_RETENTION)).thenReturn(new Parameters(Parameters.PROV_LOG_RETENTION, "0")); assertFalse(lfl.pruneRecords()); } - private File prepFile(String content){ - File file1 = new File("unit-test-logs/testFile1.txt"); + private File prepFile(String content, String fileName) throws IOException{ + File file1 = folder.newFile(fileName); try (FileWriter fileWriter = new FileWriter(file1)) { fileWriter.write(content); }catch (IOException e){ |