def test_write_stderr(self):
message = "this is the message"
serializer = mock.create_autospec(filehandler.FileHandleManager)
fh = serializer.open.return_value = mock.create_autospec(filehandler.FileHandleWrapper)
ac = ActionCommand("action.1.do", "do", serializer)
ac.write_stderr(message)
fh.write.assert_called_with(message)
def test_write_stderr(self):
message = "this is the message"
fh = turtle.Turtle()
serializer = turtle.Turtle(open=lambda fn: fh)
ac = ActionCommand("action.1.do", "do", serializer)
ac.write_stderr(message)
assert_equal(fh.write.calls, [((message,), {})])
def test_write_stderr(self):
message = "this is the message"
serializer = mock.create_autospec(filehandler.FileHandleManager)
fh = serializer.open.return_value = mock.create_autospec(
filehandler.FileHandleWrapper)
ac = ActionCommand("action.1.do", "do", serializer)
ac.write_stderr(message)
fh.write.assert_called_with(message)
def build_action(task, command=None):
"""Create an action for a task which is an Observer, and which has
properties 'task_name', 'command', and 'buffer_store'.
"""
name = '%s.%s' % (task.id, task.task_name)
command = command or task.command
action = ActionCommand(name, command, serializer=task.buffer_store)
task.watch(action)
return action
def test_init_no_serializer(self):
ac = ActionCommand("action.1.do", "do")
ac.write_stdout("something")
ac.write_stderr("else")
assert_equal(ac.stdout, filehandler.NullFileHandle)
ac.done()
def setup_command(self):
self.serializer = mock.create_autospec(filehandler.FileHandleManager)
self.serializer.open.return_value = filehandler.NullFileHandle
self.ac = ActionCommand("action.1.do", "do", self.serializer)
class ActionCommandTestCase(TestCase):
@setup
def setup_command(self):
self.serializer = mock.create_autospec(filehandler.FileHandleManager)
self.serializer.open.return_value = filehandler.NullFileHandle
self.ac = ActionCommand("action.1.do", "do", self.serializer)
def test_init(self):
assert_equal(self.ac.state, ActionCommand.PENDING)
def test_init_no_serializer(self):
ac = ActionCommand("action.1.do", "do")
ac.write_stdout("something")
ac.write_stderr("else")
assert_equal(ac.stdout, filehandler.NullFileHandle)
ac.done()
def test_started(self):
assert self.ac.started()
assert self.ac.start_time is not None
assert_equal(self.ac.state, ActionCommand.RUNNING)
def test_started_already_started(self):
self.ac.started()
assert not self.ac.started()
def test_exited(self):
self.ac.started()
assert self.ac.exited(123)
assert_equal(self.ac.exit_status, 123)
assert self.ac.end_time is not None
def test_exited_from_pending(self):
assert self.ac.exited(123)
assert_equal(self.ac.state, ActionCommand.FAILSTART)
def test_exited_bad_state(self):
self.ac.started()
self.ac.exited(123)
assert not self.ac.exited(1)
def test_write_stderr_no_fh(self):
message = "this is the message"
# Test without a stderr
self.ac.write_stderr(message)
def test_write_stderr(self):
message = "this is the message"
serializer = mock.create_autospec(filehandler.FileHandleManager)
fh = serializer.open.return_value = mock.create_autospec(filehandler.FileHandleWrapper)
ac = ActionCommand("action.1.do", "do", serializer)
ac.write_stderr(message)
fh.write.assert_called_with(message)
def test_done(self):
self.ac.started()
self.ac.exited(123)
assert self.ac.done()
def test_done_bad_state(self):
assert not self.ac.done()
def test_handle_errback(self):
message = "something went wrong"
self.ac.handle_errback(message)
assert_equal(self.ac.state, ActionCommand.FAILSTART)
assert self.ac.end_time
def test_is_failed(self):
assert not self.ac.is_failed
def test_is_failed_true(self):
self.ac.exit_status = 255
assert self.ac.is_failed
def test_is_complete(self):
assert not self.ac.is_complete
def test_is_complete_true(self):
self.ac.machine.state = self.ac.COMPLETE
assert self.ac.is_complete, self.ac.machine.state
def test_is_done(self):
self.ac.machine.state = self.ac.FAILSTART
assert self.ac.is_done, self.ac.machine.state
self.ac.machine.state = self.ac.COMPLETE
assert self.ac.is_done, self.ac.machine.state
def test_init_no_serializer(self):
ac = ActionCommand("action.1.do", "do")
ac.write_stdout("something")
ac.write_stderr("else")
assert_equal(ac.stdout, filehandler.NullFileHandle)
ac.done()
def setup_command(self):
self.serializer = mock.create_autospec(filehandler.FileHandleManager)
self.serializer.open.return_value = filehandler.NullFileHandle
self.ac = ActionCommand("action.1.do", "do", self.serializer)
class ActionCommandTestCase(TestCase):
@setup
def setup_command(self):
self.serializer = mock.create_autospec(filehandler.FileHandleManager)
self.serializer.open.return_value = filehandler.NullFileHandle
self.ac = ActionCommand("action.1.do", "do", self.serializer)
def test_init(self):
assert_equal(self.ac.state, ActionCommand.PENDING)
def test_init_no_serializer(self):
ac = ActionCommand("action.1.do", "do")
ac.write_stdout("something")
ac.write_stderr("else")
assert_equal(ac.stdout, filehandler.NullFileHandle)
ac.done()
def test_started(self):
assert self.ac.started()
assert self.ac.start_time is not None
assert_equal(self.ac.state, ActionCommand.RUNNING)
def test_started_already_started(self):
self.ac.started()
assert not self.ac.started()
def test_exited(self):
self.ac.started()
assert self.ac.exited(123)
assert_equal(self.ac.exit_status, 123)
assert self.ac.end_time is not None
def test_exited_from_pending(self):
assert self.ac.exited(123)
assert_equal(self.ac.state, ActionCommand.FAILSTART)
def test_exited_bad_state(self):
self.ac.started()
self.ac.exited(123)
assert not self.ac.exited(1)
def test_write_stderr_no_fh(self):
message = "this is the message"
# Test without a stderr
self.ac.write_stderr(message)
def test_write_stderr(self):
message = "this is the message"
serializer = mock.create_autospec(filehandler.FileHandleManager)
fh = serializer.open.return_value = mock.create_autospec(
filehandler.FileHandleWrapper)
ac = ActionCommand("action.1.do", "do", serializer)
ac.write_stderr(message)
fh.write.assert_called_with(message)
def test_done(self):
self.ac.started()
self.ac.exited(123)
assert self.ac.done()
def test_done_bad_state(self):
assert not self.ac.done()
def test_handle_errback(self):
message = "something went wrong"
self.ac.handle_errback(message)
assert_equal(self.ac.state, ActionCommand.FAILSTART)
assert self.ac.end_time
def test_is_failed(self):
assert not self.ac.is_failed
def test_is_failed_true(self):
self.ac.exit_status = 255
assert self.ac.is_failed
def test_is_complete(self):
assert not self.ac.is_complete
def test_is_complete_true(self):
self.ac.machine.state = self.ac.COMPLETE
assert self.ac.is_complete, self.ac.machine.state
def test_is_done(self):
self.ac.machine.state = self.ac.FAILSTART
assert self.ac.is_done, self.ac.machine.state
self.ac.machine.state = self.ac.COMPLETE
assert self.ac.is_done, self.ac.machine.state
def setup_command(self):
null_fh = filehandler.NullFileHandle
self.serializer = turtle.Turtle(open=lambda fn: null_fh)
self.ac = ActionCommand("action.1.do", "do", self.serializer)
class ActionCommandTestCase(TestCase):
@setup
def setup_command(self):
null_fh = filehandler.NullFileHandle
self.serializer = turtle.Turtle(open=lambda fn: null_fh)
self.ac = ActionCommand("action.1.do", "do", self.serializer)
def test_init(self):
assert_equal(self.ac.state, ActionCommand.PENDING)
def test_init_no_serializer(self):
ac = ActionCommand("action.1.do", "do")
ac.write_stdout("something")
ac.write_stderr("else")
assert_equal(ac.stdout, filehandler.NullFileHandle)
ac.done()
def test_started(self):
assert self.ac.started()
assert self.ac.start_time is not None
assert_equal(self.ac.state, ActionCommand.RUNNING)
def test_started_already_started(self):
self.ac.started()
assert not self.ac.started()
def test_exited(self):
self.ac.started()
assert self.ac.exited(123)
assert_equal(self.ac.exit_status, 123)
assert self.ac.end_time is not None
def test_exited_from_pending(self):
assert self.ac.exited(123)
assert_equal(self.ac.state, ActionCommand.FAILSTART)
def test_exited_bad_state(self):
self.ac.started()
self.ac.exited(123)
assert not self.ac.exited(1)
def test_write_stderr_no_fh(self):
message = "this is the message"
# Test without a stderr
self.ac.write_stderr(message)
def test_write_stderr(self):
message = "this is the message"
fh = turtle.Turtle()
serializer = turtle.Turtle(open=lambda fn: fh)
ac = ActionCommand("action.1.do", "do", serializer)
ac.write_stderr(message)
assert_equal(fh.write.calls, [((message,), {})])
def test_done(self):
self.ac.started()
self.ac.exited(123)
assert self.ac.done()
def test_done_bad_state(self):
assert not self.ac.done()
def test_handle_errback(self):
message = "something went wrong"
self.ac.handle_errback(message)
assert_equal(self.ac.state, ActionCommand.FAILSTART)
assert self.ac.end_time