1 files changed, 76 insertions, 0 deletions
diff --git a/source/test/StarTestUniverse.cpp b/source/test/StarTestUniverse.cpp
new file mode 100644
index 0000000..4602f34
--- /dev/null
+++ b/source/test/StarTestUniverse.cpp
@@ -0,0 +1,76 @@
+#include "StarTestUniverse.hpp"
+#include "StarFile.hpp"
+#include "StarQuests.hpp"
+#include "StarPlayerFactory.hpp"
+#include "StarPlayerStorage.hpp"
+#include "StarStatistics.hpp"
+#include "StarStatisticsService.hpp"
+#include "StarPlayer.hpp"
+#include "StarAssets.hpp"
+#include "StarWorldClient.hpp"
+
+namespace Star {
+
+TestUniverse::TestUniverse(Vec2U clientWindowSize) {
+  auto& root = Root::singleton();
+
+  m_clientWindowSize = clientWindowSize;
+
+  m_storagePath = File::temporaryDirectory();
+  auto playerStorage = make_shared<PlayerStorage>(File::relativeTo(m_storagePath, "player"));
+  auto statistics = make_shared<Statistics>(File::relativeTo(m_storagePath, "statistics"));
+  m_server = make_shared<UniverseServer>(File::relativeTo(m_storagePath, "universe"));
+  m_client = make_shared<UniverseClient>(playerStorage, statistics);
+
+  m_server->start();
+
+  m_mainPlayer = root.playerFactory()->create();
+  m_mainPlayer->finalizeCreation();
+  m_mainPlayer->setAdmin(true);
+  m_mainPlayer->setModeType(PlayerMode::Survival);
+  m_client->setMainPlayer(m_mainPlayer);
+  m_client->connect(m_server->addLocalClient(), "test", "");
+}
+
+TestUniverse::~TestUniverse() {
+  m_client = {};
+  m_server = {};
+  m_mainPlayer = {};
+  File::removeDirectoryRecursive(m_storagePath);
+}
+
+void TestUniverse::warpPlayer(WorldId worldId) {
+  m_client->warpPlayer(WarpToWorld(worldId), true);
+  while (m_mainPlayer->isTeleporting() || m_client->playerWorld().empty()) {
+    m_client->update();
+    Thread::sleep(16);
+  }
+}
+
+WorldId TestUniverse::currentPlayerWorld() const {
+  return m_client->clientContext()->playerWorldId();
+}
+
+void TestUniverse::update(unsigned times) {
+  for (unsigned i = 0; i < times; ++i) {
+    m_client->update();
+    Thread::sleep(16);
+  }
+}
+
+List<Drawable> TestUniverse::currentClientDrawables() {
+  WorldRenderData renderData;
+  auto worldClient = m_client->worldClient();
+  worldClient->centerClientWindowOnPlayer(m_clientWindowSize);
+  worldClient->render(renderData, 0);
+
+  List<Drawable> drawables;
+  for (auto& ed : renderData.entityDrawables) {
+    for (auto& p : ed.layers)
+      drawables.appendAll(move(p.second));
+  }
+
+  return drawables;
+}
+
+}