Add a MorphologySet class, for changing many morphologies

Various parts of Morph need to change a set of morphologies at once,
particularly for petrification and unpetrification. This is easiest
done by loading all the morphologies into memory at once, and changing
them there, then saving again.

1 files changed, 180 insertions, 0 deletions

diff --git a/morphlib/morphset_tests.py b/morphlib/morphset_tests.py
new file mode 100644
index 00000000..7dbc861a
--- /dev/null
+++ b/morphlib/morphset_tests.py
@@ -0,0 +1,180 @@
+# Copyright (C) 2013  Codethink Limited
+#
+# This program is free software; you can redistribute it and/or modify
+# it under the terms of the GNU General Public License as published by
+# the Free Software Foundation; version 2 of the License.
+#
+# This program is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License along
+# with this program; if not, write to the Free Software Foundation, Inc.,
+# 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
+#
+# =*= License: GPL-2 =*=
+
+
+import unittest
+
+import morphlib
+
+
+class MorphologySetTests(unittest.TestCase):
+
+    def setUp(self):
+        self.morphs = morphlib.morphset.MorphologySet()
+
+        self.system = morphlib.morph3.Morphology({
+            'kind': 'system',
+            'name': 'foo-system',
+            'strata': [
+                {
+                    'repo': 'test:morphs',
+                    'ref': 'master',
+                    'morph': 'foo-stratum',
+                },
+            ],
+        })
+        self.system.repo_url = 'test:morphs'
+        self.system.ref = 'master'
+        self.system.filename = 'foo-system.morph'
+
+        self.stratum = morphlib.morph3.Morphology({
+            'kind': 'stratum',
+            'name': 'foo-stratum',
+            'chunks': [
+                {
+                    'repo': 'test:foo-chunk',
+                    'ref': 'master',
+                    'morph': 'foo-chunk',
+                },
+            ],
+            'build-depends': [],
+        })
+        self.stratum.repo_url = 'test:morphs'
+        self.stratum.ref = 'master'
+        self.stratum.filename = 'foo-stratum.morph'
+
+    def test_is_empty_initially(self):
+        self.assertEqual(self.morphs.morphologies, [])
+        self.assertFalse(
+            self.morphs.has(
+                self.system.repo_url, self.system.ref, self.system.filename))
+
+    def test_adds_morphology(self):
+        self.morphs.add_morphology(self.system)
+        self.assertEqual(self.morphs.morphologies, [self.system])
+        self.assertTrue(
+            self.morphs.has(
+                self.system.repo_url, self.system.ref, self.system.filename))
+
+        self.morphs.add_morphology(self.stratum)
+        self.assertEqual(
+            self.morphs.morphologies,
+            [self.system, self.stratum])
+
+    def test_does_not_add_morphology_twice(self):
+        self.morphs.add_morphology(self.system)
+        self.morphs.add_morphology(self.system)
+        self.assertEqual(self.morphs.morphologies, [self.system])
+
+    def test_get_stratum_in_system(self):
+        self.morphs.add_morphology(self.system)
+        self.morphs.add_morphology(self.stratum)
+        self.assertEqual(
+            self.morphs.get_stratum_in_system(
+                self.system, self.stratum['name']),
+            self.stratum)
+
+    def test_raises_stratum_not_in_system_error(self):
+        self.morphs.add_morphology(self.system)
+        self.morphs.add_morphology(self.stratum)
+        self.assertRaises(
+            morphlib.morphset.StratumNotInSystemError,
+            self.morphs.get_stratum_in_system, self.system, 'unknown-stratum')
+
+    def test_raises_stratum_not_in_set_error(self):
+        self.morphs.add_morphology(self.system)
+        self.assertRaises(
+            morphlib.morphset.StratumNotInSetError,
+            self.morphs.get_stratum_in_system, self.system, 'foo-stratum')
+
+    def test_get_chunk_triplet(self):
+        self.morphs.add_morphology(self.system)
+        self.morphs.add_morphology(self.stratum)
+        self.assertEqual(
+            self.morphs.get_chunk_triplet(self.stratum, 'foo-chunk'),
+            ('test:foo-chunk', 'master', 'foo-chunk'))
+
+    def test_raises_chunk_not_in_stratum_error(self):
+        self.assertRaises(
+            morphlib.morphset.ChunkNotInStratumError,
+            self.morphs.get_chunk_triplet, self.stratum, 'wrong')
+
+    def test_changes_stratum_ref(self):
+        self.morphs.add_morphology(self.system)
+        self.morphs.add_morphology(self.stratum)
+        self.morphs.change_ref(
+            self.stratum.repo_url,
+            self.stratum.ref,
+            self.stratum.filename,
+            'new-ref')
+        self.assertEqual(self.stratum.ref, 'new-ref')
+        self.assertEqual(
+            self.system['strata'][0],
+            {
+                'repo': 'test:morphs',
+                'ref': 'new-ref',
+                'morph': 'foo-stratum'
+            })
+
+    def test_changes_stratum_ref_in_build_depends(self):
+        other_stratum = morphlib.morph3.Morphology({
+            'name': 'other-stratum',
+            'kind': 'stratum',
+            'chunks': [],
+            'build-depends': [
+                {
+                    'repo': self.stratum.repo_url,
+                    'ref': self.stratum.ref,
+                    'morph': self.stratum['name'],
+                },
+            ]
+        })
+
+        self.morphs.add_morphology(self.system)
+        self.morphs.add_morphology(self.stratum)
+        self.morphs.add_morphology(other_stratum)
+        self.morphs.change_ref(
+            self.stratum.repo_url,
+            self.stratum.ref,
+            self.stratum.filename,
+            'new-ref')
+        self.assertEqual(
+            other_stratum['build-depends'][0],
+            {
+                'repo': 'test:morphs',
+                'ref': 'new-ref',
+                'morph': 'foo-stratum'
+            })
+
+    def test_changes_chunk_ref(self):
+        self.morphs.add_morphology(self.system)
+        self.morphs.add_morphology(self.stratum)
+        self.morphs.change_ref(
+            'test:foo-chunk',
+            'master',
+            'foo-chunk.morph',
+            'new-ref')
+        self.assertEqual(
+            self.stratum['chunks'],
+            [
+                {
+                    'repo': 'test:foo-chunk',
+                    'ref': 'new-ref',
+                    'morph': 'foo-chunk',
+                }
+            ])
+