How to use _add_import_for_test_uuid method in tempest

Best Python code snippet using tempest_python

check_uuid.py

Source:check_uuid.py

...152        if type(node) == ast.Import:153            return node.names[0].name154        elif type(node) == ast.ImportFrom:155            return '%s.%s' % (node.module, node.names[0].name)156    def _add_import_for_test_uuid(self, patcher, src_parsed, source_path):157        with open(source_path) as f:158            src_lines = f.read().split('\n')159        line_no = 0160        tempest_imports = [node for node in src_parsed.body161                           if self._import_name(node) and162                           'tempest.' in self._import_name(node)]163        if not tempest_imports:164            import_snippet = '\n'.join(('', IMPORT_LINE, ''))165        else:166            for node in tempest_imports:167                if self._import_name(node) < DECORATOR_IMPORT:168                    continue169                else:170                    line_no = node.lineno171                    import_snippet = IMPORT_LINE172                    break173            else:174                line_no = tempest_imports[-1].lineno175                while True:176                    if (not src_lines[line_no - 1] or177                            getattr(self._next_node(src_parsed.body,178                                                    tempest_imports[-1]),179                                    'lineno') == line_no or180                            line_no == len(src_lines)):181                        break182                    line_no += 1183                import_snippet = '\n'.join((IMPORT_LINE, ''))184        patcher.add_patch(source_path, import_snippet, line_no)185    def get_tests(self):186        """Get test methods with sources from base package with metadata"""187        tests = {}188        for module_name in self._modules_search():189            tests[module_name] = {}190            module = importlib.import_module(module_name)191            source_path = '.'.join(192                (os.path.splitext(module.__file__)[0], 'py')193            )194            with open(source_path, 'r') as f:195                source = f.read()196            tests[module_name]['source_path'] = source_path197            tests[module_name]['tests'] = {}198            source_parsed = ast.parse(source)199            tests[module_name]['ast'] = source_parsed200            tests[module_name]['import_valid'] = (201                hasattr(module, DECORATOR_MODULE) and202                inspect.ismodule(getattr(module, DECORATOR_MODULE))203            )204            test_cases = (node for node in source_parsed.body205                          if self._is_test_case(module, node))206            for node in test_cases:207                for subnode in filter(self._is_test_method, node.body):208                        test_name = '%s.%s' % (node.name, subnode.name)209                        tests[module_name]['tests'][test_name] = subnode210        return tests211    @staticmethod212    def _filter_tests(function, tests):213        """Filter tests with condition 'function(test_node) == True'"""214        result = {}215        for module_name in tests:216            for test_name in tests[module_name]['tests']:217                if function(module_name, test_name, tests):218                    if module_name not in result:219                        result[module_name] = {220                            'ast': tests[module_name]['ast'],221                            'source_path': tests[module_name]['source_path'],222                            'import_valid': tests[module_name]['import_valid'],223                            'tests': {}224                        }225                    result[module_name]['tests'][test_name] = \226                        tests[module_name]['tests'][test_name]227        return result228    def find_untagged(self, tests):229        """Filter all tests without uuid in metadata"""230        def check_uuid_in_meta(module_name, test_name, tests):231            idempotent_id = self._get_idempotent_id(232                tests[module_name]['tests'][test_name])233            return not idempotent_id234        return self._filter_tests(check_uuid_in_meta, tests)235    def report_collisions(self, tests):236        """Reports collisions if there are any. Returns true if237        collisions exist.238        """239        uuids = {}240        def report(module_name, test_name, tests):241            test_uuid = self._get_idempotent_id(242                tests[module_name]['tests'][test_name])243            if not test_uuid:244                return245            if test_uuid in uuids:246                error_str = "%s:%s\n uuid %s collision: %s<->%s\n%s:%s" % (247                    tests[module_name]['source_path'],248                    tests[module_name]['tests'][test_name].lineno,249                    test_uuid,250                    test_name,251                    uuids[test_uuid]['test_name'],252                    uuids[test_uuid]['source_path'],253                    uuids[test_uuid]['test_node'].lineno,254                )255                print(error_str)256                print("cannot automatically resolve the collision, please "257                      "manually remove the duplicate value on the new test.")258                return True259            else:260                uuids[test_uuid] = {261                    'module': module_name,262                    'test_name': test_name,263                    'test_node': tests[module_name]['tests'][test_name],264                    'source_path': tests[module_name]['source_path']265                }266        return bool(self._filter_tests(report, tests))267    def report_untagged(self, tests):268        """Reports untagged tests if there are any. Returns true if269        untagged tests exist.270        """271        def report(module_name, test_name, tests):272            error_str = "%s:%s\nmissing @test.idempotent_id('...')\n%s\n" % (273                tests[module_name]['source_path'],274                tests[module_name]['tests'][test_name].lineno,275                test_name276            )277            print(error_str)278            return True279        return bool(self._filter_tests(report, tests))280    def fix_tests(self, tests):281        """Add uuids to all tests specified in tests and282        fix it in source files283        """284        patcher = SourcePatcher()285        for module_name in tests:286            add_import_once = True287            for test_name in tests[module_name]['tests']:288                if not tests[module_name]['import_valid'] and add_import_once:289                    self._add_import_for_test_uuid(290                        patcher,291                        tests[module_name]['ast'],292                        tests[module_name]['source_path']293                    )294                    add_import_once = False295                self._add_uuid_to_test(296                    patcher, tests[module_name]['tests'][test_name],297                    tests[module_name]['source_path'])298        patcher.apply_patches()299def run():300    parser = argparse.ArgumentParser()301    parser.add_argument('--package', action='store', dest='package',302                        default='tempest', type=str,303                        help='Package with tests')...

