def test_edit_ssl_protocol(self):
     """Testing that you can edit the ssl_protocol arg successfully."""
     source_out = StringIO()
     url_get_cred = get_server_location() + CREDENTIAL_URI + \
         '?name=credential1'
     url_get_source = get_server_location() + SOURCE_URI + '?name=source1'
     url_patch = get_server_location() + SOURCE_URI + '1/'
     cred_results = [{'id': 1, 'name': 'credential1', 'username': '******',
                      'password': '******'}]
     cred_data = {'count': 1, 'results': cred_results}
     results = [{'id': 1, 'name': 'source1', 'hosts': ['1.2.3.4'],
                 'credentials': [{'id': 2, 'name': 'cred2'}],
                 'ssl_protocol': 'SSLv23'}]
     source_data = {'count': 1, 'results': results}
     with requests_mock.Mocker() as mocker:
         mocker.get(url_get_source, status_code=200, json=source_data)
         mocker.get(url_get_cred, status_code=200, json=cred_data)
         mocker.patch(url_patch, status_code=200)
         aec = SourceEditCommand(SUBPARSER)
         args = Namespace(name='source1', hosts=['1.2.3.5'],
                          cred=['credential1'], ssl_protocol='SSLv23')
         with redirect_stdout(source_out):
             aec.main(args)
             self.assertEqual(source_out.getvalue(),
                              messages.SOURCE_UPDATED % 'source1' + '\n')
Exemple #2
0
 def test_edit_source_cred_err(self):
     """Testing the edit source command where cred request hits error."""
     source_out = StringIO()
     url_get_cred = BASE_URL + CREDENTIAL_URI + '?name=credential1%2Ccred2'
     url_get_source = BASE_URL + SOURCE_URI + '?name=source1'
     source_data = [{
         'id': 1,
         'name': 'source1',
         'hosts': ['1.2.3.4'],
         'credentials': [{
             'id': 2,
             'name': 'cred2'
         }]
     }]
     with requests_mock.Mocker() as mocker:
         mocker.get(url_get_source, status_code=200, json=source_data)
         mocker.get(url_get_cred, status_code=500)
         aec = SourceEditCommand(SUBPARSER)
         args = Namespace(name='source1',
                          hosts=['1.2.3.4'],
                          cred=['credential1', 'cred2'],
                          port=22)
         with self.assertRaises(SystemExit):
             with redirect_stdout(source_out):
                 aec.main(args)
                 self.assertTrue(
                     'An error occurred while processing '
                     'the "--cred" input' in source_out.getvalue())
