Python Builder.macro示例

示例#1

文件： test_builder.py 项目： mandarvaze/sdispater-orator

    def test_macros_are_called_on_builder(self):
        builder = Builder(
            QueryBuilder(flexmock(Connection), flexmock(QueryGrammar),
                         flexmock(QueryProcessor)))

        def foo_bar(builder):
            builder.foobar = True

            return builder

        builder.macro("foo_bar", foo_bar)
        result = builder.foo_bar()

        self.assertEqual(result, builder)
        self.assertTrue(builder.foobar)

示例#2

文件： test_builder.py 项目： southwestjiaotongunivercity/orator

    def test_macros_are_called_on_builder(self):
        builder = Builder(QueryBuilder(
            flexmock(Connection),
            flexmock(QueryGrammar),
            flexmock(QueryProcessor)
        ))

        def foo_bar(builder):
            builder.foobar = True

            return builder

        builder.macro('foo_bar', foo_bar)
        result = builder.foo_bar()

        self.assertEqual(result, builder)
        self.assertTrue(builder.foobar)