Next-Gen AI-Powered Visual Testing: Catch the UI surprises before your customers do

import unittest 

import time 

from selenium import webdriver 

from sbvt.visualtest import VisualTest 

API_KEY = 'BITBAR_API_KEY' 

PROJECT_TOKEN = 'VISUAL_TEST_PROJECT_TOKEN' 

class TestDiscoverHomePage(unittest.TestCase): 

    @classmethod 

    def setUpClass(cls): 

        print('---- Starting Discover Home Page Test ----') 

        import warnings 

        # turn off tracemalloc socket warnings 
        warnings.simplefilter("ignore")

        caps = { 

            'platform': 'Linux', 

            'osVersion': '18.04', 

            'browserName': 'firefox', 

            'version': 'latest', 

            'resolution': '2560x1920', 

            'bitbar_apiKey': API_KEY, 

        } 

        hubUrl = 'https://us-west-desktop-hub.bitbar.com/wd/hub' 

        cls.driver = webdriver.Remote(command_executor=hubUrl, 
        desired_capabilities=caps) 

        print( 
            f'Launching webdriver for {caps["platform"]} {caps["osVersion"]} 
            {caps["browserName"]} {caps["version"]} '
        ) 

        # load the url 

        url = 'https://discover.com' 

        print(f'Opening URL: {url}') 

        cls.driver.get(url) 

        time.sleep(5) 

    @classmethod 

    def tearDownClass(self): 

        print(f'closing webdriver') 

        self.driver.quit() 

    def setUp(self): 

        # create the Visual Test instance 

        settings = { 

            'projectToken': PROJECT_TOKEN, 

        } 

        self.visualTest = VisualTest(self.driver, settings) 

    def testDiscoverHomePage(self): 

        print(f'testDiscoverHomePage method running') 

        # take a fullpage screenshot 

        print(f'Taking fullpage screenshot') 

        screenshot = self.visualTest.capture('Home Page') 

        print(f'Fullpage screenshot result: {screenshot}') 

        # currently stored locally, but will be uploaded to a server 

        # where image regression will process to display visual bugs 


    def testDiscoverProductMenu(self): 

        print(f'testDiscoverProductMenu method running') 


        # open products link and take element screenshot 

        print(f'Finding nav menu element') 


        menuLink = self.driver.find_element_by_css_selector( 
            '#basepage-159a114a2f > div.main-container.mobile-responsive
            > div > div > div > div:nth-child(1) > div > div > div 
            > header > div > div.header-content > div.navbar-wrapper 
            > div > a'
        ) 


        print(f'Clicking nav menu element') 

        menuLink.click() 


        print(f'Finding product menu') 

        menuEl = self.driver.find_element_by_css_selector( 
            '#basepage-159a114a2f > div.main-container.mobile-responsive 
            > div > div > div > div:nth-child(1) > div > div > div > header 
            > div > div.header-content > nav'
        ) 


        print(f'Taking element screenshot') 

        self.visualTest.capture('Product Menu', {'element': menuEl}) 

if __name__ == '__main__': 

    try: 

        unittest.main() 

    except Exception as e: 

        print(f'Error starting test {e}') 
            
        

import models.ScreenshotResponse; 

import models.ScrollMethod; 

import org.openqa.selenium.remote.DesiredCapabilities; 

import org.openqa.selenium.remote.RemoteWebDriver; 

  

import java.io.IOException; 

import java.nio.file.Path; 

import java.nio.file.Paths; 

import java.util.HashMap; 

  

public class Main { 

    public static void main(String[] args) throws IOException { 

    URL hubURL = new URL("https://us-west-desktop-hub.bitbar.com/wd/hub"); 

    DesiredCapabilities capabilities = new DesiredCapabilities(); 

    capabilities.setCapability("platform", "Linux"); 

    capabilities.setCapability("osVersion", "18.04"); 

    capabilities.setCapability("browserName", "firefox"); 

    capabilities.setCapability("version", "104"); 

    capabilities.setCapability("resolution", "2560x1920"); 

    // Setting the api key 

    capabilities.setCapability("bitbar_apiKey", 
    ""); 

    driver = new RemoteWebDriver(hubURL, capabilities); 

     

    driver.get("https://www.smartbear.com"); 

  

    VisualTest visualTest = new VisualTest(driver, 
    Main.buildSettings("smartbear"), Main.buildLimits()); 

    try { 

        ScreenshotResponse result = visualTest.capture(
            String.format("Test__%s__%s__2", "fullpage", "firefox")); 

    } catch (Exception e) { 

        e.printStackTrace(); 

    } 

  

    driver.quit(); 

    } 

    public static HashMap buildLimits() { 

    HashMap limits = new HashMap<>(); 

    limits.put("scrollMethod", ScrollMethod.CSS_TRANSLATE); 

    return limits; 

    } 

  

    public static HashMap 
        buildSettings(String suiteName) { 

  

    HashMap settings = new HashMap<>(); 

    settings.put("testrun", "Testing java"); 

    settings.put("debugImages", true); 

    settings.put("projectToken", ""); 

    Path currentPath = Paths.get(System.getProperty("user.dir")); 

    Path filePath = Paths.get(currentPath.toString(), 
    String.format("src/test/resources/screenshots/%s", suiteName)); 

    settings.put("saveTo", filePath.toString()); 

    return settings; 

    } 

}