|
34 | 34 | from influxdb import InfluxDBClient
|
35 | 35 | from influxdb.resultset import ResultSet
|
36 | 36 |
|
| 37 | + |
37 | 38 | def _build_response_object(status_code=200, content=""):
|
38 | 39 | resp = requests.Response()
|
39 | 40 | resp.status_code = status_code
|
@@ -793,36 +794,41 @@ def test_invalid_port_fails(self):
|
793 | 794 | InfluxDBClient('host', '80/redir', 'username', 'password')
|
794 | 795 |
|
795 | 796 | def test_chunked_response(self):
|
796 |
| - example_response = u'{"results":[{"statement_id":0,"series": ' \ |
797 |
| - '[{"name":"cpu","columns":["fieldKey","fieldType"],"values":' \ |
798 |
| - '[["value","integer"]]}],"partial":true}]}\n{"results":' \ |
799 |
| - '[{"statement_id":0,"series":[{"name":"iops","columns":' \ |
800 |
| - '["fieldKey","fieldType"],"values":[["value","integer"]]}],' \ |
801 |
| - '"partial":true}]}\n{"results":[{"statement_id":0,"series":' \ |
802 |
| - '[{"name":"load","columns":["fieldKey","fieldType"],"values":' \ |
803 |
| - '[["value","integer"]]}],"partial":true}]}\n{"results":' \ |
804 |
| - '[{"statement_id":0,"series":[{"name":"memory","columns":' \ |
805 |
| - '["fieldKey","fieldType"],"values":[["value","integer"]]}]}]}\n' |
| 797 | + example_response = \ |
| 798 | + u'{"results":[{"statement_id":0,"series":' \ |
| 799 | + '[{"name":"cpu","columns":["fieldKey","fieldType"],"values":' \ |
| 800 | + '[["value","integer"]]}],"partial":true}]}\n{"results":' \ |
| 801 | + '[{"statement_id":0,"series":[{"name":"iops","columns":' \ |
| 802 | + '["fieldKey","fieldType"],"values":[["value","integer"]]}],' \ |
| 803 | + '"partial":true}]}\n{"results":[{"statement_id":0,"series":' \ |
| 804 | + '[{"name":"load","columns":["fieldKey","fieldType"],"values":' \ |
| 805 | + '[["value","integer"]]}],"partial":true}]}\n{"results":' \ |
| 806 | + '[{"statement_id":0,"series":[{"name":"memory","columns":' \ |
| 807 | + '["fieldKey","fieldType"],"values":[["value","integer"]]}]}]}\n' |
806 | 808 |
|
807 | 809 | with requests_mock.Mocker() as m:
|
808 | 810 | m.register_uri(
|
809 | 811 | requests_mock.GET,
|
810 | 812 | "http://localhost:8086/query",
|
811 | 813 | text=example_response
|
812 | 814 | )
|
813 |
| - response = list(self.cli.query('show series limit 4 offset 0', chunked=True)) |
| 815 | + response = list(self.cli.query('show series limit 4 offset 0', |
| 816 | + chunked=True)) |
814 | 817 | self.assertTrue(len(response) == 4)
|
815 | 818 | self.assertEqual(response[0].raw, ResultSet(
|
816 |
| - {"statement_id":0, |
817 |
| - "series": [{"name":"cpu","columns":["fieldKey","fieldType"], |
818 |
| - "values": [["value","integer"]]}],"partial":True} |
819 |
| - ).raw) |
| 819 | + {"statement_id": 0, |
| 820 | + "series": [{"name": "cpu", |
| 821 | + "columns": ["fieldKey", "fieldType"], |
| 822 | + "values": [["value", "integer"]]}], |
| 823 | + "partial": True} |
| 824 | + ).raw) |
820 | 825 | self.assertEqual(response[3].raw, ResultSet(
|
821 |
| - {"statement_id":0, |
822 |
| - "series":[{"name":"memory","columns": |
823 |
| - ["fieldKey","fieldType"], |
824 |
| - "values":[["value","integer"]]}]} |
825 |
| - ).raw) |
| 826 | + {"statement_id": 0, |
| 827 | + "series": [{"name": "memory", |
| 828 | + "columns": ["fieldKey", "fieldType"], |
| 829 | + "values": [["value", "integer"]]}]} |
| 830 | + ).raw) |
| 831 | + |
826 | 832 |
|
827 | 833 | class FakeClient(InfluxDBClient):
|
828 | 834 |
|
|
0 commit comments