Fix possible null access

Detected through the typing warning.
2023-05-02 18:09:11 +01:00 · 2023-05-02 18:09:11 +01:00 · 0a0bc1baa0
commit 0a0bc1baa0
parent ffc3d477e2
1 changed files with 3 additions and 3 deletions
--- a/tests/test_database.py
+++ b/tests/test_database.py
@ -53,9 +53,9 @@ def transactions(client: Client) -> list[Transaction]:
    client.insert(transactions)
    for i, transaction in enumerate(transactions):
        transaction.id = i + 1
-        transaction.split = False  # default
-    transactions[0].category.id = 1
-    transactions[0].category.selector.id = 1
+        if transaction.category:
+            transaction.category.id = 1
+            transaction.category.selector.id = 1

    return transactions