Added CollectionManager, made connection module

All connection-related functions are now in connection.py.

Created a ConnectionManager class for interacting with a collection
in the database. Top-level document classes have an instance of
a ConnectionManager (Document.collection).

Defined a 'save' method on top-level document's that uses the collection
manager's '_save_document' method to save the document to the database.

Added tests for CollectionManagers -- all unit tests now require a valid
connection to the database, which is set up in the tests' setUp method.

tests/collection.py

@@ -0,0 +1,33 @@
+import unittest
+import pymongo
+
+from mongomap.collection import CollectionManager
+from mongomap import *
+
+
+class CollectionManagerTest(unittest.TestCase):
+    
+    def setUp(self):
+        connect(db='mongotest')
+
+        class Person(Document):
+            name = StringField()
+            age = IntField()
+        self.Person = Person
+
+    def test_initialisation(self):
+        """Ensure that CollectionManager is correctly initialised.
+        """
+        class Person(Document):
+            name = StringField()
+            age = IntField()
+
+        self.assertTrue(isinstance(Person.collection, CollectionManager))
+        self.assertEqual(Person.collection._collection_name, 
+                         Person._meta['collection'])
+        self.assertTrue(isinstance(Person.collection._collection,
+                                   pymongo.collection.Collection))
+
+
+if __name__ == '__main__':
+    unittest.main()

tests/document.py

@@ -1,16 +1,22 @@
 import unittest
 
 from mongomap import *
+from mongomap.connection import _get_db
 
 
 class DocumentTest(unittest.TestCase):
     
     def setUp(self):
+        connect(db='mongomaptest')
+
         class Person(Document):
             name = StringField()
             age = IntField()
         self.Person = Person
 
+        self.db = _get_db()
+        self.db.drop_collection(self.Person._meta['collection'])
+
     def test_definition(self):
         """Ensure that document may be defined using fields.
         """
@@ -77,6 +83,45 @@ class DocumentTest(unittest.TestCase):
         self.assertTrue('content' in Comment._fields)
         self.assertFalse(hasattr(Comment, '_meta'))
 
+    def test_save(self):
+        """Ensure that a document may be saved in the database.
+        """
+        # Create person object and save it to the database
+        person = self.Person(name='Test User', age=30)
+        person.save()
+        # Ensure that the object is in the database
+        collection = self.db[self.Person._meta['collection']]
+        person_obj = collection.find_one({'name': 'Test User'})
+        self.assertEqual(person_obj['name'], 'Test User')
+        self.assertEqual(person_obj['age'], 30)
+
+    def test_save_embedded_document(self):
+        """Ensure that a document with an embedded document field may be 
+        saved in the database.
+        """
+        class EmployeeDetails(EmbeddedDocument):
+            position = StringField()
+
+        class Employee(self.Person):
+            salary = IntField()
+            details = EmbeddedDocumentField(EmployeeDetails)
+
+        # Create employee object and save it to the database
+        employee = Employee(name='Test Employee', age=50, salary=20000)
+        employee.details = EmployeeDetails(position='Developer')
+        employee.save()
+
+        # Ensure that the object is in the database
+        collection = self.db[self.Person._meta['collection']]
+        employee_obj = collection.find_one({'name': 'Test Employee'})
+        self.assertEqual(employee_obj['name'], 'Test Employee')
+        self.assertEqual(employee_obj['age'], 50)
+        # Ensure that the 'details' embedded object saved correctly
+        self.assertEqual(employee_obj['details']['position'], 'Developer')
+
+    def tearDown(self):
+        self.db.drop_collection(self.Person._meta['collection'])
+
 
 if __name__ == '__main__':
     unittest.main()

tests/fields.py

@@ -5,6 +5,9 @@ from mongomap import *
 
 class FieldTest(unittest.TestCase):
 
+    def setUp(self):
+        connect(db='mongomaptest')
+
     def test_default_values(self):
         """Ensure that default field values are used when creating a document.
         """