def setUp(self):
        self.company_id = ObjectId()
        self.user_collection = conn.user
        self.company_collection = Mock()
        self.company_collection.insert = Mock(return_value=self.company_id)
        self.company_collection.find_one = Mock(return_value={'_id': self.company_id})

        trait("versioned", {
            "v": 4
        })

        with factory("user", self.user_collection):
            fragment("prefs_email", {
                "receives_sms": True,
                "receives_email": False
            }, traits=['versioned'])

            fragment("prefs_sms", {
                "receives_sms": False,
                "receives_email": True                
            }, traits=['versioned'])

            fragment("empty_prefs", traits=['versioned'])

            default({
                "first": "John",
                "last": "Smith",
                "prefs": embed("prefs_email")
            }, traits=["common"])

            document("admin", {
                "first": "Bill",
                "last": "Jones",
                "prefs": embed("prefs_sms"),
            }, parent="default", traits=["versioned"])

            document("nameless", parent="default", traits=["versioned"])

            trait("timestamped", {
                "created": "now"
            })

            trait("common", {
                "company_id": id_of("company"),
                "email": dependent(lambda doc: "*****@*****.**" % (doc['first'], doc['last'])),
                "age": sequence(lambda n: n + 20)
            }, parent="timestamped")

        
        with factory("company", self.company_collection):
            default({
                "name": "GloboCorp"
            })

            document('pharma', {
                "name": "Pfizer"
            })

        with factory("fake"):
            default({
                "fake": True
            })
Exemple #2
0
    def setUp(self):
        self.company_id = ObjectId()
        self.user_collection = conn.user
        self.company_collection = Mock()
        self.company_collection.insert = Mock(return_value=self.company_id)
        self.company_collection.find_one = Mock(
            return_value={'_id': self.company_id})

        trait("versioned", {"v": 4})

        with factory("user", self.user_collection):
            fragment("prefs_email", {
                "receives_sms": True,
                "receives_email": False
            },
                     traits=['versioned'])

            fragment("prefs_sms", {
                "receives_sms": False,
                "receives_email": True
            },
                     traits=['versioned'])

            fragment("empty_prefs", traits=['versioned'])

            default(
                {
                    "first": "John",
                    "last": "Smith",
                    "prefs": embed("prefs_email")
                },
                traits=["common"])

            document("admin", {
                "first": "Bill",
                "last": "Jones",
                "prefs": embed("prefs_sms"),
            },
                     parent="default",
                     traits=["versioned"])

            document("nameless", parent="default", traits=["versioned"])

            trait("timestamped", {"created": "now"})

            trait("common", {
                "company_id":
                id_of("company"),
                "email":
                dependent(lambda doc: "*****@*****.**" %
                          (doc['first'], doc['last'])),
                "age":
                sequence(lambda n: n + 20)
            },
                  parent="timestamped")

        with factory("company", self.company_collection):
            default({"name": "GloboCorp"})

            document('pharma', {"name": "Pfizer"})

        with factory("fake"):
            default({"fake": True})
 def test_fragment_not_callable_outside_factory_context(self):
     with self.assertRaises(FactoryContextException):
         fragment("test", {"a": "b"})
Exemple #4
0
 def test_fragment_not_callable_outside_factory_context(self):
     with self.assertRaises(FactoryContextException):
         fragment("test", {"a": "b"})