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)
