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)