Attachment 1029400 Details for Bug 1222509 – client file containing GET and POST of crush map for which mon fails.

client file containing GET and POST of crush map for which mon fails.

CalAPITester.py (text/x-python), 5.46 KB, created by Harish NV Rao on 2015-05-25 09:35:20 UTC

(hide)

Description:

Filename:

MIME Type:

Creator: Harish NV Rao

Created: 2015-05-25 09:35:20 UTC

Size: 5.46 KB

patch

obsolete

>#!/usr/bin/env python
>
>import json
>import logging
>import requests
>
>log = logging.getLogger(__name__)
>
>class AuthenticatedHttpClient(requests.Session):
>    """
>    Client for the calamari REST API, principally exists to do
>    authentication, but also helpfully prefixes
>    URLs in requests with the API base URL and JSONizes
>    POST data.
>    """
>    def __init__(self, api_url, username, password):
>        super(AuthenticatedHttpClient, self).__init__()
>        self._username = username
>        self._password = password
>        self._api_url = api_url
>        self.headers = {
>            'Content-type': "application/json; charset=UTF-8"
>        }
>
>    @property
>    def prefix(self):
>        return self._api_url
>
>    def request(self, method, url, **kwargs):
>        url = self._api_url + url
>        print "REQUEST: **kwargs = ",kwargs
>	print "REQUEST: url = ", url
>	print "REQUEST: self = ", self
>	print "REQUEST: method = ", method
>	print "REQUEST: self.headers = ", self.headers
>	print "REQUEST: self.headers['Content-type'] = ", self.headers['Content-type']
>	if url == "http://10.16.184.71/api/v2/cluster/cbc3750d-763e-42f2-a929-b7043734f4f0/crush_map":
>		if method == "POST": 
>			self.headers['Content-type'] = "text/plain"
>			print "REQUEST: After modification : self.headers['Content-type'] = ", self.headers['Content-type']
>
>        response = super(AuthenticatedHttpClient, self).request(method, url, **kwargs)
>        if response.status_code >= 400:
>            # For the benefit of test logs
>            print "%s: %s" % (response.status_code, response.content)
>        return response
>
>    def post(self, url, data=None, **kwargs):
>	print "dict = ", dict
>        if isinstance(data, dict):
>            data = json.dumps(data)
>
>        print "POST: **kwargs = ",kwargs
>	print "POST: url = ", url
>	print "POST: self = ", self
>	print "POST: data = ", data
>
>        return super(AuthenticatedHttpClient, self).post(url, data, **kwargs)
>
>    def patch(self, url, data=None, **kwargs):
>        if isinstance(data, dict) or isinstance(data, list):
>            data = json.dumps(data)
>        return super(AuthenticatedHttpClient, self).patch(url, data, **kwargs)
>
>    def delete(self, url, data=None, **kwargs):
>        if isinstance(data, dict):
>            data = json.dumps(data)
>        return super(AuthenticatedHttpClient, self).delete(url, data=data, **kwargs)
>
>    def login(self):
>        """
>        Authenticate with the Django auth system as
>        it is exposed in the Calamari REST API.
>        """
>        log.info("Logging in as %s" % self._username)
>        response = self.get("auth/login/")
>        response.raise_for_status()
>        self.headers['X-XSRF-TOKEN'] = response.cookies['XSRF-TOKEN']
>
>        response = self.post("auth/login/", {
>            'next': "/",
>            'username': self._username,
>            'password': self._password
>        })
>        response.raise_for_status()
>        # XSRF token rotates on login
>        self.headers['X-XSRF-TOKEN'] = response.cookies['XSRF-TOKEN']
>
>        # Check we're allowed in now.
>        response = self.get("cluster")
>        response.raise_for_status()
>
>if __name__ == "__main__":
>
>    import argparse
>
>    p = argparse.ArgumentParser()
>    p.add_argument('-u', '--uri', default='http://mira035/api/v1/')
>    p.add_argument('--user', default='admin')
>    p.add_argument('--pass', dest='password', default='admin')
>    args, remainder = p.parse_known_args()
>
>    # Authicate the user.
>    # Run this script: python temp.py -u http://10.8.128.6/api/v2/ --user TEST12 --pass ghj
>
>    c = AuthenticatedHttpClient(args.uri, args.user, args.password)
>    c.login()
>    response = c.request('GET', ''.join(remainder)).json()
>    print json.dumps(response, indent=2)
>
>    # Get the crushmap
>    print "######################## GETting the original crushmap  #########################################"
>    response_GET = c.request('GET', 'cluster/cbc3750d-763e-42f2-a929-b7043734f4f0/crush_map').json()
>#    print json.dumps(response_GET, indent=2)
>    print response_GET
>
>#    print "######################## POSTing the Modified crushmap #########################################"
>#    response_PUT = c.request('POST', 'cluster/cbc3750d-763e-42f2-a929-b7043734f4f0/crush_map').json()
>#    print json.dumps(response_PUT, indent=2)
>
>    response = c.post('cluster/cbc3750d-763e-42f2-a929-b7043734f4f0/crush_map', '# begin crush map. \ntunable choose_local_tries 0\ntunable choose_local_fallback_tries 0\ntunable choose_total_tries 68\ntunable chooseleaf_descend_once 1\ntunable straw_calc_version 1\n\n# devices\ndevice 0 osd.0\ndevice 1 osd.1\ndevice 2 osd.2\ndevice 3 osd.3\n\n# types\ntype 0 osd\ntype 1 host\ntype 2 chassis\ntype 3 rack\ntype 4 row\ntype 5 pdu\ntype 6 pod\ntype 7 room\ntype 8 datacenter\ntype 9 region\ntype 10 root\n\n# buckets\nhost osd0 {\n\tid -2\t\t# do not change unnecessarily\n\t# weight 1.820\n\talg straw\n\thash 0\t# rjenkins1\n\titem osd.0 weight 0.910\n\titem osd.2 weight 0.910\n}\nhost osd1 {\n\tid -3\t\t# do not change unnecessarily\n\t# weight 1.820\n\talg straw\n\thash 0\t# rjenkins1\n\titem osd.1 weight 0.910\n\titem osd.3 weight 0.910\n}\nroot default {\n\tid -1\t\t# do not change unnecessarily\n\t# weight 3.640\n\talg straw\n\thash 0\t# rjenkins1\n\titem osd0 weight 1.820\n\titem osd1 weight 1.820\n}\n\n# rules\nrule replicated_ruleset {\n\truleset 0\n\ttype replicated\n\tmin_size 1\n\tmax_size 10\n\tstep take default\n\tstep chooseleaf firstn 0 type host\n\tstep emit\n}\n\n# end crush map\n').json()
>
>    print json.dumps(response, indent=2)
>
>

Actions: View

Attachments on bug 1222509: 1026671 | 1029386 | 1029400 | 1029402