def test_can_Pipeline_multiple_calls(self):
response = "We can't leave him here. May I keep him?"
def first_function(param):
return 'first', param
def second_function(param):
return 'second', param
action = FakeAction(instruction_provider=lambda: response)
action_types = [
Pipeline.Fitting(
action=Call,
enclose=first_function,
),
Pipeline.Fitting(
action=Call,
enclose=second_function,
)
]
pipeline = Pipeline(action, *action_types, should_raise=True)
result = pipeline.perform(should_raise=True)
self.assertIsInstance(result, Result)
self.assertEqual(result.value, ('second', ('first', response)))
def test_Pipeline_Fitting(self, mock_input, mock_exists, mock_file):
filename = 'this/file.txt'
file = FakeFile()
question = b'How are you?'
reply = 'I\'m fine.'
mock_file.return_value = file
mock_exists.return_value = True
mock_input.side_effect = [filename, reply]
read_input = ReadInput('Which file?')
action_types = [
Pipeline.Fitting(
action=Write,
reaction=Call(Closure(bytes.decode)),
**{'append': True, 'to_write': question},
),
Read, # retrieve question from FakeFile
ReadInput # pose question to user
]
pipeline = Pipeline(read_input, *action_types, should_raise=True)
result = pipeline.perform(should_raise=True)
self.assertEqual(file.read(), question + b'\n')
self.assertIsInstance(result, Result)
self.assertEqual(result.value, reply)
def test_can_repeatedly_perform_Pipeline_with_Receiver(self, mock_input, mock_exists, mock_file):
filename = 'this/file.txt'
file = FakeFile()
reply = b'The machines have worked perfectly up till now.'
mock_file.return_value = file
mock_exists.return_value = True
mock_input.return_value = reply
read_input = ReadInput('What would you like to record?')
action_types = [
Pipeline.Fitting(
action=Write,
reaction=Call(Closure(bytes.decode)),
**{'append': True, 'filename': filename, 'to_write': Pipeline.Receiver},
)
]
pipeline = Pipeline(read_input, *action_types, should_raise=True)
result = pipeline.perform(should_raise=True)
self.assertEqual(file.read(), reply + b'\n')
self.assertIsInstance(result, Result)
self.assertEqual(result.value, f'{cwd()}/{filename}')
second_result = pipeline.perform(should_raise=True)
self.assertEqual(file.read(), reply + b'\n' + reply + b'\n')
self.assertIsInstance(second_result, Result)
self.assertEqual(second_result.value, f'{cwd()}/{filename}')
def test_first_Pipeline_failure_prevents_further_execution(self, mock_exists):
bad_filename = 'bad/filename.txt'
mock_exists.return_value = False
pipeline = Pipeline(Read(bad_filename), ReadInput)
result = pipeline.perform()
self.assertIsInstance(result, Result)
self.assertIsInstance(result.value, FileNotFoundError)
self.assertEqual(str(result.value), bad_filename)
def test_Pipeline(self, mock_input, mock_exists, mock_file):
filename = 'this/file.txt'
contents = b"What's wrong with him? ...My first thought would be, 'a lot'."
file = FakeFile(contents)
mock_file.return_value = file
mock_exists.return_value = True
mock_input.return_value = filename
pipeline = Pipeline(ReadInput('Which file?'), Read)
result = pipeline.perform()
self.assertIsInstance(result, Result)
self.assertEqual(result.value, contents)
def test_can_perform_Pipeline_repeatedly(self, mock_input, mock_exists, mock_file):
filename = 'this/file.txt'
contents = b'Look at the little Om. Isn\'t he sweet?'
file = FakeFile(contents)
mock_file.return_value = file
mock_exists.return_value = True
mock_input.return_value = filename
pipeline = Pipeline(ReadInput('Which file?'), Read)
result = pipeline.perform()
self.assertIsInstance(result, Result)
self.assertEqual(result.value, contents)
self.assertEqual(file.read(), contents)
second_result = pipeline.perform()
self.assertIsInstance(second_result, Result)
self.assertEqual(second_result.value, contents)
self.assertEqual(file.read(), contents)
def test_Pipeline_Fitting_defines_instruction_replacement(self):
result = Pipeline.Fitting.instruction(Pipeline.Fitting(FakeAction()))
self.assertTrue(result.successful)
result = Pipeline.Fitting.instruction(Pipeline.Fitting(FakeAction(), reaction=FakeAction()))
self.assertTrue(result.successful)
def test_can_Pipeline_constitutes_iterator_over_Action_types(self):
action_types = [FakeAction] * 3
pipeline = Pipeline(FakeAction(), *action_types)
self.assertEqual(list(pipeline), action_types)
def test_instantiation_fails_with_invalid_action_types(self):
pipeline = Pipeline(ReadInput('Which file?'), 'not an ActionType')
self.assertIsInstance(pipeline, Action.Construct)
self.assertIsInstance(pipeline.failure, TypeError)