Exemplo n.º 1
0
class TestMockDataError(TestCase):

    def setUp(self):
        super(TestMockDataError, self).setUp()
        self.mdlwr = MockDataApiMiddleware({
            'a': {'data': {'result': 'a'}},
            'b': {'data': {'result': 'b'}},
            'c': {'filter': {'method': 'post'}, 'data': {'result': 'b'}},
        })
        self.cnx = ApiConnexion(url='http://localapi/v2/')
        self.cnx.push_middleware(self.mdlwr, 3)

    def test_not_found(self):
        with self.assertRaises(Exception) as e:
            self.cnx.get('dada')
        self.assertEqual(e.exception.args, ("the query 'dada' was not provided as mocked data: "
                                            "urls was %s" % ['a', 'b', 'c'], ))

    def test_not_found_abs(self):
        with self.assertRaises(Exception) as e:
            self.cnx.get('/dada')
        self.assertEqual(e.exception.args, ("the query 'http://localapi/dada' was not provided as mocked data: "
                                            "urls was %s" % ['a', 'b', 'c'],))

    def test_found_bad_filters(self):
        with self.assertRaises(Exception) as e:
            self.cnx.get('c')
        self.assertTrue(e.exception.args[0].startswith("the query 'c' was not provided as mocked data: "
                                                       "1 fixture for this url, but filter did not match. "
                                                       "got "
                                                       ))

    def test_ok(self):
        self.cnx.post('c', data={})
        self.cnx.get('a')
class TestLocalApiHandler(TestCase):
    fixtures = ['data.json']

    def setUp(self):
        self.live_server_url = LocalApiAdapter.SPECIAL_URL
        self.client = ApiConnexion(self.live_server_url + "/api/v2/",
                                   auth=None)

    def test_api_connectionerror(self):
        c = ApiConnexion("http://127.0.0.1:7777")
        self.assertRaisesMessage(FakeDatabaseDbAPI2.OperationalError,
                                 'Is the API', c.get, '')

    def test_api_get(self):
        r = self.client.get("pizza/1/")
        self.assertEqual(r.json()['pizza']['id'], 1)

    def test_api_filter(self):
        r = self.client.get('pizza/',
                            params={
                                'filter{id}': '1',
                                'include[]': 'toppings.*'
                            })
        data = r.json()
        self.assertEqual(data['pizzas'][0]['id'], 1)
        self.assertEqual(len(data['toppings']),
                         len(data['pizzas'][0]['toppings']))

    def test_api_post(self):

        r = self.client.post('pizza/',
                             json=dict(
                                 pizza={
                                     "toppings": [2, 3, 5],
                                     "from_date": "2016-11-15",
                                     "price": 13.0,
                                     "to_date": "2016-11-20T08:46:02.016000",
                                     "name": "chévre",
                                     "id": 1,
                                 }))
        self.assertEqual(r.status_code, 201)
        data = r.json()
        self.assertGreater(data['pizza']['id'],
                           3)  # 3 is the id of the fixture

    def test_api_put(self):
        rget = self.client.get('pizza/1')
        data = rget.json()
        self.assertEqual(data['pizza']['price'], 10.0)
        data['pizza']['price'] = 0.
        rput = self.client.put('pizza/1', json=data)
        self.assertEqual(rput.status_code, 200)
        self.assertEqual(rput.json()['pizza']['price'], 0)
        rget2 = self.client.get('pizza/1')
        self.assertEqual(rget2.json()['pizza']['price'], 0)

    def test_api_patch(self):
        rget = self.client.get('pizza/1')
        data = rget.json()
        self.assertEqual(data['pizza']['price'], 10.0)
        rput = self.client.patch('pizza/1', json={'pizza': {'price': 0}})
        self.assertEqual(rput.status_code, 200)
        self.assertEqual(rput.json()['pizza']['price'], 0)
        rget2 = self.client.get('pizza/1')
        self.assertEqual(rget2.json()['pizza']['price'], 0)

    def test_api_delete(self):
        rget = self.client.get('pizza/1')
        self.assertEqual(rget.status_code, 200)
        self.client.delete('pizza/1')
        rget2 = self.client.get('pizza/1')
        self.assertEqual(rget2.status_code, 404)

    def test_api_options(self):
        rget = self.client.options('pizza/1')
        self.assertEqual(rget.status_code, 200)
        data = rget.json()
        self.assertIn('include[]', data['features'])

    def test_auth_forbiden(self):
        c = ApiConnexion(self.live_server_url + "/api/v2/", auth=None)

        self.assertRaises(ProgrammingError,
                          c.patch,
                          'authpizza/1',
                          json={'pizza': {
                              'price': 0
                          }})

    def test_auth_admin(self):
        c = ApiConnexion(self.live_server_url + "/api/v2/",
                         auth=('admin', 'admin'))
        r = c.patch('authpizza/1', json={'pizza': {'price': 0}})
        self.assertEqual(r.status_code, 200)

    def test_auth_no_rigth(self):
        c = ApiConnexion(self.live_server_url + "/api/v2/",
                         auth=('user1', 'user1'))
        self.assertRaises(ProgrammingError,
                          c.patch,
                          'authpizza/1',
                          json={'pizza': {
                              'price': 0
                          }})