morphlib: add cachekeycomputer class

1 files changed, 95 insertions, 0 deletions

diff --git a/morphlib/cachekeycomputer_tests.py b/morphlib/cachekeycomputer_tests.py
new file mode 100644
index 00000000..5304307a
--- /dev/null
+++ b/morphlib/cachekeycomputer_tests.py
@@ -0,0 +1,95 @@
+# Copyright (C) 2012  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.
+
+
+import unittest
+import json
+
+import morphlib
+
+
+class CacheKeyComputerTests(unittest.TestCase):
+
+    def setUp(self):
+        self.env = {"USER": "foouser",
+                    "USERNAME": "foouser",
+                    "LOGNAME": "foouser",
+                    "TOOLCHAIN_TARGET": "dummy-baserock-linux-gnu",
+                    "PREFIX": "/baserock",
+                    "BOOTSTRAP": "false",
+                    "CFLAGS": "-O4"}
+        self.ckc = morphlib.cachekeycomputer.CacheKeyComputer(self.env)
+        pool = morphlib.sourcepool.SourcePool()
+        self.sources = {}
+        for name, text in {
+            'chunk.morph': '''{
+                "name": "chunk",
+                "kind": "chunk"
+             }''',
+            'stratum.morph': '''{
+                "name": "stratum",
+                "kind": "stratum",
+                "sources": [
+                    {
+                        "name": "chunk",
+                        "repo": "repo",
+                        "ref": "original/ref"
+                    }
+                ]
+             }''',
+         }.iteritems():
+            source = morphlib.source.Source('repo', 'original/ref', 'sha',
+                                   morphlib.morph2.Morphology(text), name)
+            pool.add(source)
+            self.sources[name] = source
+
+        morphlib.buildgraph.BuildGraph().compute_build_order(pool)
+
+    def _valid_sha256(self, s):
+        validchars = '0123456789abcdef'
+        return len(s) == 64 and all(c in validchars for c in s)
+
+    def test_get_cache_key_hashes_all_types(self):
+        runcount = {'thing': 0, 'dict': 0, 'list': 0, 'tuple': 0}
+        def inccount(func, name):
+            def f(sha, item):
+                runcount[name] = runcount[name] + 1
+                func(sha, item)
+            return f
+        self.ckc._hash_thing = inccount(self.ckc._hash_thing, 'thing')
+        self.ckc._hash_dict = inccount(self.ckc._hash_dict, 'dict')
+        self.ckc._hash_list = inccount(self.ckc._hash_list, 'list')
+        self.ckc._hash_tuple = inccount(self.ckc._hash_tuple, 'tuple')
+        self.ckc.get_cache_key(self.sources['stratum.morph'])
+        self.assertNotEqual(runcount['thing'], 0)
+        self.assertNotEqual(runcount['dict'], 0)
+        self.assertNotEqual(runcount['list'], 0)
+        self.assertNotEqual(runcount['tuple'], 0)
+
+    def test_get_cache_key_returns_sha256(self):
+        self.assertTrue(self._valid_sha256(
+                        self.ckc.get_cache_key(self.sources['stratum.morph'])))
+
+    def test_get_cache_id_returns_dict(self):
+        print json.dumps(self.ckc.get_cache_id(
+                           self.sources['stratum.morph']),
+                         indent=2)
+        self.assertEqual('', )
+
+    def test_different_env_gives_different_key(self):
+        oldsha = self.ckc.get_cache_key(self.sources['stratum.morph'])
+        self.ckc._env['CFLAGS'] = "-Os"
+        self.assertNotEqual(oldsha,
+                            self.ckc.get_cache_key(self.sources['stratum.morph']))