Fix up tests and flake8 warnings

8 files changed, 29 insertions, 30 deletions

diff --git a/tests/unittests/helpers.py b/tests/unittests/helpers.py
index 7b4d44e8..8d46a8bf 100644
--- a/tests/unittests/helpers.py
+++ b/tests/unittests/helpers.py
@@ -45,6 +45,7 @@ else:
         if _PY_MINOR == 4 and _PY_MICRO < 3:
             FIX_HTTPRETTY = True
 
+
 # Makes the old path start
 # with new base instead of whatever
 # it previously had
diff --git a/tests/unittests/test__init__.py b/tests/unittests/test__init__.py
index a9b35afe..0154784a 100644
--- a/tests/unittests/test__init__.py
+++ b/tests/unittests/test__init__.py
@@ -1,7 +1,6 @@
 import os
 import shutil
 import tempfile
-import unittest2
 
 from cloudinit import handlers
 from cloudinit import helpers
diff --git a/tests/unittests/test_datasource/test_cloudsigma.py b/tests/unittests/test_datasource/test_cloudsigma.py
index 7950fc52..2a42ce0c 100644
--- a/tests/unittests/test_datasource/test_cloudsigma.py
+++ b/tests/unittests/test_datasource/test_cloudsigma.py
@@ -6,7 +6,6 @@ from cloudinit.cs_utils import Cepko
 from cloudinit.sources import DataSourceCloudSigma
 
 from .. import helpers as test_helpers
-from ..helpers import SkipTest
 
 SERVER_CONTEXT = {
     "cpu": 1000,
diff --git a/tests/unittests/test_datasource/test_configdrive.py b/tests/unittests/test_datasource/test_configdrive.py
index 1db50798..5395e544 100644
--- a/tests/unittests/test_datasource/test_configdrive.py
+++ b/tests/unittests/test_datasource/test_configdrive.py
@@ -368,30 +368,32 @@ class TestNetJson(TestCase):
         self.assertEqual(myds.network_config, network_config)
 
     def test_network_config_conversions(self):
-        """Tests a bunch of input network json and checks the expected conversions."""
+        """Tests a bunch of input network json and checks the
+           expected conversions."""
         in_datas = [
             NETWORK_DATA,
             {
                 'services': [{'type': 'dns', 'address': '172.19.0.12'}],
-                'networks': [
-                    {'network_id': 'dacd568d-5be6-4786-91fe-750c374b78b4',
-                     'type': 'ipv4', 'netmask': '255.255.252.0', 
-                     'link': 'tap1a81968a-79', 
-                     'routes': [
-                        {
-                            'netmask': '0.0.0.0',
-                            'network': '0.0.0.0', 
-                            'gateway': '172.19.3.254'
-                        },
-                      ],
-                      'ip_address': '172.19.1.34', 
-                      'id': 'network0',
+                'networks': [{
+                    'network_id': 'dacd568d-5be6-4786-91fe-750c374b78b4',
+                    'type': 'ipv4',
+                    'netmask': '255.255.252.0',
+                    'link': 'tap1a81968a-79',
+                    'routes': [{
+                        'netmask': '0.0.0.0',
+                        'network': '0.0.0.0',
+                        'gateway': '172.19.3.254',
+                    }],
+                    'ip_address': '172.19.1.34',
+                    'id': 'network0',
+                }],
+                'links': [{
+                    'type': 'bridge',
+                    'vif_id': '1a81968a-797a-400f-8a80-567f997eb93f',
+                    'ethernet_mac_address': 'fa:16:3e:ed:9a:59',
+                    'id': 'tap1a81968a-79',
+                    'mtu': None,
                 }],
-                'links': [
-                    {'type': 'bridge',
-                     'vif_id': '1a81968a-797a-400f-8a80-567f997eb93f', 
-                     'ethernet_mac_address': 'fa:16:3e:ed:9a:59', 
-                     'id': 'tap1a81968a-79', 'mtu': None}]
             },
         ]
         out_datas = [
@@ -440,6 +442,7 @@ class TestNetJson(TestCase):
                                 'address': '172.19.1.34',
                                 'netmask': '255.255.252.0',
                                 'type': 'static',
+                                'ipv4': True,
                                 'routes': [{
                                     'gateway': '172.19.3.254',
                                     'netmask': '0.0.0.0',
diff --git a/tests/unittests/test_datasource/test_nocloud.py b/tests/unittests/test_datasource/test_nocloud.py
index 077603b4..b0fa1130 100644
--- a/tests/unittests/test_datasource/test_nocloud.py
+++ b/tests/unittests/test_datasource/test_nocloud.py
@@ -7,6 +7,8 @@ import os
 import shutil
 import tempfile
 
+import yaml
+
 
 class TestNoCloudDataSource(TestCase):
 
diff --git a/tests/unittests/test_datasource/test_smartos.py b/tests/unittests/test_datasource/test_smartos.py
index 2f159ac4..28f56039 100644
--- a/tests/unittests/test_datasource/test_smartos.py
+++ b/tests/unittests/test_datasource/test_smartos.py
@@ -42,7 +42,7 @@ from cloudinit import helpers as c_helpers
 from cloudinit.util import b64e
 
 from .. import helpers
-from ..helpers import mock, SkipTest
+from ..helpers import mock
 
 MOCK_RETURNS = {
     'hostname': 'test-host',
diff --git a/tests/unittests/test_net.py b/tests/unittests/test_net.py
index faf0f0fb..7998111a 100644
--- a/tests/unittests/test_net.py
+++ b/tests/unittests/test_net.py
@@ -1,13 +1,11 @@
 from cloudinit import net
-from cloudinit import util
-
-from cloudinit import net
 from cloudinit.net import cmdline
 from cloudinit.net import eni
 from cloudinit.net import network_state
+from cloudinit import util
 
-from .helpers import TestCase
 from .helpers import mock
+from .helpers import TestCase
 
 import base64
 import copy
diff --git a/tests/unittests/test_rh_subscription.py b/tests/unittests/test_rh_subscription.py
index e4dcc58b..891dbe77 100644
--- a/tests/unittests/test_rh_subscription.py
+++ b/tests/unittests/test_rh_subscription.py
@@ -10,13 +10,10 @@
 #    You should have received a copy of the GNU General Public License
 #    along with this program.  If not, see <http://www.gnu.org/licenses/>.
 
-
 import logging
-import mock
-import unittest
-
 
 from cloudinit.config import cc_rh_subscription
+from cloudinit import util
 
 from .helpers import TestCase, mock