def test_program_start(self, mock_socket):
program = Program.objects.get(name="program")
program_start(program.pk)
mock_socket.connect().assert_called_once()
send_calls = [
mock.call(
"prog.add {0}\n".format(program.pk).encode(encoding='utf_8')),
mock.call("exit\n".encode(encoding='utf_8'))
]
mock_socket.send().has_calls(send_calls)
mock_socket.close().assert_called_once()
def test_program_start_no_exists(self, mock_socket):
program_start(300000)
self.assertFalse(mock_socket.connect().called)
self.assertFalse(mock_socket.send().called)
self.assertFalse(mock_socket.close().called)
