From 12d7ee2cb6589b866ab26b508b15c65326481d6c Mon Sep 17 00:00:00 2001
From: Joshua Harlow <harlowja@gmail.com>
Date: Wed, 11 May 2016 16:47:50 -0700
Subject: Use a fake serial module that will allow tests to contine

Instead of aborting all serial using tests instead just
create a serial module in cloudinit that will create a fake
and broken serial class when pyserial is not actually installed.

This allows for using the datasource and tests that exist in
a more functional and tested manner (even when pyserial is not
found).
---
 tests/unittests/test_datasource/test_cloudsigma.py | 23 +++++++---------------
 tests/unittests/test_datasource/test_smartos.py    | 15 ++++----------
 2 files changed, 11 insertions(+), 27 deletions(-)

(limited to 'tests/unittests/test_datasource')

diff --git a/tests/unittests/test_datasource/test_cloudsigma.py b/tests/unittests/test_datasource/test_cloudsigma.py
index 11968796..7950fc52 100644
--- a/tests/unittests/test_datasource/test_cloudsigma.py
+++ b/tests/unittests/test_datasource/test_cloudsigma.py
@@ -2,14 +2,8 @@
 
 import copy
 
-try:
-    # Serial does not work on py2.6 (anymore)
-    import pyserial
-    from cloudinit.cs_utils import Cepko
-    from cloudinit.sources import DataSourceCloudSigma
-    WILL_WORK = True
-except ImportError:
-    WILL_WORK = False
+from cloudinit.cs_utils import Cepko
+from cloudinit.sources import DataSourceCloudSigma
 
 from .. import helpers as test_helpers
 from ..helpers import SkipTest
@@ -36,20 +30,17 @@ SERVER_CONTEXT = {
 }
 
 
-if WILL_WORK:
-    class CepkoMock(Cepko):
-        def __init__(self, mocked_context):
-            self.result = mocked_context
+class CepkoMock(Cepko):
+    def __init__(self, mocked_context):
+        self.result = mocked_context
 
-        def all(self):
-            return self
+    def all(self):
+        return self
 
 
 class DataSourceCloudSigmaTest(test_helpers.TestCase):
     def setUp(self):
         super(DataSourceCloudSigmaTest, self).setUp()
-        if not WILL_WORK:
-            raise SkipTest("Datasource testing not supported")
         self.datasource = DataSourceCloudSigma.DataSourceCloudSigma("", "", "")
         self.datasource.is_running_in_cloudsigma = lambda: True
         self.datasource.cepko = CepkoMock(SERVER_CONTEXT)
diff --git a/tests/unittests/test_datasource/test_smartos.py b/tests/unittests/test_datasource/test_smartos.py
index 6b628276..f536ef4f 100644
--- a/tests/unittests/test_datasource/test_smartos.py
+++ b/tests/unittests/test_datasource/test_smartos.py
@@ -33,13 +33,8 @@ import tempfile
 import uuid
 from binascii import crc32
 
-try:
-    # Serial does not work on py2.6 (anymore)
-    import serial
-    from cloudinit.sources import DataSourceSmartOS
-    WILL_WORK = True
-except ImportError:
-    WILL_WORK = False
+from cloudinit import serial
+from cloudinit.sources import DataSourceSmartOS
 
 import six
 
@@ -81,8 +76,7 @@ def get_mock_client(mockdata):
 class TestSmartOSDataSource(helpers.FilesystemMockingTestCase):
     def setUp(self):
         super(TestSmartOSDataSource, self).setUp()
-        if not WILL_WORK:
-            raise SkipTest("This test will not work")
+
         self.tmp = tempfile.mkdtemp()
         self.addCleanup(shutil.rmtree, self.tmp)
         self.legacy_user_d = tempfile.mkdtemp()
@@ -448,8 +442,7 @@ class TestJoyentMetadataClient(helpers.FilesystemMockingTestCase):
 
     def setUp(self):
         super(TestJoyentMetadataClient, self).setUp()
-        if not WILL_WORK:
-            raise SkipTest("This test will not work")
+
         self.serial = mock.MagicMock(spec=serial.Serial)
         self.request_id = 0xabcdef12
         self.metadata_value = 'value'
-- 
cgit v1.2.3