Ejemplo n.º 1
0
class LocalDownloaderTests(base_downloader.BaseDownloaderTests):

    def setUp(self):
        super(LocalDownloaderTests, self).setUp()
        self.config.repo_plugin_config[constants.CONFIG_FEED] = 'file://' + VALID_REPO_DIR
        self.downloader = LocalDownloader(self.repo, None, self.config, self.mock_cancelled_callback)

    def test_retrieve_metadata(self):
        # Test
        docs = self.downloader.retrieve_metadata(self.mock_progress_report)

        # Verify
        self.assertEqual(1, len(docs))
        metadata = model.RepositoryMetadata()
        metadata.update_from_json(docs[0])
        self.assertEqual(2, len(metadata.modules))

        self.assertEqual(1, self.mock_progress_report.metadata_query_total_count)
        self.assertEqual(1, self.mock_progress_report.metadata_query_finished_count)
        expected_query = os.path.join(VALID_REPO_DIR, constants.REPO_METADATA_FILENAME)
        self.assertEqual(expected_query, self.mock_progress_report.metadata_current_query)
        self.assertEqual(2, self.mock_progress_report.update_progress.call_count)

    def test_retrieve_metadata_no_metadata_found(self):
        # Setup
        self.config.repo_plugin_config[constants.CONFIG_FEED] = 'file://' + INVALID_REPO_DIR

        # Test
        try:
            self.downloader.retrieve_metadata(self.mock_progress_report)
            self.fail()
        except FileNotFoundException, e:
            expected = os.path.join(INVALID_REPO_DIR, constants.REPO_METADATA_FILENAME)
            self.assertEqual(expected, e.location)
Ejemplo n.º 2
0
class LocalDownloaderTests(base_downloader.BaseDownloaderTests):

    def setUp(self):
        super(LocalDownloaderTests, self).setUp()
        self.config.repo_plugin_config[constants.CONFIG_FEED] = 'file://' + VALID_REPO_DIR
        self.downloader = LocalDownloader(self.repo, None, self.config)

    @mock.patch('nectar.config.DownloaderConfig.finalize')
    def test_retrieve_metadata(self, mock_finalize):
        # Test
        docs = self.downloader.retrieve_metadata(self.mock_progress_report)

        # Verify
        self.assertEqual(1, len(docs))
        metadata = model.RepositoryMetadata()
        metadata.update_from_json(docs[0])
        self.assertEqual(2, len(metadata.modules))

        self.assertEqual(1, self.mock_progress_report.metadata_query_total_count)
        self.assertEqual(1, self.mock_progress_report.metadata_query_finished_count)
        expected_query = os.path.join(VALID_REPO_DIR, constants.REPO_METADATA_FILENAME)
        self.assertEqual(expected_query, self.mock_progress_report.metadata_current_query)
        self.assertEqual(2, self.mock_progress_report.update_progress.call_count)

        mock_finalize.assert_called_once()

    def test_retrieve_metadata_no_metadata_found(self):
        # Setup
        self.config.repo_plugin_config[constants.CONFIG_FEED] = 'file://' + INVALID_REPO_DIR

        # Test
        try:
            self.downloader.retrieve_metadata(self.mock_progress_report)
            self.fail()
        except FileRetrievalException:
            pass

    def test_retrieve_module(self):
        # Test
        mod_path = self.downloader.retrieve_module(self.mock_progress_report, self.module)

        # Verify
        expected = os.path.join(VALID_REPO_DIR, self.module.filename())
        self.assertEqual(expected, mod_path)

    def test_retrieve_module_no_file(self):
        # Setup
        self.module.author = 'foo'

        # Test
        try:
            self.downloader.retrieve_module(self.mock_progress_report, self.module)
            self.fail()
        except FileRetrievalException:
            pass

    def test_cleanup_module(self):
        # Test
        self.downloader.cleanup_module(self.module)
Ejemplo n.º 3
0
 def setUp(self):
     super(LocalDownloaderTests, self).setUp()
     self.config.repo_plugin_config[
         constants.CONFIG_FEED] = 'file://' + VALID_REPO_DIR
     self.downloader = LocalDownloader(self.repo, None, self.config)
Ejemplo n.º 4
0
class LocalDownloaderTests(base_downloader.BaseDownloaderTests):
    def setUp(self):
        super(LocalDownloaderTests, self).setUp()
        self.config.repo_plugin_config[
            constants.CONFIG_FEED] = 'file://' + VALID_REPO_DIR
        self.downloader = LocalDownloader(self.repo, None, self.config)

    @mock.patch('nectar.config.DownloaderConfig.finalize')
    def test_retrieve_metadata(self, mock_finalize):
        # Test
        docs = self.downloader.retrieve_metadata(self.mock_progress_report)

        # Verify
        self.assertEqual(1, len(docs))
        metadata = RepositoryMetadata()
        metadata.update_from_json(docs[0])
        self.assertEqual(2, len(metadata.modules))

        self.assertEqual(1,
                         self.mock_progress_report.metadata_query_total_count)
        self.assertEqual(
            1, self.mock_progress_report.metadata_query_finished_count)
        expected_query = os.path.join(VALID_REPO_DIR,
                                      constants.REPO_METADATA_FILENAME)
        self.assertEqual(expected_query,
                         self.mock_progress_report.metadata_current_query)
        self.assertEqual(2,
                         self.mock_progress_report.update_progress.call_count)

        mock_finalize.assert_called_once()

    def test_retrieve_metadata_no_metadata_found(self):
        # Setup
        self.config.repo_plugin_config[
            constants.CONFIG_FEED] = 'file://' + INVALID_REPO_DIR

        # Test
        try:
            self.downloader.retrieve_metadata(self.mock_progress_report)
            self.fail()
        except FileRetrievalException:
            pass

    @mock.patch(MODULE_PATH + '.os.path.exists')
    def test_retrieve_module(self, mock_exists):
        mock_exists.return_value = True

        # Test
        mod_path = self.downloader.retrieve_module(self.mock_progress_report,
                                                   self.module)

        # Verify
        expected = os.path.join(VALID_REPO_DIR, self.module.filename())
        self.assertEqual(expected, mod_path)

    @mock.patch(MODULE_PATH + '.os.path.exists')
    def test_retrieve_module_no_file(self, mock_exists):
        mock_exists.return_value = False

        # Setup
        self.module.author = 'foo'

        # Test
        try:
            self.downloader.retrieve_module(self.mock_progress_report,
                                            self.module)
            self.fail()
        except FileRetrievalException:
            pass

    def test_cleanup_module(self):
        # Test
        self.downloader.cleanup_module(self.module)
Ejemplo n.º 5
0
 def setUp(self):
     super(LocalDownloaderTests, self).setUp()
     self.config.repo_plugin_config[constants.CONFIG_FEED] = 'file://' + VALID_REPO_DIR
     self.downloader = LocalDownloader(self.repo, None, self.config)