def test_main_without_worker_config(self):
args = munch.Munch({
'cmd': '',
'auction_worker_config': 'path/to/config',
'with_api_version': None,
'auction_doc_id': '1' * 32,
'debug': False
})
self.mocked_os.path.isfile.return_value = False
self.mocked_parser_obj.parse_args.return_value = args
with self.assertRaises(SystemExit):
main()
self.assertEqual(self.mocked_os.path.isfile.call_count, 1)
self.assertEqual(self.mocked_yaml.load.call_count, 0)
self.assertEqual(self.mocked_open.call_count, 0)
self.assertEqual(self.mocked_logging.config.dictConfig.call_count, 0)
self.assertEqual(self.mocked_register_utilities.call_count, 0)
self.assertEqual(self.mocked_auction_class.call_count, 0)
def test_cmd_check(self):
args = munch.Munch({
'cmd': 'check',
'auction_worker_config': 'path/to/config',
'with_api_version': 'another api version',
'auction_doc_id': '1' * 32,
'debug': False
})
self.mocked_parser_obj.parse_args.return_value = args
with self.assertRaises(SystemExit):
main()
def test_cmd_cancel(self):
args = munch.Munch({
'cmd': 'cancel',
'auction_worker_config': 'path/to/config',
'with_api_version': 'another api version',
'auction_doc_id': '1' * 32,
'debug': False
})
self.mocked_parser_obj.parse_args.return_value = args
main()
self.assertEqual(self.auction_instance.cancel_auction.call_count, 1)
self.auction_instance.cancel_auction.assert_called_with()
def test_cmd_post_auction_protocol(self):
args = munch.Munch({
'cmd': 'post_auction_protocol',
'auction_worker_config': 'path/to/config',
'with_api_version': 'another api version',
'auction_doc_id': '1' * 32,
'debug': False,
'doc_id': 'some document id'
})
self.mocked_parser_obj.parse_args.return_value = args
main()
self.assertEqual(
self.auction_instance.post_auction_protocol.call_count, 1)
self.auction_instance.post_auction_protocol.assert_called_with(
args.doc_id)
def test_without_cleanup(self):
resulted_yaml = deepcopy(self.yaml_output)
args = munch.Munch({
'cmd': '',
'auction_worker_config': 'path/to/config',
'with_api_version': 'another api version',
'auction_doc_id': '1' * 32,
'debug': False
})
self.mocked_parser_obj.parse_args.return_value = args
main()
self.assertEqual(self.mocked_os.path.isfile.call_count, 1)
self.mocked_os.path.isfile.assert_called_with(
args.auction_worker_config)
self.assertEqual(self.mocked_yaml.load.call_count, 1)
self.mocked_yaml.load.assert_called_with(self.open_result)
self.assertEqual(self.mocked_open.call_count, 1)
self.mocked_open.assert_called_with(args.auction_worker_config)
self.assertEqual(self.mocked_logging.config.dictConfig.call_count, 1)
resulted_yaml['resource_api_version'] = args.with_api_version
resulted_yaml['handlers']['journal'][
'TENDERS_API_VERSION'] = resulted_yaml['resource_api_version']
resulted_yaml['handlers']['journal'][
'TENDERS_API_URL'] = resulted_yaml['resource_api_server']
resulted_yaml['handlers']['journal']['TENDER_ID'] = args.auction_doc_id
self.mocked_logging.config.dictConfig.assert_called_with(resulted_yaml)
self.assertEqual(self.mocked_register_utilities.call_count, 1)
self.mocked_register_utilities.assert_called_with(resulted_yaml, args)
self.assertEqual(self.mocked_auction_class.call_count, 1)
self.mocked_auction_class.assert_called_with(
args.auction_doc_id,
worker_defaults=resulted_yaml,
debug=args.debug)
def test_cmd_run(self):
args = munch.Munch({
'cmd': 'run',
'auction_worker_config': 'path/to/config',
'with_api_version': 'another api version',
'auction_doc_id': '1' * 32,
'debug': False
})
self.mocked_parser_obj.parse_args.return_value = args
main()
self.assertEqual(self.mocked_SCHEDULER.start.call_count, 1)
self.mocked_SCHEDULER.start.assert_called_with()
self.assertEqual(self.mocked_SCHEDULER.shutdown.call_count, 1)
self.mocked_SCHEDULER.shutdown.assert_called_with()
self.assertEqual(self.auction_instance.schedule_auction.call_count, 1)
self.auction_instance.schedule_auction.assert_called_with()
self.assertEqual(self.auction_instance.wait_to_end.call_count, 1)
self.auction_instance.wait_to_end.assert_called_with()