def test_list_protocols_by_name(self, *args):
protocol_vos = ProtocolFactory.build_batch(10,
domain_id=self.domain_id)
list(map(lambda vo: vo.save(), protocol_vos))
params = {'name': protocol_vos[0].name, 'domain_id': self.domain_id}
self.transaction.method = 'list'
protocol_svc = ProtocolService(transaction=self.transaction)
protocol_vos, total_count = protocol_svc.list(params)
ProtocolsInfo(protocol_vos, total_count)
self.assertEqual(len(protocol_vos), 1)
self.assertIsInstance(protocol_vos[0], Protocol)
self.assertEqual(total_count, 1)
def test_list_protocols_by_protocol_type(self, *args):
internal_protocol_vos = ProtocolFactory.build_batch(
5, protocol_type='INTERNAL', domain_id=self.domain_id)
external_protocol_vos = ProtocolFactory.build_batch(
10, protocol_type='EXTERNAL', domain_id=self.domain_id)
list(map(lambda vo: vo.save(), external_protocol_vos))
params = {'protocol_type': 'EXTERNAL', 'domain_id': self.domain_id}
self.transaction.method = 'list'
protocol_svc = ProtocolService(transaction=self.transaction)
protocol_vos, total_count = protocol_svc.list(params)
ProtocolsInfo(protocol_vos, total_count)
self.assertEqual(len(protocol_vos), 10)
self.assertIsInstance(protocol_vos[0], Protocol)
self.assertEqual(total_count, 10)