1 files changed, 35 insertions, 5 deletions
diff --git a/test.py b/test.py
index 71d6d34..c1a5423 100755
--- a/test.py
+++ b/test.py
@@ -8,6 +8,8 @@ from string import Template
 import subprocess
 import unittest
 
+from geldschieberbot import Geldschieberbot, GeldschieberbotJSONEncoder
+
 alice, bob, charlie = "alice", "bob", "charlie"
 num = {alice: "+49123456", bob: "+49654321", charlie: "+49615243"}
 os.environ["GSB_GROUP_ID"] = "test"
@@ -96,11 +98,19 @@ def reset_state_string(string):
         json.dump(json.loads(string), f)
 
 
-def compare_state(comp_state):
-    with open(comp_state, "r", encoding='utf-8') as csf, \
-         open(os.environ["GSB_STATE_FILE"], "r", encoding='utf-8') as sf:
+def compare_state(expected_state_path,
+                  state=None,
+                  state_path=os.environ["GSB_STATE_FILE"]) -> bool:
+    s = ''
+    if state is None:
+        with open(state_path, "r", encoding='utf-8') as sf:
+            s = sf.read()
+    else:
+        s = json.dumps(state, cls=GeldschieberbotJSONEncoder)
+
+    with open(expected_state_path, "r", encoding='utf-8') as csf:
         cs = csf.read()
-        s = sf.read()
+
     return cs == s
 
 
@@ -1375,7 +1385,27 @@ class TestConvertState(unittest.TestCase):
         res = run_bot(self, num[alice], "!thanks")
         self.assertEqual(
             res.stdout,
-            f"You are welcome. It is a pleasure to work with you, {num['alice']}.")
+            f"You are welcome. It is a pleasure to work with you, {num['alice']}."
+        )
+
+
+class TestStateLoadStore(unittest.TestCase):
+
+    def test_default_init(self):
+        bot = Geldschieberbot()
+        self.assertTrue(
+            compare_state(os.environ['GSB_STATE_FILE'], state=bot.state))
+
+    def test_explicit_init(self):
+        sp = "test/state_3users.json"
+        bot = Geldschieberbot(sp)
+        self.assertTrue(compare_state(sp, state=bot.state))
+
+    def test_explicit_load(self):
+        sp = "test/state_3users.json"
+        bot = Geldschieberbot()
+        bot.load_state(sp)
+        self.assertTrue(compare_state(sp, state=bot.state))
 
 
 if __name__ == '__main__':