Python RHUICataloger.nectar_configの例

コード例 #1

ファイル: test_rhui.py プロジェクト: asmacdo/pulp_rpm

    def test_nectar_config(self, fake_urlopen, fake_super):
        config = Mock()
        fake_fp = Mock()
        fake_fp.read.side_effect = [ID, SIGNATURE]
        fake_urlopen.return_value = fake_fp
        fake_super().nectar_config.return_value = DownloaderConfig()

        cataloger = RHUICataloger()
        nectar_config = cataloger.nectar_config(config)

        fake_super().nectar_config.assert_called_with(config)
        fake_urlopen.assert_any_with(ID_DOC_URL)
        fake_urlopen.assert_any_with(ID_SIG_URL)

        self.assertEqual(fake_urlopen.call_count, 2)
        self.assertEqual(fake_fp.read.call_count, fake_urlopen.call_count)
        self.assertEqual(fake_fp.close.call_count, fake_urlopen.call_count)

        self.assertEqual(
            nectar_config.headers,
            {ID_DOC_HEADER: urlsafe_b64encode(ID), ID_SIG_HEADER: urlsafe_b64encode(SIGNATURE)})

コード例 #2

    def test_nectar_config(self, fake_urlopen, fake_super):
        config = Mock()
        fake_fp = Mock()
        fake_fp.read.side_effect = [ID, SIGNATURE]
        fake_urlopen.return_value = fake_fp
        fake_super().nectar_config.return_value = DownloaderConfig()

        cataloger = RHUICataloger()
        nectar_config = cataloger.nectar_config(config)

        fake_super().nectar_config.assert_called_with(config)
        fake_urlopen.assert_any_with(ID_DOC_URL)
        fake_urlopen.assert_any_with(ID_SIG_URL)

        self.assertEqual(fake_urlopen.call_count, 2)
        self.assertEqual(fake_fp.read.call_count, fake_urlopen.call_count)
        self.assertEqual(fake_fp.close.call_count, fake_urlopen.call_count)

        self.assertEqual(
            nectar_config.headers, {
                ID_DOC_HEADER: urlsafe_b64encode(ID),
                ID_SIG_HEADER: urlsafe_b64encode(SIGNATURE)
            })