// Cyphesis Online RPG Server and AI Engine
// Copyright (C) 2009 Alistair Riddoch
//
// This program is free software; you can redistribute it and/or modify
// it under the terms of the GNU General Public License as published by
// the Free Software Foundation; either version 2 of the License, or
// (at your option) any later version.
// 
// This program is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
// GNU General Public License for more details.
// 
// You should have received a copy of the GNU General Public License
// along with this program; if not, write to the Free Software Foundation,
// Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA


#ifdef NDEBUG
#undef NDEBUG
#endif
#ifndef DEBUG
#define DEBUG
#endif

#include <Python.h>

#include "../python_testers.h"

#include "client/cyclient/Python_ClientAPI.h"
#include "client/cyclient/ObserverClient.h"
#include "client/cyclient/CyPy_ObserverClient.h"

#include "pythonbase/Python_API.h"
#include "client/SimpleTypeStore.h"
#include "../NullPropertyManager.h"

#include <cassert>
#include <rules/simulation/python/CyPy_Server.h>
#include <rules/python/CyPy_Atlas.h>
#include <rules/simulation/Inheritance.h>
#include "pythonbase/PythonMalloc.h"
#include "client/cyclient/CreatorClient.h"

Atlas::Objects::Factories factories;

static bool stub_setup_fail = false;
static bool stub_createCharacter_fail = false;
static bool stub_send_wait_fail = false;
static int stub_send_wait_results = 0;
static bool stub_wait_fail = false;

int main() {
	setupPythonMalloc();
	{
		NullPropertyManager<MemEntity> propertyManager;
		SimpleTypeStore typeStore(propertyManager);

		Inheritance inheritance;
		boost::asio::io_context io_context;

		init_python_api({&CyPy_Server::init, &CyPy_Atlas::init});
		extend_client_python_api();

		ObserverClient client(io_context, factories, typeStore);

		Py::Module module("server");
		module.setAttr("testclient", CyPy_ObserverClient::wrap(&client));

		run_python_string("import atlas");
		run_python_string("import server");
		run_python_string("import types");
		run_python_string("o=server.testclient");
		run_python_string("o.setup()");
		expect_python_error("o.setup('bob')", PyExc_IndexError);
		run_python_string("o.setup('bob', 'jim')");
		run_python_string("o.setup('bob', 'jim', 'settler')");
		stub_setup_fail = true;
		expect_python_error("o.setup('bob', 'jim', 'settler')",
							PyExc_RuntimeError);
		stub_setup_fail = false;
		run_python_string("o.create_avatar('settler')");
		expect_python_error("o.create_avatar(1)", PyExc_TypeError);
		run_python_string("o.run()");
		expect_python_error("o.send()", PyExc_IndexError);
		expect_python_error("o.send('get')", PyExc_TypeError);
		run_python_string("o.send(atlas.Operation('get'))");
		expect_python_error("o.send_wait()", PyExc_IndexError);
		expect_python_error("o.send_wait('get')", PyExc_TypeError);
		run_python_string("o.send_wait(atlas.Operation('get'))");
		stub_send_wait_results = 1;
		run_python_string("assert type(o.send_wait(atlas.Operation('get'))) == atlas.Operation");
		stub_send_wait_results = 2;
		run_python_string("assert type(o.send_wait(atlas.Operation('get'))) == atlas.Operation");
		stub_send_wait_fail = true;
		// FIXME This really should fail
		// expect_python_error("o.send_wait(atlas.Operation('get'))",
		//                     PyExc_AssertionError);
		run_python_string("o.wait()");
		stub_wait_fail = true;
		expect_python_error("o.wait()", PyExc_RuntimeError);
		run_python_string("assert type(o.id) == str");
		run_python_string("o.character");
		run_python_string("o.server = 'foo'");
		expect_python_error("o.server = 23", PyExc_TypeError);
	}
	shutdown_python_api();
	return 0;
}

int ObserverClient::setup(const std::string& account, const std::string& password, const std::string& avatar) {
	if (stub_setup_fail) {
		return -1;
	}
	return 0;
}


Ref<CreatorClient> BaseClientLegacy::createCharacter(const std::string& type) {
	if (stub_createCharacter_fail) {
		return nullptr;
	}
	return Ref<CreatorClient>(new CreatorClient(1, "2", m_connection, m_typeStore));
}


int ClientConnection::wait() {
	if (stub_wait_fail) {
		return -1;
	}
	return 0;
}


int ClientConnection::sendAndWaitReply(const Operation& op, OpVector& res) {
	if (stub_send_wait_fail) {
		return -1;
	}
	for (int i = 0; i < stub_send_wait_results; ++i) {
		res.push_back(Atlas::Objects::Operation::Info());
	}
	return 0;
}
