class GmpCreateCredentialTestCase(unittest.TestCase):
def setUp(self):
self.connection = MockConnection()
self.gmp = Gmp(self.connection)
def test_create_up_credential_missing_name(self):
with self.assertRaises(RequiredArgument):
self.gmp.create_credential(name='',
credential_type='up',
login='******')
with self.assertRaises(RequiredArgument):
self.gmp.create_credential(name=None,
credential_type='up',
login='******')
def test_create_up_credential(self):
self.gmp.create_credential(
name='foo',
credential_type='up',
comment='bar',
login='******',
password='******',
)
self.connection.send.has_been_called_with('<create_credential>'
'<name>foo</name>'
'<type>up</type>'
'<comment>bar</comment>'
'<login>Max</login>'
'<password>123</password>'
'</create_credential>')
def test_create_up_credential_with_allow_insecure(self):
self.gmp.create_credential(
name='foo',
credential_type='up',
comment='bar',
login='******',
password='******',
allow_insecure=True,
)
self.connection.send.has_been_called_with(
'<create_credential>'
'<name>foo</name>'
'<type>up</type>'
'<comment>bar</comment>'
'<allow_insecure>1</allow_insecure>'
'<login>Max</login>'
'<password>123</password>'
'</create_credential>')
self.gmp.create_credential(
name='foo',
credential_type='up',
comment='bar',
login='******',
password='******',
allow_insecure=False,
)
self.connection.send.has_been_called_with(
'<create_credential>'
'<name>foo</name>'
'<type>up</type>'
'<comment>bar</comment>'
'<allow_insecure>0</allow_insecure>'
'<login>Max</login>'
'<password>123</password>'
'</create_credential>')
def test_cc_credential_missing_certificate(self):
with self.assertRaises(RequiredArgument):
self.gmp.create_credential(name='foo', credential_type='cc')
def test_create_cc_credential(self):
self.gmp.create_credential(name='foo',
credential_type='cc',
certificate='abcdef')
self.connection.send.has_been_called_with(
'<create_credential>'
'<name>foo</name>'
'<type>cc</type>'
'<certificate>abcdef</certificate>'
'</create_credential>')
def test_create_cc_credential_with_private_key(self):
self.gmp.create_credential(
name='foo',
credential_type='cc',
certificate='abcdef',
private_key='123456',
)
self.connection.send.has_been_called_with(
'<create_credential>'
'<name>foo</name>'
'<type>cc</type>'
'<certificate>abcdef</certificate>'
'<key>'
'<private>123456</private>'
'</key>'
'</create_credential>')
def test_create_usk_credential_missing_private_key(self):
with self.assertRaises(RequiredArgument):
self.gmp.create_credential(name='foo',
credential_type='usk',
login='******')
def test_create_usk_credential_missing_login(self):
with self.assertRaises(RequiredArgument):
self.gmp.create_credential(name='foo',
credential_type='usk',
private_key='123456')
def test_create_usk_credential(self):
self.gmp.create_credential(name='foo',
credential_type='usk',
private_key='123456',
login='******')
self.connection.send.has_been_called_with('<create_credential>'
'<name>foo</name>'
'<type>usk</type>'
'<login>foo</login>'
'<key>'
'<private>123456</private>'
'</key>'
'</create_credential>')
def test_create_usk_credential_with_key_phrase(self):
self.gmp.create_credential(
name='foo',
credential_type='usk',
private_key='123456',
login='******',
key_phrase='abcdef',
)
self.connection.send.has_been_called_with('<create_credential>'
'<name>foo</name>'
'<type>usk</type>'
'<login>foo</login>'
'<key>'
'<private>123456</private>'
'<phrase>abcdef</phrase>'
'</key>'
'</create_credential>')
def test_create_snmp_credential_invalid_auth_algorithm(self):
with self.assertRaises(InvalidArgument):
self.gmp.create_credential(name='foo',
credential_type='snmp',
login='******')
with self.assertRaises(InvalidArgument):
self.gmp.create_credential(
name='foo',
credential_type='snmp',
login='******',
auth_algorithm='',
)
with self.assertRaises(InvalidArgument):
self.gmp.create_credential(
name='foo',
credential_type='snmp',
login='******',
auth_algorithm='bar',
)
def test_create_snmp_credential_auth_algorithm_md5(self):
self.gmp.create_credential(
name='foo',
credential_type='snmp',
login='******',
auth_algorithm='md5',
)
self.connection.send.has_been_called_with(
'<create_credential>'
'<name>foo</name>'
'<type>snmp</type>'
'<login>foo</login>'
'<auth_algorithm>md5</auth_algorithm>'
'</create_credential>')
def test_create_snmp_credential_auth_algorithm_sha1(self):
self.gmp.create_credential(
name='foo',
credential_type='snmp',
login='******',
auth_algorithm='sha1',
)
self.connection.send.has_been_called_with(
'<create_credential>'
'<name>foo</name>'
'<type>snmp</type>'
'<login>foo</login>'
'<auth_algorithm>sha1</auth_algorithm>'
'</create_credential>')
def test_create_snmp_credential_with_community(self):
self.gmp.create_credential(
name='foo',
credential_type='snmp',
login='******',
auth_algorithm='sha1',
community='ipsum',
)
self.connection.send.has_been_called_with(
'<create_credential>'
'<name>foo</name>'
'<type>snmp</type>'
'<login>foo</login>'
'<auth_algorithm>sha1</auth_algorithm>'
'<community>ipsum</community>'
'</create_credential>')
def test_create_snmp_credential_invalid_privacy_algorithm(self):
with self.assertRaises(InvalidArgument):
self.gmp.create_credential(
name='foo',
credential_type='snmp',
login='******',
auth_algorithm='sha1',
privacy_algorithm='',
)
with self.assertRaises(InvalidArgument):
self.gmp.create_credential(
name='foo',
credential_type='snmp',
login='******',
auth_algorithm='sha1',
privacy_algorithm='foo',
)
def test_create_snmp_credential_with_privacy_algorithm_aes(self):
self.gmp.create_credential(
name='foo',
credential_type='snmp',
login='******',
auth_algorithm='sha1',
privacy_algorithm='aes',
)
self.connection.send.has_been_called_with(
'<create_credential>'
'<name>foo</name>'
'<type>snmp</type>'
'<login>foo</login>'
'<auth_algorithm>sha1</auth_algorithm>'
'<privacy>'
'<algorithm>aes</algorithm>'
'</privacy>'
'</create_credential>')
def test_create_snmp_credential_with_privacy_algorithm_des(self):
self.gmp.create_credential(
name='foo',
credential_type='snmp',
login='******',
auth_algorithm='sha1',
privacy_algorithm='des',
)
self.connection.send.has_been_called_with(
'<create_credential>'
'<name>foo</name>'
'<type>snmp</type>'
'<login>foo</login>'
'<auth_algorithm>sha1</auth_algorithm>'
'<privacy>'
'<algorithm>des</algorithm>'
'</privacy>'
'</create_credential>')
def test_create_snmp_credential_with_privacy_password(self):
self.gmp.create_credential(
name='foo',
credential_type='snmp',
login='******',
auth_algorithm='sha1',
privacy_password='******',
)
self.connection.send.has_been_called_with(
'<create_credential>'
'<name>foo</name>'
'<type>snmp</type>'
'<login>foo</login>'
'<auth_algorithm>sha1</auth_algorithm>'
'<privacy>'
'<password>123</password>'
'</privacy>'
'</create_credential>')
def test_create_credential_invalid_credential_type(self):
with self.assertRaises(InvalidArgument):
self.gmp.create_credential(name='foo', credential_type=None)
with self.assertRaises(InvalidArgument):
self.gmp.create_credential(name='foo', credential_type='')
with self.assertRaises(InvalidArgument):
self.gmp.create_credential(name='foo', credential_type='bar')
