class TestMyFlickr(unittest.TestCase): def setUp(self): self.flickr = MyFlickr("abc") def test_url_for_method_should_have_method(self): url = self.flickr.url_for_method("bob") assert re.search("method=bob", url) def test_url_for_method_should_have_api_key(self): url = self.flickr.url_for_method("test") assert re.search("api_key=abc", url) def test_dict_to_query_string(self): dict = {"a": 1, "b": "water"} assert "a=1&b=water&" == self.flickr.dict_to_query_string(dict) def test_generic_query_with_call(self): url = self.flickr.url_for_method("erik", page=1, per_page=50) assert re.search("method=erik", url) assert re.search("page=1", url) assert re.search("per_page=50", url) def test_signature(self): dict = {"b": 567, "a": 123} assert "ee63db973f62e0df82663d21c47cf56e" == self.flickr.signature(dict) def test_login_link(self): qs = "?perms=read&api_sig=375d9f0bbd53805d9c5a3b61f65ac69d&api_key=abc&" assert "http://flickr.com/services/auth/%s" % qs == self.flickr.login_link()