Uo] d dl Z d dlZd dlZd dlmZ d dlmZmZ d dlZd dl Z
d dl m Z m
Z
mZmZ ej d Zd Zej d Zdd Zd
edefd Zd edefd
Z G d d Zde
j j defdZ dS ) N)Path)ListUnion) shorten_pathget_valid_filename_pytest_expected_on_rightflatten_filesystem_dictz ^.*?\[(.*)]$c | d }| ddd | ddd d S )Nsnapshotz--snapshot-update
store_truez6Update snapshot files instead of testing against them.)actionhelpz--allow-snapshot-deletionz0Allow snapshot deletion when updating snapshots.)getgroup addoption)parsergroups h/builddir/build/BUILD/cloudlinux-venv-1.0.10/venv/lib/python3.11/site-packages/pytest_snapshot/plugin.pypytest_addoptionr sg OOJ ' 'E OO
E
OO#
? c # K t | j }t | j j j | j j j | 5 }|V d d d d S # 1 swxY w Y d S N)_get_default_snapshot_dirnodeSnapshotconfigoptionsnapshot_updateallow_snapshot_deletion)requestdefault_snapshot_dirr s r r r s 4W\BB '.'7.'?&
(
( +3 s AA A returnc F t r
| |k sJ d S || k sJ d S r )r )valuer s r
_assert_equalr$ ' s> "" ! 5 r stringc d| v rt d | dt j S )z
Returns the bytes that would be in a file created using ``path.write_text(string)``.
See universal newlines documentation.
aN Snapshot testing strings containing "\r" is not supported.
To snapshot test non-standard newlines you should convert the tested value to bytes.
Warning: git may decide to modify the newlines in the snapshot file.
To avoid this read https://docs.github.com/en/get-started/getting-started-with-git/configuring-git-to-handle-line-endings
)
ValueErrorreplaceoslinesepencode)r% s r _file_encoder. . sL
v~~ j k k k >>$
+ + 2 2 4 44r datac z | dd dd S )z
Returns the string that would be read from a file using ``path.read_text(string)``.
See universal newlines documentation.
z
r( r' )decoder* )r/ s r _file_decoder2 > s2
;;== . . 6 6tT B BBr c e Zd ZdZdZdZdZdZdZde de de
fdZ d Z d Z
ed Zej d Zd
eee
f d e
fd Zd
eeef fdZd
eeef d
eee
f fdZdedeee
f fdZdS )r Nr r snapshot_dirc Z || _ || _ || _ g | _ g | _ g | _ d S r )_snapshot_update_allow_snapshot_deletionr4 _created_snapshots_updated_snapshots_snapshots_to_delete)selfr r r4 s r __init__zSnapshot.__init__N s8 / (?% ("$ "$ $&!!!r c | S r r; s r __enter__zSnapshot.__enter__V s r c j s j s j rUd t j dg} j r;| d | fd j D j r;| d | fd j D j rv j r4 j D ]}| | d n| d | fd j D t j d
| d d S d S )
Nz#Snapshot directory was modified: {}zR (verify that the changes are expected before committing them to version control)z Created snapshots:c 3 l K | ].}d t | j z V /dS z Nstr relative_tor4 .0sr; s r z$Snapshot.__exit__.._ @ $u$uXYVc!--@Q2R2R.S.S%S$u$u$u$u$u$ur z Updated snapshots:c 3 l K | ].}d t | j z V /dS rC rD rG s r rJ z$Snapshot.__exit__..c rK r z Deleted snapshots:z^ Snapshots that should be deleted: (run pytest with --allow-snapshot-deletion to delete them)c 3 l K | ].}d t | j z V /dS rC rD rG s r rJ z$Snapshot.__exit__..n s@ $w$wXYVc!--@Q2R2R.S.S%S$w$w$w$w$w$wr r( F)pytrace)
r8 r9 r: formatr r4 appendextendr7 unlinkpytestfailjoin)r; exc_typeexc_valexc_tb
message_linespaths ` r __exit__zSnapshot.__exit__Y s " Ad&= AAZ A B I I,W[WhJiJi j j q sM&
v $$%;<<< $$$u$u$u$u]a]t$u$u$uuuu&
v $$%;<<< $$$u$u$u$u]a]t$u$u$uuuu(
x0 g $ 9 & &
!(()?@@@@!(( *f g g g $$$w$w$w$w]a]v$w$w$wwww K -00% @ @ @ @ @ @- A Ar c | j S r )
_snapshot_dirr? s r r4 zSnapshot.snapshot_dirr s !!r c R t | | _ d S r )r absoluter] r; r# s r r4 zSnapshot.snapshot_dirv s" !%[[ 1 1 3 3r
snapshot_namer! c . t |t r| }n| j | }| j |j vrBt
d t | t | j |S )zB
Returns the absolute path to the given snapshot.
z Snapshot path {} is not in {})
isinstancer r_ r4 joinpathparentsr) rO r )r; ra
snapshot_paths r _snapshot_pathzSnapshot._snapshot_pathz s mT * * F ) 2 2 4 4MM - 6 6} E EM M$9 9 9 < C C ]++\$:K-L-L N N O O
O r r# c t |t rt t t fS t |t
r t d d fS t
d )a
Returns a 3-tuple of a compare function, an encoding function, and a decoding function.
* The compare function should compare the object to the value of its snapshot,
raising an AssertionError with a useful error message if they are different.
* The encoding function should encode the value into bytes for saving to a snapshot file.
* The decoding function should decode bytes from a snapshot file into a object.
c | S r r> xs r z5Snapshot._get_compare_encode_decode.. s A r c | S r r> rj s r rl z5Snapshot._get_compare_encode_decode.. s r zvalue must be str or bytes)rc rE r$ r. r2 bytes TypeErrorr` s r _get_compare_encode_decodez#Snapshot._get_compare_encode_decode sU eS ! ! : , < <
u
%
% : ++{{: : 899 9r c t j dt }| | \ }}}| | }| r| }nE| r/t d t | d}| j
r || } || |k r || }
|
|k rt d |j
dd | | | | j | dS | j | dS dS |w || } ||| d} n&# t $ r}
t% |
} Y d}
~
nd}
~
ww xY w| 2d t | | } t | dS t d t | ) a(
Asserts that ``value`` equals the current value of the snapshot with the given ``snapshot_name``.
If pytest was run with the --snapshot-update flag, the snapshot will instead be updated to ``value``.
The test will fail if there were any changes to the snapshot.
errisinstancez%snapshot exists but is not a file: {}Nz7value is not supported by pytest-snapshot's serializer.T)re exist_okzsvalue does not match the expected value in snapshot {}
(run pytest with --snapshot-update to update snapshots)
{}zKsnapshot {} doesn't exist. (run pytest with --snapshot-update to create it))operator methodcallerAssertionErrorrp rg is_file
read_bytesexistsrO r r6 r) parentmkdir write_bytesr8 rP r9 rE )r; r# ra __tracebackhide__comparer- r1 rf encoded_expected_value
encoded_valuedecoded_encoded_valueexpected_valuesnapshot_diff_msges r assert_matchzSnapshot.assert_match sp % 1/> R R"&"A"A%"H"H ++M::
" " *%2%=%=%?%? " "
!
!
#
# * !H!O!OP\]jPkPk!l!lmm m%) " 6 "F5MMM%- BX1X1X(.}(=(=%(E11$%^___ $**4$*GGG ))-888)1+22=AAAAA+22=AAAAA 2Y1X &1!'(>!?!?-GE>222 )-%% & / / /(+A%%%%%%/
%0)hhnhn,8 ,G,GIZi\ i\ & )):;;; 10 %ahh$]335 56 6 6s , E; ;
F FF dir_dictsnapshot_dir_namec t t j dt }t |t st d | | t | } r" fd d D }nQ
r/t d t t }t | }||z
}||z
}| j r4| j fdt# | D n|s|rd t dg} |r4| d | d
|D |r4| d | d |D t d
| | D ].\ }
} | | |
/dS )a
Asserts that the values in dir_dict equal the current values in the given snapshot directory.
If pytest was run with the --snapshot-update flag, the snapshots will be updated.
The test will fail if there were any changes to the snapshots.
rr z dir_dict must be a dictionaryc h | ]=}| | >S r> )rw rF as_posix)rH psnapshot_dir_paths r z,Snapshot.assert_match_dir.. sU T T T"#qyy{{ Tamm,= > > G G I I T T Tr *z*snapshot exists but is not a directory: {}c 3 B K | ]} | V d S r )rd )rH namer s r rJ z,Snapshot.assert_match_dir.. s3 ,p,pRV->-G-G-M-M,p,p,p,p,p,pr z#Values do not match snapshots in {}zF (run pytest with --snapshot-update to update the snapshot directory)z Values without snapshots:c 3 K | ] }d |z V
dS rC r> rH rI s r rJ z,Snapshot.assert_match_dir.. s& (I(I!(I(I(I(I(I(Ir z Snapshots without values:c 3 K | ] }d |z V
dS rC r> r s r rJ z,Snapshot.assert_match_dir.. s& (K(K!(K(K(K(K(K(Kr r( N)rt ru rv rc dictro rg r is_dirrglobry rO r setr6 r: rQ sortedrP rU itemsr rd )
r; r r r} values_by_filenameexisting_namesnames added_names
removed_namesrY r r# r s
@r assert_match_dirzSnapshot.assert_match_dir s % 1/> R R(D)) = ;<< < / /0A B B 4X > > # # % % # T T T T'8'>'>s'C'C T T TNN
%
%
'
' # !M!T!TUabsUtUt!u!uvv v UUN&''n, &.
? % , ,,p,p,p,pZ`anZoZo,p,p,p p p p p
?m
?!F!M!Ml[lNmNm!n!n!i!k
J!(()FGGG!(((I(I[(I(I(IIII L!(()FGGG!(((K(K](K(K(KKKK$TYY}%=%=>>> .3355 G GKD% e%6%?%?%E%E F F F F G Gr )__name__
__module__ __qualname__r6 r7 r8 r9 r: r] boolr r< r@ r[ propertyr4 setterr rE rg rn rp r r r r> r r r r F ss # M' 't '[_ ' ' ' ' A A A2 " " X" 4 4 4 E#t),< :c5j0A : : : : 16%U
"3 16E#t)DT 16 16 16 16f(G (G%T BR (G (G (G (G (G (Gr r r c | j }| j j }d| j vr
| j }d}nV| j }t
| j }|
J d | d }t | }| d|| }|| | }t t | S )zE
Returns the default snapshot directory for the pytest test.
[Nz@Expected request.node.name to be of format TEST_FUNCTION[PARAMS] snapshots) fspathdirpath purebasenamer originalnamePARAMETRIZED_TEST_REGEXmatchr r rU r rE )r test_module_dir test_module test_nameparametrize_nameparametrize_matchr s r r r s k))++O+*K
$)I % 3 9 9$) D D ,,.p,,,,22155-.>??*// [)TT# 3 8 89I J J ()) * **r )r! N) rt r+ repathlibr typingr r rS _pytest.python_pytestpytest_snapshot._utilsr r r r compiler r fixturer r$ rE rn r. r2 r pythonFunctionr r> r r r s
w w w w w w w w w w w w$"*_55 ! ! ! !
5
5
5
5
5
5 Cu C C C C CoG oG oG oG oG oG oG oGd+GN$; + + + + + + +r