def test_drop_retention_policy(self):
        """Test drop retention policy for TestInfluxDBClient object."""
        cli = DataFrameClient(database='db')
        example_response = '{"results":[{}]}'

        with requests_mock.Mocker() as m:
            m.register_uri(requests_mock.POST,
                           "http://localhost:8086/query",
                           text=example_response)
            cli.drop_retention_policy('somename', 'db')
            self.assertEqual(m.last_request.qs['q'][0],
                             'drop retention policy "somename" on "db"')
    def test_drop_retention_policy(self):
        """Test drop retention policy for TestInfluxDBClient object."""
        cli = DataFrameClient(database='db')
        example_response = '{"results":[{}]}'

        with requests_mock.Mocker() as m:
            m.register_uri(
                requests_mock.POST,
                "http://localhost:8086/query",
                text=example_response
            )
            cli.drop_retention_policy('somename', 'db')
            self.assertEqual(
                m.last_request.qs['q'][0],
                'drop retention policy "somename" on "db"'
            )
 def test_drop_retention_policy_fails(self):
     """Test failed drop ret policy for TestInfluxDBClient object."""
     cli = DataFrameClient('host', 8086, 'username', 'password')
     with _mocked_session(cli, 'delete', 401):
         cli.drop_retention_policy('default', 'db')
 def test_drop_retention_policy_fails(self):
     """Test failed drop ret policy for TestInfluxDBClient object."""
     cli = DataFrameClient('host', 8086, 'username', 'password')
     with _mocked_session(cli, 'delete', 401):
         cli.drop_retention_policy('default', 'db')