def collect_links(self, env, content, url, env_base_url=''): jenv = yaml.safe_load(env) files = {} if url: self.m.StubOutWithMock(urlutils, 'urlopen') urlutils.urlopen(url).AndReturn(six.StringIO(content)) self.m.ReplayAll() template_utils.resolve_environment_urls( jenv.get('resource_registry'), files, env_base_url) if url: self.assertEqual(files[url], content)
def collect_links(self, env, content, url, env_base_url=""): jenv = yaml.safe_load(env) files = {} if url: self.m.StubOutWithMock(request, "urlopen") request.urlopen(url).AndReturn(six.BytesIO(content)) request.urlopen(url).AndReturn(six.BytesIO(content)) self.m.ReplayAll() template_utils.resolve_environment_urls(jenv.get("resource_registry"), files, env_base_url) if url: self.assertEqual(content.decode("utf-8"), files[url])
def collect_links(self, env, content, url, env_base_url=''): jenv = yaml.safe_load(env) files = {} if url: self.m.StubOutWithMock(request, 'urlopen') request.urlopen(url).AndReturn(six.BytesIO(content)) self.m.ReplayAll() template_utils.resolve_environment_urls( jenv.get('resource_registry'), files, env_base_url) if url: self.assertEqual(content.decode('utf-8'), files[url])
def collect_links(self, env, content, url, env_base_url=''): jenv = yaml.safe_load(env) files = {} if url: def side_effect(args): if url == args: return six.BytesIO(content) with mock.patch('six.moves.urllib.request.urlopen') as mock_url: mock_url.side_effect = side_effect template_utils.resolve_environment_urls( jenv.get('resource_registry'), files, env_base_url) self.assertEqual(content.decode('utf-8'), files[url]) else: template_utils.resolve_environment_urls( jenv.get('resource_registry'), files, env_base_url)