Unlock 30% off on Manual Testing Annual Plans this Holiday Season.

Copied to Clipboard!

How to use screenshot_as_png method in Selene

Best Python code snippet using selene_python

schedules_test.py

Source:schedules_test.py

1# Licensed to the Apache Software Foundation (ASF) under one2# or more contributor license agreements.  See the NOTICE file3# distributed with this work for additional information4# regarding copyright ownership.  The ASF licenses this file5# to you under the Apache License, Version 2.0 (the6# "License"); you may not use this file except in compliance7# with the License.  You may obtain a copy of the License at8#9#   http://www.apache.org/licenses/LICENSE-2.010#11# Unless required by applicable law or agreed to in writing,12# software distributed under the License is distributed on an13# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY14# KIND, either express or implied.  See the License for the15# specific language governing permissions and limitations16# under the License.17import unittest18from datetime import datetime, timedelta19from unittest.mock import Mock, patch, PropertyMock20from flask_babel import gettext as __21from selenium.common.exceptions import WebDriverException22from superset import app, db23from superset.models.core import Dashboard, Slice24from superset.models.schedules import (25    DashboardEmailSchedule,26    EmailDeliveryType,27    SliceEmailReportFormat,28    SliceEmailSchedule,29)30from superset.tasks.schedules import (31    create_webdriver,32    deliver_dashboard,33    deliver_slice,34    next_schedules,35)36from .utils import read_fixture37class SchedulesTestCase(unittest.TestCase):38    RECIPIENTS = "recipient1@superset.com, recipient2@superset.com"39    BCC = "bcc@superset.com"40    CSV = read_fixture("trends.csv")41    @classmethod42    def setUpClass(cls):43        cls.common_data = dict(44            active=True,45            crontab="* * * * *",46            recipients=cls.RECIPIENTS,47            deliver_as_group=True,48            delivery_type=EmailDeliveryType.inline,49        )50        # Pick up a random slice and dashboard51        slce = db.session.query(Slice).all()[0]52        dashboard = db.session.query(Dashboard).all()[0]53        dashboard_schedule = DashboardEmailSchedule(**cls.common_data)54        dashboard_schedule.dashboard_id = dashboard.id55        dashboard_schedule.user_id = 156        db.session.add(dashboard_schedule)57        slice_schedule = SliceEmailSchedule(**cls.common_data)58        slice_schedule.slice_id = slce.id59        slice_schedule.user_id = 160        slice_schedule.email_format = SliceEmailReportFormat.data61        db.session.add(slice_schedule)62        db.session.commit()63        cls.slice_schedule = slice_schedule.id64        cls.dashboard_schedule = dashboard_schedule.id65    @classmethod66    def tearDownClass(cls):67        db.session.query(SliceEmailSchedule).filter_by(id=cls.slice_schedule).delete()68        db.session.query(DashboardEmailSchedule).filter_by(69            id=cls.dashboard_schedule70        ).delete()71        db.session.commit()72    def test_crontab_scheduler(self):73        crontab = "* * * * *"74        start_at = datetime.now().replace(microsecond=0, second=0, minute=0)75        stop_at = start_at + timedelta(seconds=3600)76        # Fire off the task every minute77        schedules = list(next_schedules(crontab, start_at, stop_at, resolution=0))78        self.assertEqual(schedules[0], start_at)79        self.assertEqual(schedules[-1], stop_at - timedelta(seconds=60))80        self.assertEqual(len(schedules), 60)81        # Fire off the task every 10 minutes, controlled via resolution82        schedules = list(next_schedules(crontab, start_at, stop_at, resolution=10 * 60))83        self.assertEqual(schedules[0], start_at)84        self.assertEqual(schedules[-1], stop_at - timedelta(seconds=10 * 60))85        self.assertEqual(len(schedules), 6)86        # Fire off the task every 12 minutes, controlled via resolution87        schedules = list(next_schedules(crontab, start_at, stop_at, resolution=12 * 60))88        self.assertEqual(schedules[0], start_at)89        self.assertEqual(schedules[-1], stop_at - timedelta(seconds=12 * 60))90        self.assertEqual(len(schedules), 5)91    def test_wider_schedules(self):92        crontab = "*/15 2,10 * * *"93        for hour in range(0, 24):94            start_at = datetime.now().replace(95                microsecond=0, second=0, minute=0, hour=hour96            )97            stop_at = start_at + timedelta(seconds=3600)98            schedules = list(next_schedules(crontab, start_at, stop_at, resolution=0))99            if hour in (2, 10):100                self.assertEqual(len(schedules), 4)101            else:102                self.assertEqual(len(schedules), 0)103    def test_complex_schedule(self):104        # Run the job on every Friday of March and May105        # On these days, run the job at106        # 5:10 pm107        # 5:11 pm108        # 5:12 pm109        # 5:13 pm110        # 5:14 pm111        # 5:15 pm112        # 5:25 pm113        # 5:28 pm114        # 5:31 pm115        # 5:34 pm116        # 5:37 pm117        # 5:40 pm118        crontab = "10-15,25-40/3 17 * 3,5 5"119        start_at = datetime.strptime("2018/01/01", "%Y/%m/%d")120        stop_at = datetime.strptime("2018/12/31", "%Y/%m/%d")121        schedules = list(next_schedules(crontab, start_at, stop_at, resolution=60))122        self.assertEqual(len(schedules), 108)123        fmt = "%Y-%m-%d %H:%M:%S"124        self.assertEqual(schedules[0], datetime.strptime("2018-03-02 17:10:00", fmt))125        self.assertEqual(schedules[-1], datetime.strptime("2018-05-25 17:40:00", fmt))126        self.assertEqual(schedules[59], datetime.strptime("2018-03-30 17:40:00", fmt))127        self.assertEqual(schedules[60], datetime.strptime("2018-05-04 17:10:00", fmt))128    @patch("superset.tasks.schedules.firefox.webdriver.WebDriver")129    def test_create_driver(self, mock_driver_class):130        mock_driver = Mock()131        mock_driver_class.return_value = mock_driver132        mock_driver.find_elements_by_id.side_effect = [True, False]133        create_webdriver()134        create_webdriver()135        mock_driver.add_cookie.assert_called_once()136    @patch("superset.tasks.schedules.firefox.webdriver.WebDriver")137    @patch("superset.tasks.schedules.send_email_smtp")138    @patch("superset.tasks.schedules.time")139    def test_deliver_dashboard_inline(self, mtime, send_email_smtp, driver_class):140        element = Mock()141        driver = Mock()142        mtime.sleep.return_value = None143        driver_class.return_value = driver144        # Ensure that we are able to login with the driver145        driver.find_elements_by_id.side_effect = [True, False]146        driver.find_element_by_class_name.return_value = element147        element.screenshot_as_png = read_fixture("sample.png")148        schedule = (149            db.session.query(DashboardEmailSchedule)150            .filter_by(id=self.dashboard_schedule)151            .all()[0]152        )153        deliver_dashboard(schedule)154        mtime.sleep.assert_called_once()155        driver.screenshot.assert_not_called()156        send_email_smtp.assert_called_once()157    @patch("superset.tasks.schedules.firefox.webdriver.WebDriver")158    @patch("superset.tasks.schedules.send_email_smtp")159    @patch("superset.tasks.schedules.time")160    def test_deliver_dashboard_as_attachment(161        self, mtime, send_email_smtp, driver_class162    ):163        element = Mock()164        driver = Mock()165        mtime.sleep.return_value = None166        driver_class.return_value = driver167        # Ensure that we are able to login with the driver168        driver.find_elements_by_id.side_effect = [True, False]169        driver.find_element_by_id.return_value = element170        driver.find_element_by_class_name.return_value = element171        element.screenshot_as_png = read_fixture("sample.png")172        schedule = (173            db.session.query(DashboardEmailSchedule)174            .filter_by(id=self.dashboard_schedule)175            .all()[0]176        )177        schedule.delivery_type = EmailDeliveryType.attachment178        deliver_dashboard(schedule)179        mtime.sleep.assert_called_once()180        driver.screenshot.assert_not_called()181        send_email_smtp.assert_called_once()182        self.assertIsNone(send_email_smtp.call_args[1]["images"])183        self.assertEqual(184            send_email_smtp.call_args[1]["data"]["screenshot.png"],185            element.screenshot_as_png,186        )187    @patch("superset.tasks.schedules.firefox.webdriver.WebDriver")188    @patch("superset.tasks.schedules.send_email_smtp")189    @patch("superset.tasks.schedules.time")190    def test_dashboard_chrome_like(self, mtime, send_email_smtp, driver_class):191        # Test functionality for chrome driver which does not support192        # element snapshots193        element = Mock()194        driver = Mock()195        mtime.sleep.return_value = None196        type(element).screenshot_as_png = PropertyMock(side_effect=WebDriverException)197        driver_class.return_value = driver198        # Ensure that we are able to login with the driver199        driver.find_elements_by_id.side_effect = [True, False]200        driver.find_element_by_id.return_value = element201        driver.find_element_by_class_name.return_value = element202        driver.screenshot.return_value = read_fixture("sample.png")203        schedule = (204            db.session.query(DashboardEmailSchedule)205            .filter_by(id=self.dashboard_schedule)206            .all()[0]207        )208        deliver_dashboard(schedule)209        mtime.sleep.assert_called_once()210        driver.screenshot.assert_called_once()211        send_email_smtp.assert_called_once()212        self.assertEqual(send_email_smtp.call_args[0][0], self.RECIPIENTS)213        self.assertEqual(214            list(send_email_smtp.call_args[1]["images"].values())[0],215            driver.screenshot.return_value,216        )217    @patch("superset.tasks.schedules.firefox.webdriver.WebDriver")218    @patch("superset.tasks.schedules.send_email_smtp")219    @patch("superset.tasks.schedules.time")220    def test_deliver_email_options(self, mtime, send_email_smtp, driver_class):221        element = Mock()222        driver = Mock()223        mtime.sleep.return_value = None224        driver_class.return_value = driver225        # Ensure that we are able to login with the driver226        driver.find_elements_by_id.side_effect = [True, False]227        driver.find_element_by_class_name.return_value = element228        element.screenshot_as_png = read_fixture("sample.png")229        schedule = (230            db.session.query(DashboardEmailSchedule)231            .filter_by(id=self.dashboard_schedule)232            .all()[0]233        )234        # Send individual mails to the group235        schedule.deliver_as_group = False236        # Set a bcc email address237        app.config["EMAIL_REPORT_BCC_ADDRESS"] = self.BCC238        deliver_dashboard(schedule)239        mtime.sleep.assert_called_once()240        driver.screenshot.assert_not_called()241        self.assertEqual(send_email_smtp.call_count, 2)242        self.assertEqual(send_email_smtp.call_args[1]["bcc"], self.BCC)243    @patch("superset.tasks.schedules.firefox.webdriver.WebDriver")244    @patch("superset.tasks.schedules.send_email_smtp")245    @patch("superset.tasks.schedules.time")246    def test_deliver_slice_inline_image(self, mtime, send_email_smtp, driver_class):247        element = Mock()248        driver = Mock()249        mtime.sleep.return_value = None250        driver_class.return_value = driver251        # Ensure that we are able to login with the driver252        driver.find_elements_by_id.side_effect = [True, False]253        driver.find_element_by_class_name.return_value = element254        element.screenshot_as_png = read_fixture("sample.png")255        schedule = (256            db.session.query(SliceEmailSchedule)257            .filter_by(id=self.slice_schedule)258            .all()[0]259        )260        schedule.email_format = SliceEmailReportFormat.visualization261        schedule.delivery_format = EmailDeliveryType.inline262        deliver_slice(schedule)263        mtime.sleep.assert_called_once()264        driver.screenshot.assert_not_called()265        send_email_smtp.assert_called_once()266        self.assertEqual(267            list(send_email_smtp.call_args[1]["images"].values())[0],268            element.screenshot_as_png,269        )270    @patch("superset.tasks.schedules.firefox.webdriver.WebDriver")271    @patch("superset.tasks.schedules.send_email_smtp")272    @patch("superset.tasks.schedules.time")273    def test_deliver_slice_attachment(self, mtime, send_email_smtp, driver_class):274        element = Mock()275        driver = Mock()276        mtime.sleep.return_value = None277        driver_class.return_value = driver278        # Ensure that we are able to login with the driver279        driver.find_elements_by_id.side_effect = [True, False]280        driver.find_element_by_class_name.return_value = element281        element.screenshot_as_png = read_fixture("sample.png")282        schedule = (283            db.session.query(SliceEmailSchedule)284            .filter_by(id=self.slice_schedule)285            .all()[0]286        )287        schedule.email_format = SliceEmailReportFormat.visualization288        schedule.delivery_type = EmailDeliveryType.attachment289        deliver_slice(schedule)290        mtime.sleep.assert_called_once()291        driver.screenshot.assert_not_called()292        send_email_smtp.assert_called_once()293        self.assertEqual(294            send_email_smtp.call_args[1]["data"]["screenshot.png"],295            element.screenshot_as_png,296        )297    @patch("superset.tasks.schedules.urllib.request.OpenerDirector.open")298    @patch("superset.tasks.schedules.urllib.request.urlopen")299    @patch("superset.tasks.schedules.send_email_smtp")300    def test_deliver_slice_csv_attachment(301        self, send_email_smtp, mock_open, mock_urlopen302    ):303        response = Mock()304        mock_open.return_value = response305        mock_urlopen.return_value = response306        mock_urlopen.return_value.getcode.return_value = 200307        response.content = self.CSV308        schedule = (309            db.session.query(SliceEmailSchedule)310            .filter_by(id=self.slice_schedule)311            .all()[0]312        )313        schedule.email_format = SliceEmailReportFormat.data314        schedule.delivery_type = EmailDeliveryType.attachment315        deliver_slice(schedule)316        send_email_smtp.assert_called_once()317        file_name = __("%(name)s.csv", name=schedule.slice.slice_name)318        self.assertEqual(send_email_smtp.call_args[1]["data"][file_name], self.CSV)319    @patch("superset.tasks.schedules.urllib.request.urlopen")320    @patch("superset.tasks.schedules.urllib.request.OpenerDirector.open")321    @patch("superset.tasks.schedules.send_email_smtp")322    def test_deliver_slice_csv_inline(self, send_email_smtp, mock_open, mock_urlopen):323        response = Mock()324        mock_open.return_value = response325        mock_urlopen.return_value = response326        mock_urlopen.return_value.getcode.return_value = 200327        response.content = self.CSV328        schedule = (329            db.session.query(SliceEmailSchedule)330            .filter_by(id=self.slice_schedule)331            .all()[0]332        )333        schedule.email_format = SliceEmailReportFormat.data334        schedule.delivery_type = EmailDeliveryType.inline335        deliver_slice(schedule)336        send_email_smtp.assert_called_once()337        self.assertIsNone(send_email_smtp.call_args[1]["data"])...

