o <‚†jã ã @sØdZddlZddlmmZddlZddlZddl Z ddl mZmZddl m Z mZmZej dej ej ej e¡¡¡¡ddlZddd„ZGdd „d ƒZGd d„dƒZGdd „d e jƒZedkrje ¡dSdS)as test_resistance_freeze_bugfix.py Regression test for the "frozen resistance" bug found in production logs on 2026-08-19: CLSK entered 5 times between 11:27 and 13:24 citing the identical "breakout of resistance $11.70" every time, despite trading as high as $12.20 in between. Root cause: monitor.py's _run_intraday_full_rescan() re-inserted a symbol's OLD dict from self.premarket_20 (with a stale pm_high/resistance) into the new pool whenever that symbol had an open position but fell outside the top full_rescan_pool_size health cut on that pass -- instead of using the symbol's freshly-rescored dict (which the same rescan pass had already computed, with a current pm_high). This test simulates two consecutive full rescans where the held symbol's health score dips low enough to fall out of the top-N cut both times, while its price genuinely makes new highs between rescans, and asserts that self.premarket_20's entry for that symbol reflects the NEW price high after the fix (it did NOT, before the fix). Run: python -m pytest tests/test_resistance_freeze_bugfix.py -v or: python tests/test_resistance_freeze_bugfix.py éN)ÚpatchÚ MagicMock)ÚdatetimeÚ timedeltaÚtimezoneÚWATCHc Cs*||||d|dur|n||dddœdœS)z¢Build a minimal scored-candidate dict shaped like scorer.score_premarket_candidate()'s output plus the fields premarket_scanner.scan()/monitor.py bolt on.gð?NT)Úmeets_min_volumeÚ spread_ok)ÚsymbolÚtotal_scoreÚpm_highÚpm_lowÚhealth_scoreÚhealth_stateÚflags©)r rrrrrrúB/var/www/screener/premarket/tests/test_resistance_freeze_bugfix.pyÚ_scored&sùrc@óeZdZdd„Zdd„ZdS)ÚFakePositionManagercCst|ƒ|_dS©N©ÚlistÚ_open)ÚselfÚopen_symbolsrrrÚ__init__6szFakePositionManager.__init__cCs t|jƒSrr©rrrrÚget_open_symbols9s z$FakePositionManager.get_open_symbolsN)Ú__name__Ú __module__Ú__qualname__rrrrrrr5órc@r)Ú FakeStreamcCódSrr)rÚsymbolsrrrÚadd_symbols>szFakeStream.add_symbolscs"t tj¡‰‡fdd„tdƒDƒS)Nc s(g|]}ˆt|dddddddœ‘qS))Úminutesééd)ÚtÚoÚhÚlÚcÚv)r)Ú.0Úi©ÚnowrrÚ Dsÿÿz'FakeStream.get_bars..é)rr3rÚutcÚrange)rr rr2rÚget_barsAs þzFakeStream.get_barsN)rr r!r&r8rrrrr#=r"r#c@s,eZdZdd„Zdd„Zdd„Zdd„Zd S) ÚTestResistanceFreezeBugfixc Cs˜tj tj¡|_ddddœddidœ|j_gd¢|j_td gd |j_tƒ|j_ dd„|j_ td d ddddtddddddtddddddg|j_dS)Nér(é)Úfull_rescan_pool_sizeÚfull_rescan_lookback_hoursÚfull_rescan_interval_minutesÚmin_avg_daily_volumei ¡)Úintraday_healthÚuniverse)ÚCLSKÚAAAÚBBBrB)rcSr$rr)ÚpoolrrrÚXsz2TestResistanceFreezeBugfix.setUp..gffffff'@é(r©rrrrrCg4@éZÚHEALTHYrDg>@éU) ÚmonitorÚSessionOrchestratorÚ__new__ÚorchÚcfgÚ_prefiltered_universerÚposition_mgrr#ÚstreamÚ_sync_opening_volume_baselinerÚpremarket_20rrrrÚsetUpKsý ú ýz TestResistanceFreezeBugfix.setUpc std|ˆdtddddtddddg}t td ¡h}t td ¡I}t td¡3}||j_‡fdd „}||j_d|j_i|j _d|j _i|j_d|j_d|j _|j ¡Wdƒn1s^wYWdƒn1smwYWdƒdSWdƒdS1s…wYdS)aGSimulate one full rescan pass where the whole universe gets rescored fresh (CLSK's price has moved to clsk_fresh_pm_high), but CLSK's health_score is deliberately kept low so it does NOT make the top-2 (full_rescan_pool_size) cut -- forcing the 'keep open-position symbol' fallback path to fire.rB)rrrCg€4@é[rDg€>@éVÚpremarket_scannerr@Ú data_storecs*tƒ}|dkr ˆnd|_d|_|ddifS)NrBé_rÚstate)rrÚconfirmed_state)r ÚbarsÚavg_vol_baselinerÚreading©Úclsk_health_scorerrÚfake_evaluate_symbolzszITestResistanceFreezeBugfix._run_rescan_with..fake_evaluate_symbolTN)rrÚobjectrLÚscanÚreturn_valueÚevaluate_symbolÚside_effectÚis_eligible_for_entryÚload_health_stateÚsave_health_stateÚload_watchlistÚsave_watchlistÚwrite_top_stocksrOÚ_run_intraday_full_rescan)rÚclsk_fresh_pm_highrbÚrescored_universeÚmock_pmsÚmock_healthÚmock_dsrcrrarÚ_run_rescan_withcs0ý ÿ þæ€Pz+TestResistanceFreezeBugfix._run_rescan_withcCsp|jdddtdd„|jjDƒƒ}| |ddd¡|jdd dtd d„|jjDƒƒ}| |ddd¡dS)Ngffffff(@é#)rprbcsó |]}|ddkr|VqdS©r rBNr©r0ÚrrrrÚ ó€zdTestResistanceFreezeBugfix.test_open_position_symbol_gets_fresh_pm_high_after_fix..raCLSK's pm_high/resistance should reflect this rescan's fresh price high (12.20), not the old stale value (11.70) -- this is exactly the bug: resistance froze at $11.70 across 5 real CLSK entries spanning 11:27-13:24 today despite price trading up to $12.20.g33333³(@é&csrwrxrryrrrr{ r|ztpm_high should keep tracking fresh price action on subsequent rescans too, not just refresh once and then re-freeze.)ruÚnextrOrUÚassertEqual)rÚ clsk_entryrrrÚ6test_open_position_symbol_gets_fresh_pm_high_after_fix‹sþ þzQTestResistanceFreezeBugfix.test_open_position_symbol_gets_fresh_pm_high_after_fixcCsštƒ|j_tdddddd}t td¡}||j_|j d¡Wdƒn1s)wYd d „|jj Dƒ}| t|ƒdd¡| |d ddd¡dS)z‚_handle_slot_freed should replace (not skip) an existing stale entry for a symbol chosen as a fresh replacement candidate.rBg)@éFrJrHÚ top_stocksÚSOME_EXITED_SYMBOLNcSsg|] }|ddkr|‘qS)r rBrryrrrr4³szRTestResistanceFreezeBugfix.test_slot_freed_upserts_stale_entry..r(z-should not end up with duplicate CLSK entriesrrz—the fresh replacement candidate's pm_high should overwrite the stale entry, not be silently discarded because CLSK was already present in premarket_20.) r#rOrSrrrdrLÚfind_replacementrfÚ_handle_slot_freedrUrÚlen)rÚfresh_candidateÚmock_top_stocksÚclsk_entriesrrrÚ#test_slot_freed_upserts_stale_entry§s ÿþþz>TestResistanceFreezeBugfix.test_slot_freed_upserts_stale_entryN)rr r!rVrurr‹rrrrr9Js (r9Ú__main__)Nr)Ú__doc__ÚbuiltinsÚ@py_builtinsÚ_pytest.assertion.rewriteÚ assertionÚrewriteÚ @pytest_arÚsysÚosÚunittestÚ unittest.mockrrrrrÚpathÚinsertÚdirnameÚabspathÚ__file__rLrrr#ÚTestCaser9rÚmainrrrrÚs"& sÿ