fixup! [Core] Add entityTraits

Add explicit business logic test that kWrite and kRead are the same except for VersionTrait Signed-off-by: David Feltell <[email protected]>
OpenAssetIO · Dec 4, 2023 · 713c6ff · 713c6ff
1 parent b4212c5
commit 713c6ff
Showing 1 changed file with 25 additions and 0 deletions.
diff --git a/tests/bal_business_logic_suite.py b/tests/bal_business_logic_suite.py
@@ -422,6 +422,31 @@ def test_when_existing_entity_queried_for_write_then_VersionTrait_not_imbued(sel
 
         self.assertNotIn(VersionTrait.kId, result)
 
+    def test_traits_for_read_are_same_as_for_write_but_with_VersionTrait(self):
+        # Missing entities are writable with unrestricted trait set.
+
+        results = [None]
+
+        self._manager.entityTraits(
+            [self._manager.createEntityReference("bal:///anAsset⭐︎")],
+            EntityTraitsAccess.kRead,
+            self.createTestContext(),
+            lambda idx, value: operator.setitem(results, idx, value),
+            lambda idx, error: self.fail("entityTraits should not fail"),
+        )
+        [read_result] = results
+
+        self._manager.entityTraits(
+            [self._manager.createEntityReference("bal:///anAsset⭐︎")],
+            EntityTraitsAccess.kWrite,
+            self.createTestContext(),
+            lambda idx, value: operator.setitem(results, idx, value),
+            lambda idx, error: self.fail("entityTraits should not fail"),
+        )
+        [write_result] = results
+
+        self.assertSetEqual(read_result, write_result | {VersionTrait.kId})
+
 
 class Test_resolve(FixtureAugmentedTestCase):
     """