Ejemplo n.º 1
0
 def test_load_from_config(self, mock_service):
     autospec_method(self.collection.get_names)
     autospec_method(self.collection.add)
     service_configs = {'a': mock.Mock(), 'b': mock.Mock()}
     context = mock.create_autospec(command_context.CommandContext)
     result = list(self.collection.load_from_config(service_configs, context))
     expected = [mock.call(config, context)
                 for config in service_configs.itervalues()]
     assert_mock_calls(expected, mock_service.from_config.mock_calls)
     expected = [mock.call(s) for s in result]
     assert_mock_calls(expected, self.collection.add.mock_calls)
Ejemplo n.º 2
0
 def test_load_from_config(self, mock_service):
     autospec_method(self.collection.get_names)
     autospec_method(self.collection.add)
     service_configs = {'a': mock.Mock(), 'b': mock.Mock()}
     context = mock.create_autospec(command_context.CommandContext)
     result = list(self.collection.load_from_config(
         service_configs, context,
     ))
     expected = [mock.call(config, context)
                 for config in six.itervalues(service_configs)]
     assert_mock_calls(expected, mock_service.from_config.mock_calls)
     expected = [mock.call(s) for s in result]
     assert_mock_calls(expected, self.collection.add.mock_calls)
Ejemplo n.º 3
0
 def test_load_from_config(self):
     autospec_method(self.collection.jobs.filter_by_name)
     autospec_method(self.collection.add)
     factory = mock.create_autospec(job.JobSchedulerFactory)
     job_configs = {'a': mock.Mock(), 'b': mock.Mock()}
     result = self.collection.load_from_config(job_configs, factory, True)
     result = list(result)
     self.collection.jobs.filter_by_name.assert_called_with(job_configs)
     expected_calls = [mock.call(v) for v in job_configs.itervalues()]
     assert_mock_calls(expected_calls, factory.build.mock_calls)
     assert_length(self.collection.add.mock_calls, len(job_configs) * 2)
     assert_length(result, len(job_configs))
     job_schedulers = [call[1][0] for call in self.collection.add.mock_calls[::2]]
     for job_scheduler in job_schedulers:
         job_scheduler.schedule.assert_called_with()
         job_scheduler.get_job.assert_called_with()
Ejemplo n.º 4
0
 def test_load_from_config(self):
     autospec_method(self.collection.jobs.filter_by_name)
     autospec_method(self.collection.add)
     factory = mock.create_autospec(JobSchedulerFactory)
     job_configs = {'a': mock.Mock(), 'b': mock.Mock()}
     result = self.collection.load_from_config(job_configs, factory, True)
     result = list(result)
     self.collection.jobs.filter_by_name.assert_called_with(job_configs)
     expected_calls = [mock.call(v) for v in job_configs.values()]
     assert_mock_calls(expected_calls, factory.build.mock_calls)
     assert_length(self.collection.add.mock_calls, len(job_configs) * 2)
     assert_length(result, len(job_configs))
     job_schedulers = [
         call[1][0] for call in self.collection.add.mock_calls[::2]
     ]
     for job_scheduler in job_schedulers:
         job_scheduler.schedule.assert_called_with()
         job_scheduler.get_job.assert_called_with()
Ejemplo n.º 5
0
 def test_filter_by_name(self):
     autospec_method(self.collection.remove)
     self.collection.update(dict.fromkeys(['c', 'd', 'e']))
     self.collection.filter_by_name(['a', 'c'])
     expected = [mock.call(name) for name in ['d', 'e']]
     assert_mock_calls(expected, self.collection.remove.mock_calls)
Ejemplo n.º 6
0
 def test_filter_by_name(self):
     autospec_method(self.collection.remove)
     self.collection.update(dict.fromkeys(['c', 'd', 'e']))
     self.collection.filter_by_name(['a', 'c'])
     expected = [mock.call(name) for name in ['d', 'e']]
     assert_mock_calls(expected, self.collection.remove.mock_calls)