ɣh~0dZddlZddlmcmZddlZddlZddl Z ddl Z ddl Z ddl m Z ddlmZddlmZe j$ddZe j$dd Ze j$dd Ze j$dd Zd Ze j$dd Ze j$ddZe j$ddZe j$ddZe j$ddZe j$ddZe j$ddZdZe j$ddZ y)aH conftest.py Purpose: Pytest configuration and fixtures for the automated testing system. Provides isolated test environments with in-memory databases, temporary directories, and pre-configured test users and clients. Fixtures: - app: Flask application instance configured for testing - client: Flask test client for HTTP requests - db: In-memory SQLite database with test schema - temp_static: Temporary directory for image file testing - test_users: Pre-created test users with different permission levels - authenticated_clients: Pre-authenticated test clients N)Flask) create_app)get_dbfunction)scopec#Ki}gd}|D]$}tjj|||<&tj\}}tj }tj j|d}tj|d|tjd<|tjd<dtjd<d tjd <d tjd <d tjd<d tjd<t}d|jd<d|jd<|j5tddd||j5ddl m}t|dr&|j j#t%|ddddtj"|tj&|ddl} | j+|d|D]>}||||tj|<tjj-|d@y#1swYxYw#1swYxYww)z6Create and configure a Flask app instance for testing.) DATABASE_PATHUSER_PHOTOS_DIR_REAL SECRET_KEY ENABLE_HSTSz User-photosT)exist_okr r z test-secret-key-for-testing-onlyr 0r 10000LOGIN_MAX_FAILS1LOGIN_WINDOW_SECONDSLOGIN_LOCK_SECONDSTESTINGFWTF_CSRF_ENABLEDNr)g _database) ignore_errors)osenvirongettempfilemkstempmkdtemppathjoinmakedirsrconfig app_contextinit_test_databaseflaskrhasattrrclosedelattrunlinkshutilrmtreepop) original_envenv_keyskeydb_fddb_path temp_statictemp_user_photosapprr*s AC:\Users\algun\Documents\ceba web\Ceba - Github\tests\conftest.pyr4r4sLUHJJNN3/ S%%'NE7""$Kww||K?KK 40#*BJJ)9BJJ%&ABJJ| #BJJ}%,BJJ !),BJJ%&'*BJJ#$ ,C CJJy%*CJJ!"    I   1k " KK    A{ # HHUOIIg MM+TM2   (*3/BJJsO JJNN3 % '    s7EI H>I89I 1B I>II IIc"|jS)zCreate a Flask test client.) test_clientr4s r5clientr9Xs ?? c#nK|j5tdddy#1swYyxYww)z)Get database connection for the test app.N)r#rr8s r5dbr<^s$  h   s5 ) 525c@tjjdS)z$Get temporary static directory path.r )rrrr8s r5r2r2es ::>>0 11r:ctjjd}tj|}tj |_|jd|jd|jd|jd|jd|jd|jd|j|jy ) z6Initialize the test database with the required schema.r z CREATE TABLE IF NOT EXISTS users ( id INTEGER PRIMARY KEY AUTOINCREMENT, username TEXT UNIQUE NOT NULL, password_hash TEXT NOT NULL, is_admin BOOLEAN NOT NULL DEFAULT 0 ) a CREATE TABLE IF NOT EXISTS cameras ( id INTEGER PRIMARY KEY AUTOINCREMENT, user_id INTEGER NOT NULL, camera_id INTEGER NOT NULL, camera_name TEXT NOT NULL, file_paths TEXT, added_to_map INTEGER DEFAULT 0, location TEXT, FOREIGN KEY (user_id) REFERENCES users (id) ) a CREATE TABLE IF NOT EXISTS markers ( id INTEGER PRIMARY KEY AUTOINCREMENT, user_id INTEGER NOT NULL, type TEXT NOT NULL, camera_id TEXT, latitude REAL NOT NULL, longitude REAL NOT NULL, added_to_map INTEGER DEFAULT 0, name TEXT, FOREIGN KEY (user_id) REFERENCES users (id) ) a& CREATE TABLE IF NOT EXISTS login_attempts ( id INTEGER PRIMARY KEY AUTOINCREMENT, username TEXT NOT NULL, ip TEXT NOT NULL, fail_count INTEGER DEFAULT 0, last_failed_at INTEGER, locked_until INTEGER ) z CREATE TABLE IF NOT EXISTS login_ip_attempts ( ip TEXT PRIMARY KEY, fail_count INTEGER DEFAULT 0, last_failed_at INTEGER, locked_until INTEGER ) aX CREATE TABLE IF NOT EXISTS auth_log ( id INTEGER PRIMARY KEY AUTOINCREMENT, ts INTEGER NOT NULL, ip TEXT NOT NULL, user_id INTEGER, username TEXT, event TEXT NOT NULL, detail TEXT, FOREIGN KEY (user_id) REFERENCES users (id) ) a CREATE TABLE IF NOT EXISTS admin_audit ( id INTEGER PRIMARY KEY AUTOINCREMENT, ts INTEGER NOT NULL, ip TEXT NOT NULL, admin_user_id INTEGER NOT NULL, target_user_id INTEGER, action TEXT NOT NULL, detail TEXT, FOREIGN KEY (admin_user_id) REFERENCES users (id), FOREIGN KEY (target_user_id) REFERENCES users (id) ) N) rrrsqlite3connectRow row_factoryexecutecommitr')r1r<s r5r$r$ksjjnn_-G  !B[[BNJJ JJ   JJ   JJ   JJ JJ   JJ   IIKHHJr:c|j5tjjd}t j |}tj |_i}tjdjdtjjd}|jdd|df}|jdddd|d<tjd jdtjjd}|jdd |d f}|jd d d d|d <tjd jdtjjd}|jdd|df}|jdd dd|d<|j|j!|cdddS#1swYyxYw)z3Create test users with different permission levels.r testpass123zutf-8zFINSERT INTO users (username, password_hash, is_admin) VALUES (?, ?, ?)testuserF)idusernamepasswordis_adminregular adminpass123 testadminTadminlimitedpass123 limiteduserlimitedN)r#rrrr?r@rArBbcrypthashpwencodegensaltdecoderC lastrowidrDr')r4r1r<usersregular_password_hashcursoradmin_password_hashlimited_password_hashs r5 test_usersr^s  **..1 __W % !' m.B.B7.KV^^M] ^ e efm n T . 6  """%  i%mmN,A,A',JFNNL\]ddelm T -t 4  ""#&  g!' .>.E.Eg.NPVP^P^P` a h hip q T 15 9  ""%(  i   a   s F:GGcH|d}|jd|d|dd|S)z5Create an authenticated test client for regular user.rL/loginrIrJrIrJdatapostr9r^users r5authenticated_client_regularrh: i D KK$$ K Mr:cH|d}|jd|d|dd|S)z3Create an authenticated test client for admin user.rOr`rIrJrarbrdrfs r5authenticated_client_adminrks: g D KK$$ K Mr:cH|d}|jd|d|dd|S)z5Create an authenticated test client for limited user.rRr`rIrJrarbrdrfs r5authenticated_client_limitedrmrir:c0dddddddddddddd dd S) z'Provide sample camera data for testing. 123456789012z Test Camera 1) camera_id camera_name 987654321098z Test Camera 212345zInvalid Camera 12345678901a)validvalid_2invalid_short_idinvalid_non_numericinvalid_empty_namer{r:r5sample_camera_datar|'sM (* (* !+ (+ ( # r:cdddddddddS) z&Provide sample image data for testing.z%PICT_20240101_120000_123456789012.jpgz image/jpegsJFIFHH)filename content_typercz$IMG_20240101_130000_123456789012.pngz image/pngsPNG  IHDR)jpgpngr{r{r:r5sample_image_datarBs- @(] ?'Z  r:c.ddddddddd d d d d ddd dS)z'Provide sample marker data for testing.camerarog[?g͏F@g3@zTest Camera Marker)typerplatitude longitudenamefaunagDܜJF@gOY3@zWild Boar Sighting)rrrr structureg-y<-?F@g>e3@zFeeding Station) camera_marker fauna_markerstructure_markerr{r{r:r5sample_marker_datarSsH '!"(  !"(   !"%   r:cd|j5tjjd}t j |}tj |_|jd|jd|j|jdddy#1swYyxYw)z,Helper function to clear rate limiting data.r zDELETE FROM login_attemptszDELETE FROM login_ip_attemptsN) r#rrrr?r@rArBrCrDr')r4r1r<s r5clear_rate_limit_datarmsq  **..1 __W %  /0 23      s B B&&B/c#LKtfd}|tyw)z4Clear rate limiting data before and after each test.cty)Nrr8sr5 clear_nowz$clear_rate_limits..clear_nows c"r:Nr)r4rs` r5clear_rate_limitsr{s'## O#s!$)!__doc__builtins @py_builtins_pytest.assertion.rewrite assertionrewrite @pytest_arrrr?pytestrSr%rapp_modules.initializerrapp_modules.dbrfixturer4r9r<r2r$r^rhrkrmr|rrrrr{r:r5rs"  .!j!8&"8&vj!" j!" j!2"2 bJj!2"2jj!"j!"j!"j!"4j! "  j!"2 j! " r: