How to use _create_add_remove method in autotest

Best Python code snippet using autotest_python

action_common_unittest.py

Source:action_common_unittest.py

...187#188# Adding or Removing users or hosts from a topic(ACL or label)189#190class atest_add_or_remove_unittest(cli_mock.cli_unittest):191    def _create_add_remove(self, items, users=None, hosts=None):192        def _items():193            return [items]194        addrm = action_common.atest_add_or_remove()195        addrm.afe = rpc.afe_comm()196        if users:197            addrm.users = users198        if hosts:199            addrm.hosts = hosts200        addrm.topic = 'acl_group'201        addrm.msg_topic = 'ACL'202        addrm.op_action = 'add'203        addrm.msg_done = 'Added to'204        addrm.get_items = _items205        return addrm206    def test__add_remove_uh_to_topic(self):207        acl_addrm = self._create_add_remove('acl0',208                                        users=['user0', 'user1'])209        self.mock_rpcs([('acl_group_add_users',210                         {'id': 'acl0',211                          'users': ['user0', 'user1']},212                         True,213                         None)])214        acl_addrm._add_remove_uh_to_topic('acl0', 'users')215        self.god.check_playback()216    def test__add_remove_uh_to_topic_raise(self):217        acl_addrm = self._create_add_remove('acl0',218                                        users=['user0', 'user1'])219        self.assertRaises(AttributeError,220                          acl_addrm._add_remove_uh_to_topic,221                          'acl0', 'hosts')222    def test_execute_add_or_remove_uh_to_topic_acl_users(self):223        acl_addrm = self._create_add_remove('acl0',224                                        users=['user0', 'user1'])225        self.mock_rpcs([('acl_group_add_users',226                         {'id': 'acl0',227                          'users': ['user0', 'user1']},228                         True,229                         None)])230        execute_result = acl_addrm.execute()231        self.god.check_playback()232        self.assertEqualNoOrder(['acl0'], execute_result['users'])233        self.assertEqual([], execute_result['hosts'])234    def test_execute_add_or_remove_uh_to_topic_acl_users_hosts(self):235        acl_addrm = self._create_add_remove('acl0',236                                            users=['user0', 'user1'],237                                            hosts=['host0', 'host1'])238        self.mock_rpcs([('acl_group_add_users',239                         {'id': 'acl0',240                          'users': ['user0', 'user1']},241                         True,242                         None),243                        ('acl_group_add_hosts',244                         {'id': 'acl0',245                          'hosts': ['host0', 'host1']},246                         True,247                         None)])248        execute_result = acl_addrm.execute()249        self.god.check_playback()250        self.assertEqualNoOrder(['acl0'], execute_result['users'])251        self.assertEqualNoOrder(['acl0'], execute_result['hosts'])252    def test_execute_add_or_remove_uh_to_topic_acl_bad_users(self):253        acl_addrm = self._create_add_remove('acl0',254                                            users=['user0', 'user1'])255        self.mock_rpcs([('acl_group_add_users',256                         {'id': 'acl0',257                          'users': ['user0', 'user1']},258                         False,259                         'DoesNotExist: The following users do not exist: '260                         'user0, user1')])261        execute_result = acl_addrm.execute()262        self.god.check_playback()263        self.assertEqual([], execute_result['users'])264        self.assertEqual([], execute_result['hosts'])265        self.assertOutput(acl_addrm, execute_result,266                          err_words_ok=['DoesNotExist',267                                        'acl_group_add_users',268                                        'user0', 'user1'],269                          err_words_no = ['acl_group_add_hosts'])270    def test_execute_add_or_remove_uh_to_topic_acl_bad_users_partial(self):271        acl_addrm = self._create_add_remove('acl0',272                                            users=['user0', 'user1'])273        self.mock_rpcs([('acl_group_add_users',274                         {'id': 'acl0',275                          'users': ['user0', 'user1']},276                         False,277                         'DoesNotExist: The following users do not exist: '278                         'user0'),279                        ('acl_group_add_users',280                         {'id': 'acl0',281                          'users': ['user1']},282                         True,283                         None)])284        execute_result = acl_addrm.execute()285        self.god.check_playback()286        self.assertEqual(['acl0'], execute_result['users'])287        self.assertEqual([], execute_result['hosts'])288        self.assertOutput(acl_addrm, execute_result,289                          out_words_ok=["Added to ACL 'acl0'", 'user1'],290                          err_words_ok=['DoesNotExist',291                                        'acl_group_add_users',292                                        'user0'],293                          err_words_no = ['acl_group_add_hosts'])294    def test_execute_add_or_remove_uh_to_topic_acl_bad_u_partial_kill(self):295        acl_addrm = self._create_add_remove('acl0',296                                            users=['user0', 'user1'])297        acl_addrm.kill_on_failure = True298        self.mock_rpcs([('acl_group_add_users',299                         {'id': 'acl0',300                          'users': ['user0', 'user1']},301                         False,302                         'DoesNotExist: The following users do not exist: '303                         'user0')])304        sys.exit.expect_call(1).and_raises(cli_mock.ExitException)305        self.god.mock_io()306        self.assertRaises(cli_mock.ExitException, acl_addrm.execute)307        (out, err) = self.god.unmock_io()308        self.god.check_playback()309        self._check_output(out=out, err=err,310                          err_words_ok=['DoesNotExist',311                                        'acl_group_add_users',312                                        'user0'],313                          err_words_no = ['acl_group_add_hosts'])314    def test_execute_add_or_remove_uh_to_topic_acl_bad_users_good_hosts(self):315        acl_addrm = self._create_add_remove('acl0',316                                        users=['user0', 'user1'],317                                        hosts=['host0', 'host1'])318        self.mock_rpcs([('acl_group_add_users',319                         {'id': 'acl0',320                          'users': ['user0', 'user1']},321                         False,322                         'DoesNotExist: The following users do not exist: '323                         'user0, user1'),324                        ('acl_group_add_hosts',325                         {'id': 'acl0',326                          'hosts': ['host0', 'host1']},327                         True,328                         None)])329        execute_result = acl_addrm.execute()330        self.god.check_playback()331        self.assertEqual([], execute_result['users'])332        self.assertEqual(['acl0'], execute_result['hosts'])333        self.assertOutput(acl_addrm, execute_result,334                          out_words_ok=["Added to ACL 'acl0' hosts:",335                                        "host0", "host1"],336                          err_words_ok=['DoesNotExist',337                                        'acl_group_add_users',338                                        'user0', 'user1'],339                          err_words_no = ['acl_group_add_hosts'])340    def test_execute_add_or_remove_uh_to_topic_acl_good_users_bad_hosts(self):341        acl_addrm = self._create_add_remove('acl0 with space',342                                        users=['user0', 'user1'],343                                        hosts=['host0', 'host1'])344        self.mock_rpcs([('acl_group_add_users',345                         {'id': 'acl0 with space',346                          'users': ['user0', 'user1']},347                         True,348                         None),349                        ('acl_group_add_hosts',350                         {'id': 'acl0 with space',351                          'hosts': ['host0', 'host1']},352                         False,353                         'DoesNotExist: The following hosts do not exist: '354                         'host0, host1')])355        execute_result = acl_addrm.execute()356        self.god.check_playback()357        self.assertEqual(['acl0 with space'], execute_result['users'])358        self.assertEqual([], execute_result['hosts'])359        self.assertOutput(acl_addrm, execute_result,360                          out_words_ok=["Added to ACL 'acl0 with space' users:",361                                        "user0", "user1"],362                          err_words_ok=['DoesNotExist',363                                        'acl_group_add_hosts',364                                        'host0', 'host1'],365                          err_words_no = ['acl_group_add_users'])366    def test_exe_add_or_remove_uh_to_topic_acl_good_u_bad_hosts_partial(self):367        acl_addrm = self._create_add_remove('acl0',368                                        users=['user0', 'user1'],369                                        hosts=['host0', 'host1'])370        self.mock_rpcs([('acl_group_add_users',371                         {'id': 'acl0',372                          'users': ['user0', 'user1']},373                         True,374                         None),375                        ('acl_group_add_hosts',376                         {'id': 'acl0',377                          'hosts': ['host0', 'host1']},378                         False,379                         'DoesNotExist: The following hosts do not exist: '380                         'host1'),381                        ('acl_group_add_hosts',382                         {'id': 'acl0',383                          'hosts': ['host0']},384                         True,385                         None)])386        execute_result = acl_addrm.execute()387        self.god.check_playback()388        self.assertEqual(['acl0'], execute_result['users'])389        self.assertEqual(['acl0'], execute_result['hosts'])390        self.assertOutput(acl_addrm, execute_result,391                          out_words_ok=["Added to ACL 'acl0' users:",392                                        "user0", "user1", "host0"],393                          err_words_ok=['DoesNotExist',394                                        'acl_group_add_hosts',395                                        'host1'],396                          err_words_no = ['acl_group_add_users'])397    def test_execute_add_or_remove_uh_to_topic_acl_bad_users_bad_hosts(self):398        acl_addrm = self._create_add_remove('acl0',399                                        users=['user0', 'user1'],400                                        hosts=['host0', 'host1'])401        self.mock_rpcs([('acl_group_add_users',402                         {'id': 'acl0',403                          'users': ['user0', 'user1']},404                         False,405                         'DoesNotExist: The following users do not exist: '406                         'user0, user1'),407                        ('acl_group_add_hosts',408                         {'id': 'acl0',409                          'hosts': ['host0', 'host1']},410                         False,411                         'DoesNotExist: The following hosts do not exist: '412                         'host0, host1')])413        execute_result = acl_addrm.execute()414        self.god.check_playback()415        self.assertEqual([], execute_result['users'])416        self.assertEqual([], execute_result['hosts'])417        self.assertOutput(acl_addrm, execute_result,418                          err_words_ok=['DoesNotExist',419                                        'acl_group_add_hosts',420                                        'host0', 'host1',421                                        'acl_group_add_users',422                                        'user0', 'user1'])423    def test_execute_add_or_remove_uh_to_topic_acl_bad_u_bad_h_partial(self):424        acl_addrm = self._create_add_remove('acl0',425                                        users=['user0', 'user1'],426                                        hosts=['host0', 'host1'])427        self.mock_rpcs([('acl_group_add_users',428                         {'id': 'acl0',429                          'users': ['user0', 'user1']},430                         False,431                         'DoesNotExist: The following users do not exist: '432                         'user0'),433                        ('acl_group_add_users',434                         {'id': 'acl0',435                          'users': ['user1']},436                         True,437                         None),438                        ('acl_group_add_hosts',439                         {'id': 'acl0',440                          'hosts': ['host0', 'host1']},441                         False,442                         'DoesNotExist: The following hosts do not exist: '443                         'host1'),444                        ('acl_group_add_hosts',445                         {'id': 'acl0',446                          'hosts': ['host0']},447                         True,448                         None)])449        execute_result = acl_addrm.execute()450        self.god.check_playback()451        self.assertEqual(['acl0'], execute_result['users'])452        self.assertEqual(['acl0'], execute_result['hosts'])453        self.assertOutput(acl_addrm, execute_result,454                          out_words_ok=["Added to ACL 'acl0' user:",455                                        "Added to ACL 'acl0' host:",456                                        'user1', 'host0'],457                          err_words_ok=['DoesNotExist',458                                        'acl_group_add_hosts',459                                        'host1',460                                        'acl_group_add_users',461                                        'user0'])462    def test_execute_add_or_remove_to_topic_bad_acl_uh(self):463        acl_addrm = self._create_add_remove('acl0',464                                        users=['user0', 'user1'],465                                        hosts=['host0', 'host1'])466        self.mock_rpcs([('acl_group_add_users',467                         {'id': 'acl0',468                          'users': ['user0', 'user1']},469                         False,470                         'DoesNotExist: acl_group matching '471                         'query does not exist.'),472                        ('acl_group_add_hosts',473                         {'id': 'acl0',474                          'hosts': ['host0', 'host1']},475                         False,476                         'DoesNotExist: acl_group matching '477                         'query does not exist.')])...

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.