示例#1
0
    def test_clear_all_with_error(self):
        """Testing the clear source command successfully.

        With stubbed data list of sources with delete error
        """
        source_out = StringIO()
        get_url = get_server_location() + SOURCE_URI
        delete_url = get_server_location() + SOURCE_URI + '1/'
        source_entry = {
            'id': 1,
            'name': 'source1',
            'hosts': ['1.2.3.4'],
            'credential': ['credential1', 'cred2'],
            'port': 22
        }
        results = [source_entry]
        data = {'count': 1, 'results': results}
        err_data = {'error': ['Server Error']}
        with requests_mock.Mocker() as mocker:
            mocker.get(get_url, status_code=200, json=data)
            mocker.delete(delete_url, status_code=500, json=err_data)
            ncc = SourceClearCommand(SUBPARSER)
            args = Namespace(name=None)
            with self.assertRaises(SystemExit):
                with redirect_stdout(source_out):
                    ncc.main(args)
                    expected = 'Some sources were removed, however and' \
                               ' error occurred removing the following' \
                               ' credentials:'
                    self.assertTrue(expected in source_out.getvalue())
示例#2
0
    def test_clear_by_name_err(self):
        """Test the clear source command successfully.

        With stubbed data when specifying a name with an error response
        """
        source_out = StringIO()
        get_url = get_server_location() + SOURCE_URI + '?name=source1'
        delete_url = get_server_location() + SOURCE_URI + '1/'
        source_entry = {
            'id': 1,
            'name': 'source1',
            'hosts': ['1.2.3.4'],
            'credential': ['credential1', 'cred2'],
            'port': 22
        }
        results = [source_entry]
        data = {'count': 1, 'results': results}
        err_data = {'error': ['Server Error']}
        with requests_mock.Mocker() as mocker:
            mocker.get(get_url, status_code=200, json=data)
            mocker.delete(delete_url, status_code=500, json=err_data)
            ncc = SourceClearCommand(SUBPARSER)
            args = Namespace(name='source1')
            with self.assertRaises(SystemExit):
                with redirect_stdout(source_out):
                    ncc.main(args)
                    expected = 'Failed to remove source "source1"'
                    self.assertTrue(expected in source_out.getvalue())
示例#3
0
    def test_clear_by_name(self):
        """Testing the clear source command.

        Successfully with stubbed data when specifying a name
        """
        source_out = StringIO()
        get_url = get_server_location() + SOURCE_URI + '?name=source1'
        delete_url = get_server_location() + SOURCE_URI + '1/'
        source_entry = {
            'id': 1,
            'name': 'source1',
            'hosts': ['1.2.3.4'],
            'credential': ['credential1', 'cred2'],
            'port': 22
        }
        results = [source_entry]
        data = {'count': 1, 'results': results}
        with requests_mock.Mocker() as mocker:
            mocker.get(get_url, status_code=200, json=data)
            mocker.delete(delete_url, status_code=204)
            ncc = SourceClearCommand(SUBPARSER)
            args = Namespace(name='source1')
            with redirect_stdout(source_out):
                ncc.main(args)
                expected = messages.SOURCE_REMOVED % 'source1' + '\n'
                self.assertEqual(source_out.getvalue(), expected)
示例#4
0
 def test_clear_source_internal_err(self):
     """Testing the clear source command with an internal error."""
     source_out = StringIO()
     url = get_server_location() + SOURCE_URI + '?name=source1'
     with requests_mock.Mocker() as mocker:
         mocker.get(url, status_code=500, json={'error': ['Server Error']})
         ncc = SourceClearCommand(SUBPARSER)
         args = Namespace(name='source1')
         with self.assertRaises(SystemExit):
             with redirect_stdout(source_out):
                 ncc.main(args)
                 self.assertEqual(source_out.getvalue(), 'Server Error')
示例#5
0
 def test_clear_source_ssl_err(self):
     """Testing the clear 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)
         ncc = SourceClearCommand(SUBPARSER)
         args = Namespace(name='source1')
         with self.assertRaises(SystemExit):
             with redirect_stdout(source_out):
                 ncc.main(args)
                 self.assertEqual(source_out.getvalue(), SSL_ERROR_MSG)
示例#6
0
 def test_clear_source_empty(self):
     """Testing the clear source command successfully with empty data."""
     source_out = StringIO()
     url = get_server_location() + SOURCE_URI + '?name=source1'
     with requests_mock.Mocker() as mocker:
         mocker.get(url, status_code=200, json={'count': 0})
         ncc = SourceClearCommand(SUBPARSER)
         args = Namespace(name='source1')
         with self.assertRaises(SystemExit):
             with redirect_stdout(source_out):
                 ncc.main(args)
                 self.assertEqual(source_out.getvalue(),
                                  'Source "source1" was not found\n')
示例#7
0
    def test_clear_all_empty(self):
        """Test the clear source command successfully.

        With stubbed data empty list of sources
        """
        source_out = StringIO()
        get_url = get_server_location() + SOURCE_URI
        with requests_mock.Mocker() as mocker:
            mocker.get(get_url, status_code=200, json={'count': 0})
            ncc = SourceClearCommand(SUBPARSER)
            args = Namespace(name=None)
            with self.assertRaises(SystemExit):
                with redirect_stdout(source_out):
                    ncc.main(args)
                    expected = 'No sources exist to be removed\n'
                    self.assertEqual(source_out.getvalue(), expected)
示例#8
0
 def test_clear_all(self):
     """Testing the clear source command successfully with stubbed data."""
     source_out = StringIO()
     get_url = BASE_URL + SOURCE_URI
     delete_url = BASE_URL + SOURCE_URI + '1/'
     source_entry = {
         'id': 1,
         'name': 'source1',
         'hosts': ['1.2.3.4'],
         'credential': ['credential1', 'cred2'],
         'port': 22
     }
     data = [source_entry]
     with requests_mock.Mocker() as mocker:
         mocker.get(get_url, status_code=200, json=data)
         mocker.delete(delete_url, status_code=204)
         ncc = SourceClearCommand(SUBPARSER)
         args = Namespace(name=None)
         with redirect_stdout(source_out):
             ncc.main(args)
             expected = 'All sources were removed\n'
             self.assertEqual(source_out.getvalue(), expected)