feat(booking): wip booking testr

Author: vmscw

scaleway_qaas_client/v1alpha1/client.py

@@ -306,7 +306,9 @@ def create_session(
 
         if booking_demand_started_at and booking_demand_finished_at:
             booking_demand_started_at_timestamp = booking_demand_started_at.timestamp()
-            booking_demand_finished_at_timestamp = booking_demand_finished_at.timestamp()
+            booking_demand_finished_at_timestamp = (
+                booking_demand_finished_at.timestamp()
+            )
             # seconds = int(booking_demand_started_at_timestamp)
             # nanos = int(booking_demand_started_at_timestamp % 1 * 1e9)
 
@@ -1012,9 +1014,3 @@ def list_bookings(
         _raise_on_error(response)
 
         return response.parsed.bookings
-
-
-# def _datetime_to_timestamp(time : datetime) -> Tuple[int, int]:
-#     timestamp = time.timestamp()
-#     seconds = int(timestamp)
-#     nanos = int(booking_demand_started_at_timestamp % 1 * 1e9)

tests/test_api.py

@@ -172,26 +172,27 @@ def _get_client() -> QaaSClient:
 
 #     assert len(bookings) > 0
 
+
 def test_create_and_cancel_booking():
     client = _get_client()
 
-    try:
-        platforms = client.list_platforms(name=_TEST_PLATFORM_NAME)
+    platforms = client.list_platforms(name=_TEST_PLATFORM_NAME)
 
-        assert len(platforms) > 0
+    assert len(platforms) > 0
 
-        target_platform = platforms[0]
+    target_platform = platforms[0]
 
+    try:
         now = datetime.now(timezone.utc)
-        booking_start = now + timedelta(days=7)
+        booking_start = now + timedelta(seconds=15)
         booking_finish = booking_start + timedelta(hours=1)
         booking_description = "my lovely booking"
 
         session = client.create_session(
             platform_id=target_platform.id,
             booking_demand_started_at=booking_start,
             booking_demand_finished_at=booking_finish,
-            booking_demand_description=booking_description
+            booking_demand_description=booking_description,
         )
 
         assert session is not None
@@ -210,9 +211,12 @@ def test_create_and_cancel_booking():
 
         while True:
             time.sleep(2)
+            session = client.get_session(session_id=session.id)
             booking = client.get_booking(booking_id=session.booking_id)
 
-            assert booking.status in ["starting", "running"]
+            print(session.status, booking.status)
+
+            assert session.status in ["starting", "running"]
             assert booking.status in ["waiting", "validating", "validated"]
 
             if booking.status == "validated":
@@ -225,12 +229,176 @@ def test_create_and_cancel_booking():
             time.sleep(2)
             booking = client.get_booking(booking_id=session.booking_id)
 
-            assert booking.status in ["running", "stopping", "stopped"]
+            print(booking.status)
+
             assert booking.status in ["cancelling", "cancelled"]
 
             if booking.status == "cancelled":
                 break
 
+
+def test_create_overlaping_booking():
+    client = _get_client()
+
+    try:
+        platforms = client.list_platforms(name=_TEST_PLATFORM_NAME)
+
+        assert len(platforms) > 0
+
+        target_platform = platforms[0]
+
+        now = datetime.now(timezone.utc)
+        booking_start = now + timedelta(days=1)
+        booking_finish = booking_start + timedelta(hours=1)
+        booking_description = "my overlaping booking"
+
+        session = client.create_session(
+            platform_id=target_platform.id,
+            booking_demand_started_at=booking_start,
+            booking_demand_finished_at=booking_finish,
+            booking_demand_description=booking_description,
+        )
+
+        assert session == None
+    finally:
+        client.delete_session(session.id)
+
+
+def test_create_too_long_booking():
+    client = _get_client()
+
+    try:
+        platforms = client.list_platforms(name=_TEST_PLATFORM_NAME)
+
+        assert len(platforms) > 0
+
+        target_platform = platforms[0]
+
+        now = datetime.now(timezone.utc)
+        booking_start = now + timedelta(minutes=1)
+        booking_finish = booking_start + timedelta(hours=10)
+        booking_description = "my too long booking"
+
+        session = client.create_session(
+            platform_id=target_platform.id,
+            booking_demand_started_at=booking_start,
+            booking_demand_finished_at=booking_finish,
+            booking_demand_description=booking_description,
+        )
+
+        assert session == None
+    finally:
+        client.delete_session(session.id)
+
+
+def test_create_too_short_booking():
+    client = _get_client()
+
+    try:
+        platforms = client.list_platforms(name=_TEST_PLATFORM_NAME)
+
+        assert len(platforms) > 0
+
+        target_platform = platforms[0]
+
+        now = datetime.now(timezone.utc)
+        booking_start = now + timedelta(minutes=1)
+        booking_finish = booking_start + timedelta(seconds=10)
+        booking_description = "my lovely booking"
+
+        session = client.create_session(
+            platform_id=target_platform.id,
+            booking_demand_started_at=booking_start,
+            booking_demand_finished_at=booking_finish,
+            booking_demand_description=booking_description,
+        )
+
+        assert session == None
+    finally:
+        client.delete_session(session.id)
+
+
+def test_create_too_short_booking():
+    client = _get_client()
+
+    try:
+        platforms = client.list_platforms(name=_TEST_PLATFORM_NAME)
+
+        assert len(platforms) > 0
+
+        target_platform = platforms[0]
+
+        now = datetime.now(timezone.utc)
+        booking_start = now + timedelta(minutes=1)
+        booking_finish = booking_start + timedelta(seconds=10)
+        booking_description = "my lovely booking"
+
+        session = client.create_session(
+            platform_id=target_platform.id,
+            booking_demand_started_at=booking_start,
+            booking_demand_finished_at=booking_finish,
+            booking_demand_description=booking_description,
+        )
+
+        assert session == None
+    finally:
+        client.delete_session(session.id)
+
+
+def test_create_too_far_booking():
+    client = _get_client()
+
+    try:
+        platforms = client.list_platforms(name=_TEST_PLATFORM_NAME)
+
+        assert len(platforms) > 0
+
+        target_platform = platforms[0]
+
+        now = datetime.now(timezone.utc)
+        booking_start = now + timedelta(days=30)
+        booking_finish = booking_start + timedelta(hours=2)
+        booking_description = "my lovely booking"
+
+        session = client.create_session(
+            platform_id=target_platform.id,
+            booking_demand_started_at=booking_start,
+            booking_demand_finished_at=booking_finish,
+            booking_demand_description=booking_description,
+        )
+
+        assert session == None
+    finally:
+        client.delete_session(session.id)
+
+
+def test_create_too_close_booking():
+    client = _get_client()
+
+    try:
+        platforms = client.list_platforms(name=_TEST_PLATFORM_NAME)
+
+        assert len(platforms) > 0
+
+        target_platform = platforms[0]
+
+        now = datetime.now(timezone.utc)
+        booking_start = now + timedelta(seconds=10)
+        booking_finish = booking_start + timedelta(hours=2)
+        booking_description = "my lovely booking"
+
+        session = client.create_session(
+            platform_id=target_platform.id,
+            booking_demand_started_at=booking_start,
+            booking_demand_finished_at=booking_finish,
+            booking_demand_description=booking_description,
+        )
+
+        assert session == None
+    finally:
+        client.delete_session(session.id)
+
+
 # def test_create_session_same_deduplication_id():
 #     client = _get_client()