tests/test_proxy.py

   1 #!/usr/bin/python
   2
   3 import unittest
   4 from mic.utils import proxy
   5
   6 def suite():
   7     return unittest.makeSuite(ProxyTest)
   8
   9 class ProxyTest(unittest.TestCase):
  10
  11     def test_proxy(self):
  12         proxy.set_proxies('http://proxy.some.com:11', '1.2.3.4')
  13         self.assertEqual(proxy.get_proxy_for('http://1.2.3.4'), None)
  14         self.assertEqual(proxy.get_proxy_for('http://download.tizen.org'), 'http://proxy.some.com:11')
  15
  16         proxy.set_proxies('http://proxy.some.com:11', 'download.am.org')
  17         self.assertEqual(proxy.get_proxy_for('http://download.am.org'), None)
  18         self.assertEqual(proxy.get_proxy_for('https://download.am.org'), None)
  19         self.assertEqual(proxy.get_proxy_for('http://download.tizen.org'), 'http://proxy.some.com:11')
  20
  21         proxy.set_proxies('http://proxy.some.com:11', '1.2.3.0/24')
  22         self.assertEqual(proxy.get_proxy_for('http://1.2.3.4'), None)
  23         self.assertEqual(proxy.get_proxy_for('http://1.2.3.0'), None)
  24         self.assertEqual(proxy.get_proxy_for('http://1.2.3.255'), None)
  25         self.assertEqual(proxy.get_proxy_for('http://download.tizen.org'), 'http://proxy.some.com:11')
  26
  27         proxy.set_proxies('http://proxy.some.com:11', '.hello.com')
  28         self.assertEqual(proxy.get_proxy_for('http://linux.hello.com'), None)
  29         self.assertEqual(proxy.get_proxy_for('http://linux.hello.com.org'), 'http://proxy.some.com:11')
  30
  31
  32
  33 if __name__ == "__main__":
  34     unittest.main()