def test_multiple_dbs(self): @self.app.get('/') def test(people, people_2): self.assertEqual(type(people), type(mongo.connect(TEST_DB_NAME))) self.assertEqual(type(people_2), type(mongo.connect(TEST_DB_NAME))) person = Person("Jesus", 33) person.save() self._dbs.append(people) self.plugin = self.app.install( mongoengine.Plugin(TEST_DB_NAME, alias='people', keyword='people')) self.plugin_2 = self.app.install( mongoengine.Plugin(TEST_DB_NAME + '2', alias='people_2', keyword='people_2'))
def test_with_keyword(self): self.plugin = self.app.install(mongoengine.Plugin(TEST_DB_NAME)) @self.app.get('/') def test(db): self.assertEqual(type(db), type(mongo.connect(TEST_DB_NAME))) resp = self.test_app.get('/', expect_errors=True) self.assertEqual(resp.status_int, 200)
def test_uri_ipv6_error(self): @self.app.get('/') def test(db): pass self.plugin = self.app.install( mongoengine.Plugin(db=TEST_DB_NAME, alias='refailied', host='mongodb://1080:0:0:0:8:800:200C:417A')) resp = self.test_app.get('/', expect_errors=True) self.assertEqual(resp.status_int, 500)
def test_uri_schema_error(self): @self.app.get('/') def test(db): pass self.plugin = self.app.install( mongoengine.Plugin(db=TEST_DB_NAME, alias='failed', host='http://notexists@')) resp = self.test_app.get('/', expect_errors=True) self.assertEqual(resp.status_int, 500)
def test_save(self): @self.app.get('/') def test(db): person = Person("Jesus", 33) person.save() self._db = db self.plugin = self.app.install(mongoengine.Plugin(TEST_DB_NAME)) resp = self.test_app.get('/', expect_errors=True) self.assertEqual(resp.status_int, 200) self.assertEqual("Jesus", Person.objects()[0].name) self.assertEqual(33, Person.objects()[0].age)
def test_without_keyword(self): self.plugin = self.app.install(mongoengine.Plugin(TEST_DB_NAME)) @self.app.get('/') def test(): pass resp = self.test_app.get('/', expect_errors=True) self.assertEqual(resp.status_int, 200) @self.app.get('/2') def test_kw(**kw): self.assertFalse('db' in kw) resp = self.test_app.get('/2', expect_errors=True) self.assertEqual(resp.status_int, 200)