How to use _get_column_names_for_sqlite3 method in autotest

Best Python code snippet using autotest_python

rpc_interface_unittest_fixme.py

Source:rpc_interface_unittest_fixme.py

...101        return false_result102    # sqlite takes any columns that don't have aliases and names them103    # "table_name"."column_name".  we map these to just column_name.104    _SQLITE_AUTO_COLUMN_ALIAS_RE = re.compile(r'".+"\."(.+)"')105    def _get_column_names_for_sqlite3(self, cursor):106        names = [column_info[0] for column_info in cursor.description]107        # replace all "table_name"."column_name" constructs with just108        # column_name109        for i, name in enumerate(names):110            match = self._SQLITE_AUTO_COLUMN_ALIAS_RE.match(name)111            if match:112                names[i] = match.group(1)113        return names114    def _create_initial_data(self):115        machine = models.Machine.objects.create(hostname='myhost')116        # create basic objects117        kernel_name = 'mykernel1'118        kernel1 = models.Kernel.objects.create(kernel_hash=kernel_name,119                                               base=kernel_name,...

rpc_interface_unittest.py

Source:rpc_interface_unittest.py

...188    def test_get_num_test_views(self):189        self.assertEquals(rpc_interface.get_num_test_views(), 3)190        self.assertEquals(rpc_interface.get_num_test_views(191            job_name='myjob1', test_name='mytest1'), 1)192    def _get_column_names_for_sqlite3(self, cursor):193        names = [column_info[0] for column_info in cursor.description]194        # replace all "table_name"."column_name" constructs with just195        # column_name196        for i, name in enumerate(names):197            if '.' in name:198                field_name = name.split('.', 1)[1]199                names[i] = field_name.strip('"')200        return names201    def test_get_group_counts(self):202        self._god.stub_with(models.TempManager, '_get_column_names',203                            self._get_column_names_for_sqlite3)204        self.assertEquals(rpc_interface.get_num_groups(['job_name']), 2)205        counts = rpc_interface.get_group_counts(['job_name'])206        groups = counts['groups']...

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.