def test_get_urls_no_match(self): self.assertEqual(set([]), urls_for_zone("euro-west1-a",json.loads(url_map_json)))
def test_get_urls_incorrect_format(self): with self.assertRaises(ValueError): urls_for_zone("weird-format-a",json.loads(url_map_json))
def test_get_urls(self): self.assertEqual( set(["https://datalab-us-east1.cloud.google.com","https://datalab-us-west1.cloud.google.com"]), urls_for_zone("us-east1-a",json.loads(url_map_json)))