From 19c1da7b405af521bd5f1286ffa51a563f00aab6 Mon Sep 17 00:00:00 2001 From: Jeremy Thurgood Date: Thu, 2 Jul 2015 12:17:58 +0200 Subject: [PATCH] Remove unnecessary caching of get_tag_info. --- go/vumitools/routing.py | 2 +- go/vumitools/tests/test_utils.py | 31 ------------------------------- go/vumitools/utils.py | 7 ------- 3 files changed, 1 insertion(+), 39 deletions(-) diff --git a/go/vumitools/routing.py b/go/vumitools/routing.py index 76d6ab292..113d379fd 100644 --- a/go/vumitools/routing.py +++ b/go/vumitools/routing.py @@ -381,7 +381,7 @@ def get_config(self, msg): if msg_mdh.has_user_account(): user_account_key = msg_mdh.get_account_key() elif msg_mdh.tag is not None: - tag_info = yield msg_mdh.get_tag_info() + tag_info = yield self.vumi_api.mdb.get_tag_info(msg_mdh.tag) user_account_key = tag_info.metadata['user_account'] if user_account_key is None: raise UnownedTagError( diff --git a/go/vumitools/tests/test_utils.py b/go/vumitools/tests/test_utils.py index 59b202312..93d51e6f6 100644 --- a/go/vumitools/tests/test_utils.py +++ b/go/vumitools/tests/test_utils.py @@ -264,10 +264,8 @@ def test_clear_object_cache(self): md.set_tag(["pool", "tag"]) self.assertEqual(md._store_objects, {}) md_conv = yield md.get_conversation() - tag_info = yield md.get_tag_info() self.assertEqual(md._store_objects, { 'conversation': md_conv, - 'tag_info': tag_info, }) md.clear_object_cache() self.assertEqual(md._store_objects, {}) @@ -459,40 +457,11 @@ def test_rewrap(self): self.assertEqual({}, other_md._store_objects) self.assertEqual(md._go_metadata, other_md._go_metadata) - def test_get_tag_info_no_tag(self): - md = self.mk_md() - self.assertEqual(None, md.tag) - self.assertRaises(ValueError, md.get_tag_info) - def test_get_tagpool_metadata_no_tag(self): md = self.mk_md() self.assertEqual(None, md.tag) self.assertRaises(ValueError, md.get_tagpool_metadata) - @inlineCallbacks - def test_get_tag_info(self): - md = self.mk_md() - md.set_tag(["pool", "tagname"]) - - tag_info = yield md.get_tag_info() - self.assertEqual(("pool", "tagname"), tag_info.tag) - - @inlineCallbacks - def test_tag_info_caching(self): - md = self.mk_md() - md.set_tag(["pool", "tagname"]) - - self.assertEqual({}, md._store_objects) - tag_info = yield md.get_tag_info() - self.assertEqual(("pool", "tagname"), tag_info.tag) - self.assertEqual({'tag_info': tag_info}, md._store_objects) - - # Stash a fake thing in the cache to make sure that what we get is - # actually the thing in the cache. - md._store_objects['tag_info'] = "I am the cached tag_info" - cached_tag_info = yield md.get_tag_info() - self.assertEqual(cached_tag_info, "I am the cached tag_info") - @inlineCallbacks def test_get_tagpool_metadata(self): yield self.vumi_helper.setup_tagpool("pool", ["tagname"], metadata={ diff --git a/go/vumitools/utils.py b/go/vumitools/utils.py index 4deebb31c..d633a0fba 100644 --- a/go/vumitools/utils.py +++ b/go/vumitools/utils.py @@ -188,13 +188,6 @@ def set_tag(self, tag): TaggingMiddleware.add_tag_to_msg(self.message, tag) self.tag = TaggingMiddleware.map_msg_to_tag(self.message) - def get_tag_info(self): - if self.tag is None: - raise ValueError("No tag to look up metadata for.") - - return self._get_if_not_stashed( - 'tag_info', self.vumi_api.mdb.get_tag_info, tuple(self.tag)) - def get_tagpool_metadata(self): if self.tag is None: raise ValueError("No tag to look up metadata for.")