Best Python code snippet using locust
admin_route.py
Source:admin_route.py  
1from run import app2from flask import request3from general_handle import sqli, output, token_handle4from Data_handle.admin import account, borrow, documents, category5###########6# Account #7###########8@app.route('/api/admin/users/all', methods=['POST', 'GET'])9def viewall_user():10    if sqli.check(request.args):11        return output.error("SQL Injection!")12    13    if 'token' in request.args:14        token = request.args['token']15        if (token_handle.checktoken(token)):16            return account.viewall(token)17        else:18            return output.error("Incorrect Token!")19    else:20        return output.page40421@app.route('/api/admin/users/create', methods=['POST', 'GET'])22def create_user():23    if sqli.check(request.args):24        return output.error("SQL Injection!")25    26    if 'token' in request.args and 'StudentID' in request.args and 'Password' in request.args and 'Fullname' in request.args and 'Specialization' in request.args and 'Class' in request.args and 'Admin' in request.args:27        token = request.args['token']28        StudentID = request.args['StudentID']29        Password = request.args['Password']30        Fullname = request.args['Fullname']31        Specialization = request.args['Specialization']32        Class = request.args['Class']33        Admin = request.args['Admin']34        if 'PhoneNumber' in request.args:35            PhoneNumber = request.args['PhoneNumber']36        else:37            PhoneNumber = "0"38        if (token_handle.checktoken(token)):39            return account.createuser(token, StudentID, Password, Fullname, PhoneNumber, Specialization, Class, Admin)40        else:41            return output.error("Incorrect Token!")42    else:43        return output.page40444@app.route('/api/admin/users/delete', methods=['POST', 'GET'])45def delete_user():46    if sqli.check(request.args):47        return output.error("SQL Injection!")48    49    if 'token' in request.args and 'StudentID' in request.args:50        token = request.args['token']51        StudentID = request.args['StudentID']52        if (token_handle.checktoken(token)):53            return account.deluser(token,StudentID)54        else:55            return output.error("Incorrect Token!")56    else:57        return output.page40458@app.route('/api/admin/users/disable', methods=['POST', 'GET'])59def disable_user():60    if sqli.check(request.args):61        return output.error("SQL Injection!")62    63    if 'token' in request.args and 'StudentID' in request.args:64        token = request.args['token']65        StudentID = request.args['StudentID']66        if (token_handle.checktoken(token)):67            return account.disableuser(token,StudentID)68        else:69            return output.error("Incorrect Token!")70    else:71        return output.page40472@app.route('/api/admin/users/active', methods=['POST', 'GET'])73def active_user():74    if sqli.check(request.args):75        return output.error("SQL Injection!")76    77    if 'token' in request.args and 'StudentID' in request.args:78        token = request.args['token']79        StudentID = request.args['StudentID']80        if (token_handle.checktoken(token)):81            return account.activeuser(token,StudentID)82        else:83            return output.error("Incorrect Token!")84    else:85        return output.page40486@app.route('/api/admin/users/update', methods=['POST', 'GET'])87def update_user():88    if sqli.check(request.args):89        return output.error("SQL Injection!")90    91    if 'token' in request.args and 'StudentID' in request.args and 'Password' in request.args and 'Fullname' in request.args and 'Specialization' in request.args and 'Class' in request.args and 'Expiry' in request.args:92        token = request.args['token']93        StudentID = request.args['StudentID']94        Password = request.args['Password']95        Fullname = request.args['Fullname']96        Specialization = request.args['Specialization']97        Class = request.args['Class']98        Expiry = request.args['Expiry']99        if 'PhoneNumber' in request.args:100            PhoneNumber = request.args['PhoneNumber']101        else:102            PhoneNumber = "0"103        if (token_handle.checktoken(token)):104            return account.update(token, StudentID, Password, Fullname, PhoneNumber, Specialization, Class, Expiry)105        else:106            return output.error("Incorrect Token!")107    else:108        return output.page404109@app.route('/api/admin/users/extend', methods=['POST', 'GET'])110def Extend():111    if sqli.check(request.args):112        return output.error("SQL Injection!")113    114    if 'token' in request.args and 'StudentID' in request.args and 'Expiry' in request.args:115        token = request.args['token']116        StudentID = request.args['StudentID']117        Expiry = request.args['Expiry']118        if (token_handle.checktoken(token)):119            return account.Extend(token, StudentID, Expiry)120        else:121            return output.error("Incorrect Token!")122    else:123        return output.page404124#############125# Documents #126#############127@app.route('/api/admin/documents/add', methods=['POST', 'GET'])128def add_documents():129    if sqli.check(request.args):130        return output.error("SQL Injection!")131    if 'token' in request.args and 'Name' in request.args and 'Author' in request.args and 'Description' in request.args and 'ListCategoryID' in request.args:132        token = request.args['token']133        Name = request.args['Name']134        Author = request.args['Author']135        Description = request.args['Description']136        ListCategoryID = request.args['ListCategoryID']137        if (token_handle.checktoken(token)):138            return documents.add(token, Name, Author, Description, ListCategoryID)139        else:140            return output.error("Incorrect Token!")141    else:142        return output.page404143@app.route('/api/admin/documents/delete', methods=['POST', 'GET'])144def delete_documents():145    if sqli.check(request.args):146        return output.error("SQL Injection!")147    if 'token' in request.args and 'DocumentsID' in request.args:148        token = request.args['token']149        DocumentsID = request.args['DocumentsID']150        if (token_handle.checktoken(token)):151            return documents.delete(token, DocumentsID)152        else:153            return output.error("Incorrect Token!")154    else:155        return output.page404156@app.route('/api/admin/documents/update', methods=['POST', 'GET'])157def update_documents():158    if sqli.check(request.args):159        return output.error("SQL Injection!")160    if 'token' in request.args and 'ID' in request.args and 'Name' in request.args and 'Author' in request.args and 'Description' in request.args and 'ListCategoryID' in request.args:161        token = request.args['token']162        ID = request.args['ID']163        Name = request.args['Name']164        Author = request.args['Author']165        Description = request.args['Description']166        ListCategoryID = request.args['ListCategoryID']167        if (token_handle.checktoken(token)):168            return documents.update(token, ID, Name, Author, Description, ListCategoryID)169        else:170            return output.error("Incorrect Token!")171    else:172        return output.page404173@app.route('/api/admin/documents/addDoc', methods=['POST', 'GET'])174def add_document():175    if sqli.check(request.args):176        return output.error("SQL Injection!")177    if 'token' in request.args and 'ID' in request.args and 'DocumentsID' in request.args and 'Reprint' in request.args and 'Price' in request.args:178        token = request.args['token']179        ID = request.args['ID']180        DocumentsID = request.args['DocumentsID']181        Reprint = request.args['Reprint']182        Price = request.args['Price']183        if (token_handle.checktoken(token)):184            return documents.addDoc(token, ID, DocumentsID, Reprint, Price)185        else:186            return output.error("Incorrect Token!")187    else:188        return output.page404189############190# Category #191############192@app.route('/api/admin/category/add', methods=['POST', 'GET'])193def add_category():194    if sqli.check(request.args):195        return output.error("SQL Injection!")196    if 'token' in request.args and 'Name' in request.args and 'Description' in request.args:197        token = request.args['token']198        Name = request.args['Name']199        Description = request.args['Description']200        if (token_handle.checktoken(token)):201            return category.add(token, Name, Description)202        else:203            return output.error("Incorrect Token!")204    else:205        return output.page404206@app.route('/api/admin/category/delete', methods=['POST', 'GET'])207def delete_category():208    if sqli.check(request.args):209        return output.error("SQL Injection!")210    if 'token' in request.args and 'ID' in request.args:211        token = request.args['token']212        ID = request.args['ID']213        if (token_handle.checktoken(token)):214            return category.delete(token, ID)215        else:216            return output.error("Incorrect Token!")217    else:218        return output.page404219@app.route('/api/admin/category/update', methods=['POST', 'GET'])220def update_category():221    if sqli.check(request.args):222        return output.error("SQL Injection!")223    if 'token' in request.args and 'ID' in request.args and 'Name' in request.args and 'Description' in request.args:224        token = request.args['token']225        ID = request.args['ID']226        Name = request.args['Name']227        Description = request.args['Description']228        if (token_handle.checktoken(token)):229            return category.update(token, ID, Name, Description)230        else:231            return output.error("Incorrect Token!")232    else:233        return output.page404234##########235# Borrow #236##########237@app.route('/api/admin/borrow/all', methods=['POST', 'GET'])238def viewall_borrow():239    if sqli.check(request.args):240        return output.error("SQL Injection!")241    242    if 'token' in request.args:243        token = request.args['token']244        if (token_handle.checktoken(token)):245            return borrow.viewall(token)246        else:247            return output.error("Incorrect Token!")248    else:249        return output.page404250@app.route('/api/admin/borrow/search', methods=['POST', 'GET'])251def searchbyStudentID():252    if sqli.check(request.args):253        return output.error("SQL Injection!")254    255    if 'token' in request.args and 'StudentID' in request.args:256        token = request.args['token']257        StudentID = request.args['StudentID']258        if (token_handle.checktoken(token)):259            return borrow.SearchbyStudentID(token, StudentID)260        else:261            return output.error("Incorrect Token!")262    else:263        return output.page404264@app.route('/api/admin/borrow/confirm', methods=['POST', 'GET'])265def confirm_borrow():266    if sqli.check(request.args):267        return output.error("SQL Injection!")268    269    if 'token' in request.args and 'borrowID' in request.args:270        token = request.args['token']271        borrowID = request.args['borrowID']272        if (token_handle.checktoken(token)):273            return borrow.confirm(token,borrowID)274        else:275            return output.error("Incorrect Token!")276    else:277        return output.page404278@app.route('/api/admin/borrow/return', methods=['POST', 'GET'])279def return_borrow():280    if sqli.check(request.args):281        return output.error("SQL Injection!")282    283    if 'token' in request.args and 'documentID' in request.args:284        token = request.args['token']285        documentID = request.args['documentID']286        if (token_handle.checktoken(token)):287            return borrow.returndocument(token,documentID)288        else:289            return output.error("Incorrect Token!")290    else:291        return output.page404292@app.route('/api/admin/borrow/lose', methods=['POST', 'GET'])293def lose_borrow():294    if sqli.check(request.args):295        return output.error("SQL Injection!")296    297    if 'token' in request.args and 'documentID' in request.args:298        token = request.args['token']299        documentID = request.args['documentID']300        if (token_handle.checktoken(token)):301            return borrow.lose(token,documentID)302        else:303            return output.error("Incorrect Token!")304    else:...test_habr2.py
Source:test_habr2.py  
1import time2from random import randint3link = "https://habr.com/ru/all/page"4# пÑовеÑка ÑÑÑÐ°Ð½Ð¸Ñ Ñ 1 по 505# number = 16# def test_guest_should_see_1_50_pages(browser):7#     for number in range (1,51):8#         browser.get(link + str(number))9#         time.sleep(randint(1,3))10#пÑовеÑка -1 ÑÑÑаниÑÑ11def test_guest_should_see_minus1_page(browser):12    browser.get(link + str(-1))13    time.sleep(randint(1,3))14    page404 = browser.find_element_by_tag_name("h1").text15    assert "СÑÑаниÑа не найдена" != page404, "Page not found"16#пÑовеÑка 0 ÑÑÑаниÑÑ17def test_guest_should_see_0_page(browser):18    browser.get(link + str(0))19    time.sleep(randint(1, 3))20    page404 = browser.find_element_by_tag_name("h1").text21    assert "СÑÑаниÑа не найдена" != page404, "Page not found"22#пÑовеÑка 1 ÑÑÑаниÑÑ23def test_guest_should_see_1_page(browser):24    browser.get(link + str(1))25    time.sleep(randint(1, 3))26    page404 = browser.find_element_by_tag_name("h1").text27    assert "СÑÑаниÑа не найдена" != page404, "Page not found"28#пÑовеÑка 25 ÑÑÑаниÑÑ29def test_guest_should_see_25_page(browser):30    browser.get(link + str(25))31    time.sleep(randint(1, 3))32    page404 = browser.find_element_by_tag_name("h1").text33    assert "СÑÑаниÑа не найдена" != page404, "Page not found"34#пÑовеÑка 49 ÑÑÑаниÑÑ35def test_guest_should_see_49_page(browser):36    browser.get(link + str(49))37    time.sleep(randint(1, 3))38    page404 = browser.find_element_by_tag_name("h1").text39    assert "СÑÑаниÑа не найдена" != page404, "Page not found"40#пÑовеÑка 50 ÑÑÑаниÑÑ41def test_guest_should_see_50_page(browser):42    browser.get(link + str(50))43    time.sleep(randint(1, 3))44    page404 = browser.find_element_by_tag_name("h1").text45    assert "СÑÑаниÑа не найдена" != page404, "Page not found"46#пÑовеÑка 51 ÑÑÑаниÑÑ47def test_guest_should_see_51_page(browser):48    browser.get(link + str(51))49    time.sleep(randint(1, 3))50    page4041 = browser.find_element_by_css_selector("div.tm-articles-subpage > div > div > div > span").text...test_Page404.py
Source:test_Page404.py  
1from Pages.DashboadPage import DashboardPage2from Pages.IndicesPage import IndicesPage3from Pages.Page404 import Page4044from Tests.Base import BaseTest5from Pages.LoginPage import LoginPage6from Pages.ConfigurationPage import ConfigurationPage7from Config.config import TestData8#test9class Test_Page404(BaseTest):10    def test_404_page_displaying_before_login_and_redirected_back_to_login(self):11          self.configpage = ConfigurationPage(self.driver)12          self.configpage.click_on_getStarted_button()13          self.page404 = Page404(self.driver)14          assert self.page404.access_non_exist_page()15          self.page404.click_go_to_login_button()16          self.loginpage = LoginPage(self.driver)17          assert self.loginpage.is_login_header_text_displaying()18    def test_404_page_displaying_after_login_on_indices_and_redirected_back_to_indices(self):19          self.configpage = ConfigurationPage(self.driver)20          self.loginpage = LoginPage(self.driver)21          self.configpage.click_on_getStarted_button()22          self.loginpage.do_login(TestData.USERNAME, TestData.PASSWORD)23          self.page404 = Page404(self.driver)24          assert self.page404.access_non_exist_page()25          self.page404.click_go_to_login_button()26          self.indicespage = IndicesPage(self.driver)27          assert self.indicespage.check_indices_header_text_displaying()28    def test_404_page_displaying_after_login_on_dashboard_and_redirected_back_to_dashboard(self):29         self.configpage = ConfigurationPage(self.driver)30         self.loginpage = LoginPage(self.driver)31         self.configpage.click_on_getStarted_button()32         self.loginpage.do_login(TestData.USERNAME, TestData.PASSWORD)33         self.indicespage = IndicesPage(self.driver)34         self.indicespage.index_pattern(TestData.INDEXPATTERN)35         self.indicespage.image_key_field(TestData.IMAGEKEYFIELD)36         self.indicespage.continue_button_click()37         self.page404 = Page404(self.driver)38         assert self.page404.access_non_exist_page()39         self.page404.click_go_to_login_button()40         self.dashboard = DashboardPage(self.driver)...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.
You could also refer to video tutorials over LambdaTest YouTube channel to get step by step demonstration from industry experts.
Get 100 minutes of automation test minutes FREE!!
