diff options
author | Jonathan Tushman <jtushman@pipewave.com> | 2013-09-07 16:30:25 -0400 |
---|---|---|
committer | Raphaƫl Barrois <raphael.barrois@polytechnique.org> | 2013-09-13 02:03:35 +0200 |
commit | 676e02fd916a4e41a6080d29ef8f89925c499f35 (patch) | |
tree | 86b458a92d7f4c5d755449c22d56459e1c71a46c /tests | |
parent | ca393990b35062c5151bc529622131dc93bbed84 (diff) | |
download | factory-boy-676e02fd916a4e41a6080d29ef8f89925c499f35.tar factory-boy-676e02fd916a4e41a6080d29ef8f89925c499f35.tar.gz |
Adding factory support for mongoengine (Closes #89).
Diffstat (limited to 'tests')
-rw-r--r-- | tests/__init__.py | 1 | ||||
-rw-r--r-- | tests/test_mongoengine.py | 62 |
2 files changed, 63 insertions, 0 deletions
diff --git a/tests/__init__.py b/tests/__init__.py index 9960382..d823a87 100644 --- a/tests/__init__.py +++ b/tests/__init__.py @@ -9,3 +9,4 @@ from .test_fuzzy import * from .test_using import * from .test_utils import * from .test_alchemy import * +from .test_mongoengine import * diff --git a/tests/test_mongoengine.py b/tests/test_mongoengine.py new file mode 100644 index 0000000..e078aed --- /dev/null +++ b/tests/test_mongoengine.py @@ -0,0 +1,62 @@ +# -*- coding: utf-8 -*- +# Copyright (c) 2013 Romain Command& +# +# Permission is hereby granted, free of charge, to any person obtaining a copy +# of this software and associated documentation files (the "Software"), to deal +# in the Software without restriction, including without limitation the rights +# to use, copy, modify, merge, publish, distribute, sublicense, and/or sell +# copies of the Software, and to permit persons to whom the Software is +# furnished to do so, subject to the following conditions: +# +# The above copyright notice and this permission notice shall be included in +# all copies or substantial portions of the Software. +# +# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR +# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, +# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE +# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER +# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, +# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN +# THE SOFTWARE. + +"""Tests for factory_boy/SQLAlchemy interactions.""" + +import factory +from .compat import unittest + + +try: + import mongoengine +except ImportError: + mongoengine = None + +if mongoengine: + from factory.mongoengine import MongoEngineFactory + + class Person(mongoengine.Document): + name = mongoengine.StringField() + + class PersonFactory(MongoEngineFactory): + FACTORY_FOR = Person + + name = factory.Sequence(lambda n: 'name%d' % n) + + + +@unittest.skipIf(mongoengine is None, "mongoengine not installed.") +class MongoEngineTestCase(unittest.TestCase): + + def setUp(self): + mongoengine.connect('factory_boy_test') + + def test_build(self): + std = PersonFactory.build() + self.assertEqual('name0', std.name) + self.assertIsNone(std.id) + + def test_creation(self): + std1 = PersonFactory.create() + self.assertEqual('name1', std1.name) + self.assertIsNotNone(std1.id) + + |