Commit and legacy backpopulate pattern from unit test

galaxyproject · Dec 25, 2023 · ccdd733 · ccdd733
1 parent 8bab912
commit ccdd733
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/test/integration/test_display_applications.py b/test/integration/test_display_applications.py
@@ -6,6 +6,7 @@
 
 
 class TestDisplayApplicationTestCase(ContainerizedIntegrationTestCase):
+    dataset_populator: DatasetPopulator
     container_type = "docker"
 
     def setUp(self):

diff --git a/test/unit/data/test_galaxy_mapping.py b/test/unit/data/test_galaxy_mapping.py
@@ -60,7 +60,7 @@ def persist(cls, *args, **kwargs):
         for arg in args:
             session.add(arg)
             if flush:
-                session.flush()
+                session.commit()
         if kwargs.get("expunge", not flush):
             cls.expunge()
         return arg  # Return last or only arg.
@@ -699,7 +699,7 @@ def test_current_session(self):
         self.persist(user, galaxy_session)
         assert user.current_galaxy_session == galaxy_session
         new_galaxy_session = model.GalaxySession()
-        new_galaxy_session.user = user
+        user.galaxy_sessions.append(new_galaxy_session)
         self.persist(user, new_galaxy_session)
         assert user.current_galaxy_session == new_galaxy_session