add embedded router

Author: timgraham

django_mongodb_backend/routers.py

@@ -0,0 +1,10 @@
+from django.apps import apps
+
+from django_mongodb_backend.models import EmbeddedModel
+
+
+class MongoDBRouter:
+    def allow_migrate(self, db, app_label, model_name=None, **hints):
+        """EmbeddedModels don't have their own collectionn."""
+        model = apps.get_model(app_label, model_name)
+        return False if issubclass(model, EmbeddedModel) else None

tests/models_/models.py

@@ -1,5 +1,11 @@
+from django.db import models
+
 from django_mongodb_backend.models import EmbeddedModel
 
 
 class Embed(EmbeddedModel):
     pass
+
+
+class PlainModel(models.Model):
+    pass

tests/models_/test_embedded_model.py

@@ -1,6 +1,8 @@
 from django.db import NotSupportedError
 from django.test import SimpleTestCase
 
+from django_mongodb_backend.routers import MongoDBRouter
+
 from .models import Embed
 
 
@@ -45,3 +47,10 @@ def test_get_or_create(self):
         msg = "'EmbeddedModelManager' object has no attribute 'get_or_create'"
         with self.assertRaisesMessage(AttributeError, msg):
             Embed.objects.get_or_create()
+
+
+class TestRouter(SimpleTestCase):
+    def test_allow_migrate(self):
+        router = MongoDBRouter()
+        self.assertIsNone(router.allow_migrate("db", "models_", "plainmodel"))
+        self.assertIs(router.allow_migrate("db", "models_", "embed"), False)