update tests

Signed-off-by: Victor Garcia Reolid <[email protected]>

timely_beliefs/tests/test_belief_persistence.py

@@ -49,9 +49,31 @@ def test_adding_to_session(
     assert len(bdf) == len(new_bdf)
 
 
-@pytest.mark.parametrize("bulk_save_objects", [False, True])
-def test_fail_adding_to_session(
-    bulk_save_objects: bool,
+def test_adding_to_session_succeeds(
+    time_slot_sensor: DBSensor,
+    rolling_day_ahead_beliefs_about_time_slot_events,
+):
+
+    # Retrieve some data from the database
+    bdf = DBTimedBelief.search_session(
+        session=session,
+        sensor=time_slot_sensor,
+    )
+
+    # Attempting to save the same data should not fail, even if we expunge everything from the session
+    try:
+        DBTimedBelief.add_to_session(
+            session,
+            bdf,
+            expunge_session=True,
+            bulk_save_objects=True,
+            commit_transaction=True,
+        )
+    except IntegrityError as exception:
+        raise pytest.fail("DID RAISE {0}".format(exception))
+
+
+def test_adding_to_session_fails(
     time_slot_sensor: DBSensor,
     rolling_day_ahead_beliefs_about_time_slot_events,
 ):
@@ -69,7 +91,7 @@ def test_fail_adding_to_session(
             session,
             bdf,
             expunge_session=True,
-            bulk_save_objects=bulk_save_objects,
+            bulk_save_objects=False,
             commit_transaction=True,
         )
         bdf = DBTimedBelief.search_session(