From 0926e31db898b576164698962f7d75395b5399ab Mon Sep 17 00:00:00 2001
From: Ese <17943860+dlpbc@users.noreply.github.com>
Date: Wed, 5 Feb 2025 12:16:51 +0000
Subject: [PATCH] ongoing work to fix tests (part 2)

---
 tests/test_routes/test_usage.py | 38 +++++++++++++++++++++++++--------
 1 file changed, 29 insertions(+), 9 deletions(-)

diff --git a/tests/test_routes/test_usage.py b/tests/test_routes/test_usage.py
index da205c9..ad73be3 100644
--- a/tests/test_routes/test_usage.py
+++ b/tests/test_routes/test_usage.py
@@ -214,8 +214,8 @@ def test_write_usage(
         cost=75.34,
         amortised_cost=0.0,
         total_cost=75.34,
-        first_usage=datetime.date.today(),
-        latest_usage=datetime.date.today(),
+        first_usage=datetime.datetime(2024, 2, 2),
+        latest_usage=datetime.datetime(2024, 2, 4),
         remaining=130.0 - 75.34,
         desired_status_info=None,
         abolished=False,
@@ -248,21 +248,33 @@ def test_write_usage(
 
         assert (
             api_calls.create_usage(
-                client, token, constants.TEST_SUB_UUID, cost=50.0
+                client,
+                token,
+                constants.TEST_SUB_UUID,
+                cost=50.0,
+                date=datetime.datetime(2024, 2, 2),
             ).status_code
             == 200
         )
 
         assert (
             api_calls.create_usage(
-                client, token, constants.TEST_SUB_UUID, cost=20.0
+                client,
+                token,
+                constants.TEST_SUB_UUID,
+                cost=20.00,
+                date=datetime.datetime(2024, 2, 3),
             ).status_code
             == 200
         )
 
         assert (
             api_calls.create_usage(
-                client, token, constants.TEST_SUB_UUID, cost=5.34
+                client,
+                token,
+                constants.TEST_SUB_UUID,
+                cost=5.340,
+                date=datetime.datetime(2024, 2, 4),
             ).status_code
             == 200
         )
@@ -285,8 +297,8 @@ def test_greater_budget(
         amortised_cost=0.0,
         total_cost=150.0,
         remaining=130.0 - 150.0,
-        first_usage=datetime.date.today(),
-        latest_usage=datetime.date.today(),
+        first_usage=datetime.datetime(2024, 2, 2),
+        latest_usage=datetime.datetime(2024, 2, 3),
         desired_status_info=BillingStatus.OVER_BUDGET,
         abolished=False,
     )
@@ -318,14 +330,22 @@ def test_greater_budget(
 
         assert (
             api_calls.create_usage(
-                client, token, constants.TEST_SUB_UUID, cost=100.0
+                client,
+                token,
+                constants.TEST_SUB_UUID,
+                cost=100.0,
+                date=datetime.datetime(2024, 2, 2),
             ).status_code
             == 200
         )
 
         assert (
             api_calls.create_usage(
-                client, token, constants.TEST_SUB_UUID, cost=50.0
+                client,
+                token,
+                constants.TEST_SUB_UUID,
+                cost=50.0,
+                date=datetime.datetime(2024, 2, 3),
             ).status_code
             == 200
         )