def test_params_err(self):
"parses params errors and returns the descriptive message"
def urlopen(req):
url = req.get_full_url()
body = io.BytesIO(
json.dumps({
"errors": {
"params": {
"name": ["is missing", "is required"],
"value": ["is not a number", "is too big"]
}
}
}))
raise urllib2.HTTPError(url, 400, "Bad Request", {}, body)
urllib2.urlopen = urlopen
l = librato.Librato(
{
"user": "******",
"token": "OpenSesame",
"metrics": ["cpu_time", "cpu_idle"]
}, OPTIONS)
try:
l.read()
except librato.LibratoError, e:
self.assertEqual(str(e), "name is missing")
def test_get_metrics(self):
"retrieves the full list of available metrics"
urls = []
def urlopen(req):
urls.append(req.get_full_url())
res = {
"metrics": [
{
"name": "cpu_time",
"display_name": "hello"
},
{
"name": "cpu_idle",
"display_name": "world"
},
]
}
return io.BytesIO(json.dumps(res))
urllib2.urlopen = urlopen
l = librato.Librato({
"user": "******",
"token": "OpenSesame",
}, OPTIONS)
metrics = l.get_metrics()
self.assertEqual(metrics, ["cpu_time", "cpu_idle"])
self.assertEqual(urls, ["https://metrics-api.librato.com/v1/metrics"])
def test_pagination(self):
"_read() continues to read the same metric with the new start time"
times = [100, 200]
urls = []
def urlopen(req):
urls.append(req.get_full_url())
nexttime = times.pop(0) if len(times) > 0 else None
res = {"query": {"next_time": nexttime}}
return io.BytesIO(json.dumps(res))
urllib2.urlopen = urlopen
l = librato.Librato(
{
"user": "******",
"token": "OpenSesame",
"metrics": ["cpu_time"]
}, OPTIONS)
self.assertIsNotNone(l.read()) # first start time
self.assertIsNotNone(l.read()) # 100
self.assertIsNotNone(l.read()) # 200; last
self.assertIsNone(l.read()) # done.
# urls[0] has the default start time; irrelevant for pagination
qs = parseqs(urls[1])
self.assertEqual(qs["start_time"], ["100"])
qs = parseqs(urls[2])
self.assertEqual(qs["start_time"], ["200"])
def test_results(self):
"parse results json and breakdown all measurements to a list of results"
def urlopen(req):
return io.BytesIO(json.dumps(RESULTS))
urllib2.urlopen = urlopen
l = librato.Librato(
{
"user": "******",
"token": "OpenSesame",
"metrics": ["cpu_time"]
}, OPTIONS)
res = l.read()
self.assertEqual(res, EXPECTED)
def test_done(self):
"_read() returns None when all results where consumed"
def urlopen(req):
return io.BytesIO("{}")
urllib2.urlopen = urlopen
l = librato.Librato(
{
"user": "******",
"token": "OpenSesame",
"metrics": ["cpu_time", "cpu_idle"]
}, OPTIONS)
self.assertIsNotNone(l.read()) # first metric
self.assertIsNotNone(l.read()) # second metric
self.assertIsNone(l.read()) # should be done by now
def test_auth(self):
"user and token are encoded correclty in HTTP authorization header"
auths = []
def urlopen(req):
auths.append(req.get_header("Authorization"))
return io.BytesIO("{}")
urllib2.urlopen = urlopen
l = librato.Librato(
{
"user": "******",
"token": "OpenSesame",
"metrics": ["cpu_time"]
}, OPTIONS)
l.read()
self.assertEqual(auths[0], "Basic QWxhZGRpbjpPcGVuU2VzYW1l")
def test_failed_authentication(self):
"fail to authenticate request"
urls = []
def urlopen(req):
raise Exception("fail")
urllib2.urlopen = urlopen
l = librato.Librato({
"user": "******",
"token": "OpenSesame",
}, OPTIONS)
try:
metrics = l.get_metrics()
except Exception as e:
err_message = ('Authentication failed.'
' Please check your credentials and try again')
self.assertEqual(e.args[0], err_message)
def test_request_err(self):
"parses request errors and returns the descriptive message"
def urlopen(req):
url = req.get_full_url()
body = io.BytesIO(
json.dumps({"errors": {
"request": ["one two", "three four"]
}}))
raise urllib2.HTTPError(url, 400, "Bad Request", {}, body)
urllib2.urlopen = urlopen
l = librato.Librato(
{
"user": "******",
"token": "OpenSesame",
"metrics": ["cpu_time", "cpu_idle"]
}, OPTIONS)
try:
l.read()
except librato.LibratoError, e:
self.assertEqual(str(e), "one two")
def test_url(self):
"valid url suffixed with the metric name"
urls = []
def urlopen(req):
urls.append(req.get_full_url())
return io.BytesIO("{}")
urllib2.urlopen = urlopen
l = librato.Librato(
{
"user": "******",
"token": "OpenSesame",
"metrics": ["cpu_time", "cpu_idle"]
}, OPTIONS)
l.read() # first metric
l.read() # second metric
expected = "https://metrics-api.librato.com/v1/metrics/cpu_time?"
self.assertTrue(urls[0].startswith(expected))
expected = "https://metrics-api.librato.com/v1/metrics/cpu_idle?"
self.assertTrue(urls[1].startswith(expected))
def test_unknown_err(self):
"forwards all other errors as-is"
raised = {}
def urlopen(req):
url = req.get_full_url()
body = io.BytesIO(json.dumps({}))
e = urllib2.HTTPError(url, 400, "Bad Request", {}, body)
raised["error"] = e
raise e
urllib2.urlopen = urlopen
l = librato.Librato(
{
"user": "******",
"token": "OpenSesame",
"metrics": ["cpu_time", "cpu_idle"]
}, OPTIONS)
try:
l.read()
except Exception, e:
self.assertEqual(e, raised["error"])
