From a12d7a606397fdb354ae1ce785ef9b01773dfbfb Mon Sep 17 00:00:00 2001
From: Lucas Meneghel Rodrigues <lmr@redhat.com>
Date: Fri, 13 Jun 2014 22:55:06 -0300
Subject: [PATCH] selftest/unittest: Add another sysinfo unittest

This time, using the Sysinfo Logger class.

Signed-off-by: Lucas Meneghel Rodrigues <lmr@redhat.com>
---
 .../all/unit/avocado/sysinfo_unittest.py      | 50 ++++++++++++++++++-
 1 file changed, 49 insertions(+), 1 deletion(-)

diff --git a/selftests/all/unit/avocado/sysinfo_unittest.py b/selftests/all/unit/avocado/sysinfo_unittest.py
index 2fda0e97..3a1da9f7 100755
--- a/selftests/all/unit/avocado/sysinfo_unittest.py
+++ b/selftests/all/unit/avocado/sysinfo_unittest.py
@@ -15,9 +15,10 @@
 # Author: Lucas Meneghel Rodrigues <lmr@redhat.com>
 
 
-import unittest
 import os
 import sys
+import unittest
+import tempfile
 
 # simple magic for using scripts within a source tree
 basedir = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
@@ -30,6 +31,9 @@ from avocado import sysinfo
 
 class SysinfoTest(unittest.TestCase):
 
+    def setUp(self):
+        self.tmpdir = tempfile.mkdtemp(prefix="sysinfo_unittest")
+
     def testLoggablesEqual(self):
         cmd1 = sysinfo.Command("ls -l")
         cmd2 = sysinfo.Command("ls -l")
@@ -68,5 +72,49 @@ class SysinfoTest(unittest.TestCase):
 
         self.assertEqual(len(container), 5)
 
+    def test_logger_job_hooks(self):
+        jobdir = os.path.join(self.tmpdir, 'job')
+        sysinfo_logger = sysinfo.SysInfo(basedir=jobdir)
+        sysinfo_logger.start_job_hook()
+        self.assertTrue(os.path.isdir(jobdir))
+        self.assertEqual(len(os.listdir(jobdir)), 1,
+                         "Job does not have 'pre' dir")
+        job_predir = os.path.join(jobdir, 'pre')
+        self.assertTrue(os.path.isdir(job_predir))
+        self.assertGreater(len(os.listdir(job_predir)), 0,
+                           "Job pre dir is empty")
+        sysinfo_logger.end_job_hook()
+        job_postdir = os.path.join(jobdir, 'post')
+        self.assertTrue(os.path.isdir(job_postdir))
+        self.assertGreater(len(os.listdir(job_postdir)), 0,
+                           "Job post dir is empty")
+
+    def test_logger_test_hooks(self):
+        testdir = os.path.join(self.tmpdir, 'job', 'test1')
+        sysinfo_logger = sysinfo.SysInfo(basedir=testdir)
+        sysinfo_logger.start_test_hook()
+        self.assertTrue(os.path.isdir(testdir))
+        self.assertEqual(len(os.listdir(testdir)), 1,
+                         "Test does not have 'pre' dir")
+        test_predir = os.path.join(testdir, 'pre')
+        self.assertTrue(os.path.isdir(test_predir))
+        # By default, there are no pre test files
+        self.assertEqual(len(os.listdir(test_predir)), 0,
+                         "Test pre dir is not empty")
+        sysinfo_logger.end_test_hook()
+        self.assertEqual(len(os.listdir(testdir)), 2,
+                         "Test does not have 'pre' dir")
+        job_postdir = os.path.join(testdir, 'post')
+        self.assertTrue(os.path.isdir(job_postdir))
+        # By default, there are no post test files
+        self.assertEqual(len(os.listdir(job_postdir)), 0,
+                         "Test post dir is not empty")
+
+    def tearDown(self):
+        try:
+            os.rmdir(self.tmpdir)
+        except OSError:
+            pass
+
 if __name__ == '__main__':
     unittest.main()
-- 
GitLab