
    ri                        d Z ddlZddlZej                  j	                  dej                  j                  ej                  j                  e      d             ddlmZm	Z	m
Z
 ddlmZ ddlmZmZ ddlmZ d Zd	 Zd
 Zd Zd Zd Zd Zd Zd Zedk(  r, e         e         e         e         e         ed       yy)z/Tests for the eligibility determination engine.    Nz../../scripts)datetime	timedeltatimezone)determine_prop_eligibility)ValidationStateAvailabilityStatusnormalize_player_namec                      | t        |       |dS )N)player_namenormalized_nameteamr	   )namer   s     N/var/www/html/rainmaker/backend/tests/publishing_integrity/test_eligibility.pymake_rosterr      s    06     c                 T    | t        |       ||j                  j                         ddS )Ntest)r   r   availability_statusdetailed_statussource)r
   valuelower)r   statuss     r   
make_availr      s-    06%!<<--/ r   c                  |   t        dd      g} t        dt        j                        g}t	        ddddddt        j                         t        d	      z   d
| |g g       }|j                  t        j                  t        j                  fv sJ |j                  t        j                  k7  sJ |j                  dk7  sJ y )NzLeBron JamesLALpointsznba-lal-bos-20260310nbazLos Angeles LakersBoston Celtics   hourstest123r   	prop_typeevent_idleague	home_team	away_team
game_startrun_idroster_entriesavailability_reportslineup_datamarket_propsSUPPRESSED_TEAM_MISMATCH)r   r   r   	AVAILABLEr   r   nowr   validation_stater   VERIFIEDVERIFIED_PENDING_LINEUPr1   suppress_reasonrosteravaileligs      r   test_fully_verified_playerr<      s    .%01F(:(D(DEFE%"'&"<<>IA$66"D   _%=%=?f?f$gggg  O$L$LLLL#====r   c                  <   t        dd      g} t        dt        j                        g}t	        ddddddt        j                         t        d	      z   d
| |g g       }|j                  t        j                  k(  sJ |j                  rJ |j                  dk(  sJ y )NzKevin DurantPHXr   znba-phx-bos-20260310r   zPhoenix Sunsr    r!   r"   r$   r%   SUPPRESSED_UNAVAILABLE)r   r   r   OUTr   r   r3   r   r4   r   r?   publish_allowedr7   r8   s      r   test_out_player_suppressedrB   7   s    .%01F(:(>(>?@E%"' "<<>IA$66"D   O$J$JJJJ#####;;;;r   c                     t        dd      g} t        dt        j                        g}t	        ddddddt        j                         t        d	      z   d
| |g g       }|j                  t        j                  k(  sJ |j                  rJ y)z)Player traded but prop still on old team.zJimmy ButlerGSWr   znba-mia-bos-20260310r   z
Miami Heatr    r!   r"   r$   r%   N)r   r   r   r2   r   r   r3   r   r4   r   r1   rA   r8   s      r   test_team_mismatch_suppressedrE   O   s    .%01F(:(D(DEFE%"'"<<>IA$66"D   O$L$LLLL#####r   c                  >   t        dd      g} t        dt        j                        g}t	        ddddddt        j                  t        j                        t        d	      z   d
| |g g       }|j                  t        j                  k7  sJ |j                  dk7  sJ y )NzNikita KucherovTBLshots_on_goalznhl-nsh-tb-20260329nhlzTampa Bay LightningzNashville Predatorsr!   r"   r$   r%   r1   r   r   r   r2   r   r   r3   r   utcr   r4   r   r1   r7   r8   s      r   8test_nhl_team_aliases_do_not_trigger_false_team_mismatchrL   g   s    +U34F)+=+G+GHIE%%!&''<<-	0BB"D   O$L$LLLL#====r   c                  >   t        dd      g} t        dt        j                        g}t	        ddddddt        j                  t        j                        t        d	      z   d
| |g g       }|j                  t        j                  k7  sJ |j                  dk7  sJ y )NzJunior CamineroTBRhitszmlb-tb-stl-20260329mlbzSt. Louis CardinalszTampa Bay Raysr!   r"   r$   r%   r1   rJ   r8   s      r   8test_mlb_team_aliases_do_not_trigger_false_team_mismatchrQ   ~   s    +U34F)+=+G+GHIE%%&'"<<-	0BB"D   O$L$LLLL#====r   c                      t        ddddddt        j                         t        d      z   d	g g g g 
      } | j                  t
        j                  k(  sJ | j                  rJ y )N r   znba-test-20260310r   zTeam AzTeam Br!   r"   r$   r%   )r   r   r3   r   r4   r   SUPPRESSED_NO_CANONICAL_IDrA   )r;   s    r   test_empty_name_suppressedrU      sn    %$<<>IA$66D   O$N$NNNN#####r   c                     t        dd      g} t        dt        j                        g}t	        ddddddt        j                         t        d	      z   d
| |g g       }|j                  t        j                  k(  sJ |j                  rJ y )NzDraymond GreenrD   r   znba-gsw-bos-20260310r   zGolden State Warriorsr    r!   r"   r$   r%   )r   r   r   	SUSPENDEDr   r   r3   r   r4   r   r?   rA   r8   s      r    test_suspended_player_suppressedrX      s    *E23F(*<*F*FGHE%$')"<<>IA$66"D   O$J$JJJJ#####r   __main__zAll eligibility tests passed!)__doc__sysospathinsertjoindirname__file__r   r   r    publishing_integrity.eligibilityr   publishing_integrity.enumsr   r   "publishing_integrity.normalizationr
   r   r   r<   rB   rE   rL   rQ   rU   rX   __name__print r   r   <module>rh      s    5 
 	 277<< 9?K L 2 2 G J D>0<0$0>.>.$($. z  !# $&	
)* r   