added shutdown function to remove test db

doctrine · Dec 20, 2011 · 37eb381 · 37eb381
1 parent 68e772b
commit 37eb381
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 13 deletions.
diff --git a/Tests/Validator/Constraints/UniqueValidatorTest.php b/Tests/Validator/Constraints/UniqueValidatorTest.php
@@ -34,14 +34,9 @@ protected function setUp()
             __DIR__ . '/../DependencyInjection/Fixtures/Bundles/AnnotationsBundle/Document'
         ));
 
-        $this->dropDocumentCollection();
-    }
-
-    protected function tearDown()
-    {
-        if ($this->documentManager) {
-            $this->dropDocumentCollection();
-        }
+        $this->documentManager
+            ->getDocumentCollection('Symfony\Bundle\DoctrineMongoDBBundle\Tests\Fixtures\Validator\Document')
+            ->drop();
     }
 
     public function testValidateUniquenessForScalarField()
@@ -223,11 +218,6 @@ public function testEmbeddedDocumentIsNotUnsupported()
         $violationsList = $validator->validate(new EmbeddedDocument());
     }
 
-    private function dropDocumentCollection()
-    {
-        $this->documentManager->getDocumentCollection('Symfony\Bundle\DoctrineMongoDBBundle\Tests\Fixtures\Validator\Document')->drop();
-    }
-
     private function createMockContainer($documentManagerId = self::DEFAULT_DOCUMENT_MANAGER)
     {
         $container = $this->getMock('Symfony\Component\DependencyInjection\ContainerInterface');

diff --git a/Tests/bootstrap.php b/Tests/bootstrap.php
@@ -14,3 +14,11 @@
 } elseif (file_exists($file = __DIR__.'/autoload.php.dist')) {
     require_once $file;
 }
+
+register_shutdown_function(function() {
+    try {
+        $mongo = new Mongo();
+        $mongo->doctrine->drop();
+    } catch (\MongoException $e) {
+    }
+});