Set Morphology.needs_artifact_metadata_cached

1 files changed, 12 insertions, 0 deletions

diff --git a/morphlib/morphologyfactory_tests.py b/morphlib/morphologyfactory_tests.py
index 4aa3c018..91768843 100644
--- a/morphlib/morphologyfactory_tests.py
+++ b/morphlib/morphologyfactory_tests.py
@@ -209,3 +209,15 @@ class MorphologyFactoryTests(unittest.TestCase):
         morph = self.mf.get_morphology('reponame', 'sha1', 'system.morph')
         self.assertEqual(morph.needs_staging_area, False)
 
+    def test_does_not_set_needs_artifact_metadata_cached_for_chunk(self):
+        morph = self.mf.get_morphology('reponame', 'sha1', 'chunk.morph')
+        self.assertEqual(morph.needs_artifact_metadata_cached, False)
+
+    def test_sets_artifact_metadata_cached_for_stratum(self):
+        morph = self.mf.get_morphology('reponame', 'sha1', 'stratum.morph')
+        self.assertEqual(morph.needs_artifact_metadata_cached, True)
+
+    def test_does_not_set_artifact_metadata_cached_for_system(self):
+        morph = self.mf.get_morphology('reponame', 'sha1', 'system.morph')
+        self.assertEqual(morph.needs_artifact_metadata_cached, False)
+