github.com/qsunny/k8s@v0.0.0-20220101153623-e6dca256d5bf/examples-master/staging/selenium/selenium-test.py (about)

     1  #!/usr/bin/env python
     2  
     3  # Copyright 2015 The Kubernetes Authors.
     4  #
     5  # Licensed under the Apache License, Version 2.0 (the "License");
     6  # you may not use this file except in compliance with the License.
     7  # You may obtain a copy of the License at
     8  #
     9  #     http://www.apache.org/licenses/LICENSE-2.0
    10  #
    11  # Unless required by applicable law or agreed to in writing, software
    12  # distributed under the License is distributed on an "AS IS" BASIS,
    13  # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
    14  # See the License for the specific language governing permissions and
    15  # limitations under the License.
    16  
    17  from selenium import webdriver
    18  from selenium.webdriver.common.desired_capabilities import DesiredCapabilities
    19  
    20  def check_browser(browser):
    21    driver = webdriver.Remote(
    22      command_executor='http://selenium-hub:4444/wd/hub',
    23      desired_capabilities=getattr(DesiredCapabilities, browser)
    24    )
    25    driver.get("http://google.com")
    26    assert "google" in driver.page_source
    27    driver.quit()
    28    print("Browser %s checks out!" % browser)
    29  
    30  
    31  check_browser("FIREFOX")
    32  check_browser("CHROME")
    33