def setUp(self):
     # create some items
     self.person = Person(firstname="Steve", lastname="Loria")
     self.person2 = Person(firstname="Monty", lastname="Python")
     self.item = Item(name="Foo")
     self.item.save()
     self.person.items.append(self.item)
     self.person.save()
     self.item2 = Item(name="Bar")
     self.person2.save()
     self.item2.save()
Exemple #2
0
 def setUp(self):
     # create some items
     self.person = Person(firstname="Steve", lastname="Loria")
     self.person2 = Person(firstname="Monty", lastname="Python")
     self.item = Item(name="Foo")
     self.item.save()
     self.person.items.append(self.item)
     self.person.save()
     self.item2 = Item(name="Bar")
     self.person2.save()
     self.item2.save()
class TestMongoengineAPI(TestCase):
    TESTING = True
    MONGODB_SETTINGS = {
        "DB": "_test_inventory"
    }
    DEBUG = True

    def create_app(self):
        app.config.from_object(self)
        return app

    def setUp(self):
        # create some items
        self.person = Person(firstname="Steve", lastname="Loria")
        self.person2 = Person(firstname="Monty", lastname="Python")
        self.item = Item(name="Foo")
        self.item.save()
        self.person.items.append(self.item)
        self.person.save()
        self.item2 = Item(name="Bar")
        self.person2.save()
        self.item2.save()

    def tearDown(self):
        drop_collections()

    def test_get_items(self):
        url = "/api/v1/items/"
        res = self.client.get(url)
        data = res.json
        assert_equal(res.status_code, 200)
        assert_equal(len(data['items']), 2)
        assert_equal(data['items'][0]['name'], self.item2.name)

    def test_get_item(self):
        url = '/api/v1/items/{0}'.format(self.item.id)
        res = self.client.get(url)
        data = res.json
        assert_equal(res.status_code, 200)
        assert_equal(data['name'], self.item.name)
        assert_equal(data['person']['id'], str(self.person.id))

    def test_get_persons(self):
        res = self.client.get('/api/v1/people/')
        assert_equal(res.status_code, 200)
        assert_equal(len(res.json['people']), len(Person.objects))

    def test_get_person(self):
        res = self.client.get('/api/v1/people/{0}'.format(self.person.id))
        assert_equal(res.status_code, 200)
        assert_equal(res.json['name'], "{0}, {1}".format(self.person.lastname,
                                                        self.person.firstname))
        assert_equal(res.json['n_items'], 1)

    def test_get_nonexistent_person(self):
        res = self.client.get("/api/v1/people/10")
        assert_equal(res.status_code, 404)

    def _post_json(self, url, data):
        return self.client.post(url,
                                data=json.dumps(data),
                                content_type='application/json')

    def _put_json(self, url, data):
        return self.client.put(url,
                                data=json.dumps(data),
                                content_type='application/json')

    def test_post_item(self):
        res = self._post_json("/api/v1/items/", {"name": "Ipad"})
        assert_equal(res.status_code, 201)
        item = Item.objects.order_by("-updated").first()
        assert_true(item is not None)
        assert_equal(item.name, "Ipad")

    def test_post_item_with_person_id(self):
        res = self._post_json('/api/v1/items/',
                              {"name": "Ipod", "person_id": str(self.person.id)})
        assert_equal(res.status_code, 201)
        item = Item.objects.first()
        person = get_item_person(item)
        assert_equal(person, self.person)

    def test_post_person(self):
        res = self._post_json('/api/v1/people/',
                            {'firstname': 'Foo', 'lastname': 'Bar'})
        assert_equal(res.status_code, 201)
        person = Person.objects.order_by("-created").first()
        assert_equal(person.firstname, "Foo")
        assert_equal(person.lastname, "Bar")

    def test_delete_item(self):
        all_items = Item.objects
        assert_in(self.item, all_items)
        res = self.client.delete("/api/v1/items/{0}".format(self.item.id))
        all_items = Item.objects
        assert_not_in(self.item, all_items)

    def test_put_item(self):
        res = self._put_json("/api/v1/items/{0}".format(self.item.id),
                            {"checked_out": True,
                            "person_id": str(self.person2.id)})
        item = Item.objects(id=self.item.id).first()
        assert_true(item.checked_out)
        item_person = get_item_person(item)
        assert_equal(item_person, self.person2)

    def test_delete_person(self):
        all_persons = Person.objects
        assert_in(self.person, all_persons)
        self.client.delete('/api/v1/people/{0}'.format(self.person.id))
        all_persons = Person.objects
        assert_not_in(self.person, all_persons)

    def test_recent(self):
        self.item.checked_out = True
        self.item2.checked_out = False
        self.item.save()
        self.item2.save()
        res = self.client.get("/api/v1/recentcheckouts/")
        assert_in(ItemDocSerializer(self.item._data).data, res.json['items'])
        assert_not_in(ItemDocSerializer(self.item2._data).data, res.json['items'])
