def test_uri_connect(self): conn = connect(uri="mongodb://localhost/%s" % DBNAME) self.assertTrue(isinstance(conn, pymongo.Connection)) connection = Connection.instance() self.assertEqual(connection._database, DBNAME) conn.disconnect() # overriding the database name conn = connect(DBNAME, uri="mongodb://localhost/foobar") self.assertTrue(isinstance(conn, pymongo.Connection)) connection = Connection.instance() self.assertEqual(connection._database, DBNAME) conn.disconnect()
#!/usr/bin/python3.2 """ Various tests for the Model class """ from mago import Field, ReferenceField, Model, connect, UnSet, OneToMany, ManyToOne from bson import ObjectId import unittest import mago _mongo_connection = connect("__test_model") class Bar(object): bar = "open bar" def __init__(self): self.close = "almost a model" class NotAModel(object): class_attr = "class...ic" def __init__(self): self.obj_attr = "objective" self._so = Bar() def method(self): return "just a {wait_for_it}".format(wait_for_it=self.obj_attr) class Foo(Model): field = Field() default = Field(default="default")
def setUp(self): super(MagoFieldTests, self).setUp() self._mongo_connection = connect("__test_change_field_name")
def setUp(self): super().setUp() self._mongo_connection = mago.connect("__test_model") Foo.collection().remove({})
def setUp(self): super().setUp() # self._mongo_connection.drop_database("__test_model") self._mongo_connection = mago.connect("__test_model") Foo.collection().remove({}) tran.Transaction.collection().remove({})
def setUp(self): self._conn = connect(DBNAME)