feat: add more tests

src/endstone_test/event_listener.py

@@ -2,13 +2,21 @@
 from functools import partial
 
 from endstone import ColorFormat, Server
-from endstone.event import event_handler, PlayerJoinEvent
+from endstone.event import event_handler, PlayerLoginEvent, PlayerJoinEvent
 from endstone.plugin import Plugin
 
 
 class EventListener:
     def __init__(self, plugin: Plugin):
         self._plugin = plugin
+        self._should_kick = True
+
+    @event_handler
+    def on_player_login(self, event: PlayerLoginEvent) -> None:
+        if self._should_kick:
+            event.kick_message = "Player::kick is working. Please join again."
+            event.cancelled = True
+            self._should_kick = False
 
     @event_handler
     def on_player_join(self, event: PlayerJoinEvent) -> None:

src/endstone_test/plugin.py

@@ -3,7 +3,6 @@
 
 from endstone import __minecraft_version__
 from endstone.plugin import Plugin
-
 from endstone_test.event_listener import EventListener
 
 
@@ -48,6 +47,14 @@ def on_enable(self) -> None:
         assert self.server.get_player("non-existent") is None
         assert self.server.get_player(uuid.uuid4()) is None
 
+        for level in self.server.levels:
+            self.logger.info(f"Level: {level.name}")
+            assert self.server.get_level(level.name) is level
+
+            for dimension in level.dimensions:
+                self.logger.info(f"\tDimension: {dimension.name}")
+                assert level.get_dimension(dimension.name) is dimension
+
         self.server.broadcast_message("Hello!")
         self.server.scheduler.run_task_timer(self, self.send_debug_message, delay=0, period=10)