1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
|
#!/usr/bin/python3
#
# This file is part of nss-tls.
#
# Copyright (C) 2018, 2019 Dima Krasner
#
# This library is free software; you can redistribute it and/or
# modify it under the terms of the GNU Lesser General Public
# License as published by the Free Software Foundation; either
# version 2.1 of the License, or (at your option) any later version.
#
# This library is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
# Lesser General Public License for more details.
#
# You should have received a copy of the GNU Lesser General Public
# License along with this library; if not, write to the Free Software
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
from selenium import webdriver
from selenium.webdriver.firefox.firefox_binary import FirefoxBinary
from selenium.webdriver.firefox.options import Options
import os
import unittest
import sys
opts = Options()
if os.getenv("CI"):
opts.headless = True
# use the native resolver (https://wiki.mozilla.org/Trusted_Recursive_Resolver)
prof = webdriver.FirefoxProfile()
prof.set_preference("network.trr.mode", 5)
driver = webdriver.Firefox(firefox_binary=FirefoxBinary(sys.argv[1]),
options=opts,
firefox_profile=prof)
class FirefoxTest(unittest.TestCase):
def test_sanity(self):
for d in sys.argv[2:]:
url = "https://" + d
driver.get(url)
assert d in driver.current_url
if __name__ == "__main__":
suite = unittest.TestLoader().loadTestsFromTestCase(FirefoxTest)
result = unittest.TextTestRunner(verbosity=2).run(suite)
driver.quit()
sys.exit(not result.wasSuccessful())
|
