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)