#!/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())