summaryrefslogtreecommitdiffstats
path: root/testing/mozbase/mozprofile/tests/server_locations.py
diff options
context:
space:
mode:
authorMatt A. Tobin <mattatobin@localhost.localdomain>2018-02-02 04:16:08 -0500
committerMatt A. Tobin <mattatobin@localhost.localdomain>2018-02-02 04:16:08 -0500
commit5f8de423f190bbb79a62f804151bc24824fa32d8 (patch)
tree10027f336435511475e392454359edea8e25895d /testing/mozbase/mozprofile/tests/server_locations.py
parent49ee0794b5d912db1f95dce6eb52d781dc210db5 (diff)
downloadUXP-5f8de423f190bbb79a62f804151bc24824fa32d8.tar
UXP-5f8de423f190bbb79a62f804151bc24824fa32d8.tar.gz
UXP-5f8de423f190bbb79a62f804151bc24824fa32d8.tar.lz
UXP-5f8de423f190bbb79a62f804151bc24824fa32d8.tar.xz
UXP-5f8de423f190bbb79a62f804151bc24824fa32d8.zip
Add m-esr52 at 52.6.0
Diffstat (limited to 'testing/mozbase/mozprofile/tests/server_locations.py')
-rw-r--r--testing/mozbase/mozprofile/tests/server_locations.py151
1 files changed, 151 insertions, 0 deletions
diff --git a/testing/mozbase/mozprofile/tests/server_locations.py b/testing/mozbase/mozprofile/tests/server_locations.py
new file mode 100644
index 000000000..5aa5c0f5e
--- /dev/null
+++ b/testing/mozbase/mozprofile/tests/server_locations.py
@@ -0,0 +1,151 @@
+#!/usr/bin/env python
+
+# This Source Code Form is subject to the terms of the Mozilla Public
+# License, v. 2.0. If a copy of the MPL was not distributed with this file,
+# You can obtain one at http://mozilla.org/MPL/2.0/.
+
+import mozfile
+import unittest
+from mozprofile.permissions import ServerLocations, \
+ MissingPrimaryLocationError, MultiplePrimaryLocationsError, \
+ DuplicateLocationError, BadPortLocationError, LocationsSyntaxError
+
+
+class ServerLocationsTest(unittest.TestCase):
+ """test server locations"""
+
+ locations = """# This is the primary location from which tests run.
+#
+http://mochi.test:8888 primary,privileged
+
+# a few test locations
+http://127.0.0.1:80 privileged
+http://127.0.0.1:8888 privileged
+https://test:80 privileged
+http://example.org:80 privileged
+http://test1.example.org privileged
+
+ """
+
+ locations_no_primary = """http://secondary.test:80 privileged
+http://tertiary.test:8888 privileged
+"""
+
+ locations_bad_port = """http://mochi.test:8888 primary,privileged
+http://127.0.0.1:80 privileged
+http://127.0.0.1:8888 privileged
+http://test:badport privileged
+http://example.org:80 privileged
+"""
+
+ def compare_location(self, location, scheme, host, port, options):
+ self.assertEqual(location.scheme, scheme)
+ self.assertEqual(location.host, host)
+ self.assertEqual(location.port, port)
+ self.assertEqual(location.options, options)
+
+ def create_temp_file(self, contents):
+ f = mozfile.NamedTemporaryFile()
+ f.write(contents)
+ f.flush()
+ return f
+
+ def test_server_locations(self):
+ # write a permissions file
+ f = self.create_temp_file(self.locations)
+
+ # read the locations
+ locations = ServerLocations(f.name)
+
+ # ensure that they're what we expect
+ self.assertEqual(len(locations), 6)
+ i = iter(locations)
+ self.compare_location(i.next(), 'http', 'mochi.test', '8888',
+ ['primary', 'privileged'])
+ self.compare_location(i.next(), 'http', '127.0.0.1', '80',
+ ['privileged'])
+ self.compare_location(i.next(), 'http', '127.0.0.1', '8888',
+ ['privileged'])
+ self.compare_location(i.next(), 'https', 'test', '80', ['privileged'])
+ self.compare_location(i.next(), 'http', 'example.org', '80',
+ ['privileged'])
+ self.compare_location(i.next(), 'http', 'test1.example.org', '8888',
+ ['privileged'])
+
+ locations.add_host('mozilla.org')
+ self.assertEqual(len(locations), 7)
+ self.compare_location(i.next(), 'http', 'mozilla.org', '80',
+ ['privileged'])
+
+ # test some errors
+ self.assertRaises(MultiplePrimaryLocationsError, locations.add_host,
+ 'primary.test', options='primary')
+
+ # We no longer throw these DuplicateLocation Error
+ try:
+ locations.add_host('127.0.0.1')
+ except DuplicateLocationError:
+ self.assertTrue(False, "Should no longer throw DuplicateLocationError")
+
+ self.assertRaises(BadPortLocationError, locations.add_host, '127.0.0.1',
+ port='abc')
+
+ # test some errors in locations file
+ f = self.create_temp_file(self.locations_no_primary)
+
+ exc = None
+ try:
+ ServerLocations(f.name)
+ except LocationsSyntaxError as e:
+ exc = e
+ self.assertNotEqual(exc, None)
+ self.assertEqual(exc.err.__class__, MissingPrimaryLocationError)
+ self.assertEqual(exc.lineno, 3)
+
+ # test bad port in a locations file to ensure lineno calculated
+ # properly.
+ f = self.create_temp_file(self.locations_bad_port)
+
+ exc = None
+ try:
+ ServerLocations(f.name)
+ except LocationsSyntaxError as e:
+ exc = e
+ self.assertNotEqual(exc, None)
+ self.assertEqual(exc.err.__class__, BadPortLocationError)
+ self.assertEqual(exc.lineno, 4)
+
+ def test_server_locations_callback(self):
+ class CallbackTest(object):
+ last_locations = None
+
+ def callback(self, locations):
+ self.last_locations = locations
+
+ c = CallbackTest()
+ f = self.create_temp_file(self.locations)
+ locations = ServerLocations(f.name, c.callback)
+
+ # callback should be for all locations in file
+ self.assertEqual(len(c.last_locations), 6)
+
+ # validate arbitrary one
+ self.compare_location(c.last_locations[2], 'http', '127.0.0.1', '8888',
+ ['privileged'])
+
+ locations.add_host('a.b.c')
+
+ # callback should be just for one location
+ self.assertEqual(len(c.last_locations), 1)
+ self.compare_location(c.last_locations[0], 'http', 'a.b.c', '80',
+ ['privileged'])
+
+ # read a second file, which should generate a callback with both
+ # locations.
+ f = self.create_temp_file(self.locations_no_primary)
+ locations.read(f.name)
+ self.assertEqual(len(c.last_locations), 2)
+
+
+if __name__ == '__main__':
+ unittest.main()