import unittest from proxy import app class FlaskTestCase(unittest.TestCase): def setUp(self): """Set up for tests: create a test client.""" self.app = app.test_client() self.app.testing = True def test_health_check(self): """Check health endpoint.""" response = self.app.get('/health') self.assertEqual(response.status_code, 200) self.assertIn(b'healthy', response.data) def test_proxy(self): """Test RSS proxying.""" # Sample URL for proxying url = 'http://example.com/rss' response = self.app.get(f'/proxy?url={url}') self.assertEqual(response.status_code, 200) self.assertIn(b'