Python find_root_thriftsの例

コード例 #1

ファイル: test_thrift_util.py プロジェクト: CaitieM20/pants

  def test_find_root_thrifts(self):
    with temporary_dir() as dir:
      root_1 = self.write(os.path.join(dir, 'root_1.thrift'), '# noop')
      root_2 = self.write(os.path.join(dir, 'root_2.thrift'), '# noop')
      self.assertEquals({root_1, root_2},
                        find_root_thrifts(basedirs=[], sources=[root_1, root_2]))

    with temporary_dir() as dir:
      root_1 = self.write(os.path.join(dir, 'root_1.thrift'), 'include "mid_1.thrift"')
      self.write(os.path.join(dir, 'mid_1.thrift'), 'include "leaf_1.thrift"')
      self.write(os.path.join(dir, 'leaf_1.thrift'), '# noop')
      root_2 = self.write(os.path.join(dir, 'root_2.thrift'), 'include "root_1.thrift"')
      self.assertEquals({root_2}, find_root_thrifts(basedirs=[], sources=[root_1, root_2]))

コード例 #2

    def test_find_root_thrifts(self):
        with temporary_dir() as dir:
            root_1 = self.write(os.path.join(dir, 'root_1.thrift'), '# noop')
            root_2 = self.write(os.path.join(dir, 'root_2.thrift'), '# noop')
            self.assertEquals({root_1, root_2},
                              find_root_thrifts(basedirs=[],
                                                sources=[root_1, root_2]))

        with temporary_dir() as dir:
            root_1 = self.write(os.path.join(dir, 'root_1.thrift'),
                                'include "mid_1.thrift"')
            self.write(os.path.join(dir, 'mid_1.thrift'),
                       'include "leaf_1.thrift"')
            self.write(os.path.join(dir, 'leaf_1.thrift'), '# noop')
            root_2 = self.write(os.path.join(dir, 'root_2.thrift'),
                                'include "root_1.thrift"')
            self.assertEquals({root_2},
                              find_root_thrifts(basedirs=[],
                                                sources=[root_1, root_2]))