summaryrefslogtreecommitdiff
path: root/tests/unittests
diff options
context:
space:
mode:
authorDaniel Watkins <daniel.watkins@canonical.com>2014-11-13 10:35:46 +0000
committerDaniel Watkins <daniel.watkins@canonical.com>2014-11-13 10:35:46 +0000
commit5d168a9d11e905dd19e24d0596b7d1145e6b8aa1 (patch)
tree6da28da6be87300e9299de78aa1b60be4aaf8997 /tests/unittests
parent4ecca95e5973707f08fefc43448fec9e0f984966 (diff)
downloadvyos-cloud-init-5d168a9d11e905dd19e24d0596b7d1145e6b8aa1.tar.gz
vyos-cloud-init-5d168a9d11e905dd19e24d0596b7d1145e6b8aa1.zip
Use test_helpers.TestCase for test_sshutil tests.
As requested by harlowja.
Diffstat (limited to 'tests/unittests')
-rw-r--r--tests/unittests/test_sshutil.py7
1 files changed, 3 insertions, 4 deletions
diff --git a/tests/unittests/test_sshutil.py b/tests/unittests/test_sshutil.py
index cd576e8f..3b317121 100644
--- a/tests/unittests/test_sshutil.py
+++ b/tests/unittests/test_sshutil.py
@@ -1,7 +1,6 @@
-from unittest import TestCase
-
from mock import patch
+from . import helpers as test_helpers
from cloudinit import ssh_util
@@ -38,7 +37,7 @@ TEST_OPTIONS = ("no-port-forwarding,no-agent-forwarding,no-X11-forwarding,"
'user \"root\".\';echo;sleep 10"')
-class TestAuthKeyLineParser(TestCase):
+class TestAuthKeyLineParser(test_helpers.TestCase):
def test_simple_parse(self):
# test key line with common 3 fields (keytype, base64, comment)
parser = ssh_util.AuthKeyLineParser()
@@ -101,7 +100,7 @@ class TestAuthKeyLineParser(TestCase):
self.assertFalse(key.valid())
-class TestParseSSHConfig(TestCase):
+class TestParseSSHConfig(test_helpers.TestCase):
def setUp(self):
self.load_file_patch = patch('cloudinit.ssh_util.util.load_file')