def test_resolve_file(self):
content = json.dumps({'universe': {'galaxy': {'system': 'sun'}}})
cache = FakeCache()
resolver = URIResolver(None, cache)
with tempfile.NamedTemporaryFile(dir=os.getcwd()) as fh:
fh.write(content)
fh.flush()
self.assertEqual(resolver.resolve('file:%s' % fh.name), content)
def test_resolve_file(self):
content = json.dumps({"universe": {"galaxy": {"system": "sun"}}})
cache = FakeCache()
resolver = URIResolver(None, cache)
with tempfile.NamedTemporaryFile(mode="w+", dir=os.getcwd()) as fh:
fh.write(content)
fh.flush()
self.assertEqual(resolver.resolve("file:%s" % fh.name), content)
def test_resolve_file(self):
content = json.dumps({'universe': {'galaxy': {'system': 'sun'}}})
cache = FakeCache()
resolver = URIResolver(None, cache)
with tempfile.NamedTemporaryFile(dir=os.getcwd()) as fh:
fh.write(content)
fh.flush()
self.assertEqual(
resolver.resolve('file:%s' % fh.name), content)
def test_resolve_file(self):
content = json.dumps({"universe": {"galaxy": {"system": "sun"}}})
cache = FakeCache()
resolver = URIResolver(None, cache)
with tempfile.NamedTemporaryFile(mode="w+", dir=os.getcwd(), delete=False) as fh:
self.addCleanup(os.unlink, fh.name)
fh.write(content)
fh.flush()
self.assertEqual(resolver.resolve("file:%s" % fh.name), content)
def test_handle_content_encoding(self):
session_factory = self.replay_flight_data("test_s3_resolver")
cache = FakeCache()
resolver = URIResolver(session_factory, cache)
uri = "http://httpbin.org/gzip"
with vcr.use_cassette('tests/data/vcr_cassettes/test_resolver.yaml'):
response = urlopen(uri)
content = resolver.handle_response_encoding(response)
data = json.loads(content)
self.assertEqual(data['gzipped'], True)
self.assertEqual(response.headers['Content-Encoding'], 'gzip')
def test_resolve_s3(self):
session_factory = self.replay_flight_data('test_s3_resolver')
session = session_factory()
client = session.client('s3')
resource = session.resource('s3')
bname = 'custodian-byebye'
client.create_bucket(Bucket=bname)
self.addCleanup(destroyBucket, client, bname)
key = resource.Object(bname, 'resource.json')
content = json.dumps({'moose': {'soup': 'duck'}})
key.put(Body=content, ContentLength=len(content),
ContentType='application/json')
cache = FakeCache()
resolver = URIResolver(session_factory, cache)
uri = 's3://%s/resource.json?RequestPayer=requestor' % bname
data = resolver.resolve(uri)
self.assertEqual(content, data)
self.assertEqual(cache.state.keys(), [('uri-resolver', uri)])
def test_resolve_s3(self):
session_factory = self.replay_flight_data("test_s3_resolver")
session = session_factory()
client = session.client("s3")
resource = session.resource("s3")
bname = "custodian-byebye"
client.create_bucket(Bucket=bname)
self.addCleanup(destroyBucket, client, bname)
key = resource.Object(bname, "resource.json")
content = json.dumps({"moose": {"soup": "duck"}})
key.put(
Body=content, ContentLength=len(content), ContentType="application/json"
)
cache = FakeCache()
resolver = URIResolver(session_factory, cache)
uri = "s3://%s/resource.json?RequestPayer=requestor" % bname
data = resolver.resolve(uri)
self.assertEqual(content, data)
self.assertEqual(list(cache.state.keys()), [("uri-resolver", uri)])
def test_resolve_s3(self):
session_factory = self.replay_flight_data("test_s3_resolver")
session = session_factory()
client = session.client("s3")
resource = session.resource("s3")
bname = "custodian-byebye"
client.create_bucket(Bucket=bname)
self.addCleanup(destroyBucket, client, bname)
key = resource.Object(bname, "resource.json")
content = json.dumps({"moose": {"soup": "duck"}})
key.put(
Body=content, ContentLength=len(content), ContentType="application/json"
)
cache = FakeCache()
resolver = URIResolver(session_factory, cache)
uri = "s3://%s/resource.json?RequestPayer=requestor" % bname
data = resolver.resolve(uri)
self.assertEqual(content, data)
self.assertEqual(list(cache.state.keys()), [("uri-resolver", uri)])
