Advertisement
Guest User

Untitled

a guest
Jan 23rd, 2017
114
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 6.03 KB | None | 0 0
  1. package logger.tests;
  2.  
  3. import static org.junit.Assert.*;
  4.  
  5. import java.io.BufferedReader;
  6. import java.io.File;
  7. import java.io.FileInputStream;
  8. import java.io.FileNotFoundException;
  9. import java.io.FileReader;
  10. import java.io.IOException;
  11. import java.io.InputStreamReader;
  12. import java.io.PrintWriter;
  13. import java.io.UnsupportedEncodingException;
  14. import java.util.ArrayList;
  15.  
  16. import javax.swing.JCheckBox;
  17. import javax.swing.JFrame;
  18. import javax.swing.JTextArea;
  19.  
  20. import org.junit.After;
  21. import org.junit.AfterClass;
  22. import org.junit.Before;
  23. import org.junit.FixMethodOrder;
  24. import org.junit.Rule;
  25. import org.junit.Test;
  26. import org.junit.rules.TemporaryFolder;
  27. import org.junit.runners.MethodSorters;
  28. import org.mockito.Mockito;
  29.  
  30. import logger.server.ClientHandler;
  31. import logger.server.ClientWrapper;
  32. import logger.server.ServerFrame;
  33. import logger.server.ServerUtilities;
  34. import logger.server.SimpleServerFrame;
  35.  
  36. public class TestServerUtilities {
  37. private ClientHandler client;
  38.  
  39. @Rule
  40. public TemporaryFolder folder = new TemporaryFolder();
  41.  
  42. @Before
  43. public void initialiseClientHandler() {
  44. client = new ClientHandler(null, 0, null);
  45. JCheckBox box = new JCheckBox();
  46. client.setBox(box);
  47. JTextArea ignored = Mockito.mock(JTextArea.class);
  48. ServerUtilities.setOutput(ignored);
  49. ServerUtilities.setServerFrame(Mockito.mock(ServerFrame.class));
  50.  
  51. }
  52.  
  53. @Test
  54. public void testEnableFileSaving() throws IOException {
  55. ServerUtilities.setDirPath(folder.newFolder("TEST").getPath());
  56. ServerUtilities.enableFileSaving();
  57. assertTrue(ServerUtilities.getFullLogFile().exists());
  58.  
  59. File tmpDir = new File(ServerUtilities.getDirPath());
  60. File[] files = tmpDir.listFiles();
  61. assertEquals(1, files.length);
  62. assertEquals(ServerUtilities.getFullLogFile(), files[0]);
  63. ServerUtilities.getFileWriter().close();
  64. ServerUtilities.getFullLogFile().delete();
  65. ServerUtilities.deleteFilesFromDirectory(ServerUtilities.getDirPath());
  66. }
  67.  
  68. @Test
  69. public void testSetName() {
  70. client.setName("test");
  71. assertEquals("test", client.getName());
  72. }
  73.  
  74. @Test
  75. public void testAddToList() {
  76. ArrayList<ClientHandler> testList = ServerUtilities.getClientList();
  77. ServerUtilities.addToList(client);
  78. assertEquals(1, testList.size());
  79. }
  80.  
  81. @Test
  82. public void testDeleteClientLogFile() {
  83. client.createLogFile("test");
  84. File testFile = client.getLogFile();
  85. assertTrue(testFile.exists());
  86. ServerUtilities.deleteClientLogFile(client);
  87. assertFalse(testFile.exists());
  88. }
  89.  
  90. @Test
  91. public void testParseLogs() throws IOException {
  92. File test1 = folder.newFile("test2.txt");
  93.  
  94. PrintWriter writer = new PrintWriter(test1, "UTF-8");
  95. writer.println("test");
  96. writer.println("test2");
  97. writer.println("hello");
  98. writer.println("end");
  99. writer.close();
  100. BufferedReader br = new BufferedReader(new InputStreamReader(new FileInputStream(test1), "UTF-8"));
  101. ServerUtilities.parseLogs(br, client);
  102. test1.delete();
  103.  
  104. }
  105.  
  106. @Test
  107. public void testDeleteFilesFromDirectory() throws IOException {
  108. File test1 = folder.newFile("test1.txt");
  109. File test2 = folder.newFile("test2.txt");
  110. assertTrue(test1.exists());
  111. assertTrue(test2.exists());
  112. ServerUtilities.deleteFilesFromDirectory(folder.getRoot().toString());
  113. assertFalse(test1.exists());
  114. assertFalse(test2.exists());
  115. }
  116.  
  117. @Test
  118. public void testSetVisibility() {
  119. assertTrue(client.isVisible());
  120. ServerUtilities.setVisibility(client, false);
  121. assertFalse(client.isVisible());
  122. ServerUtilities.setVisibility(client, true);
  123. assertTrue(client.isVisible());
  124.  
  125. }
  126.  
  127. @Test
  128. public void testCopyFile() throws IOException {
  129. File test1 = folder.newFile("test1.txt");
  130. assertTrue(test1.exists());
  131. PrintWriter writer = new PrintWriter(test1);
  132. writer.println("Hello World");
  133. writer.close();
  134. File test2 = folder.newFile("test2.txt");
  135. ServerUtilities.deleteFilesFromDirectory("test");
  136. assertTrue(test2.exists());
  137. ServerUtilities.copyFile(test1, test2);
  138. assertTrue(test2.exists());
  139. BufferedReader reader = new BufferedReader(new InputStreamReader(new FileInputStream(test2)));
  140. String hello = reader.readLine();
  141. assertEquals("Hello World", hello);
  142. reader.close();
  143. test2.delete();
  144. test1.delete();
  145. }
  146.  
  147. @Test
  148. public void testSaveLogs() throws IOException {
  149. File test1 = folder.newFile();
  150. assertTrue(ServerUtilities.getNotAllowedList().isEmpty());
  151. ServerUtilities.enableFileSaving();
  152. File fullLogFile = ServerUtilities.getFullLogFile();
  153. PrintWriter writer = new PrintWriter(fullLogFile);
  154. writer.println("TEST");
  155. writer.close();
  156. ServerUtilities.saveLogs(test1);
  157. BufferedReader reader = new BufferedReader(new InputStreamReader(new FileInputStream(test1)));
  158. String test = reader.readLine();
  159. reader.close();
  160. assertEquals("TEST", test);
  161. fullLogFile.delete();
  162. test1.delete();
  163.  
  164. }
  165.  
  166. @Test
  167. public void testLoadLogs() throws IOException {
  168. File test = folder.newFile();
  169. assertTrue(ServerUtilities.loadLogs(test));
  170. }
  171.  
  172. @Test
  173. public void testChangeWindowOutput() throws IOException {
  174. ServerUtilities.enableFileSaving();
  175. File testFile = ServerUtilities.getFullLogFile();
  176. PrintWriter writer = new PrintWriter(testFile, "UTF-8");
  177. writer.println("test1 : ABRAKADABRA");
  178. writer.println("test2 : ABECADŁO");
  179. writer.println("test3 : XYZ");
  180. writer.println("test1 : XZY");
  181. writer.println("test2 : ABRAKADABRA");
  182. writer.println("test1 : ABAB");
  183. writer.close();
  184. ServerUtilities.getNotAllowedList().add("test1");
  185. ServerUtilities.changeWindowOutput();
  186. File finalTestFile = new File(ServerUtilities.getDirPath() + "/current.txt");
  187. BufferedReader reader = new BufferedReader(new InputStreamReader(new FileInputStream(finalTestFile), "UTF-8"));
  188. assertEquals("test2 : ABECADŁO", reader.readLine());
  189. assertEquals("test3 : XYZ", reader.readLine());
  190. assertEquals("test2 : ABRAKADABRA", reader.readLine());
  191. reader.close();
  192. finalTestFile.delete();
  193. testFile.delete();
  194. ServerUtilities.getFileWriter().close();
  195. ServerUtilities.getFullLogFile().delete();
  196. }
  197.  
  198. }
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement