o @h~0@sndZddlZddlmmZddlZddlZddl Z ddl Z ddl Z ddl m Z ddlmZddlmZe jdddd Ze jddd d Ze jddd d Ze jddddZddZe jddddZe jddddZe jddddZe jddddZe jddddZe jddddZe jddddZd d!Ze jddd"d#Z dS)$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 csi}gd}|D] }tj|||<q t\}}t}tj|d}tj|dd|tjd<|tjd<dtjd<d tjd <d tjd <d tjd<d tjd<t }d|j d<d|j d<| t Wdn1snwY|V| ddl m}t|dr|jt|dWdn1swYt|t|ddl} | j|dd|D]}||dur||tj|<qtj|dqdS)z6Create and configure a Flask app instance for testing.) DATABASE_PATHUSER_PHOTOS_DIR_REAL SECRET_KEY ENABLE_HSTSz User-photosT)exist_okrrz 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(r3/var/www/html/tests/conftest.pyr2sN                   r2cCs|S)zCreate a Flask test client.) test_clientr2r3r3r4clientXsr7ccs8| tVWddS1swYdS)z)Get database connection for the test app.N)r!rr6r3r3r4db^s  "r8cCs tjdS)z$Get temporary static directory path.r)rrrr6r3r3r4r0es r0cCsxtjd}t|}tj|_|d|d|d|d|d|d|d|| d S) z6Initialize the test database with the required schema.rz 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%)r/r8r3r3r4r"ks       r"cCs(|tjd}t|}tj|_i}t d dt  d}| dd|df}|jdddd|d<t d dt  d}| dd |d f}|jd d d d|d <t d dt  d}| dd|df}|jdd dd|d<|||WdS1swYdS)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!rrrr9r:r;r<bcrypthashpwencodegensaltdecoder= lastrowidr>r%)r2r/r8usersregular_password_hashcursoradmin_password_hashlimited_password_hashr3r3r4 test_userssN      $rWcC(|d}|jd|d|ddd|S)z5Create an authenticated test client for regular user.rE/loginrBrCrBrCdatapostr7rWuserr3r3r4authenticated_client_regular  racCrX)z3Create an authenticated test client for admin user.rHrYrBrCrZr[r]r_r3r3r4authenticated_client_adminrbrccCrX)z5Create an authenticated test client for limited user.rKrYrBrCrZr[r]r_r3r3r4authenticated_client_limitedrbrdcCs.dddddddddddddd 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_namer3r3r3r3r4sample_camera_data's rqcCsdddddddddS) z&Provide sample image data for testing.z%PICT_20240101_120000_123456789012.jpgz image/jpegsJFIFHH)filename content_typer\z$IMG_20240101_130000_123456789012.pngz image/pngsPNG  IHDR)jpgpngr3r3r3r3r4sample_image_dataBsrvcCs,ddddddddd d d d d ddd dS)z'Provide sample marker data for testing.camerareg[?g͏F@g3@zTest Camera Marker)typerflatitude longitudenamefaunagDܜJF@gOY3@zWild Boar Sighting)rxryrzr{ structureg-y<-?F@g>e3@zFeeding Station) camera_marker fauna_markerstructure_markerr3r3r3r3r4sample_marker_dataSs"rcCsp|*tjd}t|}tj|_|d|d| | WddS1s1wYdS)z,Helper function to clear rate limiting data.rzDELETE FROM login_attemptszDELETE FROM login_ip_attemptsN) r!rrrr9r:r;r<r=r>r%)r2r/r8r3r3r4clear_rate_limit_datams      "rc#s(tfdd}|VtdS)z4Clear rate limiting data before and after each test.cs tdS)Nrr3r6r3r4 clear_nows z$clear_rate_limits..clear_nowNr)r2rr3r6r4clear_rate_limits{s   r)!__doc__builtins @py_builtins_pytest.assertion.rewrite assertionrewrite @pytest_arrrr9pytestrLr#rapp_modules.initializerrapp_modules.dbrfixturer2r7r8r0r"rWrarcrdrqrvrrrr3r3r3r4sF"      ;      e  5