def test_returns_given_terms_if_place_vids_do_not_exist(self):
# prepare state
search = Search(self.lib)
# testing
ret = search.expand_place_ids('California')
self.assertEquals(ret, 'California')
def test_returns_place_vids(self):
# first assert signatures of the functions we are going to mock did not change.
assert_spec(Search.search_identifiers, ['self', 'search_phrase', 'limit'])
assert_spec(GVid.parse, ['cls', 'gvid'])
# prepare state
search = Search(self.lib)
score = 1
vid = 'vid-1'
t = 'type'
name = 'California1'
fake_search = fudge.Fake().expects_call().returns([(score, vid, t, name)])
fake_parse = fudge.Fake().expects_call().returns([])
# testing
with fudge.patched_context(Search, 'search_identifiers', fake_search):
with fudge.patched_context(GVid, 'parse', fake_parse):
ret = search.expand_place_ids('California')
self.assertEquals(ret, [vid])