Exemple #3
0
 def test_edit_vc_source(self):
     """Testing the edit vcenter source command successfully."""
     source_out = StringIO()
     url_get_cred = BASE_URL + CREDENTIAL_URI + '?name=credential1'
     url_get_source = BASE_URL + SOURCE_URI + '?name=source1'
     url_patch = BASE_URL + SOURCE_URI + '1/'
     cred_data = [{
         'id': 1,
         'name': 'credential1',
         'username': '******',
         'password': '******'
     }]
     source_data = [{
         'id': 1,
         'name': 'source1',
         'hosts': ['1.2.3.4'],
         'credentials': [{
             'id': 2,
             'name': 'cred2'
         }]
     }]
     with requests_mock.Mocker() as mocker:
         mocker.get(url_get_source, status_code=200, json=source_data)
         mocker.get(url_get_cred, status_code=200, json=cred_data)
         mocker.patch(url_patch, status_code=200)
         aec = SourceEditCommand(SUBPARSER)
         args = Namespace(name='source1',
                          hosts=['1.2.3.5'],
                          cred=['credential1'])
         with redirect_stdout(source_out):
             aec.main(args)
             self.assertEqual(source_out.getvalue(),
                              'Source "source1" was updated\n')
 def test_edit_source_exclude_host(self):
     """Testing edit network source command by adding an excluded host."""
     source_out = StringIO()
     url_get_cred = get_server_location() + CREDENTIAL_URI + \
         '?name=credential1'
     url_get_source = get_server_location() + SOURCE_URI + '?name=source1'
     url_patch = get_server_location() + SOURCE_URI + '1/'
     cred_results = [{'id': 1, 'name': 'credential1', 'username': '******',
                      'password': '******'}]
     cred_data = {'count': 1, 'results': cred_results}
     results = [{'id': 1, 'name': 'source1', 'hosts': ['1.2.3.4'],
                 'credentials':[{'id': 2, 'name': 'cred2'}]}]
     source_data = {'count': 1, 'results': results}
     with requests_mock.Mocker() as mocker:
         mocker.get(url_get_source, status_code=200, json=source_data)
         mocker.get(url_get_cred, status_code=200, json=cred_data)
         mocker.patch(url_patch, status_code=200)
         aec = SourceEditCommand(SUBPARSER)
         args = Namespace(name='source1', hosts=['1.2.3.[0:255]'],
                          exclude_hosts=['1.2.3.4'], cred=['credential1'],
                          port=22)
         with redirect_stdout(source_out):
             aec.main(args)
             self.assertEqual(source_out.getvalue(),
                              messages.SOURCE_UPDATED % 'source1' + '\n')
 def test_edit_source_ssl_err(self):
     """Testing the edit source command with a connection error."""
     source_out = StringIO()
     url = get_server_location() + SOURCE_URI + '?name=source1'
     with requests_mock.Mocker() as mocker:
         mocker.get(url, exc=requests.exceptions.SSLError)
         aec = SourceEditCommand(SUBPARSER)
         args = Namespace(name='source1', hosts=['1.2.3.4'],
                          cred=['credential1'], port=22)
         with self.assertRaises(SystemExit):
             with redirect_stdout(source_out):
                 aec.main(args)
                 self.assertEqual(source_out.getvalue(), SSL_ERROR_MSG)
 def test_edit_source_no_val(self):
     """Testing the edit source command with a server error."""
     source_out = StringIO()
     url_get_source = get_server_location() + SOURCE_URI + '?name=source1'
     with requests_mock.Mocker() as mocker:
         mocker.get(url_get_source, status_code=500, json=None)
         aec = SourceEditCommand(SUBPARSER)
         args = Namespace(name='source1', hosts=['1.2.3.4'],
                          cred=['credential1'], port=22)
         with self.assertRaises(SystemExit):
             with redirect_stdout(source_out):
                 aec.main(args)
                 self.assertEqual(source_out.getvalue(),
                                  messages.SERVER_INTERNAL_ERROR)
 def test_edit_source_none(self):
     """Testing the edit cred command for none existing cred."""
     source_out = StringIO()
     url = get_server_location() + SOURCE_URI + '?name=source_none'
     with requests_mock.Mocker() as mocker:
         mocker.get(url, status_code=200, json={'count': 0})
         aec = SourceEditCommand(SUBPARSER)
         args = Namespace(name='source_none', hosts=['1.2.3.4'],
                          cred=['credential1'], port=22)
         with self.assertRaises(SystemExit):
             with redirect_stdout(source_out):
                 aec.main(args)
                 aec.main(args)
                 self.assertTrue('Source "source_none" does not exist'
                                 in source_out.getvalue())
Exemple #8
0
 def test_edit_source_no_val(self):
     """Testing the edit source command with source doesn't exist."""
     source_out = StringIO()
     url_get_source = BASE_URL + SOURCE_URI + '?name=source1'
     with requests_mock.Mocker() as mocker:
         mocker.get(url_get_source, status_code=500, json=[])
         aec = SourceEditCommand(SUBPARSER)
         args = Namespace(name='source1',
                          hosts=['1.2.3.4'],
                          cred=['credential1'],
                          port=22)
         with self.assertRaises(SystemExit):
             with redirect_stdout(source_out):
                 aec.main(args)
                 self.assertEqual(source_out.getvalue(),
                                  'Source "source1" does not exist\n')
Exemple #9
0
 def test_edit_source_cred_nf(self):
     """Testing the edit source command where cred is not found."""
     source_out = StringIO()
     url_get_cred = get_server_location() + CREDENTIAL_URI + \
         '?name=credential1%2Ccred2'
     url_get_source = get_server_location() + SOURCE_URI + '?name=source1'
     cred_results = [{
         'id': 1,
         'name': 'credential1',
         'username': '******',
         'password': '******'
     }]
     cred_data = {'count': 1, 'results': cred_results}
     results = [{
         'id': 1,
         'name': 'source1',
         'hosts': ['1.2.3.4'],
         'credentials': [{
             'id': 2,
             'name': 'cred2'
         }]
     }]
     source_data = {'count': 1, 'results': results}
     with requests_mock.Mocker() as mocker:
         mocker.get(url_get_source, status_code=200, json=source_data)
         mocker.get(url_get_cred, status_code=200, json=cred_data)
         aec = SourceEditCommand(SUBPARSER)
         args = Namespace(name='source1',
                          hosts=['1.2.3.4'],
                          cred=['credential1', 'cred2'],
                          port=22)
         with self.assertRaises(SystemExit):
             with redirect_stdout(source_out):
                 aec.main(args)
                 self.assertTrue(
                     'An error occurred while processing '
                     'the "--cred" input' in source_out.getvalue())