\j<ddlZddlZddlZddlZddlZddlZddlmZddlm Z ddl m Z ddl m Z ddlmZmZdZ ddlZdZd d gZd Zd Zd ZdZdadadadZdZdZdZdZ dZ!dZ"dZ#dZ$dZ%dZ&dZ'dZ(dZ)dZ*dZ+dZ,d Z-da.d!Z/d"Z0y#e$rYXwxYw)#N)LDClient)Config)InMemoryFeatureStore)Files)FEATURESSEGMENTSFTflag1flag2af { "flags": { "flag1": { "key": "flag1", "on": true, "fallthrough": { "variation": 2 }, "variations": [ "fall", "off", "on" ] } }, "flagValues": { "flag2": "value2" }, "segments": { "seg1": { "key": "seg1", "include": ["user1"] } } } z --- flags: flag1: key: flag1 "on": true flagValues: flag2: value2 segments: seg1: key: seg1 include: ["user1"] z { "flags": { "flag1": { "key": "flag1", "on": true, "fallthrough": { "variation": 2 }, "variations": [ "fall", "off", "on" ] } } } zm { "segments": { "seg1": { "key": "seg1", "include": ["user1"] } } } c@tatjayN)rstore threadingEventreadyG/root/env/lib/python3.12/site-packages/testing/test_file_data_source.pysetup_functionr\s "E OO Erc:ttjyyr ) data_sourcestoprrrteardown_functionrasrc jtjdi|tdttat S)NSDK_KEYr)rnew_data_sourcerr rr)kwargss rmake_data_sourceres+1%''1&1&2CUERK rctj\}}tj||j dtj ||S)Nzlatin-1)tempfilemkstemposwriteencodeclose)contentfpaths rmake_temp_filer(js; GAtHHQy)*HHQK Krcht|d5}|j|dddy#1swYyxYw)Nw)openr")r'r%r&s r replace_filer,ps. dCA s(1ctd} t|}tjdusJ|j dusJt jdusJ t j|y#t j|wxYw)Nz{"flagValues":{"key":"value"}}pathsF)r(rris_set initializedr r!remover'sources r&test_does_not_load_data_prior_to_startr5tsr : ;D!$/||~&&&!!#u,,,  E))) $ $s A A00Bcltt} t|}|jtj dusJt ttjtdjtk(sJ tj|y#tj|wxYw)Nr.Tc|Sr rxs rz5test_loads_flags_on_start_from_json..r)r(all_properties_jsonrstartr r1sortedlistallrkeys all_flag_keysr!r2r3s r#test_loads_flags_on_start_from_jsonrC~s| - .D!$/   D(((d599X{;@@BCD UUU $ $s A4BB3ctstjdtt} t |}|j tjdusJtttjtdjtk(sJ tj |y#tj |wxYw)NBskipping file source test with YAML because pyyaml isn't availabler.Tc|Sr rr8s rr:z5test_loads_flags_on_start_from_yaml..r;r) have_yamlpytestskipr(all_properties_yamlrr=r r1r>r?r@rrArBr!r2r3s r#test_loads_flags_on_start_from_yamlrKs  XY - .D!$/   D(((d599X{;@@BCD UUU $ $s A4B77Cctt} t|}|j|j dusJt j dusJ tj|y#tj|wxYw)Nr.T) r(r<rr=r1rr0r!r2r3s r8test_sets_ready_event_and_initialized_on_successful_loadrMsg - .D!$/ !!#t+++||~%%% $ $s AA00Bcd}t|}|j|jdusJtj dusJy)Nz no-such-filer.FT)rr=r1rr0) bad_file_pathr4s rGtest_sets_ready_event_and_does_not_set_initialized_on_unsuccessful_loadrPsE"M m 4F LLN    5 (( ( <<>T !! !rctt}tt} t||g}|j t t jtddk(sJt t jtddk(sJ tj|tj|y#tj|tj|wxYw)Nr.c|Sr rr8s rr:z.test_can_load_multiple_files..rc|Sr rr8s rr:z.test_can_load_multiple_files..rSr) r(flag_only_jsonsegment_only_jsonrr=lenr r@rrr!r2path1path2r4s rtest_can_load_multiple_filesr\s > *E , -E!E5*:; 599X{34999599X{34999 % % % %s A0B<<,C(ctt}tt} t||g}|jt t j tddk(sJ tj|tj|y#tj|tj|wxYw)Nr.c|Sr rr8s rr:z4test_does_not_allow_duplicate_keys..rSrr) r(rVrr=rXr r@rr!r2rYs r"test_does_not_allow_duplicate_keysr_s > *E > *E!E5*:; 599X{34999 % % % %s AB,B?ctt} t|}|jt t j tddk(sJtjdt|ttjdt t j tddk(sJ tj|y#tj|wxYw)Nr.c|Sr rr8s rr:zJtest_does_not_reload_modified_file_if_auto_update_is_off..rSrr?c|Sr rr8s rr:zJtest_does_not_reload_modified_file_if_auto_update_is_off..rSrr(rVrr=rXr r@rtimesleepr,rWr!r2r3s r8test_does_not_reload_modified_file_if_auto_update_is_offrgs . )D !$/ 599X{34999 3T,- 3599X{34999 $ $s B(CC'crtt}||d< td i|}|jt t j tddk(sJtjdt|ttjdz}tj|krktjdt t j tddk(r tj|ytj|krkJd #tj|wxYw) Nr/c|Sr rr8s rr:z%do_auto_update_test..rSrrrb 皙?c|Sr rr8s rr:z%do_auto_update_test..rSrrTz(Flags were not reloaded after 10 secondsrrd)optionsr'r4deadlines rdo_auto_update_testros . )DGG !,G, 599X{34999 3T,-99;#iikH$ JJsO599X{349 $ iikH$ A@@u $sCDDDD6ctddiy)N auto_updateTrorrr/test_reloads_modified_file_if_auto_update_is_onrss-/0rc"tddddy)NTrk)rq force_polling poll_intervalrrrrr*test_reloads_modified_file_in_polling_moderwsWZ\]rcdtt} tj|}t t d|d}|j dddid }|d k(sJ tj|||jyy#tj||jwwxYw) Nr.rFupdate_processor_class send_eventsconfigr keyuseron r(r<rrrr variationr!r2r$r'factoryclientvalues r0test_evaluates_full_flag_with_client_as_expectedrs - .D''5 Gch!ij  E6*;R@}} $   LLN  $   LLN  A B*B/cdtt} tj|}t t d|d}|j dddid }|d k(sJ tj|||jyy#tj||jwwxYw) Nr.rFryr|r r~rrvalue2rrs r6test_evaluates_simplified_flag_with_client_as_expectedrs - .D''5 Gch!ij  E6*;R@    $   LLN  $   LLN rcday)NT)&unsafe_yaml_caused_method_to_be_calledr8s r!arbitrary_method_called_from_yamlrs-1*rcrtstjdd}t|} t j |}t td|d}tj|||j tdk(sJy#tj||jwwxYw)NrEz^ !!python/object/apply:testing.test_file_data_source.arbitrary_method_called_from_yaml ["hi"] r.rFryr|) rGrHrIr(rrrrr!r2r$r) unsafe_yamlr'rrs rtest_does_not_allow_unsafe_yamlrs  XYK + &D''5 Gch!ij $   LLN 1U :: : $   LLN s .B *B6)1jsonr!rHrrreldclient.clientrldclient.configrldclient.feature_storerldclient.integrationsrldclient.versioned_data_kindrrrGyaml ImportErrorrBr<rJrVrWrr rrrrr(r,r5rCrKrMrPr\r_rgrorsrwrrrrrrrrrs  $"7';   I 7$ 0          "   $1^  */&2;[  sBBB