Exemplo n.º 1
0
 def test_rest_action_connection(self):
     """Test the rest timeout indefinitely"""
     # This will try to actually connect to an invalid service
     test_base = Base(self.CERT, self.URLBASE)
     rest_function = test_base._session.get
     with self.assertRaises(requests.ConnectionError):
         test_base.rest_action(rest_function, self.URLBASE)
Exemplo n.º 2
0
 def test_rest_action_not_json(self):
     """Test the rest timeout indefinitely"""
     self._register_uri(body='stuff')
     test_base = Base(self.CERT, self.URLBASE)
     rest_function = test_base._session.get
     with self.assertRaises(ValueError):
         test_base.rest_action(rest_function, self.URLBASE)
Exemplo n.º 3
0
 def test_rest_action_success(self):
     """Test the rest action"""
     payload = {'a': 'b'}
     self._register_uri(body=json.dumps(payload))
     test_base = Base(self.CERT, self.URLBASE)
     rest_function = test_base._session.get
     response = test_base.rest_action(rest_function, self.URLBASE)
     self.assertEqual(payload, response)
Exemplo n.º 4
0
 def test_url_format(self):
     """Verify url format does the right thing"""
     base = Base(self.CERT, self.URLBASE)
     service = 'assignment'
     self.assertEqual(
         '{0}{1}'.format(self.URLBASE, service),
         base._url_format(service)
     )
     self.assertTrue(type(base._url_format(service)), str)
Exemplo n.º 5
0
 def test_data_to_json(self):
     """Verify that we convert python data to json"""
     data = dict(a='b')
     self.assertEqual(
         Base._data_to_json(data),
         json.dumps(data)
     )
     self.assertEqual('a', Base._data_to_json('a'))
     self.assertEqual('a', Base._data_to_json(unicode('a')))
Exemplo n.º 6
0
 def test_rest_action_timeout(self):
     """Test the rest timeout indefinitely"""
     self._register_uri(timeout=True)
     test_base = Base(self.CERT, self.URLBASE)
     rest_function = test_base._session.get
     with self.assertRaisesRegexp(
         requests.ConnectionError,
         'Max retries exceeded with url'
     ):
         test_base.rest_action(rest_function, self.URLBASE)
Exemplo n.º 7
0
 def test_rest_action_success_eventually(self):
     """Wait some time and then succeed"""
     payload = {'a': 'b'}
     self._register_uri(
         responses=[
             httpretty.Response(body=raise_timeout),
             httpretty.Response(body=raise_timeout),
             httpretty.Response(body=json.dumps(payload)),
         ]
     )
     test_base = Base(self.CERT, self.URLBASE)
     rest_function = test_base._session.get
     response = test_base.rest_action(rest_function, self.URLBASE)
     self.assertEqual(payload, response)
Exemplo n.º 8
0
    def test_delete_success(self):
        """Verify that delete works as expected."""
        service = 'supercool'
        full_url = '{0}{1}'.format(self.URLBASE, service)
        response = dict(a='b')

        # Apparently the service always returns JSON, but this seems
        # like a really suspicious assumption.  I would really like to
        # verify it.
        httpretty.register_uri(
            httpretty.DELETE,
            full_url,
            body=json.dumps(response)
        )
        test_base = Base(self.CERT, self.URLBASE)
        response_json = test_base.delete(service)
        self.assertEqual(response, response_json)
Exemplo n.º 9
0
    def test_post_success(self):
        """Verify that post works as expected."""
        service = 'supercool'
        full_url = '{0}{1}'.format(self.URLBASE, service)
        response = dict(a='b')
        post_data = {'a': 'b'}

        # Apparently the service always returns JSON, but this seems
        # like a really suspicious assumption.  I would really like to
        # verify it.
        httpretty.register_uri(
            httpretty.POST,
            full_url,
            body=json.dumps(response)
        )
        test_base = Base(self.CERT, self.URLBASE)
        response_json = test_base.post(service, data=post_data)
        self.assertEqual(response, response_json)
        last_request = httpretty.last_request()
        self.assertEqual(last_request.parsed_body, post_data)
Exemplo n.º 10
0
    def test_get_success(self):
        """Verify that get works as expected"""
        service = 'notreal'
        data = dict(a='b')
        full_url = '{0}{1}'.format(self.URLBASE, service)
        # Using array here because httpretty auto arrays params
        params = dict(c=['d'])
        httpretty.register_uri(
            httpretty.GET,
            full_url,
            body=json.dumps(data),
        )
        test_base = Base(self.CERT, self.URLBASE)
        response_json = test_base.get(service, params=params)
        self.assertEqual(data, response_json)
        last_request = httpretty.last_request()
        self.assertEqual(last_request.querystring, params)

        # Now without params
        response_json = test_base.get(service)
        self.assertEqual(data, response_json)
        last_request = httpretty.last_request()
        self.assertEqual(last_request.querystring, {})
Exemplo n.º 11
0
 def test_delete_failure(self):
     """Verify we are raising properly if a get request fails."""
     test_base = Base(self.CERT, self.URLBASE)
     with self.assertRaises(requests.ConnectionError):
         test_base.delete('my-special-beans')
Exemplo n.º 12
0
 def test_post_failure(self):
     """Verify we are raising properly if a get request fails."""
     test_base = Base(self.CERT, self.URLBASE)
     with self.assertRaises(requests.ConnectionError):
         test_base.post('make-beans', json.dumps(dict(beans='pinto')))
Exemplo n.º 13
0
 def test_get_failure(self):
     """Verify we are raising properly if a get request fails."""
     test_base = Base(self.CERT, self.URLBASE)
     with self.assertRaises(requests.ConnectionError):
         test_base.get('pinto_beans')