← Back to branch summary

~ntt-pf-lab/nova/monkey_patch_notification

« back to all changes in this revision

Viewing changes to vendor/Twisted-10.0.0/twisted/conch/test/test_ckeygen.py

  • Committer: Jesse Andrews
  • Date: 2010-05-28 06:05:26 UTC
  • Revision ID: git-v1:bf6e6e718cdc7488e2da87b21e258ccc065fe499
initial commit

Show diffs side-by-side

added added

removed removed

Lines of Context:
vendor/Twisted-10.0.0/twisted/conch/test/test_ckeygen.py
 
1
# Copyright (c) 2008 Twisted Matrix Laboratories.
 
2
# See LICENSE for details.
 
3
 
 
4
"""
 
5
Tests for L{twisted.conch.scripts.ckeygen}.
 
6
"""
 
7
 
 
8
import sys
 
9
from StringIO import StringIO
 
10
 
 
11
try:
 
12
    import Crypto
 
13
    import pyasn1
 
14
except ImportError:
 
15
    skip = "PyCrypto and pyasn1 required for twisted.conch.scripts.ckeygen."
 
16
else:
 
17
    from twisted.conch.ssh.keys import Key
 
18
    from twisted.conch.scripts.ckeygen import printFingerprint, _saveKey
 
19
 
 
20
from twisted.python.filepath import FilePath
 
21
from twisted.trial.unittest import TestCase
 
22
from twisted.conch.test.keydata import publicRSA_openssh, privateRSA_openssh
 
23
 
 
24
 
 
25
 
 
26
class KeyGenTests(TestCase):
 
27
    """
 
28
    Tests for various functions used to implement the I{ckeygen} script.
 
29
    """
 
30
    def setUp(self):
 
31
        """
 
32
        Patch C{sys.stdout} with a L{StringIO} instance to tests can make
 
33
        assertions about what's printed.
 
34
        """
 
35
        self.stdout = StringIO()
 
36
        self.patch(sys, 'stdout', self.stdout)
 
37
 
 
38
 
 
39
    def test_printFingerprint(self):
 
40
        """
 
41
        L{printFingerprint} writes a line to standard out giving the number of
 
42
        bits of the key, its fingerprint, and the basename of the file from it
 
43
        was read.
 
44
        """
 
45
        filename = self.mktemp()
 
46
        FilePath(filename).setContent(publicRSA_openssh)
 
47
        printFingerprint({'filename': filename})
 
48
        self.assertEqual(
 
49
            self.stdout.getvalue(),
 
50
            '768 3d:13:5f:cb:c9:79:8a:93:06:27:65:bc:3d:0b:8f:af temp\n')
 
51
 
 
52
 
 
53
    def test_saveKey(self):
 
54
        """
 
55
        L{_saveKey} writes the private and public parts of a key to two
 
56
        different files and writes a report of this to standard out.
 
57
        """
 
58
        base = FilePath(self.mktemp())
 
59
        base.makedirs()
 
60
        filename = base.child('id_rsa').path
 
61
        key = Key.fromString(privateRSA_openssh)
 
62
        _saveKey(
 
63
            key.keyObject,
 
64
            {'filename': filename, 'pass': 'passphrase'})
 
65
        self.assertEqual(
 
66
            self.stdout.getvalue(),
 
67
            "Your identification has been saved in %s\n"
 
68
            "Your public key has been saved in %s.pub\n"
 
69
            "The key fingerprint is:\n"
 
70
            "3d:13:5f:cb:c9:79:8a:93:06:27:65:bc:3d:0b:8f:af\n" % (
 
71
                filename,
 
72
                filename))
 
73
        self.assertEqual(
 
74
            key.fromString(
 
75
                base.child('id_rsa').getContent(), None, 'passphrase'),
 
76
            key)
 
77
        self.assertEqual(
 
78
            Key.fromString(base.child('id_rsa.pub').getContent()),
 
79
            key.public())
 
80