Compare commits
10
Commits
| Author | SHA1 | Date | |
|---|---|---|---|
 tejasshasane
|
48d0d786b2 | ||
|
tejasshasane
|
d681002b0f | ||
|
tejasshasane
|
eb8a831edb | ||
|
tejasshasane
|
2454ca3180 | ||
|
tejasshasane
|
771b755ca1 | ||
|
tejasshasane
|
d9218709f0 | ||
|
tejasshasane
|
385a9e22cc | ||
|
tejasshasane
|
24f2c1fe6a | ||
|
tejasshasane
|
631f64b84f | ||
|
tejasshasane
|
c42cc83aa1 |
@@ -21,3 +21,6 @@ reports/logs/*
|
|||||||
|
|
||||||
reports/html/*
|
reports/html/*
|
||||||
!reports/html/.gitkeep
|
!reports/html/.gitkeep
|
||||||
|
|
||||||
|
# CAPTCHA stress-test screenshots and CSV reports
|
||||||
|
reports/captcha_login_stress/
|
||||||
|
|||||||
+241
-175
@@ -1,183 +1,249 @@
|
|||||||
{
|
{
|
||||||
"url": "https://utilseastern.pxil.in/Pxil/",
|
"url": "https://utilseastern.pxil.in/Pxil/",
|
||||||
"stagingurl": "https://utilseastern.pxil.in/Pxil/",
|
"stagingurl": "https://utilseastern.pxil.in/Pxil/",
|
||||||
"uaturl": "https://utilsuat.pxil.in/Pxil/",
|
"uaturl": "https://utilsuat.pxil.in/Pxil/",
|
||||||
"login": {
|
"login": {
|
||||||
"username": "Tejas",
|
"username": "Tejas",
|
||||||
"password": "Pxil@123"
|
"password": "Pxil@123"
|
||||||
},
|
},
|
||||||
"login_captcha_wait_seconds": 10,
|
"login_captcha_wait_seconds": 10,
|
||||||
"login_captcha_max_retries": 10,
|
"login_captcha_max_retries": 10,
|
||||||
"login_post_submit_wait_seconds": 10,
|
"login_post_submit_wait_seconds": 10,
|
||||||
"step_delay_seconds": 1,
|
"step_delay_seconds": 1,
|
||||||
"setup": {
|
"setup": {
|
||||||
"run_session": true,
|
"run_session": true,
|
||||||
"run_token": false,
|
"run_token": false,
|
||||||
"session": {
|
"session": {
|
||||||
"start_offset_minutes": 5,
|
"time_mode": "offset",
|
||||||
"end_time": "20:30"
|
"start_offset_minutes": 5,
|
||||||
}
|
"start_time": "14:30",
|
||||||
},
|
"end_time": "20:30"
|
||||||
"products": {
|
|
||||||
"IDAS": {
|
|
||||||
"enabled": true,
|
|
||||||
"run_session": true,
|
|
||||||
"run_token": false,
|
|
||||||
"me_instance": "55",
|
|
||||||
"session": {},
|
|
||||||
"token": {
|
|
||||||
"market_code": "NATIONAL",
|
|
||||||
"product_type": "IDAS",
|
|
||||||
"product_code": "IDAS4",
|
|
||||||
"product_codes": "IDAS,IDASTEST01,IDASTEST02,IDASTEST03,IDASTEST04,IDASTEST05,IDASTEST06",
|
|
||||||
"available_codes": ",IDASTEST06,IDASTEST07,IDASTEST08,IDASTEST09,IDASTEST10,IDASTEST11,IDASTEST12,IDASTEST13,IDASTEST14,IDASTEST15,IDASTEST16,IDASTEST17,IDASTEST18,IDASTEST19,IDASTEST20,IDASTEST21,IDASTEST22,IDASTEST23,IDASTEST24,IDASTEST25,IDASTEST26,IDASTEST27,IDASTEST28,IDASTEST29,IDASTEST30",
|
|
||||||
"delivery_offset_days": 1,
|
|
||||||
"expiry_time": "20:30",
|
|
||||||
"time_sessions": {
|
|
||||||
"ORDER_NOR": {
|
|
||||||
"start_offset_minutes": 10,
|
|
||||||
"end_time": "19:30"
|
|
||||||
},
|
|
||||||
"ORDER_EXCH": {
|
|
||||||
"start_time": "19:30",
|
|
||||||
"end_time": "19:45"
|
|
||||||
},
|
|
||||||
"TRADE": {
|
|
||||||
"start_time": "19:45",
|
|
||||||
"end_time": "19:50"
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
}
|
|
||||||
},
|
},
|
||||||
"RTM": {
|
"products": {
|
||||||
"enabled": false,
|
"IDAS": {
|
||||||
"run_session": false,
|
"enabled": true,
|
||||||
"run_token": true,
|
"run_session": true,
|
||||||
"me_instance": "52",
|
"run_token": false,
|
||||||
"session": {
|
"me_instance": "55",
|
||||||
"start_offset_minutes": 7,
|
"session": {},
|
||||||
"end_time": "20:30"
|
"token": {
|
||||||
},
|
"market_code": "NATIONAL",
|
||||||
"token": {
|
"product_type": "IDAS",
|
||||||
"market_code": "NATIONAL",
|
"product_code": "IDAS4",
|
||||||
"product_type": "RTM",
|
"product_codes": "IDAS,IDASTEST01,IDASTEST02,IDASTEST03,IDASTEST04,IDASTEST05,IDASTEST06",
|
||||||
"product_codes": [
|
"available_codes": ",IDASTEST06,IDASTEST07,IDASTEST08,IDASTEST09,IDASTEST10,IDASTEST11,IDASTEST12,IDASTEST13,IDASTEST14,IDASTEST15,IDASTEST16,IDASTEST17,IDASTEST18,IDASTEST19,IDASTEST20,IDASTEST21,IDASTEST22,IDASTEST23,IDASTEST24,IDASTEST25,IDASTEST26,IDASTEST27,IDASTEST28,IDASTEST29,IDASTEST30",
|
||||||
"RTM"
|
"delivery_offset_days": 1,
|
||||||
],
|
"expiry_time": "20:30",
|
||||||
"generation_mode": "RTM",
|
"time_sessions": {
|
||||||
"delivery_offset_days": 0,
|
"ORDER_NOR": {
|
||||||
"delivery_from_time": "00:00",
|
"start_offset_minutes": 10,
|
||||||
"delivery_to_time": "23:59",
|
"end_time": "19:30"
|
||||||
"session_start_offset_minutes": 5,
|
},
|
||||||
"session_end_time": "20:00"
|
"ORDER_EXCH": {
|
||||||
}
|
"start_time": "19:30",
|
||||||
},
|
"end_time": "19:45"
|
||||||
"UPC": {
|
},
|
||||||
"enabled": true,
|
"TRADE": {
|
||||||
"run_session": true,
|
"start_time": "19:45",
|
||||||
"run_token": true,
|
"end_time": "19:50"
|
||||||
"session_variants": {
|
}
|
||||||
"DAILY": {
|
}
|
||||||
"enabled": true,
|
}
|
||||||
"run_session": false,
|
|
||||||
"run_token": true,
|
|
||||||
"me_instance": "58",
|
|
||||||
"token": {
|
|
||||||
"generation_mode": "AnyDayUPC",
|
|
||||||
"create_product_codes": [
|
|
||||||
"DailyC_E"
|
|
||||||
],
|
|
||||||
"available_product_codes": [
|
|
||||||
"DailyC_D",
|
|
||||||
"DailyGR_D",
|
|
||||||
"DailyC_RTC",
|
|
||||||
"DailyC_E",
|
|
||||||
"DailyC_M"
|
|
||||||
]
|
|
||||||
}
|
|
||||||
},
|
},
|
||||||
"WEEKLY": {
|
"RTM": {
|
||||||
"enabled": true,
|
"enabled": false,
|
||||||
"run_session": false,
|
"run_session": false,
|
||||||
"run_token": true,
|
"run_token": true,
|
||||||
"me_instance": "59",
|
"me_instance": "52",
|
||||||
"token": {
|
"session": {
|
||||||
"generation_mode": "WeeklyUPC",
|
"time_mode": "offset",
|
||||||
"create_product_codes": [
|
"start_offset_minutes": 7,
|
||||||
"WeeklyC_E"
|
"start_time": "14:30",
|
||||||
],
|
"end_time": "20:30"
|
||||||
"available_product_codes": [
|
},
|
||||||
"WeeklyC_D",
|
"token": {
|
||||||
"WeeklyC_N",
|
"market_code": "NATIONAL",
|
||||||
"WeeklyGR_D",
|
"product_type": "RTM",
|
||||||
"WeeklyC_RTC"
|
"product_codes": [
|
||||||
]
|
"RTM"
|
||||||
}
|
],
|
||||||
|
"generation_mode": "RTM",
|
||||||
|
"delivery_offset_days": 0,
|
||||||
|
"delivery_from_time": "00:00",
|
||||||
|
"delivery_to_time": "23:59",
|
||||||
|
"time_mode": "offset",
|
||||||
|
"session_start_offset_minutes": 5,
|
||||||
|
"session_start_time": "14:30",
|
||||||
|
"session_end_time": "20:00"
|
||||||
|
}
|
||||||
},
|
},
|
||||||
"MONTHLY": {
|
"UPC": {
|
||||||
"enabled": true,
|
"enabled": true,
|
||||||
"run_session": false,
|
"run_session": true,
|
||||||
"run_token": true,
|
"run_token": true,
|
||||||
"me_instance": "60",
|
"session_variants": {
|
||||||
"token": {
|
"DAILY": {
|
||||||
"generation_mode": "MonthlyUPC",
|
"enabled": true,
|
||||||
"create_product_codes": [
|
"run_session": false,
|
||||||
"MonthlyC_N"
|
"run_token": true,
|
||||||
],
|
"me_instance": "58",
|
||||||
"available_product_codes": [
|
"token": {
|
||||||
"MonthlyC_D",
|
"generation_mode": "AnyDayUPC",
|
||||||
"MonthlyGR_D",
|
"create_product_codes": [
|
||||||
"MonthlyC_RTC"
|
"DailyC_E"
|
||||||
]
|
],
|
||||||
}
|
"available_product_codes": [
|
||||||
|
"DailyC_D",
|
||||||
|
"DailyGR_D",
|
||||||
|
"DailyC_RTC",
|
||||||
|
"DailyC_E",
|
||||||
|
"DailyC_M"
|
||||||
|
]
|
||||||
|
}
|
||||||
|
},
|
||||||
|
"WEEKLY": {
|
||||||
|
"enabled": true,
|
||||||
|
"run_session": false,
|
||||||
|
"run_token": true,
|
||||||
|
"me_instance": "59",
|
||||||
|
"token": {
|
||||||
|
"generation_mode": "WeeklyUPC",
|
||||||
|
"create_product_codes": [
|
||||||
|
"WeeklyC_E"
|
||||||
|
],
|
||||||
|
"available_product_codes": [
|
||||||
|
"WeeklyC_D",
|
||||||
|
"WeeklyC_N",
|
||||||
|
"WeeklyGR_D",
|
||||||
|
"WeeklyC_RTC"
|
||||||
|
]
|
||||||
|
}
|
||||||
|
},
|
||||||
|
"MONTHLY": {
|
||||||
|
"enabled": true,
|
||||||
|
"run_session": false,
|
||||||
|
"run_token": true,
|
||||||
|
"me_instance": "60",
|
||||||
|
"token": {
|
||||||
|
"generation_mode": "MonthlyUPC",
|
||||||
|
"create_product_codes": [
|
||||||
|
"MonthlyC_N"
|
||||||
|
],
|
||||||
|
"available_product_codes": [
|
||||||
|
"MonthlyC_D",
|
||||||
|
"MonthlyGR_D",
|
||||||
|
"MonthlyC_RTC"
|
||||||
|
]
|
||||||
|
}
|
||||||
|
}
|
||||||
|
},
|
||||||
|
"session": {},
|
||||||
|
"token": {
|
||||||
|
"market_code": "NATIONAL",
|
||||||
|
"product_type": "UPC",
|
||||||
|
"product_codes": [],
|
||||||
|
"delivery_offset_days": 1,
|
||||||
|
"expiry_time": "20:30",
|
||||||
|
"upc_times": {
|
||||||
|
"time_mode": "manual",
|
||||||
|
"order_normal_start_time": "13:00",
|
||||||
|
"order_normal_start_offset_minutes": 10,
|
||||||
|
"order_normal_end_time": "17:00",
|
||||||
|
"trade_session_start_time": "17:00",
|
||||||
|
"trade_session_end_time": "22:00"
|
||||||
|
}
|
||||||
|
}
|
||||||
|
},
|
||||||
|
"RA": {
|
||||||
|
"enabled": false,
|
||||||
|
"run_session": true,
|
||||||
|
"run_token": false,
|
||||||
|
"me_instance": "64",
|
||||||
|
"session": {},
|
||||||
|
"token": {
|
||||||
|
"market_code": "NATIONAL",
|
||||||
|
"product_type": "RA",
|
||||||
|
"product_codes": [],
|
||||||
|
"delivery_offset_days": 1,
|
||||||
|
"expiry_time": "20:30",
|
||||||
|
"time_sessions": {}
|
||||||
|
}
|
||||||
|
},
|
||||||
|
"REC": {
|
||||||
|
"enabled": false,
|
||||||
|
"run_session": true,
|
||||||
|
"run_token": true,
|
||||||
|
"me_instance": "25",
|
||||||
|
"session": {},
|
||||||
|
"token": {
|
||||||
|
"market_code": "NATIONAL",
|
||||||
|
"product_type": "REC",
|
||||||
|
"product_codes": [
|
||||||
|
"REC"
|
||||||
|
],
|
||||||
|
"delivery_offset_days": 1,
|
||||||
|
"expiry_time": "20:30",
|
||||||
|
"type_prefix": "REC",
|
||||||
|
"time_sessions": {
|
||||||
|
"ORDER_NOR": {
|
||||||
|
"time_mode": "offset",
|
||||||
|
"start_offset_minutes": 10,
|
||||||
|
"start_time": "14:30",
|
||||||
|
"end_time": "19:30"
|
||||||
|
},
|
||||||
|
"ORDER_EXCH": {
|
||||||
|
"start_time": "19:30",
|
||||||
|
"end_time": "19:45"
|
||||||
|
},
|
||||||
|
"TRADE": {
|
||||||
|
"start_time": "19:45",
|
||||||
|
"end_time": "19:50"
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
},
|
||||||
|
"INTRADAY": {
|
||||||
|
"enabled": true,
|
||||||
|
"run_session": true,
|
||||||
|
"run_token": true,
|
||||||
|
"me_instance": "53",
|
||||||
|
"session": {},
|
||||||
|
"token": {
|
||||||
|
"market_code": "NATIONAL",
|
||||||
|
"product_type": "INTRADAY",
|
||||||
|
"product_codes": [
|
||||||
|
"I_DayStatic_C"
|
||||||
|
],
|
||||||
|
"generation_mode": "Intraday24by7",
|
||||||
|
"delivery_start_offset_minutes": 5,
|
||||||
|
"trading_start_time": "00:20",
|
||||||
|
"trading_start_days_before_delivery": 1,
|
||||||
|
"duration_before_delivery_end_time": 150,
|
||||||
|
"delivery_to_time": "23:59"
|
||||||
|
}
|
||||||
|
},
|
||||||
|
"DAC": {
|
||||||
|
"enabled": false,
|
||||||
|
"run_session": true,
|
||||||
|
"run_token": true,
|
||||||
|
"me_instance": "57",
|
||||||
|
"session": {
|
||||||
|
"time_mode": "offset",
|
||||||
|
"start_offset_minutes": 5,
|
||||||
|
"end_time": "23:00"
|
||||||
|
},
|
||||||
|
"token": {
|
||||||
|
"market_code": "NATIONAL",
|
||||||
|
"product_type": "DAC",
|
||||||
|
"product_codes": ["DACStatic"],
|
||||||
|
"generation_mode": "DAC",
|
||||||
|
"delivery_start_offset_minutes": 5,
|
||||||
|
"delivery_to_time": "23:59",
|
||||||
|
"session_start_time": "13:00",
|
||||||
|
"session_end_time": "23:00",
|
||||||
|
"delivery_time_period": 15
|
||||||
|
}
|
||||||
}
|
}
|
||||||
},
|
|
||||||
"session": {},
|
|
||||||
"token": {
|
|
||||||
"market_code": "NATIONAL",
|
|
||||||
"product_type": "UPC",
|
|
||||||
"product_codes": [],
|
|
||||||
"delivery_offset_days": 1,
|
|
||||||
"expiry_time": "20:30",
|
|
||||||
"upc_times": {
|
|
||||||
"order_normal_start_time": "13:00",
|
|
||||||
"order_normal_end_time": "17:00",
|
|
||||||
"trade_session_start_time": "17:00",
|
|
||||||
"trade_session_end_time": "22:00"
|
|
||||||
},
|
|
||||||
"time_sessions": {}
|
|
||||||
}
|
|
||||||
},
|
|
||||||
"RA": {
|
|
||||||
"enabled": false,
|
|
||||||
"run_session": true,
|
|
||||||
"run_token": false,
|
|
||||||
"me_instance": "64",
|
|
||||||
"session": {},
|
|
||||||
"token": {
|
|
||||||
"market_code": "NATIONAL",
|
|
||||||
"product_type": "RA",
|
|
||||||
"product_codes": [],
|
|
||||||
"delivery_offset_days": 1,
|
|
||||||
"expiry_time": "20:30",
|
|
||||||
"time_sessions": {}
|
|
||||||
}
|
|
||||||
},
|
|
||||||
"REC": {
|
|
||||||
"enabled": false,
|
|
||||||
"run_session": true,
|
|
||||||
"run_token": true,
|
|
||||||
"me_instance": "25",
|
|
||||||
"session": {},
|
|
||||||
"token": {
|
|
||||||
"market_code": "NATIONAL",
|
|
||||||
"product_type": "REC",
|
|
||||||
"product_codes": ["REC"],
|
|
||||||
"delivery_offset_days": 1,
|
|
||||||
"expiry_time": "20:30",
|
|
||||||
"type_prefix": "REC",
|
|
||||||
"time_sessions": {}
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -12,3 +12,9 @@ markers =
|
|||||||
reports: report download coverage
|
reports: report download coverage
|
||||||
order_book: order book coverage
|
order_book: order book coverage
|
||||||
trade_book: trade book coverage
|
trade_book: trade book coverage
|
||||||
|
captcha: CAPTCHA login stress test
|
||||||
|
|
||||||
|
filterwarnings =
|
||||||
|
ignore::DeprecationWarning:easyocr.*
|
||||||
|
ignore::DeprecationWarning:torch.*
|
||||||
|
ignore::UserWarning:torch.*
|
||||||
|
|||||||
+10
-2
@@ -1,6 +1,14 @@
|
|||||||
# Core test runner
|
# Core test runner
|
||||||
pytest==8.4.1
|
pytest==9.1.1
|
||||||
pytest-html==4.2.0
|
pytest-html==4.2.0
|
||||||
|
|
||||||
# Browser automation
|
# Browser automation
|
||||||
selenium==4.34.2
|
selenium==4.50.0
|
||||||
|
|
||||||
|
# CAPTCHA OCR and image processing
|
||||||
|
easyocr==1.7.2
|
||||||
|
opencv-python==5.0.0.93
|
||||||
|
Pillow==12.3.0
|
||||||
|
numpy==2.5.3
|
||||||
|
torch==2.14.1
|
||||||
|
torchvision==0.29.1
|
||||||
|
|||||||
+10
-1
@@ -1,4 +1,7 @@
|
|||||||
|
import os
|
||||||
|
|
||||||
from selenium import webdriver
|
from selenium import webdriver
|
||||||
|
from selenium.webdriver.chrome.service import Service
|
||||||
|
|
||||||
|
|
||||||
def create_chrome_driver():
|
def create_chrome_driver():
|
||||||
@@ -10,5 +13,11 @@ def create_chrome_driver():
|
|||||||
options.add_argument("--disable-features=CalculateNativeWinOcclusion")
|
options.add_argument("--disable-features=CalculateNativeWinOcclusion")
|
||||||
options.add_argument("--disable-hang-monitor")
|
options.add_argument("--disable-hang-monitor")
|
||||||
options.add_argument("--disable-popup-blocking")
|
options.add_argument("--disable-popup-blocking")
|
||||||
|
options.add_argument("--disable-background-networking")
|
||||||
|
options.add_argument("--disable-component-update")
|
||||||
|
options.add_argument("--disable-sync")
|
||||||
|
options.add_argument("--disable-default-apps")
|
||||||
|
options.add_experimental_option("excludeSwitches", ["enable-logging"])
|
||||||
|
|
||||||
return webdriver.Chrome(options=options)
|
service = Service(log_output=os.devnull)
|
||||||
|
return webdriver.Chrome(service=service, options=options)
|
||||||
|
|||||||
@@ -0,0 +1,353 @@
|
|||||||
|
"""
|
||||||
|
CAPTCHA Solver for PXIL Login using EasyOCR.
|
||||||
|
Extracts base64 from input's CSS background-image, runs EasyOCR OCR, fills the input.
|
||||||
|
"""
|
||||||
|
import re
|
||||||
|
import base64
|
||||||
|
import time
|
||||||
|
from typing import Optional
|
||||||
|
|
||||||
|
from PIL import Image
|
||||||
|
import cv2
|
||||||
|
import numpy as np
|
||||||
|
from selenium.webdriver.common.by import By
|
||||||
|
from selenium.webdriver.remote.webdriver import WebDriver
|
||||||
|
from selenium.webdriver.remote.webelement import WebElement
|
||||||
|
from selenium.webdriver.support.ui import WebDriverWait
|
||||||
|
from selenium.webdriver.support import expected_conditions as EC
|
||||||
|
from selenium.common.exceptions import StaleElementReferenceException, TimeoutException
|
||||||
|
|
||||||
|
|
||||||
|
# Initialize EasyOCR once (module level)
|
||||||
|
# First run downloads models, then works offline
|
||||||
|
try:
|
||||||
|
import easyocr
|
||||||
|
_EASYOCR_READER = easyocr.Reader(['en'], gpu=False)
|
||||||
|
except ImportError:
|
||||||
|
_EASYOCR_READER = None
|
||||||
|
|
||||||
|
|
||||||
|
# Character confusion correction map for common OCR errors
|
||||||
|
CONFUSION_MAP = {
|
||||||
|
'0': 'O', 'O': '0', 'D': '0',
|
||||||
|
'1': 'I', 'I': '1', 'L': '1', 'l': '1',
|
||||||
|
'5': 'S', 'S': '5',
|
||||||
|
'8': 'B', 'B': '8',
|
||||||
|
'2': 'Z', 'Z': '2',
|
||||||
|
'6': 'G', 'G': '6',
|
||||||
|
'9': 'g', 'g': '9',
|
||||||
|
'Q': 'O',
|
||||||
|
'U': 'V', 'V': 'U',
|
||||||
|
'n': 'h', 'h': 'n',
|
||||||
|
'r': 'f', 'f': 'r',
|
||||||
|
'c': 'e', 'e': 'c',
|
||||||
|
'o': '0', '0': 'o',
|
||||||
|
's': '5', '5': 's',
|
||||||
|
'z': '2', '2': 'z',
|
||||||
|
'i': '1', '1': 'i',
|
||||||
|
'l': '1', '1': 'l',
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
def find_visible_captcha_input(driver: WebDriver) -> WebElement:
|
||||||
|
"""Find the visible captcha input element (name='captcha')."""
|
||||||
|
wait = WebDriverWait(driver, 15)
|
||||||
|
inputs = wait.until(EC.presence_of_all_elements_located((By.NAME, "captcha")))
|
||||||
|
for inp in inputs:
|
||||||
|
if inp.is_displayed():
|
||||||
|
return inp
|
||||||
|
raise RuntimeError("No visible captcha input found")
|
||||||
|
|
||||||
|
|
||||||
|
def extract_captcha_base64(captcha_input: WebElement) -> bytes:
|
||||||
|
"""Extract base64 image data from input's style attribute. Returns raw JPEG bytes."""
|
||||||
|
style = captcha_input.get_attribute("style") or ""
|
||||||
|
match = re.search(r'background:\s*url\(["\']?data:image/[^;]+;base64,([^"\')]+)["\']?\)', style)
|
||||||
|
if not match:
|
||||||
|
raise RuntimeError(f"No base64 background-image found in style: {style[:200]}")
|
||||||
|
|
||||||
|
b64_data = match.group(1)
|
||||||
|
if not b64_data:
|
||||||
|
raise RuntimeError("Base64 data is empty (captcha not loaded yet)")
|
||||||
|
|
||||||
|
b64_data += "=" * ((4 - len(b64_data) % 4) % 4) # Fix padding
|
||||||
|
return base64.b64decode(b64_data)
|
||||||
|
|
||||||
|
|
||||||
|
def wait_for_captcha_load(driver: WebDriver, captcha_input: WebElement, timeout: int = 15) -> bool:
|
||||||
|
"""Wait for captcha base64 to be populated with actual image data."""
|
||||||
|
end_time = time.time() + timeout
|
||||||
|
while time.time() < end_time:
|
||||||
|
try:
|
||||||
|
style = captcha_input.get_attribute("style") or ""
|
||||||
|
match = re.search(r'background:\s*url\(["\']?data:image/[^;]+;base64,([^"\')]+)["\']?\)', style)
|
||||||
|
if match and match.group(1):
|
||||||
|
return True
|
||||||
|
except StaleElementReferenceException:
|
||||||
|
captcha_input = find_visible_captcha_input(driver)
|
||||||
|
time.sleep(0.3)
|
||||||
|
return False
|
||||||
|
|
||||||
|
|
||||||
|
def preprocess_captcha_image(image_bytes: bytes) -> Image.Image:
|
||||||
|
"""Preprocess image for better OCR accuracy - gentle preprocessing."""
|
||||||
|
nparr = np.frombuffer(image_bytes, np.uint8)
|
||||||
|
img = cv2.imdecode(nparr, cv2.IMREAD_COLOR) # Load as color (BGR)
|
||||||
|
|
||||||
|
if img is None:
|
||||||
|
raise RuntimeError("Failed to decode image bytes")
|
||||||
|
|
||||||
|
# Convert BGR to RGB
|
||||||
|
img = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)
|
||||||
|
|
||||||
|
# Convert to grayscale
|
||||||
|
gray = cv2.cvtColor(img, cv2.COLOR_RGB2GRAY)
|
||||||
|
|
||||||
|
# Simple threshold - keep it simple
|
||||||
|
_, gray = cv2.threshold(gray, 128, 255, cv2.THRESH_BINARY)
|
||||||
|
|
||||||
|
# Light denoise
|
||||||
|
gray = cv2.fastNlMeansDenoising(gray, h=3)
|
||||||
|
|
||||||
|
# Very light morphological close - only connect broken parts
|
||||||
|
kernel = np.ones((1, 1), np.uint8)
|
||||||
|
gray = cv2.morphologyEx(gray, cv2.MORPH_CLOSE, kernel)
|
||||||
|
|
||||||
|
# Resize 6x for better recognition (larger for 6-char captcha)
|
||||||
|
gray = cv2.resize(gray, None, fx=6, fy=6, interpolation=cv2.INTER_CUBIC)
|
||||||
|
|
||||||
|
# Convert back to RGB (3-channel) for EasyOCR
|
||||||
|
rgb = cv2.cvtColor(gray, cv2.COLOR_GRAY2RGB)
|
||||||
|
|
||||||
|
return Image.fromarray(rgb)
|
||||||
|
|
||||||
|
|
||||||
|
def apply_confusion_corrections(text: str) -> str:
|
||||||
|
"""Apply character confusion corrections based on common OCR errors."""
|
||||||
|
corrected = []
|
||||||
|
for ch in text:
|
||||||
|
corrected.append(CONFUSION_MAP.get(ch, ch))
|
||||||
|
return ''.join(corrected)
|
||||||
|
|
||||||
|
|
||||||
|
def ocr_captcha(image: Image.Image) -> str:
|
||||||
|
"""Run OCR on preprocessed captcha image using EasyOCR."""
|
||||||
|
if _EASYOCR_READER is None:
|
||||||
|
raise RuntimeError("EasyOCR not installed. Run: pip install easyocr")
|
||||||
|
|
||||||
|
img_array = np.array(image)
|
||||||
|
|
||||||
|
# EasyOCR with alphanumeric allowlist for 6-char captcha
|
||||||
|
results = _EASYOCR_READER.readtext(
|
||||||
|
img_array,
|
||||||
|
detail=0,
|
||||||
|
allowlist='ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789',
|
||||||
|
width_ths=0.5,
|
||||||
|
paragraph=False
|
||||||
|
)
|
||||||
|
|
||||||
|
# Combine all detected text
|
||||||
|
result = ''.join(results)
|
||||||
|
|
||||||
|
# Clean result - keep only alphanumeric, max 6 chars
|
||||||
|
cleaned = ''.join(filter(str.isalnum, result))[:6]
|
||||||
|
|
||||||
|
print(f" DEBUG OCR: result='{result}', cleaned='{cleaned}', len={len(cleaned)}")
|
||||||
|
return cleaned
|
||||||
|
|
||||||
|
|
||||||
|
def find_refresh_button(driver: WebDriver, captcha_input: WebElement) -> Optional[WebElement]:
|
||||||
|
"""Find captcha refresh button near the captcha input."""
|
||||||
|
parent = captcha_input.find_element(By.XPATH, "..")
|
||||||
|
for sibling in parent.find_elements(By.XPATH, "*"):
|
||||||
|
onclick = (sibling.get_attribute("onclick") or "").lower()
|
||||||
|
class_attr = (sibling.get_attribute("class") or "").lower()
|
||||||
|
id_attr = (sibling.get_attribute("id") or "").lower()
|
||||||
|
if any(kw in onclick for kw in ["refresh", "captcha", "reload", "newcaptcha"]):
|
||||||
|
return sibling
|
||||||
|
if any(kw in class_attr for kw in ["refresh", "captcha-refresh", "reload"]):
|
||||||
|
return sibling
|
||||||
|
if any(kw in id_attr for kw in ["refresh", "captcha", "reload"]):
|
||||||
|
return sibling
|
||||||
|
|
||||||
|
# Fallback: search globally for refresh-like elements
|
||||||
|
try:
|
||||||
|
candidates = driver.find_elements(By.XPATH, "//*[self::a or self::button or self::img or self::span][contains(translate(@onclick, 'ABCDEFGHIJKLMNOPQRSTUVWXYZ', 'abcdefghijklmnopqrstuvwxyz'), 'refresh') or contains(translate(@class, 'ABCDEFGHIJKLMNOPQRSTUVWXYZ', 'abcdefghijklmnopqrstuvwxyz'), 'refresh') or contains(translate(@title, 'ABCDEFGHIJKLMNOPQRSTUVWXYZ', 'abcdefghijklmnopqrstuvwxyz'), 'refresh')]")
|
||||||
|
for cand in candidates:
|
||||||
|
if cand.is_displayed():
|
||||||
|
return cand
|
||||||
|
except:
|
||||||
|
pass
|
||||||
|
return None
|
||||||
|
|
||||||
|
|
||||||
|
def wait_for_new_captcha(driver: WebDriver, captcha_input: WebElement, timeout: int = 10) -> bool:
|
||||||
|
"""Wait for captcha background-image to change (new challenge loaded)."""
|
||||||
|
old_style = captcha_input.get_attribute("style") or ""
|
||||||
|
end_time = time.time() + timeout
|
||||||
|
while time.time() < end_time:
|
||||||
|
try:
|
||||||
|
new_style = captcha_input.get_attribute("style") or ""
|
||||||
|
if new_style != old_style and "data:image" in new_style:
|
||||||
|
return True
|
||||||
|
except StaleElementReferenceException:
|
||||||
|
captcha_input = find_visible_captcha_input(driver)
|
||||||
|
old_style = captcha_input.get_attribute("style") or ""
|
||||||
|
time.sleep(0.3)
|
||||||
|
return False
|
||||||
|
|
||||||
|
|
||||||
|
def solve_and_fill_captcha(driver: WebDriver, max_retries: int = 5, attempt_callback=None) -> bool:
|
||||||
|
"""
|
||||||
|
Main entry point: solve captcha and fill input.
|
||||||
|
Returns True if filled successfully, False if max retries exceeded.
|
||||||
|
Each retry gets a fresh captcha via refresh button.
|
||||||
|
"""
|
||||||
|
for attempt in range(1, max_retries + 1):
|
||||||
|
captcha_text = ""
|
||||||
|
try:
|
||||||
|
captcha_input = find_visible_captcha_input(driver)
|
||||||
|
|
||||||
|
# Wait for captcha base64 to be populated (async load)
|
||||||
|
if not wait_for_captcha_load(driver, captcha_input):
|
||||||
|
print(f" [Attempt {attempt}] Captcha failed to load in time, retrying...")
|
||||||
|
if attempt < max_retries:
|
||||||
|
refresh_btn = find_refresh_button(driver, captcha_input)
|
||||||
|
if refresh_btn:
|
||||||
|
refresh_btn.click()
|
||||||
|
wait_for_new_captcha(driver, captcha_input)
|
||||||
|
if attempt_callback:
|
||||||
|
attempt_callback(attempt, "", "", "not_loaded")
|
||||||
|
continue
|
||||||
|
|
||||||
|
image_bytes = extract_captcha_base64(captcha_input)
|
||||||
|
pil_image = preprocess_captcha_image(image_bytes)
|
||||||
|
captcha_text = ocr_captcha(pil_image)
|
||||||
|
|
||||||
|
if not captcha_text or len(captcha_text) < 6:
|
||||||
|
print(f" [Attempt {attempt}] OCR weak result: '{captcha_text}' (len={len(captcha_text)}), retrying...")
|
||||||
|
if attempt < max_retries:
|
||||||
|
refresh_btn = find_refresh_button(driver, captcha_input)
|
||||||
|
if refresh_btn:
|
||||||
|
refresh_btn.click()
|
||||||
|
wait_for_new_captcha(driver, captcha_input)
|
||||||
|
if attempt_callback:
|
||||||
|
attempt_callback(attempt, captcha_text, "", "ocr_failed")
|
||||||
|
continue
|
||||||
|
|
||||||
|
# Apply confusion corrections to get best variant
|
||||||
|
corrected = apply_confusion_corrections(captcha_text)
|
||||||
|
|
||||||
|
# Generate case variants for case-sensitive captcha (mixed case + numbers)
|
||||||
|
variants = [
|
||||||
|
captcha_text, # as-is (from OCR)
|
||||||
|
corrected, # with confusion corrections
|
||||||
|
captcha_text.upper(), # all uppercase
|
||||||
|
captcha_text.capitalize(), # first letter uppercase
|
||||||
|
captcha_text.title(), # each word capitalized
|
||||||
|
captcha_text.swapcase(), # swap case
|
||||||
|
]
|
||||||
|
|
||||||
|
# Add variations with first char variations
|
||||||
|
if len(captcha_text) >= 1:
|
||||||
|
variants.append(captcha_text[0].upper() + captcha_text[1:])
|
||||||
|
variants.append(captcha_text[0].lower() + captcha_text[1:])
|
||||||
|
if len(captcha_text) >= 2:
|
||||||
|
variants.append(captcha_text[:2].upper() + captcha_text[2:])
|
||||||
|
variants.append(captcha_text[:2].lower() + captcha_text[2:])
|
||||||
|
if len(captcha_text) >= 3:
|
||||||
|
variants.append(captcha_text[:3].upper() + captcha_text[3:])
|
||||||
|
variants.append(captcha_text[:3].lower() + captcha_text[3:])
|
||||||
|
|
||||||
|
# Remove duplicates while preserving order, keep only 6-char variants
|
||||||
|
seen = set()
|
||||||
|
unique_variants = []
|
||||||
|
for v in variants:
|
||||||
|
if v not in seen and len(v) == 6:
|
||||||
|
seen.add(v)
|
||||||
|
unique_variants.append(v)
|
||||||
|
variants = unique_variants
|
||||||
|
|
||||||
|
if attempt_callback:
|
||||||
|
screenshot_path = attempt_callback(attempt, captcha_text, "", "before_entry") or ""
|
||||||
|
|
||||||
|
success = False
|
||||||
|
# Try ALL variants before giving up on this attempt
|
||||||
|
for variant in variants:
|
||||||
|
if len(variant) == 6:
|
||||||
|
print(f" [Attempt {attempt}] Trying variant: '{variant}'")
|
||||||
|
captcha_input.clear()
|
||||||
|
captcha_input.send_keys(variant)
|
||||||
|
success = True
|
||||||
|
if attempt_callback:
|
||||||
|
attempt_callback(attempt, captcha_text, variant, "filled")
|
||||||
|
break
|
||||||
|
|
||||||
|
if not success:
|
||||||
|
print(f" [Attempt {attempt}] No valid 6-char variant found for '{captcha_text}', retrying...")
|
||||||
|
if attempt < max_retries:
|
||||||
|
refresh_btn = find_refresh_button(driver, captcha_input)
|
||||||
|
if refresh_btn:
|
||||||
|
refresh_btn.click()
|
||||||
|
wait_for_new_captcha(driver, captcha_input)
|
||||||
|
continue
|
||||||
|
|
||||||
|
# Click submit
|
||||||
|
submit_btn = WebDriverWait(driver, 10).until(
|
||||||
|
EC.element_to_be_clickable((By.XPATH, "//button[@value='Login']"))
|
||||||
|
)
|
||||||
|
submit_btn.click()
|
||||||
|
|
||||||
|
# Wait a bit for response
|
||||||
|
time.sleep(2)
|
||||||
|
|
||||||
|
# Check for invalid captcha error using the exact xpath
|
||||||
|
try:
|
||||||
|
# First handle any alerts that might be open
|
||||||
|
try:
|
||||||
|
alert = driver.switch_to.alert
|
||||||
|
alert_text = alert.text
|
||||||
|
print(f" DEBUG: Alert detected after submit: {alert_text}")
|
||||||
|
if "logout" in alert_text.lower() or "session" in alert_text.lower() or "previous" in alert_text.lower():
|
||||||
|
alert.accept()
|
||||||
|
print(" DEBUG: Accepted session/logout alert")
|
||||||
|
time.sleep(2)
|
||||||
|
except:
|
||||||
|
pass
|
||||||
|
|
||||||
|
invalid_captcha_elem = WebDriverWait(driver, 3).until(
|
||||||
|
EC.visibility_of_element_located((
|
||||||
|
By.XPATH,
|
||||||
|
"//div[@id='error']//strong[contains(normalize-space(), 'Invalid Captcha')]",
|
||||||
|
))
|
||||||
|
)
|
||||||
|
print(f" [Attempt {attempt}] Invalid captcha detected: {invalid_captcha_elem.text}")
|
||||||
|
if attempt < max_retries:
|
||||||
|
refresh_btn = find_refresh_button(driver, captcha_input)
|
||||||
|
if refresh_btn:
|
||||||
|
refresh_btn.click()
|
||||||
|
wait_for_new_captcha(driver, captcha_input)
|
||||||
|
continue
|
||||||
|
except TimeoutException:
|
||||||
|
# No invalid captcha error - likely success
|
||||||
|
print(f" [Attempt {attempt}] No invalid captcha error detected, assuming success")
|
||||||
|
return True
|
||||||
|
except Exception as e:
|
||||||
|
# Handle other exceptions like NoSuchElementException
|
||||||
|
print(f" [Attempt {attempt}] Error checking invalid captcha: {type(e).__name__}: {e}")
|
||||||
|
if attempt < max_retries:
|
||||||
|
refresh_btn = find_refresh_button(driver, captcha_input)
|
||||||
|
if refresh_btn:
|
||||||
|
refresh_btn.click()
|
||||||
|
wait_for_new_captcha(driver, captcha_input)
|
||||||
|
continue
|
||||||
|
|
||||||
|
except Exception as e:
|
||||||
|
print(f" [Attempt {attempt}] Error: {e}")
|
||||||
|
if attempt < max_retries:
|
||||||
|
refresh_btn = find_refresh_button(driver, captcha_input)
|
||||||
|
if refresh_btn:
|
||||||
|
refresh_btn.click()
|
||||||
|
wait_for_new_captcha(driver, captcha_input)
|
||||||
|
|
||||||
|
return False
|
||||||
@@ -67,7 +67,10 @@ def get_setup_configs(mode, config=None):
|
|||||||
session = setup.get("run_session", True)
|
session = setup.get("run_session", True)
|
||||||
token = setup.get("run_token", True)
|
token = setup.get("run_token", True)
|
||||||
matches = {
|
matches = {
|
||||||
"combined": session and token or (name.startswith("UPC_") and token),
|
# The full setup run owns every enabled product that has work to do.
|
||||||
|
# Products without tokens (for example RA) still belong in this
|
||||||
|
# shared browser flow; the combined test skips only their token step.
|
||||||
|
"combined": session or token,
|
||||||
"session": session and not token,
|
"session": session and not token,
|
||||||
"token": token and not session,
|
"token": token and not session,
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -47,6 +47,30 @@ def pytest_configure(config):
|
|||||||
config.option.self_contained_html = True
|
config.option.self_contained_html = True
|
||||||
|
|
||||||
|
|
||||||
|
def pytest_collection_modifyitems(config, items):
|
||||||
|
"""Keep the full setup run limited to the shared combined flow."""
|
||||||
|
invocation = " ".join(
|
||||||
|
str(arg).replace("\\", "/") for arg in config.invocation_params.args
|
||||||
|
).lower()
|
||||||
|
full_setup_run = (
|
||||||
|
"tests/setup" in invocation
|
||||||
|
and "tests/setup/sessions" not in invocation
|
||||||
|
and "tests/setup/tokens" not in invocation
|
||||||
|
and "test_session.py" not in invocation
|
||||||
|
and "test_trading_token.py" not in invocation
|
||||||
|
)
|
||||||
|
if not full_setup_run:
|
||||||
|
return
|
||||||
|
|
||||||
|
deselected = [
|
||||||
|
item for item in items
|
||||||
|
if "/tests/setup/sessions/" in f"/{item.nodeid.replace('\\', '/')}"
|
||||||
|
or "/tests/setup/tokens/" in f"/{item.nodeid.replace('\\', '/')}"
|
||||||
|
]
|
||||||
|
items[:] = [item for item in items if item not in deselected]
|
||||||
|
config.hook.pytest_deselected(items=deselected)
|
||||||
|
|
||||||
|
|
||||||
# -----------------------------
|
# -----------------------------
|
||||||
# WebDriver Fixture
|
# WebDriver Fixture
|
||||||
# -----------------------------
|
# -----------------------------
|
||||||
|
|||||||
+221
-37
@@ -57,43 +57,216 @@ def enter_pressed():
|
|||||||
return pressed
|
return pressed
|
||||||
|
|
||||||
|
|
||||||
def login(driver, username, password):
|
def check_captcha_error(driver) -> str:
|
||||||
|
"""Check if there's a CAPTCHA error message on the page."""
|
||||||
|
try:
|
||||||
|
# First handle any alerts
|
||||||
|
try:
|
||||||
|
alert = driver.switch_to.alert
|
||||||
|
alert_text = alert.text
|
||||||
|
print(f" DEBUG: Alert detected in check_captcha_error: {alert_text}")
|
||||||
|
if "logout" in alert_text.lower() or "session" in alert_text.lower() or "previous" in alert_text.lower():
|
||||||
|
alert.accept()
|
||||||
|
print(" DEBUG: Accepted session/logout alert")
|
||||||
|
time.sleep(2)
|
||||||
|
except:
|
||||||
|
pass
|
||||||
|
|
||||||
|
error_selectors = [
|
||||||
|
"//div[@id='error']//strong[normalize-space()='Invalid Captcha']",
|
||||||
|
"//div[@id='error']//span[contains(@class,'help-block')]//strong[contains(text(),'Invalid')]",
|
||||||
|
"//*[contains(@class,'error') and contains(text(),'Invalid Captcha')]",
|
||||||
|
]
|
||||||
|
for selector in error_selectors:
|
||||||
|
elements = driver.find_elements(By.XPATH, selector)
|
||||||
|
for el in elements:
|
||||||
|
if el.is_displayed() and el.text.strip():
|
||||||
|
return el.text.strip()
|
||||||
|
except Exception as e:
|
||||||
|
print(f" DEBUG: check_captcha_error exception: {e}")
|
||||||
|
return ""
|
||||||
|
|
||||||
|
|
||||||
|
def check_login_success(driver) -> bool:
|
||||||
|
"""Check if login was successful using multiple indicators."""
|
||||||
|
try:
|
||||||
|
# Handle any pending alerts first
|
||||||
|
try:
|
||||||
|
alert = driver.switch_to.alert
|
||||||
|
alert_text = alert.text
|
||||||
|
print(f" DEBUG: Alert detected: {alert_text}")
|
||||||
|
if "logout" in alert_text.lower() or "session" in alert_text.lower() or "previous" in alert_text.lower():
|
||||||
|
alert.accept()
|
||||||
|
print(" DEBUG: Accepted session/logout alert")
|
||||||
|
time.sleep(2)
|
||||||
|
except:
|
||||||
|
pass
|
||||||
|
|
||||||
|
# Wait a bit for page to load after submit
|
||||||
|
time.sleep(3)
|
||||||
|
|
||||||
|
# Print current URL and page title for debugging
|
||||||
|
print(f" DEBUG: Current URL: {driver.current_url}")
|
||||||
|
print(f" DEBUG: Page title: {driver.title}")
|
||||||
|
|
||||||
|
# Multiple dashboard indicators
|
||||||
|
dashboard_indicators = [
|
||||||
|
"//span[normalize-space()='TRADING MAINTENANCE']",
|
||||||
|
"//a[contains(@href, 'tradingMaintenance')]",
|
||||||
|
"//span[contains(text(), 'Dashboard')]",
|
||||||
|
"//h1[contains(text(), 'Dashboard')]",
|
||||||
|
"//div[contains(@class, 'dashboard')]",
|
||||||
|
"//a[contains(text(), 'Trading')]",
|
||||||
|
"//div[contains(@class, 'navbar')]//span[contains(text(), 'Trading')]",
|
||||||
|
]
|
||||||
|
|
||||||
|
wait = WebDriverWait(driver, 15)
|
||||||
|
for indicator in dashboard_indicators:
|
||||||
|
try:
|
||||||
|
wait.until(EC.visibility_of_element_located((By.XPATH, indicator)))
|
||||||
|
print(f" DEBUG: Found dashboard indicator: {indicator}")
|
||||||
|
return True
|
||||||
|
except:
|
||||||
|
continue
|
||||||
|
|
||||||
|
# URL-based check
|
||||||
|
if "home" in driver.current_url.lower() or "dashboard" in driver.current_url.lower():
|
||||||
|
print(" DEBUG: Login success detected via URL")
|
||||||
|
return True
|
||||||
|
|
||||||
|
return False
|
||||||
|
except Exception as e:
|
||||||
|
print(f" DEBUG: check_login_success failed: {e}")
|
||||||
|
return False
|
||||||
|
|
||||||
|
|
||||||
|
def login(
|
||||||
|
driver,
|
||||||
|
username,
|
||||||
|
password,
|
||||||
|
max_captcha_retries=5,
|
||||||
|
captcha_metrics=None,
|
||||||
|
max_login_duration_seconds=1200,
|
||||||
|
):
|
||||||
|
|
||||||
wait = WebDriverWait(driver, 20)
|
wait = WebDriverWait(driver, 20)
|
||||||
|
|
||||||
# username
|
# --- AUTOMATED CAPTCHA SOLVING WITH RETRY ON FAILURE ---
|
||||||
userid_field = wait.until(
|
from tests.captcha_solver import solve_and_fill_captcha, find_refresh_button, wait_for_new_captcha
|
||||||
EC.visibility_of_element_located((By.XPATH, Locator.userid))
|
|
||||||
)
|
login_deadline = time.monotonic() + max_login_duration_seconds
|
||||||
userid_field.send_keys(username)
|
captcha_attempt = 0
|
||||||
|
while True:
|
||||||
|
captcha_attempt += 1
|
||||||
|
|
||||||
# password
|
# Refreshing after a failed CAPTCHA clears the login form, so refill
|
||||||
password_field = driver.find_element(By.XPATH, Locator.password)
|
# both credentials before every automatic or manual retry.
|
||||||
password_field.send_keys(password)
|
userid_field = wait.until(
|
||||||
|
EC.visibility_of_element_located((By.XPATH, Locator.userid))
|
||||||
|
)
|
||||||
|
userid_field.clear()
|
||||||
|
userid_field.send_keys(username)
|
||||||
|
|
||||||
print(f"Solve captcha within {LOGIN_CAPTCHA_WAIT_SECONDS} seconds...")
|
password_field = wait.until(
|
||||||
|
EC.visibility_of_element_located((By.XPATH, Locator.password))
|
||||||
|
)
|
||||||
|
password_field.clear()
|
||||||
|
password_field.send_keys(password)
|
||||||
|
|
||||||
# wait for manual captcha solve
|
if captcha_metrics is not None:
|
||||||
time.sleep(LOGIN_CAPTCHA_WAIT_SECONDS)
|
captcha_metrics["tries"] = captcha_attempt
|
||||||
|
captcha_metrics["max_retries"] = max_captcha_retries
|
||||||
|
print(f"Solving CAPTCHA (attempt {captcha_attempt})...")
|
||||||
|
|
||||||
# click submit
|
# Once automatic solving has run for the configured duration, ask for
|
||||||
submit_btn = wait.until(
|
# the CAPTCHA value in a browser prompt and continue the same login flow.
|
||||||
EC.element_to_be_clickable((By.XPATH, Locator.submitbtn))
|
if time.monotonic() >= login_deadline:
|
||||||
)
|
print("Automatic CAPTCHA retries timed out; waiting for manual CAPTCHA entry.")
|
||||||
submit_btn.click()
|
manual_value = driver.execute_script(
|
||||||
|
"return window.prompt('Automatic CAPTCHA solving timed out. Enter the CAPTCHA shown on the page:');"
|
||||||
|
)
|
||||||
|
if not manual_value:
|
||||||
|
raise RuntimeError("Manual CAPTCHA entry was cancelled")
|
||||||
|
|
||||||
time.sleep(LOGIN_POST_SUBMIT_WAIT_SECONDS)
|
captcha_input = wait.until(
|
||||||
|
EC.visibility_of_element_located((By.NAME, "captcha"))
|
||||||
|
)
|
||||||
|
captcha_input.clear()
|
||||||
|
captcha_input.send_keys(manual_value.strip())
|
||||||
|
wait.until(EC.element_to_be_clickable((By.XPATH, Locator.submitbtn))).click()
|
||||||
|
time.sleep(LOGIN_POST_SUBMIT_WAIT_SECONDS)
|
||||||
|
accept_browser_alert_if_present(driver)
|
||||||
|
handle_transient_popups(driver)
|
||||||
|
|
||||||
accept_browser_alert_if_present(driver)
|
captcha_error = check_captcha_error(driver)
|
||||||
|
if captcha_error:
|
||||||
|
print(f" Manual CAPTCHA rejected: {captcha_error}")
|
||||||
|
continue
|
||||||
|
|
||||||
# wait for dashboard
|
if check_login_success(driver):
|
||||||
wait.until(
|
print("Login successful after manual CAPTCHA entry")
|
||||||
EC.visibility_of_element_located((By.XPATH, Locator.dashboard_marker))
|
if captcha_metrics is not None:
|
||||||
)
|
captcha_metrics["correct_attempt"] = captcha_attempt
|
||||||
|
captcha_metrics.setdefault("captcha_attempts", []).append({
|
||||||
close_tac_expiry_modal_if_present(driver)
|
"attempt": captcha_attempt,
|
||||||
|
"captcha_text": "manual",
|
||||||
print("Login successful")
|
"variant": manual_value.strip(),
|
||||||
|
"status": "manual",
|
||||||
|
"screenshot": "",
|
||||||
|
"error": "",
|
||||||
|
})
|
||||||
|
close_tac_expiry_modal_if_present(driver)
|
||||||
|
return True
|
||||||
|
continue
|
||||||
|
|
||||||
|
def attempt_callback(attempt_num, captcha_text, variant, status, screenshot_path="", error=""):
|
||||||
|
if captcha_metrics is not None:
|
||||||
|
captcha_metrics.setdefault("captcha_attempts", []).append({
|
||||||
|
"attempt": attempt_num,
|
||||||
|
"captcha_text": captcha_text,
|
||||||
|
"variant": variant,
|
||||||
|
"status": status,
|
||||||
|
"screenshot": screenshot_path,
|
||||||
|
"error": error
|
||||||
|
})
|
||||||
|
|
||||||
|
if not solve_and_fill_captcha(driver, max_retries=1, attempt_callback=attempt_callback):
|
||||||
|
print(" Automatic CAPTCHA solve failed; retrying until the login deadline.")
|
||||||
|
driver.refresh()
|
||||||
|
time.sleep(3)
|
||||||
|
continue
|
||||||
|
|
||||||
|
# solve_and_fill_captcha already clicks submit, so just wait and check result
|
||||||
|
time.sleep(LOGIN_POST_SUBMIT_WAIT_SECONDS)
|
||||||
|
|
||||||
|
accept_browser_alert_if_present(driver)
|
||||||
|
|
||||||
|
# Handle any popups that appear
|
||||||
|
handle_transient_popups(driver)
|
||||||
|
|
||||||
|
# Check for CAPTCHA error (invalid captcha message)
|
||||||
|
captcha_error = check_captcha_error(driver)
|
||||||
|
if captcha_error:
|
||||||
|
print(f" CAPTCHA validation failed: {captcha_error}")
|
||||||
|
# Reload the page to get a fresh CAPTCHA and keep retrying until
|
||||||
|
# the overall login deadline is reached.
|
||||||
|
print(f" Reloading page to get fresh CAPTCHA (attempt {captcha_attempt + 1})...")
|
||||||
|
driver.refresh()
|
||||||
|
time.sleep(3)
|
||||||
|
continue
|
||||||
|
|
||||||
|
# Check for successful login - use improved detection
|
||||||
|
if check_login_success(driver):
|
||||||
|
print("Login successful")
|
||||||
|
if captcha_metrics is not None:
|
||||||
|
captcha_metrics["correct_attempt"] = captcha_attempt
|
||||||
|
close_tac_expiry_modal_if_present(driver)
|
||||||
|
return True
|
||||||
|
|
||||||
|
# If we get here, something else is wrong
|
||||||
|
print(" Login not confirmed, retrying...")
|
||||||
|
driver.refresh()
|
||||||
|
time.sleep(3)
|
||||||
|
|
||||||
|
|
||||||
def accept_browser_alert_if_present(driver, timeout=5):
|
def accept_browser_alert_if_present(driver, timeout=5):
|
||||||
@@ -309,23 +482,34 @@ def open_trading_maintenance_page(driver, menu_xpath):
|
|||||||
|
|
||||||
wait = WebDriverWait(driver, 20)
|
wait = WebDriverWait(driver, 20)
|
||||||
|
|
||||||
trading_maintenance = wait.until(
|
def visible_elements(xpath):
|
||||||
EC.presence_of_element_located((By.XPATH, Locator.tradingmaintenance))
|
return [element for element in driver.find_elements(By.XPATH, xpath) if element.is_displayed()]
|
||||||
)
|
|
||||||
driver.execute_script("arguments[0].scrollIntoView({block: 'center'});", trading_maintenance)
|
|
||||||
|
|
||||||
submenu = driver.find_elements(By.XPATH, menu_xpath)
|
submenu_visible = any(
|
||||||
submenu_visible = any(item.is_displayed() for item in submenu)
|
item.is_enabled() for item in visible_elements(menu_xpath)
|
||||||
|
)
|
||||||
|
|
||||||
if not submenu_visible:
|
if not submenu_visible:
|
||||||
|
maintenance_items = visible_elements(Locator.tradingmaintenance)
|
||||||
|
if not maintenance_items:
|
||||||
|
wait.until(lambda current_driver: visible_elements(Locator.tradingmaintenance))
|
||||||
|
maintenance_items = visible_elements(Locator.tradingmaintenance)
|
||||||
|
|
||||||
|
maintenance = maintenance_items[0]
|
||||||
|
driver.execute_script(
|
||||||
|
"arguments[0].scrollIntoView({block: 'center'});", maintenance
|
||||||
|
)
|
||||||
try:
|
try:
|
||||||
wait.until(EC.element_to_be_clickable((By.XPATH, Locator.tradingmaintenance)))
|
wait.until(lambda current_driver: maintenance.is_enabled())
|
||||||
trading_maintenance.click()
|
maintenance.click()
|
||||||
except Exception:
|
except Exception:
|
||||||
driver.execute_script("arguments[0].click();", trading_maintenance)
|
driver.execute_script("arguments[0].click();", maintenance)
|
||||||
|
|
||||||
menu_item = wait.until(
|
menu_item = wait.until(
|
||||||
EC.element_to_be_clickable((By.XPATH, menu_xpath))
|
lambda current_driver: next(
|
||||||
|
(item for item in visible_elements(menu_xpath) if item.is_enabled()),
|
||||||
|
False,
|
||||||
|
)
|
||||||
)
|
)
|
||||||
try:
|
try:
|
||||||
menu_item.click()
|
menu_item.click()
|
||||||
|
|||||||
+15
-2
@@ -4,11 +4,13 @@ class Locator(object):
|
|||||||
userid="//input[@placeholder='User ID']"
|
userid="//input[@placeholder='User ID']"
|
||||||
password="//input[@placeholder='Password']"
|
password="//input[@placeholder='Password']"
|
||||||
submitbtn="//button[@value='Login']"
|
submitbtn="//button[@value='Login']"
|
||||||
dashboard_marker="//span[normalize-space()='TRADING MAINTENANCE']"
|
dashboard_marker="//*[contains(normalize-space(), 'TRADING MAINTENANCE')]"
|
||||||
tacexpiryclosebtn="//div[contains(@class,'modal') and contains(@style,'display: block')]//button[contains(@class,'close')]"
|
tacexpiryclosebtn="//div[contains(@class,'modal') and contains(@style,'display: block')]//button[contains(@class,'close')]"
|
||||||
usermenutoggle="//li[contains(@class,'dropdown') and contains(@class,'navbar-user')]//a[contains(@class,'dropdown-toggle') and @data-toggle='dropdown']"
|
usermenutoggle="//li[contains(@class,'dropdown') and contains(@class,'navbar-user')]//a[contains(@class,'dropdown-toggle') and @data-toggle='dropdown']"
|
||||||
logoutlink="//li[contains(@class,'dropdown') and contains(@class,'navbar-user')]//ul[contains(@class,'dropdown-menu')]//a[@ng-click='logout()' and normalize-space()='Log Out']"
|
logoutlink="//li[contains(@class,'dropdown') and contains(@class,'navbar-user')]//ul[contains(@class,'dropdown-menu')]//a[@ng-click='logout()' and normalize-space()='Log Out']"
|
||||||
tradingmaintenance="//span[normalize-space()='TRADING MAINTENANCE']"
|
# Click the Angular menu anchor, not the nested label span. The anchor
|
||||||
|
# owns openSubMenu(item), which is required to render its child links.
|
||||||
|
tradingmaintenance="//a[.//span[normalize-space()='TRADING MAINTENANCE']]"
|
||||||
tradingsession="//a[normalize-space()='Trading Session Master']"
|
tradingsession="//a[normalize-space()='Trading Session Master']"
|
||||||
tokengeneration="//a[@href='#/tradingMaintenance/generateToken' or normalize-space()='Token Generation']"
|
tokengeneration="//a[@href='#/tradingMaintenance/generateToken' or normalize-space()='Token Generation']"
|
||||||
tokengenerator = "//a[@href='#/tradingMaintenance/tokenGenerator']"
|
tokengenerator = "//a[@href='#/tradingMaintenance/tokenGenerator']"
|
||||||
@@ -36,6 +38,17 @@ class Locator(object):
|
|||||||
tokendeliverystartdate = "//input[@placeholder='DELIVERY_START_DATE']"
|
tokendeliverystartdate = "//input[@placeholder='DELIVERY_START_DATE']"
|
||||||
tokendeliveryenddate = "//input[@placeholder='DELIVERY_END_DATE']"
|
tokendeliveryenddate = "//input[@placeholder='DELIVERY_END_DATE']"
|
||||||
tokenexpirydate = "//input[@placeholder='Token Expiry Date']"
|
tokenexpirydate = "//input[@placeholder='Token Expiry Date']"
|
||||||
|
tokentype = "//input[@name='TYPE' or @placeholder='TYPE']"
|
||||||
|
intradaydeliveryfromdate = "//input[@placeholder='Delivery From Date']"
|
||||||
|
intradaydeliverytodate = "//input[@placeholder='Delivery To Date']"
|
||||||
|
intradaytradingstarttime = "//input[@placeholder='Trading Start Time']"
|
||||||
|
intradaydaysbeforedelivery = "//input[@placeholder='Trading Start days before delivery']"
|
||||||
|
intradaydurationbeforeend = "//input[@placeholder='Duration before delivery end time']"
|
||||||
|
intradaysuccessmessage = "//*[contains(normalize-space(text()), 'Tokens generated successfully')]"
|
||||||
|
dactradingfromdate = "//input[@placeholder='Trading From Date']"
|
||||||
|
dactradingtodate = "//input[@placeholder='Trading To Date']"
|
||||||
|
dacsessionstarttime = "//input[@placeholder='Session Start Time']"
|
||||||
|
dacsessionendtime = "//input[@placeholder='Session End Time']"
|
||||||
tokentradedate = "//input[@placeholder='Trade Date']"
|
tokentradedate = "//input[@placeholder='Trade Date']"
|
||||||
upcordernormalstart = "//input[@name='ses3StartTime']"
|
upcordernormalstart = "//input[@name='ses3StartTime']"
|
||||||
upcordernormalend = "//input[@name='ses3EndTime']"
|
upcordernormalend = "//input[@name='ses3EndTime']"
|
||||||
|
|||||||
@@ -13,13 +13,21 @@ from tests.locators import Locator
|
|||||||
def get_session_window(config):
|
def get_session_window(config):
|
||||||
|
|
||||||
session_config = config.get("setup", {}).get("session", {})
|
session_config = config.get("setup", {}).get("session", {})
|
||||||
start_offset_minutes = session_config.get("start_offset_minutes", 7)
|
time_mode = session_config.get("time_mode", "offset").lower()
|
||||||
|
if time_mode == "manual":
|
||||||
|
start_time = session_config.get("start_time")
|
||||||
|
if not start_time:
|
||||||
|
raise ValueError("Session time_mode 'manual' requires start_time")
|
||||||
|
elif time_mode == "offset":
|
||||||
|
start_offset_minutes = session_config.get("start_offset_minutes", 7)
|
||||||
|
start_time = (datetime.now() + timedelta(minutes=start_offset_minutes)).strftime("%H:%M")
|
||||||
|
else:
|
||||||
|
raise ValueError("Session time_mode must be 'manual' or 'offset'")
|
||||||
end_time = session_config.get("end_time", "20:30")
|
end_time = session_config.get("end_time", "20:30")
|
||||||
|
|
||||||
start_datetime = datetime.now() + timedelta(minutes=start_offset_minutes)
|
|
||||||
today = datetime.now().strftime("%d/%m/%Y")
|
today = datetime.now().strftime("%d/%m/%Y")
|
||||||
|
|
||||||
return start_datetime.strftime("%d/%m/%Y %H:%M"), f"{today} {end_time}"
|
return f"{today} {start_time}", f"{today} {end_time}"
|
||||||
|
|
||||||
|
|
||||||
def create_session(driver, step, config):
|
def create_session(driver, step, config):
|
||||||
@@ -44,7 +52,13 @@ def create_session(driver, step, config):
|
|||||||
|
|
||||||
@pytest.mark.session
|
@pytest.mark.session
|
||||||
@pytest.mark.parametrize("config", get_setup_configs("session"), ids=lambda c: c["product_name"])
|
@pytest.mark.parametrize("config", get_setup_configs("session"), ids=lambda c: c["product_name"])
|
||||||
def test_session(driver, test_context, config):
|
def test_session(driver, test_context, config, request):
|
||||||
|
|
||||||
|
# The full setup command runs session-only products (such as RA) through
|
||||||
|
# test_combined_setup.py so they share the same browser/login context.
|
||||||
|
invocation = " ".join(str(arg).replace("\\", "/") for arg in request.config.invocation_params.args)
|
||||||
|
if "tests/setup" in invocation and "tests/setup/sessions" not in invocation and "test_session.py" not in invocation:
|
||||||
|
pytest.skip("Handled by the shared full setup flow")
|
||||||
|
|
||||||
wait = WebDriverWait(driver, 20)
|
wait = WebDriverWait(driver, 20)
|
||||||
|
|
||||||
|
|||||||
@@ -31,8 +31,9 @@ def test_session_and_token_setup(driver, test_context, config):
|
|||||||
if config.get("setup", {}).get("run_session", True):
|
if config.get("setup", {}).get("run_session", True):
|
||||||
create_session(driver, step, config)
|
create_session(driver, step, config)
|
||||||
|
|
||||||
for product_code in get_product_codes(token):
|
if config.get("setup", {}).get("run_token", True):
|
||||||
product_token = dict(token)
|
for product_code in get_product_codes(token):
|
||||||
product_token["product_code"] = product_code
|
product_token = dict(token)
|
||||||
run_token_setup(driver, step, product_token)
|
product_token["product_code"] = product_code
|
||||||
|
run_token_setup(driver, step, product_token)
|
||||||
|
|
||||||
|
|||||||
@@ -3,6 +3,7 @@ from datetime import datetime, timedelta
|
|||||||
import pytest
|
import pytest
|
||||||
from selenium.webdriver.common.by import By
|
from selenium.webdriver.common.by import By
|
||||||
from selenium.webdriver.support.ui import WebDriverWait
|
from selenium.webdriver.support.ui import WebDriverWait
|
||||||
|
from selenium.webdriver.support import expected_conditions as EC
|
||||||
|
|
||||||
from tests.config_reader import get_setup_configs, parse_product_codes
|
from tests.config_reader import get_setup_configs, parse_product_codes
|
||||||
from tests.helpers import (
|
from tests.helpers import (
|
||||||
@@ -55,15 +56,28 @@ def get_delivery_date_and_expiry(token):
|
|||||||
return delivery_date, token_expiry
|
return delivery_date, token_expiry
|
||||||
|
|
||||||
|
|
||||||
|
def get_intraday_delivery_window(token, now=None):
|
||||||
|
"""Return an Intraday delivery window that cannot start in the past."""
|
||||||
|
now = datetime.now() if now is None else now
|
||||||
|
start = now + timedelta(minutes=token.get("delivery_start_offset_minutes", 5))
|
||||||
|
start_text = start.strftime("%d/%m/%Y %H:%M")
|
||||||
|
end_text = f"{start.strftime('%d/%m/%Y')} {token.get('delivery_to_time', '23:59')}"
|
||||||
|
return start_text, end_text
|
||||||
|
|
||||||
|
|
||||||
def get_time_session_times(token, row_text, default_start_time, default_end_time):
|
def get_time_session_times(token, row_text, default_start_time, default_end_time):
|
||||||
|
|
||||||
time_session_config = token.get("time_sessions", {}).get(row_text, {})
|
time_session_config = token.get("time_sessions", {}).get(row_text, {})
|
||||||
|
time_mode = time_session_config.get("time_mode")
|
||||||
start_time = time_session_config.get("start_time")
|
start_time = time_session_config.get("start_time")
|
||||||
|
|
||||||
if start_time is None:
|
if time_mode == "offset":
|
||||||
start_offset_minutes = time_session_config.get("start_offset_minutes")
|
start_offset_minutes = time_session_config.get("start_offset_minutes")
|
||||||
if start_offset_minutes is not None:
|
if start_offset_minutes is None:
|
||||||
start_time = (datetime.now() + timedelta(minutes=start_offset_minutes)).strftime("%H:%M")
|
raise ValueError(f"{row_text} offset mode requires start_offset_minutes")
|
||||||
|
start_time = (datetime.now() + timedelta(minutes=start_offset_minutes)).strftime("%H:%M")
|
||||||
|
elif time_mode == "manual" and not start_time:
|
||||||
|
raise ValueError(f"{row_text} manual mode requires start_time")
|
||||||
|
|
||||||
return (
|
return (
|
||||||
start_time or default_start_time,
|
start_time or default_start_time,
|
||||||
@@ -71,6 +85,29 @@ def get_time_session_times(token, row_text, default_start_time, default_end_time
|
|||||||
)
|
)
|
||||||
|
|
||||||
|
|
||||||
|
def get_upc_order_normal_start_time(timings, now=None):
|
||||||
|
"""Resolve UPC order-normal start time from manual or offset mode."""
|
||||||
|
mode = timings.get("time_mode", "manual").lower()
|
||||||
|
if mode == "offset":
|
||||||
|
offset = timings.get("order_normal_start_offset_minutes")
|
||||||
|
if offset is None:
|
||||||
|
raise ValueError(
|
||||||
|
"UPC time_mode 'offset' requires "
|
||||||
|
"order_normal_start_offset_minutes"
|
||||||
|
)
|
||||||
|
now = datetime.now() if now is None else now
|
||||||
|
return (now + timedelta(minutes=offset)).strftime("%H:%M")
|
||||||
|
if mode != "manual":
|
||||||
|
raise ValueError("UPC time_mode must be 'manual' or 'offset'")
|
||||||
|
|
||||||
|
start_time = timings.get("order_normal_start_time")
|
||||||
|
if not start_time:
|
||||||
|
raise ValueError(
|
||||||
|
"UPC time_mode 'manual' requires order_normal_start_time"
|
||||||
|
)
|
||||||
|
return start_time
|
||||||
|
|
||||||
|
|
||||||
def get_product_codes(token):
|
def get_product_codes(token):
|
||||||
|
|
||||||
product_codes = parse_product_codes(token.get("create_product_codes", token.get("product_codes")))
|
product_codes = parse_product_codes(token.get("create_product_codes", token.get("product_codes")))
|
||||||
@@ -133,11 +170,11 @@ def complete_time_session_edits(driver, step, token, next_day):
|
|||||||
edit_time_session_row(
|
edit_time_session_row(
|
||||||
driver,
|
driver,
|
||||||
step,
|
step,
|
||||||
"ORDER_NOR",
|
"ORDER_NORMAL",
|
||||||
order_nor_start_time,
|
order_nor_start_time,
|
||||||
order_nor_end_time
|
order_nor_end_time
|
||||||
)
|
)
|
||||||
edit_time_session_row(driver, step, "ORDER_EXCH", order_exch_start_time, order_exch_end_time)
|
edit_time_session_row(driver, step, "ORDER_EXCHANGE", order_exch_start_time, order_exch_end_time)
|
||||||
edit_time_session_row(driver, step, "TRADE", trade_start_time, trade_end_time)
|
edit_time_session_row(driver, step, "TRADE", trade_start_time, trade_end_time)
|
||||||
|
|
||||||
click_first_grid_button(driver, "Token Details")
|
click_first_grid_button(driver, "Token Details")
|
||||||
@@ -150,6 +187,7 @@ def complete_time_session_edits(driver, step, token, next_day):
|
|||||||
def complete_time_session_edits_from_saved_token(driver, step, token):
|
def complete_time_session_edits_from_saved_token(driver, step, token):
|
||||||
|
|
||||||
step.click((By.XPATH, Locator.tokentimesessionbtn), "open_time_session")
|
step.click((By.XPATH, Locator.tokentimesessionbtn), "open_time_session")
|
||||||
|
WebDriverWait(driver, 20).until(EC.url_contains("timeSessionView"))
|
||||||
click_first_grid_button(driver, "Time Session Details")
|
click_first_grid_button(driver, "Time Session Details")
|
||||||
step.capture("open_time_session_details")
|
step.capture("open_time_session_details")
|
||||||
step.capture("before_edit_order_normal")
|
step.capture("before_edit_order_normal")
|
||||||
@@ -166,11 +204,11 @@ def complete_time_session_edits_from_saved_token(driver, step, token):
|
|||||||
edit_time_session_row(
|
edit_time_session_row(
|
||||||
driver,
|
driver,
|
||||||
step,
|
step,
|
||||||
"ORDER_NOR",
|
"ORDER_NORMAL",
|
||||||
order_nor_start_time,
|
order_nor_start_time,
|
||||||
order_nor_end_time
|
order_nor_end_time
|
||||||
)
|
)
|
||||||
edit_time_session_row(driver, step, "ORDER_EXCH", order_exch_start_time, order_exch_end_time)
|
edit_time_session_row(driver, step, "ORDER_EXCHANGE", order_exch_start_time, order_exch_end_time)
|
||||||
edit_time_session_row(driver, step, "TRADE", trade_start_time, trade_end_time)
|
edit_time_session_row(driver, step, "TRADE", trade_start_time, trade_end_time)
|
||||||
|
|
||||||
click_first_grid_button(driver, "Token Details")
|
click_first_grid_button(driver, "Token Details")
|
||||||
@@ -201,10 +239,19 @@ def open_rtm_new_token(driver, step, token):
|
|||||||
def get_rtm_token_times(token, now=None):
|
def get_rtm_token_times(token, now=None):
|
||||||
now = datetime.now() if now is None else now
|
now = datetime.now() if now is None else now
|
||||||
delivery_date = (now + timedelta(days=token.get("delivery_offset_days", 0))).strftime("%d/%m/%Y")
|
delivery_date = (now + timedelta(days=token.get("delivery_offset_days", 0))).strftime("%d/%m/%Y")
|
||||||
|
time_mode = token.get("time_mode", "offset").lower()
|
||||||
|
if time_mode == "manual":
|
||||||
|
session_start_time = token.get("session_start_time")
|
||||||
|
if not session_start_time:
|
||||||
|
raise ValueError("RTM time_mode 'manual' requires session_start_time")
|
||||||
|
elif time_mode == "offset":
|
||||||
|
session_start_time = (now + timedelta(minutes=token.get("session_start_offset_minutes", 5))).strftime("%H:%M")
|
||||||
|
else:
|
||||||
|
raise ValueError("RTM time_mode must be 'manual' or 'offset'")
|
||||||
return (
|
return (
|
||||||
f"{delivery_date} {token.get('delivery_from_time', '00:00')}",
|
f"{delivery_date} {token.get('delivery_from_time', '00:00')}",
|
||||||
f"{delivery_date} {token.get('delivery_to_time', '23:59')}",
|
f"{delivery_date} {token.get('delivery_to_time', '23:59')}",
|
||||||
(now + timedelta(minutes=token.get("session_start_offset_minutes", 5))).strftime("%H:%M"),
|
session_start_time,
|
||||||
token.get("session_end_time", "20:00"),
|
token.get("session_end_time", "20:00"),
|
||||||
)
|
)
|
||||||
|
|
||||||
@@ -225,6 +272,76 @@ def create_rtm_token(driver, step, token):
|
|||||||
handle_transient_popups(driver)
|
handle_transient_popups(driver)
|
||||||
|
|
||||||
|
|
||||||
|
def create_intraday_token(driver, step, token):
|
||||||
|
"""Create an Intraday token through the Token Generator page."""
|
||||||
|
open_trading_maintenance_page(driver, Locator.tokengenerator)
|
||||||
|
handle_transient_popups(driver)
|
||||||
|
step.capture("open_token_generator_intraday")
|
||||||
|
|
||||||
|
for element_id, value, name in (
|
||||||
|
(Locator.generatormarketcode, token["market_code"], "market_code"),
|
||||||
|
(Locator.generatorproductcode, token["product_code"], "product_code"),
|
||||||
|
(Locator.generatorgenerationmode, token["generation_mode"], "generation_mode"),
|
||||||
|
):
|
||||||
|
select_bootstrap_dropdown_by_id(driver, element_id, value, exact=True)
|
||||||
|
step.capture(f"select_{name}_intraday")
|
||||||
|
|
||||||
|
step.click((By.XPATH, Locator.addnewtokenbtn), "click_add_new_token_intraday")
|
||||||
|
|
||||||
|
delivery_start, delivery_end = get_intraday_delivery_window(token)
|
||||||
|
values = (
|
||||||
|
(Locator.intradaydeliveryfromdate, delivery_start, "delivery_from_date"),
|
||||||
|
(Locator.intradaydeliverytodate, delivery_end, "delivery_to_date"),
|
||||||
|
(Locator.intradaytradingstarttime, token.get("trading_start_time", "00:20"), "trading_start_time"),
|
||||||
|
(Locator.intradaydaysbeforedelivery, str(token.get("trading_start_days_before_delivery", 1)), "days_before_delivery"),
|
||||||
|
)
|
||||||
|
for locator, value, name in values:
|
||||||
|
type_first_visible_input(driver, locator, value)
|
||||||
|
step.capture(f"enter_{name}_intraday")
|
||||||
|
|
||||||
|
step.click((By.XPATH, Locator.savetokenbtn), "save_token_intraday")
|
||||||
|
WebDriverWait(driver, 30).until(
|
||||||
|
EC.visibility_of_element_located((By.XPATH, Locator.intradaysuccessmessage))
|
||||||
|
)
|
||||||
|
step.capture("intraday_token_generated_successfully")
|
||||||
|
handle_transient_popups(driver)
|
||||||
|
|
||||||
|
|
||||||
|
def create_dac_token(driver, step, token):
|
||||||
|
"""Create a DAC token through the Token Generator page."""
|
||||||
|
open_trading_maintenance_page(driver, Locator.tokengenerator)
|
||||||
|
handle_transient_popups(driver)
|
||||||
|
step.capture("open_token_generator_dac")
|
||||||
|
|
||||||
|
for element_id, value, name in (
|
||||||
|
(Locator.generatormarketcode, token["market_code"], "market_code"),
|
||||||
|
(Locator.generatorproductcode, token["product_code"], "product_code"),
|
||||||
|
(Locator.generatorgenerationmode, token["generation_mode"], "generation_mode"),
|
||||||
|
):
|
||||||
|
select_bootstrap_dropdown_by_id(driver, element_id, value, exact=True)
|
||||||
|
step.capture(f"select_{name}_dac")
|
||||||
|
|
||||||
|
step.click((By.XPATH, Locator.addnewtokenbtn), "click_add_new_token_dac")
|
||||||
|
|
||||||
|
trading_start, trading_end = get_intraday_delivery_window(token)
|
||||||
|
values = (
|
||||||
|
(Locator.dactradingfromdate, trading_start, "trading_from_date"),
|
||||||
|
(Locator.dactradingtodate, trading_end, "trading_to_date"),
|
||||||
|
(Locator.dacsessionstarttime, token.get("session_start_time", "13:00"), "session_start_time"),
|
||||||
|
(Locator.dacsessionendtime, token.get("session_end_time", "23:00"), "session_end_time"),
|
||||||
|
)
|
||||||
|
for locator, value, name in values:
|
||||||
|
type_first_visible_input(driver, locator, value)
|
||||||
|
step.capture(f"enter_{name}_dac")
|
||||||
|
|
||||||
|
step.click((By.XPATH, Locator.savetokenbtn), "save_token_dac")
|
||||||
|
WebDriverWait(driver, 30).until(
|
||||||
|
EC.visibility_of_element_located((By.XPATH, Locator.intradaysuccessmessage))
|
||||||
|
)
|
||||||
|
step.capture("dac_token_generated_successfully")
|
||||||
|
handle_transient_popups(driver)
|
||||||
|
|
||||||
|
|
||||||
def create_upc_token(driver, step, token):
|
def create_upc_token(driver, step, token):
|
||||||
open_trading_maintenance_page(driver, Locator.tokengenerator)
|
open_trading_maintenance_page(driver, Locator.tokengenerator)
|
||||||
handle_transient_popups(driver)
|
handle_transient_popups(driver)
|
||||||
@@ -238,7 +355,7 @@ def create_upc_token(driver, step, token):
|
|||||||
step.click((By.XPATH, Locator.addnewtokenbtn), "click_add_new_token_upc")
|
step.click((By.XPATH, Locator.addnewtokenbtn), "click_add_new_token_upc")
|
||||||
timings = token.get("upc_times", {})
|
timings = token.get("upc_times", {})
|
||||||
for name, locator, value in (
|
for name, locator, value in (
|
||||||
("order_normal_start", Locator.upcorderstart, timings.get("order_normal_start_time")),
|
("order_normal_start", Locator.upcorderstart, get_upc_order_normal_start_time(timings)),
|
||||||
("order_normal_end", Locator.upcorderend, timings.get("order_normal_end_time")),
|
("order_normal_end", Locator.upcorderend, timings.get("order_normal_end_time")),
|
||||||
("trade_start", Locator.upctradestart, timings.get("trade_session_start_time")),
|
("trade_start", Locator.upctradestart, timings.get("trade_session_start_time")),
|
||||||
("trade_end", Locator.upctradeend, timings.get("trade_session_end_time")),
|
("trade_end", Locator.upctradeend, timings.get("trade_session_end_time")),
|
||||||
@@ -252,6 +369,12 @@ def create_upc_token(driver, step, token):
|
|||||||
|
|
||||||
|
|
||||||
def run_token_setup(driver, step, token):
|
def run_token_setup(driver, step, token):
|
||||||
|
if token.get("product_type", "").upper() == "DAC":
|
||||||
|
create_dac_token(driver, step, token)
|
||||||
|
return
|
||||||
|
if token.get("product_type", "").upper() == "INTRADAY":
|
||||||
|
create_intraday_token(driver, step, token)
|
||||||
|
return
|
||||||
if token.get("product_type", "").upper() == "UPC":
|
if token.get("product_type", "").upper() == "UPC":
|
||||||
create_upc_token(driver, step, token)
|
create_upc_token(driver, step, token)
|
||||||
return
|
return
|
||||||
@@ -289,16 +412,23 @@ def create_and_activate_token(driver, step, token):
|
|||||||
|
|
||||||
step.click((By.XPATH, Locator.addnewtokenbtn), f"click_add_new_token_{product_code.lower()}")
|
step.click((By.XPATH, Locator.addnewtokenbtn), f"click_add_new_token_{product_code.lower()}")
|
||||||
|
|
||||||
next_day, token_expiry = get_delivery_date_and_expiry(token)
|
is_intraday = token.get("product_type", "").upper() == "INTRADAY"
|
||||||
|
if is_intraday:
|
||||||
|
delivery_start, delivery_end = get_intraday_delivery_window(token)
|
||||||
|
else:
|
||||||
|
next_day, token_expiry = get_delivery_date_and_expiry(token)
|
||||||
|
|
||||||
|
if token.get("product_type", "").upper() == "REC":
|
||||||
|
token_type_prefix = token.get("type_prefix", "REC")
|
||||||
|
token_type = f"{token_type_prefix}-{datetime.now().strftime('%d%m%Y')}"
|
||||||
|
step.type((By.XPATH, Locator.tokentype), token_type, f"enter_type_{product_code.lower()}")
|
||||||
|
|
||||||
if token.get("product_type", "").upper() == "UPC":
|
if token.get("product_type", "").upper() == "UPC":
|
||||||
today = datetime.now().strftime("%d/%m/%Y")
|
today = datetime.now().strftime("%d/%m/%Y")
|
||||||
if token.get("generation_mode") == "AnyDayUPC":
|
if token.get("generation_mode") == "AnyDayUPC":
|
||||||
step.type((By.XPATH, Locator.tokentradedate), today, f"enter_trade_date_{product_code.lower()}")
|
step.type((By.XPATH, Locator.tokentradedate), today, f"enter_trade_date_{product_code.lower()}")
|
||||||
timings = token.get("upc_times", {})
|
timings = token.get("upc_times", {})
|
||||||
normal_start = timings.get("order_normal_start_time")
|
normal_start = get_upc_order_normal_start_time(timings)
|
||||||
if normal_start is None and timings.get("order_normal_start_offset_minutes") is not None:
|
|
||||||
normal_start = (datetime.now() + timedelta(minutes=timings["order_normal_start_offset_minutes"])).strftime("%H:%M")
|
|
||||||
values = {
|
values = {
|
||||||
Locator.upcordernormalstart: normal_start,
|
Locator.upcordernormalstart: normal_start,
|
||||||
Locator.upcordernormalend: timings.get("order_normal_end_time"),
|
Locator.upcordernormalend: timings.get("order_normal_end_time"),
|
||||||
@@ -310,9 +440,17 @@ def create_and_activate_token(driver, step, token):
|
|||||||
continue
|
continue
|
||||||
type_first_visible_input(driver, locator, value)
|
type_first_visible_input(driver, locator, value)
|
||||||
|
|
||||||
if token.get("generation_mode") not in ("WeeklyUPC", "MonthlyUPC"):
|
if is_intraday:
|
||||||
|
step.type((By.XPATH, Locator.tokendeliverystartdate), delivery_start, f"enter_delivery_start_date_{product_code.lower()}")
|
||||||
|
step.type((By.XPATH, Locator.tokendeliveryenddate), delivery_end, f"enter_delivery_end_date_{product_code.lower()}")
|
||||||
|
elif (
|
||||||
|
token.get("generation_mode") not in ("WeeklyUPC", "MonthlyUPC")
|
||||||
|
and token.get("product_type", "").upper() != "REC"
|
||||||
|
):
|
||||||
step.type((By.XPATH, Locator.tokendeliverystartdate), next_day, f"enter_delivery_start_date_{product_code.lower()}")
|
step.type((By.XPATH, Locator.tokendeliverystartdate), next_day, f"enter_delivery_start_date_{product_code.lower()}")
|
||||||
step.type((By.XPATH, Locator.tokendeliveryenddate), next_day, f"enter_delivery_end_date_{product_code.lower()}")
|
step.type((By.XPATH, Locator.tokendeliveryenddate), next_day, f"enter_delivery_end_date_{product_code.lower()}")
|
||||||
|
|
||||||
|
if not is_intraday and token.get("generation_mode") not in ("WeeklyUPC", "MonthlyUPC"):
|
||||||
step.type((By.XPATH, Locator.tokenexpirydate), token_expiry, f"enter_token_expiry_date_{product_code.lower()}")
|
step.type((By.XPATH, Locator.tokenexpirydate), token_expiry, f"enter_token_expiry_date_{product_code.lower()}")
|
||||||
|
|
||||||
step.click((By.XPATH, Locator.savetokenbtn), f"save_token_{product_code.lower()}")
|
step.click((By.XPATH, Locator.savetokenbtn), f"save_token_{product_code.lower()}")
|
||||||
@@ -322,12 +460,19 @@ def create_and_activate_token(driver, step, token):
|
|||||||
step.click((By.XPATH, Locator.tokengeneratorresetbtn), f"reset_token_generator_{product_code.lower()}")
|
step.click((By.XPATH, Locator.tokengeneratorresetbtn), f"reset_token_generator_{product_code.lower()}")
|
||||||
return
|
return
|
||||||
|
|
||||||
complete_time_session_edits_from_saved_token(driver, step, token)
|
if not is_intraday:
|
||||||
|
complete_time_session_edits_from_saved_token(driver, step, token)
|
||||||
|
|
||||||
|
|
||||||
@pytest.mark.trading_token
|
@pytest.mark.trading_token
|
||||||
@pytest.mark.parametrize("config", get_setup_configs("token"), ids=lambda c: c["product_name"])
|
@pytest.mark.parametrize("config", get_setup_configs("token"), ids=lambda c: c["product_name"])
|
||||||
def test_trading_token_creation(driver, test_context, config):
|
def test_trading_token_creation(driver, test_context, config, request):
|
||||||
|
|
||||||
|
# The full setup command runs token-enabled products through the combined
|
||||||
|
# flow so session and token creation use one browser/login context.
|
||||||
|
invocation = " ".join(str(arg).replace("\\", "/") for arg in request.config.invocation_params.args)
|
||||||
|
if "tests/setup" in invocation and "tests/setup/tokens" not in invocation and "test_trading_token.py" not in invocation:
|
||||||
|
pytest.skip("Handled by the shared full setup flow")
|
||||||
|
|
||||||
config, step, token = get_token_context(driver, test_context, config)
|
config, step, token = get_token_context(driver, test_context, config)
|
||||||
open_app_and_login(driver, step, config)
|
open_app_and_login(driver, step, config)
|
||||||
|
|||||||
@@ -0,0 +1,124 @@
|
|||||||
|
import csv
|
||||||
|
import os
|
||||||
|
import time
|
||||||
|
from datetime import datetime
|
||||||
|
|
||||||
|
import pytest
|
||||||
|
from selenium.webdriver.common.by import By
|
||||||
|
from selenium.webdriver.support.ui import WebDriverWait
|
||||||
|
from selenium.webdriver.support import expected_conditions as EC
|
||||||
|
|
||||||
|
from tests.browser import create_chrome_driver
|
||||||
|
from tests.config_reader import load_config
|
||||||
|
from tests.helpers import login, logout
|
||||||
|
from tests.locators import Locator
|
||||||
|
|
||||||
|
|
||||||
|
LOGIN_ATTEMPTS = 50
|
||||||
|
MAX_CAPTCHA_RETRIES = 7
|
||||||
|
|
||||||
|
|
||||||
|
@pytest.mark.captcha
|
||||||
|
def test_captcha_login_stress():
|
||||||
|
"""Run bounded EasyOCR login attempts and retain auditable CAPTCHA evidence."""
|
||||||
|
config = load_config()
|
||||||
|
output_dir = os.path.abspath(os.path.join("reports", "captcha_login_stress"))
|
||||||
|
os.makedirs(output_dir, exist_ok=True)
|
||||||
|
started = datetime.now().strftime("%Y%m%d_%H%M%S")
|
||||||
|
csv_path = os.path.join(output_dir, f"captcha_login_{started}.csv")
|
||||||
|
rows = []
|
||||||
|
passed = 0
|
||||||
|
failed = 0
|
||||||
|
|
||||||
|
def record(row):
|
||||||
|
rows.append(row)
|
||||||
|
|
||||||
|
for login_number in range(1, LOGIN_ATTEMPTS + 1):
|
||||||
|
login_started = datetime.now().isoformat(timespec="seconds")
|
||||||
|
captcha_rows = []
|
||||||
|
|
||||||
|
driver = create_chrome_driver()
|
||||||
|
try:
|
||||||
|
driver.get(config["url"])
|
||||||
|
|
||||||
|
login_status = "FAILED"
|
||||||
|
logout_status = "NOT_ATTEMPTED"
|
||||||
|
error_text = ""
|
||||||
|
captcha_metrics = {"captcha_attempts": []}
|
||||||
|
try:
|
||||||
|
login(
|
||||||
|
driver,
|
||||||
|
config["login"]["username"],
|
||||||
|
config["login"]["password"],
|
||||||
|
max_captcha_retries=MAX_CAPTCHA_RETRIES,
|
||||||
|
captcha_metrics=captcha_metrics,
|
||||||
|
max_login_duration_seconds=1200,
|
||||||
|
)
|
||||||
|
login_status = "PASSED"
|
||||||
|
passed += 1
|
||||||
|
except Exception as exc:
|
||||||
|
failed += 1
|
||||||
|
error_text = f"{type(exc).__name__}: {exc}"
|
||||||
|
try:
|
||||||
|
driver.save_screenshot(os.path.join(output_dir, f"login_{login_number:03d}_failure.png"))
|
||||||
|
except Exception:
|
||||||
|
pass
|
||||||
|
finally:
|
||||||
|
# Explicitly log out after a successful authenticated flow,
|
||||||
|
# then close the browser session.
|
||||||
|
if login_status == "PASSED":
|
||||||
|
try:
|
||||||
|
logout_status = "PASSED" if logout(driver) else "FAILED"
|
||||||
|
except Exception as exc:
|
||||||
|
logout_status = "FAILED"
|
||||||
|
print(f"Logout error: {type(exc).__name__}: {exc}")
|
||||||
|
|
||||||
|
try:
|
||||||
|
driver.quit()
|
||||||
|
except Exception:
|
||||||
|
pass
|
||||||
|
|
||||||
|
captcha_rows = captcha_metrics.get("captcha_attempts", []) or [{
|
||||||
|
"attempt": 0, "captcha_text": "", "variant": "",
|
||||||
|
"status": "no_captcha_record", "screenshot": "", "error": error_text,
|
||||||
|
}]
|
||||||
|
for item in captcha_rows:
|
||||||
|
record({
|
||||||
|
"login_attempt": login_number,
|
||||||
|
"started_at": login_started,
|
||||||
|
"login_status": login_status,
|
||||||
|
"logout_status": logout_status,
|
||||||
|
"captcha_attempt": item["attempt"],
|
||||||
|
"attempted_captcha": item["variant"],
|
||||||
|
"ocr_captcha": item["captcha_text"],
|
||||||
|
"screenshot_path": item["screenshot"],
|
||||||
|
"captcha_status": item["status"],
|
||||||
|
"captcha_entered_count": sum(
|
||||||
|
1 for captcha_row in captcha_rows if captcha_row["variant"]
|
||||||
|
),
|
||||||
|
"error": item["error"] or error_text,
|
||||||
|
"attempts_to_result": len(captcha_rows),
|
||||||
|
"passed_count": passed,
|
||||||
|
"failed_count": failed,
|
||||||
|
})
|
||||||
|
finally:
|
||||||
|
# Clean up the driver after all attempts
|
||||||
|
try:
|
||||||
|
driver.quit()
|
||||||
|
except Exception:
|
||||||
|
pass
|
||||||
|
|
||||||
|
fieldnames = [
|
||||||
|
"login_attempt", "started_at", "login_status", "captcha_attempt",
|
||||||
|
"logout_status",
|
||||||
|
"attempted_captcha", "ocr_captcha", "screenshot_path", "captcha_status",
|
||||||
|
"captcha_entered_count", "error", "attempts_to_result", "passed_count",
|
||||||
|
"failed_count",
|
||||||
|
]
|
||||||
|
with open(csv_path, "w", newline="", encoding="utf-8") as report:
|
||||||
|
writer = csv.DictWriter(report, fieldnames=fieldnames)
|
||||||
|
writer.writeheader()
|
||||||
|
writer.writerows(rows)
|
||||||
|
|
||||||
|
print(f"CAPTCHA stress report: {csv_path}")
|
||||||
|
print(f"Passed: {passed}; Failed: {failed}; Total: {LOGIN_ATTEMPTS}")
|
||||||
Reference in New Issue
Block a user