Esempio n. 1
0
    def test_check_and_add_init_py(self):
        """
        Test that __init__.py is created in all directories in which it was missing
        :return:
        """
        dirs = [
            'myproj/common',
            'myproj/specific',
            'yourproj',
            'theirproj/common',
            'theirproj/common/steps',
            'theirproj/specific',
            'theirproj/.git',
            'theirproj/.git/dir',
        ]

        files = [
            # 'myproj/__init__.py'
            'myproj/common/__init__.py',
            # 'myproj/specific/__init__.py',
            'yourproj/__init__.py',
            'theirproj/__init__.py',
            # 'theirproj/common/__init__.py',
            'theirproj/common/steps/__init__.py',
            # 'theirproj/specific/__init__.py',
        ]

        f = [
            'myproj/__init__.py',
            'myproj/specific/__init__.py',
            'theirproj/common/__init__.py',
            'theirproj/specific/__init__.py',
        ]

        files_to_check = [os.path.join(self.WORKING_DIR, x) for x in f]

        # create dirs
        for d in dirs:
            os.makedirs(d)

        # create files
        for f in files:
            with open(f, 'w') as fw:
                fw.write(f)

        output = BehaveWorkingDirectory.check_and_add_init_py(os.getcwd())
        print(output)
        assert len(
            set(files_to_check).intersection(output)) == len(files_to_check)

        for f in files_to_check:
            assert os.path.exists(f)
Esempio n. 2
0
    def test_check_and_add_init_py(self):
        """
        Test that __init__.py is created in all directories in which it was missing
        :return:
        """
        dirs = [
            'myproj/common',
            'myproj/specific',
            'yourproj',
            'theirproj/common',
            'theirproj/common/steps',
            'theirproj/specific',
            'theirproj/.git',
            'theirproj/.git/dir',
        ]

        files = [
            # 'myproj/__init__.py'
            'myproj/common/__init__.py',
            # 'myproj/specific/__init__.py',
            'yourproj/__init__.py',
            'theirproj/__init__.py',
            # 'theirproj/common/__init__.py',
            'theirproj/common/steps/__init__.py',
            # 'theirproj/specific/__init__.py',
        ]

        f = [
            'myproj/__init__.py',
            'myproj/specific/__init__.py',
            'theirproj/common/__init__.py',
            'theirproj/specific/__init__.py',
        ]

        files_to_check = [os.path.join(self.WORKING_DIR, x) for x in f]

        # create dirs
        for d in dirs:
            os.makedirs(d)

        # create files
        for f in files:
            with open(f, 'w') as fw:
                fw.write(f)

        output = BehaveWorkingDirectory.check_and_add_init_py(os.getcwd())
        print(output)
        assert len(set(files_to_check).intersection(output)) == len(files_to_check)

        for f in files_to_check:
            assert os.path.exists(f)