← Back to branch summary

~smoser/ubuntu/trusty/maas/lp-1172566

« back to all changes in this revision

Viewing changes to src/maasserver/models/tests/test_sshkey.py

  • Committer: Package Import Robot
  • Author(s): Andres Rodriguez
  • Date: 2014-04-03 13:45:02 UTC
  • mto: This revision was merged to the branch mainline in revision 58.
  • Revision ID: package-import@ubuntu.com-20140403134502-8a6wvuqwyuekufh0
Tags: upstream-1.5+bzr2227
Import upstream version 1.5+bzr2227

Show diffs side-by-side

added added

removed removed

Lines of Context:
src/maasserver/models/tests/test_sshkey.py
23
23
from maasserver.models.sshkey import (
24
24
    get_html_display_for_key,
25
25
    HELLIPSIS,
 
26
    Key,
26
27
    validate_ssh_public_key,
27
28
    )
28
29
from maasserver.testing import get_data
29
30
from maasserver.testing.factory import factory
30
31
from maasserver.testing.testcase import MAASServerTestCase
 
32
from mock import Mock
31
33
from testtools.matchers import EndsWith
32
34
 
33
35
 
63
65
        self.assertRaises(
64
66
            ValidationError, validate_ssh_public_key, key_string)
65
67
 
 
68
    def test_does_not_validate_wrong_key(self):
 
69
        # If twisted.conch.ssh.keys.Key raises an exception, the validation
 
70
        # fails.
 
71
        self.patch(Key, 'fromString', Mock(side_effect=Exception()))
 
72
        self.assertRaises(
 
73
            ValidationError, validate_ssh_public_key, factory.make_name('key'))
 
74
 
66
75
    def test_does_not_validate_rsa_private_key(self):
67
76
        key_string = get_data('data/test_rsa')
68
77
        self.assertRaises(