Example #1
0
    def test_pipeline_multi_modules(self):
        def test_func(bundle):
            bundle.data['func'] = True
            return bundle

        def test_func2(bundle):
            bundle.data['func2'] = True
            return bundle

        pipelines = dict()
        pipe = FakePipeline('multi')
        pipe.add_func(test_func)
        pipe.add_func(test_func2)
        pipelines['multi'] = pipe
        thread = self._start_server(pipelines=pipelines)
        client = ProcessClient(thread.server._transport)

        data = client.process_url(str(uuid.uuid4()),
                                  'multi',
                                  dict(),
                                  'fake://blank',
                                  autosave=False,
                                  wait_for_result=True)

        self._stop_server(client, thread)

        self.assertFalse('exceptions' in data)
        self.assertTrue('func' in data)
        self.assertTrue('func2' in data)
        self.assertTrue(data['func'])
        self.assertTrue(data['func'])
Example #2
0
    def test_pipeline_multi_modules(self):
        def test_func(bundle):
            bundle.data['func'] = True
            return bundle

        def test_func2(bundle):
            bundle.data['func2'] = True
            return bundle

        pipelines = dict()
        pipe = FakePipeline('multi')
        pipe.add_func(test_func)
        pipe.add_func(test_func2)
        pipelines['multi'] = pipe
        thread = self._start_server(pipelines=pipelines)
        client = ProcessClient(thread.server._transport)

        data = client.process_url(str(uuid.uuid4()), 'multi', dict(), 'fake://blank',
            autosave=False, wait_for_result=True)

        self._stop_server(client, thread)

        self.assertFalse('exceptions' in data)
        self.assertTrue('func' in data)
        self.assertTrue('func2' in data)
        self.assertTrue(data['func'])
        self.assertTrue(data['func'])