MorphSet: add unpetrify_all() method

Add an unpetrify method to MorphSet, since it's a good place to put it
and may be needed for merge.

2 files changed, 27 insertions, 0 deletions

diff --git a/morphlib/morphset.py b/morphlib/morphset.py
index 2343cbae..3c07d58e 100644
--- a/morphlib/morphset.py
+++ b/morphlib/morphset.py
@@ -272,3 +272,16 @@ class MorphologySet(object):
 
         self._traverse_specs(process_chunk_spec, wanted_chunk_spec)
 
+    def unpetrify_all(self):
+        '''If a spec is petrified, unpetrify it.
+
+        '''
+
+        def wanted_spec(m, kind, spec):
+            return ('unpetrify-ref' in spec and
+                    morphlib.git.is_valid_sha1(spec['ref']))
+        def process_spec(m, kind, spec):
+            spec['ref'] = spec.pop('unpetrify-ref')
+            return True
+
+        self._traverse_specs(process_spec, wanted_spec)
diff --git a/morphlib/morphset_tests.py b/morphlib/morphset_tests.py
index fa8fa030..d6908844 100644
--- a/morphlib/morphset_tests.py
+++ b/morphlib/morphset_tests.py
@@ -222,3 +222,17 @@ class MorphologySetTests(unittest.TestCase):
                 }
             ])
 
+    def test_unpetrify_all(self):
+        self.morphs.add_morphology(self.system)
+        self.morphs.add_morphology(self.stratum)
+        self.morphs.petrify_chunks({('test:foo-chunk', 'master'): '0'*40})
+        self.morphs.unpetrify_all()
+        self.assertEqual(
+            self.stratum['chunks'],
+            [
+                {
+                    'repo': 'test:foo-chunk',
+                    'ref': 'master',
+                    'morph': 'foo-chunk',
+                }
+            ])