Compare_image.py

Source:Compare_image.py

1import glob2import os3from selenium import webdriver4from selenium.webdriver.common.by import By5from selenium.webdriver.chrome.service import Service6from PIL import Image, ImageChops, ImageStat7import time8s = Service(executable_path="../drivers/chromedriver")9input_logo = '../input/0345.png'10output_logo = '../output/0345.png'11comparison_result = 'output/result1.png'12def clear_result():13    files = glob.glob('output/*')14    for f in files:15        os.remove(f)16def capture_image():17    driver = webdriver.Chrome(service=s)18    driver.get("http://192.168.0.103:8102/landing")19    driver.maximize_window()20    driver.implicitly_wait(5)21    time.sleep(3)22    driver.find_element(By.XPATH,"//ion-button[@color='primary']").click()23    time.sleep(2)24    driver.find_element(By.XPATH, "//ion-input[@type='email']/input").send_keys("venkateshrv307@gmail.com")25    driver.find_element(By.XPATH, "//input[@name='ion-input-1']").send_keys("Venkat@2000")26    time.sleep(2)27    driver.find_element(By.XPATH, "//ion-button[@size='large']").click()28    time.sleep(4)29    driver.find_element(By.XPATH, "//ion-tab-button[@tab='settings']").click()30    time.sleep(2)31    driver.find_element(By.XPATH,"(//ion-label[@class='sc-ion-label-md-h sc-ion-label-md-s md hydrated'])[9]").click()32    time.sleep(2)33    driver.find_element(By.XPATH,"//input[@class='native-input sc-ion-input-md']").send_keys("0345")34    time.sleep(1)35    driver.find_element(By.XPATH,"(//ion-button[@color='primary'])[3]").click()36    time.sleep(2)37    with open(output_logo,'wb') as file:38         #file.write(driver.find_element(By.XPATH,"//img[@src='assets/img/scales/0341.png']").screenshot_as_png)39         #file.write(driver.find_element(By.XPATH,"//img[@src='assets/img/scales/0342.png']").screenshot_as_png)40         #file.write(driver.find_element(By.XPATH,"//img[@src='assets/img/scales/0343.png']").screenshot_as_png)41         file.write(driver.find_element(By.XPATH,"//img[@src='assets/img/scales/0345.png']").screenshot_as_png)42         #file.write(driver.find_element(By.XPATH,"//img[@src='assets/img/scales/0346.png']").screenshot_as_png)43         #file.write(driver.find_element(By.XPATH,"//img[@src='assets/img/scales/0347.png']").screenshot_as_png)44         #file.write(driver.find_element(By.XPATH,"//img[@src='assets/img/scales/0358.png']").screenshot_as_png)45         #file.write(driver.find_element(By.XPATH,"//img[@src='assets/img/scales/0359.png']").screenshot_as_png)46         #file.write(driver.find_element(By.XPATH,"//img[@src='assets/img/scales/0364.png']").screenshot_as_png)47         #file.write(driver.find_element(By.XPATH,"//img[@src='assets/img/scales/0369.png']").screenshot_as_png)48         #file.write(driver.find_element(By.XPATH,"//img[@src='assets/img/scales/0370.png']").screenshot_as_png)49         #file.write(driver.find_element(By.XPATH,"//img[@src='assets/img/scales/0371.png']").screenshot_as_png)50         #file.write(driver.find_element(By.XPATH,"//img[@src='assets/img/scales/0375.png']").screenshot_as_png)51         #file.write(driver.find_element(By.XPATH,"//img[@src='assets/img/scales/0376.png']").screenshot_as_png)52         #file.write(driver.find_element(By.XPATH,"(//img[@src='assets/img/scales/0383.png'])[1]").screenshot_as_png)53         #file.write(driver.find_element(By.XPATH,"//img[@src='assets/img/scales/0380.png']").screenshot_as_png)54         #file.write(driver.find_element(By.XPATH,"//img[@src='assets/img/scales/0382.png']").screenshot_as_png)55         #file.write(driver.find_element(By.XPATH,"(//img[@src='assets/img/scales/0383.png'])[2]").screenshot_as_png)56         #file.write(driver.find_element(By.XPATH,"//img[@src='assets/img/scales/0384.png']").screenshot_as_png)57         #file.write(driver.find_element(By.XPATH,"//img[@src='assets/img/scales/0385.png']").screenshot_as_png)58         #file.write(driver.find_element(By.XPATH,"//img[@src='assets/img/scales/0396.png']").screenshot_as_png)59         #file.write(driver.find_element(By.XPATH,"//img[@src='assets/img/scales/0396.png']").screenshot_as_png)60    img1 = Image.open(input_logo)61    img2 = Image.open(output_logo)62    logoresult = ImageChops.difference(img1,img2)63    stat = ImageStat.Stat(logoresult)64    diff_value = sum(stat.mean)65    print(diff_value)66    if diff_value > 1:67        result = "FAIL"68        print(result)69    else:70        result = "PASS"71        print(result)72        logoresult.show()73        logoresult.save(comparison_result)74    driver.close()75    driver.quit()76if __name__ == '__main__':77    clear_result()...

