From f352c4e7804f64a29670c23e50cff3c56a902496 Mon Sep 17 00:00:00 2001 From: Lars Wirzenius Date: Tue, 18 Dec 2012 15:23:16 +0000 Subject: Silence log messages during unit tests --- morphlib/remoteartifactcache.py | 8 ++++---- morphlib/remoteartifactcache_tests.py | 6 ++++-- 2 files changed, 8 insertions(+), 6 deletions(-) (limited to 'morphlib') diff --git a/morphlib/remoteartifactcache.py b/morphlib/remoteartifactcache.py index e50c1159..17f547f2 100644 --- a/morphlib/remoteartifactcache.py +++ b/morphlib/remoteartifactcache.py @@ -67,18 +67,18 @@ class RemoteArtifactCache(object): filename = '%s.%s' % (cachekey, name) return self._has_file(filename) - def get(self, artifact): + def get(self, artifact, log=logging.error): try: return self._get_file(artifact.basename()) except urllib2.URLError, e: - logging.error(str(e)) + log(str(e)) raise GetError(self, artifact) - def get_artifact_metadata(self, artifact, name): + def get_artifact_metadata(self, artifact, name, log=logging.error): try: return self._get_file(artifact.metadata_basename(name)) except urllib2.URLError, e: - logging.error(str(e)) + log(str(e)) raise GetArtifactMetadataError(self, artifact, name) def get_source_metadata(self, source, cachekey, name): diff --git a/morphlib/remoteartifactcache_tests.py b/morphlib/remoteartifactcache_tests.py index a4ec9370..431731b9 100644 --- a/morphlib/remoteartifactcache_tests.py +++ b/morphlib/remoteartifactcache_tests.py @@ -123,7 +123,8 @@ class RemoteArtifactCacheTests(unittest.TestCase): def test_fails_to_get_a_non_existent_artifact(self): self.assertRaises(morphlib.remoteartifactcache.GetError, - self.cache.get, self.doc_artifact) + self.cache.get, self.doc_artifact, + log=lambda *args: None) def test_get_existing_artifact_metadata(self): handle = self.cache.get_artifact_metadata( @@ -137,7 +138,8 @@ class RemoteArtifactCacheTests(unittest.TestCase): morphlib.remoteartifactcache.GetArtifactMetadataError, self.cache.get_artifact_metadata, self.runtime_artifact, - 'non-existent-meta') + 'non-existent-meta', + log=lambda *args: None) def test_get_existing_source_metadata(self): handle = self.cache.get_source_metadata( -- cgit v1.2.1