test_check_uuid.py

Source:test_check_uuid.py

...122        fake_file.close()123        class Fake_src_parsed():124            body = [TestTestChecker.get_mocked_ast_object(125                1, 4, 'unittest', 'mock', ast.ImportFrom)]126        checker._add_import_for_test_uuid(patcher, Fake_src_parsed,127                                          fake_file.name)128        patcher.apply_patches()129        with open(fake_file.name, "r") as f:130            expected_result = source_code + '\n' + TestTestChecker.IMPORT_LINE131            self.assertTrue(expected_result == f.read())132    def test_add_import_for_test_uuid_tempest(self):133        patcher = check_uuid.SourcePatcher()134        checker = check_uuid.TestChecker(importlib.import_module('tempest'))135        fake_file = tempfile.NamedTemporaryFile("w+t", delete=False)136        source_code = "from tempest import a_fake_module\n"137        fake_file.write(source_code)138        fake_file.close()139        class Fake_src_parsed:140            body = [TestTestChecker.get_mocked_ast_object(141                1, 4, 'tempest', 'a_fake_module', ast.ImportFrom)]142        checker._add_import_for_test_uuid(patcher, Fake_src_parsed,143                                          fake_file.name)144        patcher.apply_patches()145        with open(fake_file.name, "r") as f:146            expected_result = source_code + TestTestChecker.IMPORT_LINE147            self.assertTrue(expected_result == f.read())148    def test_add_import_no_import(self):149        patcher = check_uuid.SourcePatcher()150        patcher.add_patch = mock.Mock()151        checker = check_uuid.TestChecker(importlib.import_module('tempest'))152        fake_file = tempfile.NamedTemporaryFile("w+t", delete=False)153        fake_file.close()154        class Fake_src_parsed:155            body = []156        checker._add_import_for_test_uuid(patcher, Fake_src_parsed,157                                          fake_file.name)...

Automation Testing Tutorials

Learn to execute automation testing from scratch with LambdaTest Learning Hub. Right from setting up the prerequisites to run your first automation test, to following best practices and diving deeper into advanced test scenarios. LambdaTest Learning Hubs compile a list of step-by-step guides to help you be proficient with different test automation frameworks i.e. Selenium, Cypress, TestNG etc.