[#20] Start working on testing FreeBSDUtils with pytest

Test case for FreeBSDUtils.reboot

cloudbaseinit/osutils/freebsd.py

@@ -1,15 +1,15 @@
-from cloudbaseinit.osutils import base
-
-from subprocess import CalledProcessError
-import subprocess
 import datetime
+import subprocess
 import os
-import os.path
+
+from subprocess import CalledProcessError
+from cloudbaseinit.osutils import base
+
 
 class FreeBSDUtils(base.BaseOSUtils):
     def reboot(self):
-        if ( os.system('reboot') != 0 ):
-            raise Exception('Reboot failed')
+        if os.system('reboot') != 0:
+            raise OSError('Reboot failed')
 
     def user_exists(self, username):
         try:

cloudbaseinit/tests/osutils/test_freebsd.py

@@ -0,0 +1,22 @@
+import unittest
+
+try:
+    import unittest.mock as mock
+except ImportError:
+    import mock
+
+from cloudbaseinit.osutils.freebsd import FreeBSDUtils
+from pytest import raises
+
+
+class TestFreeBSDUtils():
+    def setup_method(self, method):
+        self.bsd = FreeBSDUtils()
+
+    def teardown_method(self, method):
+        del self.bsd
+
+    @mock.patch('cloudbaseinit.osutils.freebsd.os.system', return_value=1)
+    def test_reboot(self, system):
+        with raises(OSError):
+            self.bsd.reboot()

test-requirements.txt

@@ -6,3 +6,4 @@ sphinx>=1.1.2,<1.1.999
 oslosphinx
 testtools>=0.9.32
 testrepository>=0.0.18
+pytest