o wOiE@sddlZddlZddlZddlmZzddlZWney'ddlmZYnwGdddeZGdddeZ Gdd d e Z dS) N)TestClientApiBase)mockc@s,eZdZddZddZddZddZd S) TestClientApiInternalscCsHdddddd}|d||jd}||jjjdd d d S) ! Read timeout set to the default get/testkeytest modifiedIndexkeyvalueactionnodetimeoutN _mock_apiclientread assertEqual api_execute call_argsselfdresrP/home/ubuntu/.local/lib/python3.10/site-packages/etcd/tests/unit/test_request.pytest_read_default_timeouts  z0TestClientApiInternals.test_read_default_timeoutcCLdddddd}|d||jjddd ||jjjd d dd S) z( Read timeout set to the supplied value rrrr r rrrrrNrrrrrr test_read_custom_timeout z/TestClientApiInternals.test_read_custom_timeoutcCr") z Read timeout disabled rrrr r rrrr$rrNrr%rrr test_read_no_timeout+r'z+TestClientApiInternals.test_read_no_timeoutcCsVddddddd}|d||jjdddd ||jjjd td d id fdS)za Calling `write` without a value argument will omit the `value` from the API call params setTz/newdir) createdIndexdirr r rrNr,)z/v2/keys/newdirPUTr,true)params)rrwrite assertEqualsrrdictr%rrr test_write_no_params9s   z+TestClientApiInternals.test_write_no_paramsN)__name__ __module__ __qualname__r!r&r(r4rrrr r s  rc@seZdZdZedddZejdejdddZd d Z d d Z d dZ ejdejdddZ ddZ ddZddZddZddZddZddZd d!Zd"d#Zd$d%Zd&d'Zd(d)Zd*d+Zd,d-Zd.d/Zd0d1Zd2d3Zd4S)5TestClientApiInterfacez All tests defined in this class are executed also in TestClientRequest. If a test should be run only in this class, please override the method there. z&urllib3.request.RequestMethods.requestcCs4gd}d|}|d||_|||jjdS) Can request machines )http://127.0.0.1:4001zhttp://127.0.0.1:4002zhttp://127.0.0.1:4003,rN)join_prepare_response return_valuer2rmachines)rmockerdatarrrr test_machinesRs z$TestClientApiInterface.test_machineszetcd.Client.machines) new_callablecCs~gd|_tjddddd}||jdg||jd||j|jtjdddd d}|d |j||j|jd S) z6Do not overwrite the machines cache when using proxies)zhttps://10.0.0.2:4001https://10.0.0.3:4001zhttps://10.0.0.4:4001) localhost) localproxyrGhttpsT)hostprotocolallow_reconnect use_proxieszhttps://localproxy:4001zhttps://localhost:4001)rE)z10.0.0.2rGFrDN) r>etcdClientr2_machines_cache _base_uri assertNotInbase_uriassertIn)rr@crrr test_use_proxies[s$ z'TestClientApiInterface.test_use_proxiescCsBdddddgdgdgi}|d|||jjdd dd S) r9membersce2a822cea30bfcadefaultzhttp://localhost:2380zhttp://localhost:7001r:)idnamepeerURLs clientURLsrrZN)rr2rrWrrArrr test_membersus z#TestClientApiInterface.test_membersc CsDddddddddd d d d d }|d|||jjdddS)z Request for stats eca0338f4ea315668a69d5f6b7814500z#2014-10-24T13:15:51.186620747-07:00z10m59.322358947s)leader startTimeuptimenode3i8gJ|Ձ@gl͆2"@rz#2014-10-24T13:15:50.072007085-07:00 StateFollower) rZ leaderInfor[recvAppendRequestCntrecvBandwidthRate recvPkgRatesendAppendRequestCntrcstaterr[N)rr2rstatsr^rrr test_self_statss z&TestClientApiInterface.test_self_statscCs.did}|d|||jjdddS)z Request for leader stats 924e2e83e93f2560rb followersrrbN)rr2r leader_statsr^rrr test_leader_statss  z(TestClientApiInterface.test_leader_statszetcd.Client.memberscCsBddddi}||_|dddidi||jj|ddS)z Can request the leader rXrY)rZr[rrgrpN)r>rr2rrb)rr@rWrrr test_leadersz"TestClientApiInterface.test_leadercCsLdddddddd}|d ||jdd}||tjd i|d S) z Can set a value r)#2013-09-14T00:56:59.316195568+02:00rr  expirationr r ttlr rrNrrrr1r2rN EtcdResultrrrr test_set_plains z%TestClientApiInterface.test_set_plaincCsjdddddddd}|d ||jd}d |_d |d d <|d ||j|}||jd d S)zCan update a result.r)rurrwr rxrrciaorr N)rrrr updater2)rrrnewresrrr test_updates    z"TestClientApiInterface.test_updatecCsXdddddddd}|d ||jdd}d |d d <||tjdi|d S)z Can set a new value r)rurvrrwr rxrTrnewKeyNrr{rrrr test_newkeys  z"TestClientApiInterface.test_newkeycCsNdddddddd}|d ||jjddd }||tjd i|d S) z Can refresh a new value rz2016-05-31T08:27:54.660337ZrvriXr rxrr)rzNr)rrrefreshr2rNr|rrrr test_refreshs z#TestClientApiInterface.test_refreshcCs$|dd|tj|jjddS)z& Can handle server not found response iz Not foundz /somebadkeyN)r assertRaisesrN EtcdExceptionrrrrrr test_not_found_responses z.TestClientApiInterface.test_not_found_responsecCsPdddddddd}|d ||jjddd d }||tjd i|d S)z" Can set compare-and-swap a value compareAndSwaprurvrrwr rxrrtest_old prevValueNrr{rrrr test_compare_and_swaps z,TestClientApiInterface.test_compare_and_swapcCs(|td|jt|jjdddddS)< Exception will be raised if prevValue != value in test_set zTest Failed : [ 1!=3 ]/testKeyr oldbogrN)_mock_exception ValueErrorrrr1rrrr test_compare_and_swap_failures  z4TestClientApiInterface.test_compare_and_swap_failurecCs@ddddddd}|d||jdd}||jdd S) z Can append a new key createrz /testdir/190r )r+r r r rrz/testdirN)rrr1r2r+rrrr test_set_append s z&TestClientApiInterface.test_set_appendcCs|jtj|jjdddddS)z4 Creating a directory with a value raises an error. z/bar testvalyeTr-N)rrNrrr1rrrr test_set_dir_with_values z.TestClientApiInterface.test_set_dir_with_valuecCFdddddd}|d||jd}||tjd i|d S) z Can delete a value deleterr r r+rrrNr)rrrr2rNr|rrrr test_delete!  z"TestClientApiInterface.test_deletec sddddddddd ddddd d }|d ||j|d d|fddtD|d|j|dddS)z Can pop a value rrrrrFNrr )rr+r r ryr rzr,)rrprevNoderrr cs$i|]}|tjjvr|t|qSr)rNr| _node_propsgetattr).0attrrrr ?s  z3TestClientApiInterface.test_pop..rr )rrpopr2r,rr r%rrr test_pop/s  zTestClientApiInterface.test_popcCr) z Can get a value rrrr r rrrNrrrrr2rNr|rrrr test_readCrz TestClientApiInterface.test_readc Cs`ddddddddd dd dgd d }|d ||jjddd}||tjdi|dS)zCan get values in dirsrrrTz/testDir/testKeyr )r r r z/testDir/testKey2test2)r r r,nodesrrz/testDir) recursiveNrrrrrr test_get_dirQs$ z#TestClientApiInterface.test_get_dircCs"|tjd|d|jvdS)# Can check if key is not in client zKey not Found : /testKey/testeyN)rrNEtcdKeyNotFound assertTruerrrrr test_not_inksz"TestClientApiInterface.test_not_incCs2dddddd}|d||d|jvd S) rrrrr r rrrN)rrrr%rrr test_inps zTestClientApiInterface.test_incCsJdddddd}|d||jjddd }||tjd i|d S) z Can watch a key rrrr r rrT)waitNrrrrrr test_watch}s z!TestClientApiInterface.test_watchcCsLdddddd}|d||jjdddd }||tjd i|d S) z/ Can watch a key starting from the given Index rrtestoldr rrT)r waitIndexNrrrrrr test_watch_indexs z'TestClientApiInterface.test_watch_indexN)r5r6r7__doc__rpatchrB PropertyMockrVr_rnrsrtr}rrrrrrrrrrrrrrrrrrrr r8Ls8     r8c@steZdZddZdddZ  dddZd d Zd d Zd dZddZ ddZ ddZ ddZ ddZ ddZdS)TestClientRequestcCstjdd|_dS)N abcdef1234)expected_cluster_id)rNrOrrrrr setUpszTestClientRequest.setUpNcCs@|||}|p d|j_tj|d|jj_tj|d|jj_dS)Nrr>) r= getheaderr>r MagicMockrhttprequest_encode_bodyrequest)rstatusr cluster_idresprrr rs   zTestClientRequest._mock_apir.cCsX|d|||d}|p d|j_tj|jjj|d|jj_tj|jjj|d|jj_dS)Ni) errorCodemessagecauserr) r=rr>rcreate_autospecrrrr)r error_codemsgrmethodfieldsrrrrr _mock_errors     zTestClientRequest._mock_errorcCs2|jdddddid|jt|jjdddd d S) rrz Test Failedz[ 1!=3 ]rr)rrr rN)rrrrr1rrrr rs z/TestClientRequest.test_compare_and_swap_failurecCsBtj|jjjtj|jjddd|jj_|t j |jj ddS)rfoozRead timed out side_effectrN) rrrrrurllib3 exceptionsReadTimeoutErrorrrNEtcdWatchTimedOutwatchrrrr test_watch_timeouts  z$TestClientRequest.test_watch_timeoutcCs|t|jjd|jjdS)zE Exception will be raised if a path without a trailing slash is used z testpath/barN)rrrr_MPUTrrrr test_path_without_trailing_slashs z2TestClientRequest.test_path_without_trailing_slashcCs|tj|jjdddS)z@ Exception will be raised if an unsupported HTTP method is used z /testpath/barTRACEN)rrNrrrrrrr test_api_method_not_supporteds z/TestClientRequest.test_api_method_not_supportedcCsJdddddddd}|jd |d d |tj|jjd|jdd S) rr)rur~rrwr rxrr notabcd1234)rN)rrrNEtcdClusterIdChangedrrr%rrr test_read_cluster_id_changeds z.TestClientRequest.test_read_cluster_id_changedcCsLtj|jjjtd|jj_|tj |jj d|tj |jj dddS)Nrz /somethingz/aGET) rrrrrsocketerrorrrNEtcdConnectionFailedrrrrrr test_read_connection_errors  z,TestClientRequest.test_read_connection_errorcCdSNrrrrr rzTestClientRequest.test_not_incCrrrrrrr rrzTestClientRequest.test_incCs\dddddddd}tjdi|}d d d dd }|d |d|_|t|jj|dS)z Non-atomic updates fail r)rur~rrwr rxrezCompare failedz[ != bar] [7 != 6])rrrindexibarNr)rNr|rr rrrr)rrrrrrr test_update_failss"  z#TestClientRequest.test_update_failsr)r.NN)r5r6r7rrrrrrrrrrrrrrrr rs     r) rrrNetcd.tests.unitrr ImportErrorunittestrr8rrrrr s   ?P