Method test_check_updates_mock

tests/TestUpdate.py:44–81

Unit test for function check_updates() using mock

(self)

    """Test case for module Update"""

    def test_check_updates_mock(self):
        """Unit test for function check_updates() using mock"""
        wa = '<winapp2 url="http://katana.oooninja.com/bleachbit/winapp2.ini" sha512="ce9e18252f608c8aff28811e372124d29a86404f328d3cd51f1f220578744bb8b15f55549eabfe8f1a80657fc940f6d6deece28e0532b3b0901a4c74110f7ba7"/>'
        update_tests = [
            ('<updates><stable ver="0.8.4">http://084</stable><beta ver="0.8.5beta">http://085beta</beta>%s</updates>' % wa,
             (('0.8.4', 'http://084'), ('0.8.5beta', 'http://085beta'))),
            ('<updates><stable ver="0.8.4">http://084</stable>%s</updates>' % wa,
             (('0.8.4', 'http://084'), )),
            ('<updates><beta ver="0.8.5beta">http://085beta</beta>%s</updates>' % wa,
             (('0.8.5beta', 'http://085beta'), )),
            ('<updates></updates>', ())]

        with patch('bleachbit.Update.fetch_url') as mock_fetch:
            # Configure mock responses
            mock_response = Mock()
            mock_response.status_code = 200
            mock_response.content = b''

            for xml, expected in update_tests:
                # Set up mock response content
                mock_response.text = xml
                mock_response.content = xml.encode()
                mock_fetch.return_value = mock_response

                # Run test
                updates = check_updates(True, False, None, None)
                self.assertEqual(updates, expected)

                # check_updates() must send update-specific headers
                mock_fetch.assert_called()
                _, fetch_kwargs = mock_fetch.call_args
                sent_headers = fetch_kwargs['headers']
                self.assertIn('X-BleachBit-Version', sent_headers)
                self.assertIn('X-OS-Type', sent_headers)
                self.assertIn('X-OS-Version', sent_headers)
                self.assertIn('X-Locale', sent_headers)
                if 'X-GTK-Version' in sent_headers:
                    self.assertIsInstance(sent_headers['X-GTK-Version'], str)

    def test_check_updates_real_network(self):
        """Unit test for function check_updates() using real network"""