sample tests

lichess-org · Mar 2, 2024 · 8f39bab · 8f39bab
1 parent 9d330c8
commit 8f39bab
Show file tree

Hide file tree

Showing 3 changed files with 32 additions and 13 deletions.
diff --git a/integration/sample.py b/integration/sample.py
diff --git a/integration/test_account.py b/integration/test_account.py
@@ -0,0 +1,24 @@
+import berserk
+
+def test_account_get():
+    session = berserk.TokenSession('lip_bobby')
+    client = berserk.Client(session, base_url="http://lila:9663")
+
+    me = client.account.get()
+    assert me['id'] == 'bobby'
+
+
+def test_account_get_email():
+    session = berserk.TokenSession('lip_bobby')
+    client = berserk.Client(session, base_url="http://lila:9663")
+
+    assert client.account.get_email() == 'bobby@localhost'
+
+
+def test_account_get_set_kid_mode():
+    session = berserk.TokenSession('lip_bobby')
+    client = berserk.Client(session, base_url="http://lila:9663")
+
+    assert client.account.get_kid_mode() == False
+    client.account.set_kid_mode(True)
+    assert client.account.get_kid_mode() == True
diff --git a/integration/test_users.py b/integration/test_users.py
@@ -0,0 +1,8 @@
+import berserk
+
+def test_account_get():
+    client = berserk.Client(base_url="http://lila:9663")
+
+    top_10 = client.users.get_all_top_10()
+
+    assert len(top_10['bullet']) == 10