From 69ad4669a214dcb402d41cdd2a9c455403a0791a Mon Sep 17 00:00:00 2001
From: Christoph Kappestein <christoph.kappestein@gmail.com>
Date: Sat, 30 Mar 2024 18:20:42 +0100
Subject: [PATCH] update main

---
 integration/python/main.py | 92 ++++++++++++++++++--------------------
 1 file changed, 44 insertions(+), 48 deletions(-)

diff --git a/integration/python/main.py b/integration/python/main.py
index b05213f..31b0ca4 100644
--- a/integration/python/main.py
+++ b/integration/python/main.py
@@ -1,70 +1,66 @@
-import json
-import sdkgen
-from pathlib import Path
+from sdkgen import Anonymous
 
+from client import Client
+from error_exception import ErrorException
+from todo import Todo
 
-credentials = Anonymous()
-client = Client('http://127.0.0.1:1080', credentials)
 
-assertGetHello(client)
-assertGetEntries(client)
-assertInsert(client)
-assertThrowException(client)
+def assert_get_hello(client: Client):
+    message = client.test().get_hello()
 
-def assertGetHello(client: Client)
-{
-    message = client.test().getHello()
+    if message.message != 'Hello World!':
+        raise Exception('Test assertGetHello failed: Message, got: ' + message.message)
 
-    if (message.message != 'Hello World!') {
-        raise Exception('Test assertGetHello failed: Message, got: ' + message.message);
-    }
-}
 
-def assertGetEntries(client: Client)
-{
-    todos = client.test().getEntries(0, 16);
+def assert_get_entries(client: Client):
+    todos = client.test().get_entries(0, 16)
 
-    if todos.totalResults != 4
-        raise Exception("Test assertGetEntries failed: TotalResults, got: " + todos.totalResults);
+    if todos.total_results != 4:
+        raise Exception("Test assertGetEntries failed: TotalResults, got: " + str(todos.total_results))
 
-    if todos.startIndex != 0
-        raise Exception("Test assertGetEntries failed: StartIndex, got: " + todos.startIndex;
+    if todos.start_index != 0:
+        raise Exception("Test assertGetEntries failed: StartIndex, got: " + str(todos.start_index))
 
-    if todos.itemsPerPage != 16
-        raise Exception("Test assertGetEntries failed: ItemsPerPage, got: " + todos.itemsPerPage);
+    if todos.items_per_page != 16:
+        raise Exception("Test assertGetEntries failed: ItemsPerPage, got: " + str(todos.items_per_page))
 
-    if count(todos.entry) != 2
-        raise Exception("Test assertGetEntries failed: Entry, got: " + count(todos.entry));
+    if len(todos.entry) != 2:
+        raise Exception("Test assertGetEntries failed: Entry, got: " + str(len(todos.entry)))
 
-    if todos.entry[0].title != 'foo'
-        raise Exception("Test assertGetEntries failed: Entry.0.Title, got: " + todos.entry[0].title);
+    if todos.entry[0].title != 'foo':
+        raise Exception("Test assertGetEntries failed: Entry.0.Title, got: " + todos.entry[0].title)
 
-    if todos.entry[1].title() != "bar"
-        raise Exception("Test assertGetEntries failed: Entry.1.Title, got: " + todos.entry[1].title);
-}
+    if todos.entry[1].title != "bar":
+        raise Exception("Test assertGetEntries failed: Entry.1.Title, got: " + todos.entry[1].title)
 
-def assertInsert(client: Client)
-{
+
+def assert_insert(client: Client):
     payload = Todo()
     payload.title = "baz"
 
-    message = client.test().insert(payload);
+    message = client.test().insert(payload)
+
+    if not message.success:
+        raise Exception("Test assertInsert failed: Success, got: " + str(message.success))
 
-    if message.success != true
-        raise Exception("Test assertInsert failed: Success, got: " + message.success);
+    if message.message != "Successful":
+        raise Exception("Test assertInsert failed: Message, got: " + message.message)
 
-    if message.message != "Successful"
-        raise Exception("Test assertInsert failed: Message, got: " + message.message);
-}
 
-def assertThrowException(client: Client)
-{
+def assert_throw_exception(client: Client):
     try:
-        client.test().throwException();
+        client.test().throw_exception()
 
-        raise Exception("Test assertThrowException failed: Expected an error");
-    except Exception as e:
-        if e.payload.message != "Error"
-            raise Exception("Test assertThrowException failed: Error message does not match, got: " + e.payload.message);
-}
+        raise Exception("Test assertThrowException failed: Expected an error")
+    except ErrorException as e:
+        if e.payload.message != "Error":
+            raise Exception("Test assertThrowException failed: Error message does not match, got: " + e.payload.message)
+
+
+credentials = Anonymous()
+client = Client('http://127.0.0.1:1080', credentials)
 
+assert_get_hello(client)
+assert_get_entries(client)
+assert_insert(client)
+assert_throw_exception(client)