diff options
author | Richard Maw <richard.maw@codethink.co.uk> | 2012-04-12 13:58:49 +0100 |
---|---|---|
committer | Richard Maw <richard.maw@codethink.co.uk> | 2012-04-12 17:36:26 +0100 |
commit | 8902a85ee612e98601daea8cca1e46c49511d106 (patch) | |
tree | 13f27058be25c1aef1272a1895044ddab90bc617 /morphlib/cachekeycomputer_tests.py | |
parent | f474282b188524f3830acb797404978af16fda8f (diff) | |
download | morph-8902a85ee612e98601daea8cca1e46c49511d106.tar.gz |
morphlib: add cachekeycomputer class
Diffstat (limited to 'morphlib/cachekeycomputer_tests.py')
-rw-r--r-- | morphlib/cachekeycomputer_tests.py | 95 |
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'])) |