def test_walk_file_and_dir(self):

        expected_files = ['dfoo/foo', 'dbar/bar', 'dbiz/biz']
        expected_dirs = ['dfoo', 'dbar', 'dbiz']
        actual_files = []
        create_file_tree(self.wordkir, expected_dirs, expected_files)
        source = walk.make_driver().call(
            walk.Sink(request=Observable.just(
                walk.Walk(top=self.wordkir, id='test', recursive=True))))

        class TestObserver(Observer):
            def on_next(self, i):
                actual_files.append(i)

            def on_completed(self):
                return

            def on_error(self, e):
                raise Exception(e)

        source.response \
            .filter(lambda i: type(i) is walk.WalkResponse) \
            .flat_map(lambda i: i.content.directories) \
            .flat_map(lambda i: i.files) \
            .subscribe(TestObserver())

        for f in expected_files:
            self.assertIn(os.path.join(self.wordkir, f), actual_files)
示例#2
0
    def test_walk_file(self):

        expected_files = ['foo', 'bar', 'biz']
        actual_files = []
        create_file_tree(self.wordkir, [], expected_files)
        source = walk.make_driver().call(
            walk.Sink(request=rx.just(walk.Walk(top=self.wordkir, id='test'))))

        class TestObserver(Observer):
            def on_next(self, i):
                actual_files.append(i)

            def on_completed(self):
                return

            def on_error(self, e):
                raise Exception(e)

        source.response.pipe(
            ops.filter(lambda i: type(i) is walk.WalkResponse),
            ops.flat_map(lambda i: i.content.files),
        ).subscribe(TestObserver())

        for f in expected_files:
            self.assertIn(os.path.join(self.wordkir, f), actual_files)
 def test_create(self):
     source = walk.make_driver().call(walk.Sink(request=Observable.empty()))
     self.assertIsNotNone(source)