class RunnerTests(unittest.TestCase):
# XXX : test more than just --map and --reduce
def setUp(self):
self.mock_job = MockJob()
self.runner = Runner(self.get_mock_job)
def get_mock_job(self):
return self.mock_job
@test
def runner_should_start_mapping_if_told_to(self):
self.runner.main(arguments=['--map'])
self.assertEqual(['start_map'], self.mock_job.get_actions())
@test
def runner_should_start_reducing_if_told_to(self):
self.runner.main(arguments=['--reduce'])
self.assertEqual(['start_reduce'], self.mock_job.get_actions())
def setUp(self): self.mock_job = MockJob() self.runner = Runner(self.get_mock_job)
