Patch MockContext for SA 2.0 compatibility

galaxyproject · Aug 13, 2023 · 6b24c66 · 6b24c66
1 parent cc9856d
commit 6b24c66
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/lib/galaxy/app_unittest_utils/tools_support.py b/lib/galaxy/app_unittest_utils/tools_support.py
@@ -134,6 +134,9 @@ def expunge_all(self):
     def query(self, clazz):
         return MockQuery(self.model_objects.get(clazz))
 
+    def get(self, clazz, id):
+        return self.query(clazz).get(id)
+
     def flush(self):
         self.flushed = True