From 3c5846f7b9afbeb54c51b867d350b173da38c651 Mon Sep 17 00:00:00 2001
From: Razvan Deaconescu <razvan.deaconescu@cs.pub.ro>
Date: Sat, 27 Aug 2011 09:43:50 +0300
Subject: [PATCH] ppf/new: Add two tests and setUp/tearDown to parsing test
 module.

---
 ppf/new/tests/test_parsing.py | 60 ++++++++++++++++++++++++++++++++---
 1 file changed, 56 insertions(+), 4 deletions(-)

diff --git a/ppf/new/tests/test_parsing.py b/ppf/new/tests/test_parsing.py
index 4ccb5dc..b354ccc 100644
--- a/ppf/new/tests/test_parsing.py
+++ b/ppf/new/tests/test_parsing.py
@@ -15,23 +15,75 @@ import parsing
 class LibtorrentLogParserTest(unittest.TestCase):
     """Test suite for LibtorrentLogParser class in parsing.py."""
 
+    ok_dir = "ok_dir"
+    no_dir = "no_dir"
+    status_file = "status.log"
+    verbose_file1 = "141.85.224.201_12345.log"
+    verbose_file2 = "141.85.224.202_12345.log"
+    verbose_file3 = "141.85.224.203_12345.log"
+
     def setUp(self):
-        pass
+        # Create folder and log files.
+        os.mkdir(self.ok_dir)
+        self.ok_status_file_path = os.path.join(self.ok_dir,
+                self.status_file)
+        self.ok_verbose_file_path1 = os.path.join(self.ok_dir,
+                self.verbose_file1)
+        self.ok_verbose_file_path2 = os.path.join(self.ok_dir,
+                self.verbose_file2)
+        self.ok_verbose_file_path3 = os.path.join(self.ok_dir,
+                self.verbose_file3)
+        f = open(self.ok_status_file_path, "w")
+        f.close()
+        f = open(self.ok_verbose_file_path1, "w")
+        f.close()
+        f = open(self.ok_verbose_file_path2, "w")
+        f.close()
+        f = open(self.ok_verbose_file_path3, "w")
+        f.close()
+
+        # Remove no_dir in case it exists.
+        try:
+            shutil.rmtree(self.no_dir)
+        except OSError, e:
+            pass
 
     def tearDown(self):
-        pass
+        shutil.rmtree(self.ok_dir)
+        # Remove no_dir in case it exists.
+        try:
+            shutil.rmtree(self.no_dir)
+        except OSError, e:
+            pass
 
     def test_non_existent_folder(self):
         """libtorrent log folder doesn't exist.
         Exception should be raised.
         """
-        self.assertEqual(True, False)
+        exception_raised = False
+        try:
+            p = parsing.LibtorrentLogParser(self.no_dir)
+        except OSError, e:
+            exception_raised = True
+
+        self.assertEqual(exception_raised, True)
 
     def test_no_log_files(self):
         """No log files are present in the log folder.
         Exception should be raised.
         """
-        self.assertEqual(True, False)
+        # Remove files.
+        os.remove(self.ok_status_file_path)
+        os.remove(self.ok_verbose_file_path1)
+        os.remove(self.ok_verbose_file_path2)
+        os.remove(self.ok_verbose_file_path3)
+
+        try:
+            p = parsing.LibtorrentLogParser(self.ok_dir)
+        except (IOError, OSError), e:
+            exception_raised = True
+
+        self.assertEqual(exception_raised, True)
 
     def test_get_parsing_file_no_status_log_file(self):
         """No status log file is present.
-- 
2.20.1