selenium_验证码.py

Source:selenium_验证码.py

1# -*- coding:utf-8 -*-2# @Time : 2022/5/18 15:163# Auther : shenyuming4# @File : selenium_éªè¯ç .py5# @Software : PyCharm6'''7ç¥è¯ç¹:  screenshot_as_png  ä¿ååç´ çæªå¾ ï¼ ddddocréªè¯ç è¯å«8'''9from selenium import webdriver10from time import sleep11import ddddocr12driver = webdriver.Chrome()13driver.get('http://121.41.14.39:8088/index.html')14sleep(2)15driver.find_element_by_css_selector('#username').send_keys('sq2')16driver.find_element_by_css_selector('#password').send_keys('123')17# è·å¾éªè¯ç å¾ç18ele_img = driver.find_element_by_css_selector('img')19## å¯ä»¥ä¿åå¾ç20# with open('ocr.img','wb') as f:21#     f.write(ele_img.screenshot_as_png)  # screenshot_as_png  bytesç±»åï¼ å¯ä»¥ä¿åå¾ç22## ç´æ¥è·å¾imgçææ¬ , å¯¼å¥ddddcorï¼å®ä¾åddddocr23ocr = ddddocr.DdddOcr()24text = ocr.classification(ele_img.screenshot_as_png)25## è¾å¥éªè¯ç 26driver.find_element_by_css_selector('#code').send_keys(text)27driver.find_element_by_css_selector('#submitButton').click()28sleep(1)...

Automation Testing Tutorials

Learn to execute automation testing from scratch with LambdaTest Learning Hub. Right from setting up the prerequisites to run your first automation test, to following best practices and diving deeper into advanced test scenarios. LambdaTest Learning Hubs compile a list of step-by-step guides to help you be proficient with different test automation frameworks i.e. Selenium, Cypress, TestNG etc.