Exemple #4
0
class TestMongoengineAPI(TestCase):
    TESTING = True
    MONGODB_SETTINGS = {"DB": "_test_inventory"}
    DEBUG = True

    def create_app(self):
        app.config.from_object(self)
        return app

    def setUp(self):
        # create some items
        self.person = Person(firstname="Steve", lastname="Loria")
        self.person2 = Person(firstname="Monty", lastname="Python")
        self.item = Item(name="Foo")
        self.item.save()
        self.person.items.append(self.item)
        self.person.save()
        self.item2 = Item(name="Bar")
        self.person2.save()
        self.item2.save()

    def tearDown(self):
        drop_collections()

    def test_get_items(self):
        url = "/api/v1/items/"
        res = self.client.get(url)
        data = res.json
        assert_equal(res.status_code, 200)
        assert_equal(len(data['items']), 2)
        assert_equal(data['items'][0]['name'], self.item2.name)

    def test_get_item(self):
        url = '/api/v1/items/{0}'.format(self.item.id)
        res = self.client.get(url)
        data = res.json
        assert_equal(res.status_code, 200)
        assert_equal(data['name'], self.item.name)
        assert_equal(data['person']['id'], str(self.person.id))

    def test_get_persons(self):
        res = self.client.get('/api/v1/people/')
        assert_equal(res.status_code, 200)
        assert_equal(len(res.json['people']), len(Person.objects))

    def test_get_person(self):
        res = self.client.get('/api/v1/people/{0}'.format(self.person.id))
        assert_equal(res.status_code, 200)
        assert_equal(
            res.json['name'], "{0}, {1}".format(self.person.lastname,
                                                self.person.firstname))
        assert_equal(res.json['n_items'], 1)

    def test_get_nonexistent_person(self):
        res = self.client.get("/api/v1/people/10")
        assert_equal(res.status_code, 404)

    def _post_json(self, url, data):
        return self.client.post(url,
                                data=json.dumps(data),
                                content_type='application/json')

    def _put_json(self, url, data):
        return self.client.put(url,
                               data=json.dumps(data),
                               content_type='application/json')

    def test_post_item(self):
        res = self._post_json("/api/v1/items/", {
            "name": "Ipad",
            'checked_out': True
        })
        assert_equal(res.status_code, 201)
        item = Item.objects.order_by("-updated").first()
        assert_true(item is not None)
        assert_equal(item.name, "Ipad")
        assert_true(item.checked_out)

    def test_post_item_with_person_id(self):
        res = self._post_json('/api/v1/items/', {
            "name": "Ipod",
            "person_id": str(self.person.id)
        })
        assert_equal(res.status_code, 201)
        item = Item.objects.first()
        person = get_item_person(item)
        assert_equal(person, self.person)

    def test_post_person(self):
        res = self._post_json('/api/v1/people/', {
            'firstname': 'Foo',
            'lastname': 'Bar'
        })
        assert_equal(res.status_code, 201)
        person = Person.objects.order_by("-created").first()
        assert_equal(person.firstname, "Foo")
        assert_equal(person.lastname, "Bar")

    def test_delete_item(self):
        all_items = Item.objects
        assert_in(self.item, all_items)
        res = self.client.delete("/api/v1/items/{0}".format(self.item.id))
        all_items = Item.objects
        assert_not_in(self.item, all_items)

    def test_put_item(self):
        res = self._put_json("/api/v1/items/{0}".format(self.item.id), {
            "checked_out": True,
            "person_id": str(self.person2.id)
        })
        item = Item.objects(id=self.item.id).first()
        assert_true(item.checked_out)
        item_person = get_item_person(item)
        assert_equal(item_person, self.person2)

    def test_delete_person(self):
        all_persons = Person.objects
        assert_in(self.person, all_persons)
        self.client.delete('/api/v1/people/{0}'.format(self.person.id))
        all_persons = Person.objects
        assert_not_in(self.person, all_persons)

    def test_recent(self):
        self.item.checked_out = True
        self.item2.checked_out = False
        self.item.save()
        self.item2.save()
        res = self.client.get("/api/v1/recentcheckouts/")
        assert_in(ItemDocSerializer(self.item._data).data, res.json['items'])
        assert_not_in(
            ItemDocSerializer(self.item2._data).data, res.json['items'])