formats: curve name should be a string

This commit is contained in:
Roman Zeyde
2016-05-05 22:28:06 +03:00
parent 4cbf8a9f0a
commit b2147a8418
4 changed files with 9 additions and 9 deletions

View File

@@ -29,7 +29,7 @@ class FakeConnection(object):
def get_public_node(self, n, ecdsa_curve_name=b'secp256k1'):
assert not self.closed
assert n == ADDR
assert ecdsa_curve_name in {b'secp256k1', b'nist256p1'}
assert ecdsa_curve_name in {'secp256k1', 'nist256p1'}
result = mock.Mock(spec=[])
result.node = mock.Mock(spec=[])
result.node.public_key = PUBKEY
@@ -91,7 +91,7 @@ def test_ssh_agent():
client.identity_to_string(ident))
assert challenge_hidden == BLOB
assert challenge_visual == 'VISUAL'
assert ecdsa_curve_name == b'nist256p1'
assert ecdsa_curve_name == 'nist256p1'
result = mock.Mock(spec=[])
result.public_key = PUBKEY