diff options
| author | Daniel Watkins <daniel.watkins@canonical.com> | 2014-11-13 10:35:46 +0000 |
|---|---|---|
| committer | Daniel Watkins <daniel.watkins@canonical.com> | 2014-11-13 10:35:46 +0000 |
| commit | 5d168a9d11e905dd19e24d0596b7d1145e6b8aa1 (patch) | |
| tree | 6da28da6be87300e9299de78aa1b60be4aaf8997 /tests/unittests | |
| parent | 4ecca95e5973707f08fefc43448fec9e0f984966 (diff) | |
| download | vyos-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.py | 7 |
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') |