Compare commits
544 Commits
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
9a8e34c726 | ||
|
|
552b7bf47b | ||
|
|
a50550535d | ||
|
|
0adf6df131 | ||
|
|
f145c7db94 | ||
|
|
b6bac45c7e | ||
|
|
8dc7efade2 | ||
|
|
f4219bdca2 | ||
|
|
900d9c7910 | ||
|
|
575d75f4d3 | ||
|
|
c30eff1291 | ||
|
|
e67931dff7 | ||
|
|
64edd22d73 | ||
|
|
b3f1c4befb | ||
|
|
ab44a21b46 | ||
|
|
a13842f351 | ||
|
|
8363bf60ad | ||
|
|
5406febcd9 | ||
|
|
3491c08ef6 | ||
|
|
912ef337f9 | ||
|
|
b4e2c03a92 | ||
|
|
f31ba09713 | ||
|
|
5da91837d4 | ||
|
|
35e51a50f3 | ||
|
|
f41dd5c08b | ||
|
|
49e7a052e2 | ||
|
|
a2b0cd85e3 | ||
|
|
e1751ff253 | ||
|
|
ee20ef52ec | ||
|
|
08b806fd87 | ||
|
|
db4db928d3 | ||
|
|
ee851ba974 | ||
|
|
ce416a0944 | ||
|
|
daba50c797 | ||
|
|
c3428789fb | ||
|
|
ec9d95413c | ||
|
|
13085ff679 | ||
|
|
80c3ff2aee | ||
|
|
3d89d9faca | ||
|
|
b6cc22ffdd | ||
|
|
f303125cee | ||
|
|
38da6c3f9a | ||
|
|
926541fa4d | ||
|
|
f533157981 | ||
|
|
e028fe2a38 | ||
|
|
469a32a9dd | ||
|
|
84814b2a72 | ||
|
|
908102d776 | ||
|
|
edd0e0698f | ||
|
|
cbd82e75e7 | ||
|
|
6923641626 | ||
|
|
b60d266ad7 | ||
|
|
36676384bd | ||
|
|
0441761872 | ||
|
|
fdb6a3291f | ||
|
|
b1a0a1f104 | ||
|
|
f28ab12db0 | ||
|
|
d5e1777732 | ||
|
|
afc884d786 | ||
|
|
e077ebd926 | ||
|
|
543c4e532c | ||
|
|
182d8e2ea0 | ||
|
|
4c86d86bf1 | ||
|
|
fa56b563dd | ||
|
|
896f139ddc | ||
|
|
a89c5e3a89 | ||
|
|
db6dad95be | ||
|
|
a426ad91d7 | ||
|
|
1e7477b9df | ||
|
|
db28d14b6d | ||
|
|
12b3859903 | ||
|
|
b88415b8e8 | ||
|
|
cb01da6a50 | ||
|
|
32901018ca | ||
|
|
1be94d262d | ||
|
|
6585d9cb51 | ||
|
|
fdb1f7ea4a | ||
|
|
0cd9763423 | ||
|
|
2f263d4d3f | ||
|
|
e9658b2406 | ||
|
|
10dd20405b | ||
|
|
130067799c | ||
|
|
18c1f142b5 | ||
|
|
6ccaae5772 | ||
|
|
5b13ae006a | ||
|
|
c930c69776 | ||
|
|
b1ea391842 | ||
|
|
e5070789cc | ||
|
|
60289f0398 | ||
|
|
1e78bd4d46 | ||
|
|
029c74cb3a | ||
|
|
0ce7686c64 | ||
|
|
f9f48bab95 | ||
|
|
0cfba4994d | ||
|
|
f813437154 | ||
|
|
27ac7bd717 | ||
|
|
2f7946f523 | ||
|
|
da6d3d9e62 | ||
|
|
37632e70d6 | ||
|
|
a9923507c5 | ||
|
|
1496b8d2ff | ||
|
|
ab8b99a005 | ||
|
|
ae88b5a08f | ||
|
|
49e59d41b2 | ||
|
|
e7aab567bf | ||
|
|
2195866ba7 | ||
|
|
3093a7c7cb | ||
|
|
44baec9e9c | ||
|
|
bd6decb8c0 | ||
|
|
6011df9e37 | ||
|
|
8bf8263391 | ||
|
|
222a55f8cd | ||
|
|
3d76c84b2a | ||
|
|
881094015b | ||
|
|
42cf3407c8 | ||
|
|
1a434c7c59 | ||
|
|
cef27b18d9 | ||
|
|
a9c034426b | ||
|
|
0c5eec790b | ||
|
|
47da699a87 | ||
|
|
71e0c950e1 | ||
|
|
44a58f1650 | ||
|
|
63acab96cf | ||
|
|
eb25ce420e | ||
|
|
f278e471d1 | ||
|
|
5f00966f9c | ||
|
|
b607ef144c | ||
|
|
fd02550349 | ||
|
|
0c8a22b87d | ||
|
|
a5f6d638b9 | ||
|
|
ad1167d01e | ||
|
|
faa77eab43 | ||
|
|
6a796c8668 | ||
|
|
6d91859659 | ||
|
|
5eb1edf315 | ||
|
|
d595913117 | ||
|
|
89a4eaa36c | ||
|
|
491f5ba77f | ||
|
|
84772bd8a9 | ||
|
|
a2e39ada00 | ||
|
|
46974363b4 | ||
|
|
406bed0418 | ||
|
|
16a2fea219 | ||
|
|
52400e11a1 | ||
|
|
d47bd9cf95 | ||
|
|
6bdf07e726 | ||
|
|
e3910f868b | ||
|
|
215b3b1938 | ||
|
|
67384f0f08 | ||
|
|
b291807106 | ||
|
|
8d4518287c | ||
|
|
4839e523c8 | ||
|
|
e0767cf5a1 | ||
|
|
0d93ec48a5 | ||
|
|
1f308c841f | ||
|
|
eb0cedd969 | ||
|
|
fad2ccb0e7 | ||
|
|
448a4944ff | ||
|
|
4849eacc10 | ||
|
|
9e64ed1bd8 | ||
|
|
0049dc8896 | ||
|
|
51f7b408f3 | ||
|
|
29c4c2cc0f | ||
|
|
266caf02c4 | ||
|
|
6fe85436ae | ||
|
|
9603a85f21 | ||
|
|
ef59602e39 | ||
|
|
668bd8232a | ||
|
|
4936fe90bf | ||
|
|
dba0aa94ad | ||
|
|
a7a645f85f | ||
|
|
5787acc271 | ||
|
|
4b4d74971c | ||
|
|
4b4c949997 | ||
|
|
8cf65796da | ||
|
|
7a07c02814 | ||
|
|
eca38b730b | ||
|
|
32e183baa7 | ||
|
|
3da6e858d5 | ||
|
|
2b148867e7 | ||
|
|
527cbdd18a | ||
|
|
8879f2b951 | ||
|
|
d2653eecc6 | ||
|
|
1514e9b784 | ||
|
|
98e9f5ecc1 | ||
|
|
d7316afb63 | ||
|
|
7580309e99 | ||
|
|
47c5460d2e | ||
|
|
b84e1a7ce4 | ||
|
|
c0e0554a3b | ||
|
|
d63edf7aea | ||
|
|
b756bbd2a3 | ||
|
|
b2bba67b61 | ||
|
|
6656767850 | ||
|
|
cffb9069fd | ||
|
|
0d4c84e86f | ||
|
|
8c361be219 | ||
|
|
9892e1bbe9 | ||
|
|
cd4e5db997 | ||
|
|
3ebd93480a | ||
|
|
ddf9508e15 | ||
|
|
e08284bdca | ||
|
|
5e96353797 | ||
|
|
9a9a65c617 | ||
|
|
f7b94be0db | ||
|
|
db3d3c7022 | ||
|
|
3a26119eb3 | ||
|
|
d24b4a5697 | ||
|
|
e4b7eb52e6 | ||
|
|
37d327fbd8 | ||
|
|
b32da4b409 | ||
|
|
2a1523fa85 | ||
|
|
de332c553c | ||
|
|
bfa01c6a6c | ||
|
|
9dbccbee2c | ||
|
|
e44ece0b07 | ||
|
|
b83fcbb1a9 | ||
|
|
1ffdedc7a4 | ||
|
|
3c786f5cd9 | ||
|
|
f105789e12 | ||
|
|
61e90473e6 | ||
|
|
e79c19b4d2 | ||
|
|
578a532395 | ||
|
|
4e9b52b8d6 | ||
|
|
d88596cef1 | ||
|
|
ad16d6e880 | ||
|
|
bbb6f88fae | ||
|
|
d7074ba9ee | ||
|
|
d11cf20452 | ||
|
|
baec1e05f7 | ||
|
|
003ff56cbc | ||
|
|
cbe1b9e81b | ||
|
|
0249a08201 | ||
|
|
bfaab82e26 | ||
|
|
9f64dad687 | ||
|
|
9514a67270 | ||
|
|
59d94f397b | ||
|
|
ffb372aec7 | ||
|
|
06872f7f03 | ||
|
|
5a640daf92 | ||
|
|
f9b2f711c0 | ||
|
|
31cca98656 | ||
|
|
cf2630148a | ||
|
|
55d399b75a | ||
|
|
fed738df42 | ||
|
|
6ef91357b2 | ||
|
|
36bb704c76 | ||
|
|
71514b5989 | ||
|
|
3be7f76763 | ||
|
|
83fcf7cd08 | ||
|
|
4a29fbda69 | ||
|
|
1e67c5c02c | ||
|
|
ca3cbc3f31 | ||
|
|
84db915d91 | ||
|
|
5ce53dc175 | ||
|
|
e07a6b7e77 | ||
|
|
7b82b3582f | ||
|
|
f29c7b1f15 | ||
|
|
04f2cc0fa9 | ||
|
|
c77c15e369 | ||
|
|
11d3488026 | ||
|
|
d6ab555b81 | ||
|
|
c788c7e0c1 | ||
|
|
1d21d27dda | ||
|
|
dde8837fb2 | ||
|
|
947fbe33ee | ||
|
|
2eb18f18b2 | ||
|
|
0b2900d162 | ||
|
|
578c6c2ad2 | ||
|
|
ed26975926 | ||
|
|
0e407351a4 | ||
|
|
6a672d4dd3 | ||
|
|
0cb4b2f1a6 | ||
|
|
244f17d57b | ||
|
|
65f7c54208 | ||
|
|
e2bfa9a590 | ||
|
|
4ea0288eaa | ||
|
|
336163aaa0 | ||
|
|
8a94de1c1b | ||
|
|
87c015fbcf | ||
|
|
50ea82310d | ||
|
|
c16ef0a13e | ||
|
|
e51ac1fd1c | ||
|
|
c2aefc2eda | ||
|
|
54846cd110 | ||
|
|
a742aa22d4 | ||
|
|
dfd5573c19 | ||
|
|
8a8cb6f0af | ||
|
|
7e19515d7b | ||
|
|
f645036d71 | ||
|
|
9145521dfa | ||
|
|
673f821acb | ||
|
|
1bcb85f1c1 | ||
|
|
0dd5a5c794 | ||
|
|
f50ef65dd5 | ||
|
|
24e9da6a09 | ||
|
|
0f6647b49a | ||
|
|
536b1cb8ab | ||
|
|
6d33d4fb34 | ||
|
|
454c163788 | ||
|
|
958b3d11dc | ||
|
|
1366b9132e | ||
|
|
27593796fb | ||
|
|
b8460b6001 | ||
|
|
65089ead4c | ||
|
|
717fb47c17 | ||
|
|
106343a6a2 | ||
|
|
726a6fa98d | ||
|
|
a435a3e1c3 | ||
|
|
84411b9119 | ||
|
|
397c280217 | ||
|
|
60faca0253 | ||
|
|
92a5623886 | ||
|
|
4cc5e89ebf | ||
|
|
2374f3811a | ||
|
|
75f6a3718e | ||
|
|
aa41e70523 | ||
|
|
2655bb4d86 | ||
|
|
98b030af94 | ||
|
|
775b683276 | ||
|
|
5914ea0ff4 | ||
|
|
d0b0dc3512 | ||
|
|
12b4fe3644 | ||
|
|
b0667b11d3 | ||
|
|
cf9b68c6f4 | ||
|
|
2b4ab2a9ad | ||
|
|
000391c440 | ||
|
|
0ec57781d1 | ||
|
|
85c5ee45b5 | ||
|
|
08e7f036a2 | ||
|
|
e7db0e34aa | ||
|
|
02c3df6fc9 | ||
|
|
9a2f5678de | ||
|
|
1dc45fddc1 | ||
|
|
2ebd2aaecd | ||
|
|
c3e3373df2 | ||
|
|
dae4f7f159 | ||
|
|
34f7d7ee4c | ||
|
|
1bdf035802 | ||
|
|
42cb70e9ab | ||
|
|
09897b698e | ||
|
|
e2e673edeb | ||
|
|
412d0005b8 | ||
|
|
d51c70b401 | ||
|
|
0ee8e585c2 | ||
|
|
6ed3533b73 | ||
|
|
0f37bfd7a1 | ||
|
|
5f956210f8 | ||
|
|
094a320817 | ||
|
|
db1449b79f | ||
|
|
448c8e8326 | ||
|
|
4cac781530 | ||
|
|
aa6fa7a696 | ||
|
|
f476d8bd97 | ||
|
|
149fe390d4 | ||
|
|
846b1fc25c | ||
|
|
b83389d6bc | ||
|
|
e6a4a7145e | ||
|
|
3723edc199 | ||
|
|
5d14b3458e | ||
|
|
a8197a792e | ||
|
|
7ec9c07081 | ||
|
|
7a0afcf0b9 | ||
|
|
cce4a0d179 | ||
|
|
5b42ed8cac | ||
|
|
5b0862e78e | ||
|
|
86e221eb5c | ||
|
|
3c2ce16ad8 | ||
|
|
fafa00e9e3 | ||
|
|
175481e35d | ||
|
|
93886286a3 | ||
|
|
bc2cc1ed58 | ||
|
|
2c077ba8a4 | ||
|
|
98b656f286 | ||
|
|
92970d39fd | ||
|
|
4d4a26cbbe | ||
|
|
7a4218f0ad | ||
|
|
79e5d60139 | ||
|
|
bf0aa56fbb | ||
|
|
708ce16961 | ||
|
|
1c813dc648 | ||
|
|
e7c0814512 | ||
|
|
b40223938c | ||
|
|
8b668898b8 | ||
|
|
5177e4ec53 | ||
|
|
7243b43e63 | ||
|
|
61033760e5 | ||
|
|
3210b175dd | ||
|
|
04ebbcde87 | ||
|
|
e724d66bb5 | ||
|
|
7424f6c098 | ||
|
|
894782e1b3 | ||
|
|
02fc85f635 | ||
|
|
9ff5a1755c | ||
|
|
711a45a0e9 | ||
|
|
4bffc4e492 | ||
|
|
d1c14150dc | ||
|
|
a3f3ec52bc | ||
|
|
8675a163df | ||
|
|
58a41ae7f5 | ||
|
|
c326161379 | ||
|
|
e51ad27d20 | ||
|
|
f3ca1e0372 | ||
|
|
a0951515f4 | ||
|
|
a1b4a7481c | ||
|
|
0c6f84642c | ||
|
|
33843d285b | ||
|
|
3875cc6e1b | ||
|
|
53c4694cf9 | ||
|
|
3a3f3d2214 | ||
|
|
4e5166b29a | ||
|
|
2149567ed3 | ||
|
|
2c1a1608f2 | ||
|
|
eea8a37257 | ||
|
|
38ceda4c09 | ||
|
|
95cedeefb8 | ||
|
|
894d5e1c7f | ||
|
|
43fa11cb80 | ||
|
|
4403169821 | ||
|
|
fa5e637da8 | ||
|
|
296046cc38 | ||
|
|
df30eff390 | ||
|
|
dec13a8020 | ||
|
|
f77e1ac119 | ||
|
|
e99818df08 | ||
|
|
1145c9de54 | ||
|
|
737a1b7344 | ||
|
|
134f52ff23 | ||
|
|
dc44b49128 | ||
|
|
1d866b3cb4 | ||
|
|
4a9e90a24b | ||
|
|
166821f22c | ||
|
|
4353278dc8 | ||
|
|
72ed1029be | ||
|
|
e6e8650cc9 | ||
|
|
5e055eea0f | ||
|
|
8274db2722 | ||
|
|
0351992ddb | ||
|
|
189923f7c3 | ||
|
|
6e5126d6bb | ||
|
|
e48365c5f1 | ||
|
|
34ca759f85 | ||
|
|
4322fdf7ce | ||
|
|
7ef18af49b | ||
|
|
61d435408e | ||
|
|
b66a5cb673 | ||
|
|
a548f7a3d4 | ||
|
|
38127f458b | ||
|
|
f69e1f6f04 | ||
|
|
3653984f21 | ||
|
|
85b6f182ae | ||
|
|
4ffebb04e9 | ||
|
|
4ed621b3e7 | ||
|
|
df7304ddf0 | ||
|
|
de09181b1c | ||
|
|
baf0344446 | ||
|
|
74b85d51d5 | ||
|
|
9edfc626ed | ||
|
|
3f5d6c1539 | ||
|
|
9a123bb631 | ||
|
|
8927cfe79b | ||
|
|
f1fe49784b | ||
|
|
5ac262783f | ||
|
|
9d5d4fa735 | ||
|
|
150a3ba756 | ||
|
|
6da1fdbed9 | ||
|
|
38f97a3b49 | ||
|
|
e28f49c510 | ||
|
|
080c1c6121 | ||
|
|
e52965eea4 | ||
|
|
c9a85ad321 | ||
|
|
85ca6f1d9f | ||
|
|
61feccacfe | ||
|
|
f5aeac3ccc | ||
|
|
324ed1730f | ||
|
|
2b667df14f | ||
|
|
eaae22c63b | ||
|
|
0ac56a3019 | ||
|
|
f27e8e123a | ||
|
|
25b2d77b63 | ||
|
|
b56605cc1f | ||
|
|
691ab8e783 | ||
|
|
5456f1ec04 | ||
|
|
f32e2bc372 | ||
|
|
4e672bd549 | ||
|
|
3e2967147e | ||
|
|
338b7dcd49 | ||
|
|
6198805647 | ||
|
|
9a7bac7d45 | ||
|
|
8496d58faa | ||
|
|
42d7ec2749 | ||
|
|
0e6aa6fa15 | ||
|
|
61c3ddda82 | ||
|
|
1f7f8a9e0f | ||
|
|
8b6be4ab71 | ||
|
|
13fa361421 | ||
|
|
1a7239c5ac | ||
|
|
b94dd8e12b | ||
|
|
7a7ffa615d | ||
|
|
c6ffa1e951 | ||
|
|
b9228684a4 | ||
|
|
0fe1691b74 | ||
|
|
a641337b5c | ||
|
|
041edbecba | ||
|
|
a1a21099ed | ||
|
|
25c9b90842 | ||
|
|
3dd44e6e5f | ||
|
|
1b818470fc | ||
|
|
f344eb7965 | ||
|
|
ac8fa2d645 | ||
|
|
7108c2199d | ||
|
|
02d6cc9cfd | ||
|
|
7a4150ac17 | ||
|
|
8ae9a0432a | ||
|
|
2c492155a6 | ||
|
|
097a853805 | ||
|
|
554e1fddb8 | ||
|
|
68237f78ec | ||
|
|
b47143cd10 | ||
|
|
1fda6a2310 | ||
|
|
665eb9a4b8 | ||
|
|
702c86c02c | ||
|
|
6b141d1f5b | ||
|
|
acc437af57 | ||
|
|
3b5788906d | ||
|
|
3e783f5489 | ||
|
|
44d1d40d65 | ||
|
|
c42f3d6421 | ||
|
|
8f6249e2f9 | ||
|
|
19fdf7cceb | ||
|
|
c8d3afdefd | ||
|
|
0a1471d8f9 | ||
|
|
2909e1fc8a | ||
|
|
94b9a54c8e | ||
|
|
01f659cd8b | ||
|
|
34ae214e33 | ||
|
|
1784e2c4b9 | ||
|
|
b4e32a1d83 | ||
|
|
cbc3af704f | ||
|
|
82a8a60af6 | ||
|
|
c2e7fae7d9 | ||
|
|
9603e92894 | ||
|
|
4ecf786892 | ||
|
|
78724cde8d |
4
.codecov.yml
Normal file
4
.codecov.yml
Normal file
@ -0,0 +1,4 @@
|
||||
# do not notify until at least 100 builds have been uploaded from the CI pipeline
|
||||
# you can also set after_n_builds on comments independently
|
||||
comment:
|
||||
after_n_builds: 100
|
||||
@ -3,22 +3,14 @@
|
||||
# Coverage combine merges (and removes) all the coverage files and
|
||||
# generates a new .coverage file in the current directory.
|
||||
|
||||
set -o xtrace # Write all commands first to stderr
|
||||
set -o errexit # Exit the script with error if any of the commands fail
|
||||
set -eu
|
||||
|
||||
. .evergreen/utils.sh
|
||||
# Set up the virtual env.
|
||||
. .evergreen/scripts/setup-dev-env.sh
|
||||
uv sync --group coverage
|
||||
source .venv/bin/activate
|
||||
|
||||
if [ -z "${PYTHON_BINARY:-}" ]; then
|
||||
PYTHON_BINARY=$(find_python3)
|
||||
fi
|
||||
|
||||
createvirtualenv "$PYTHON_BINARY" covenv
|
||||
# Keep in sync with run-tests.sh
|
||||
# coverage >=5 is needed for relative_files=true.
|
||||
pip install -q "coverage[toml]>=5,<=7.5"
|
||||
|
||||
pip list
|
||||
ls -la coverage/
|
||||
|
||||
python -m coverage combine coverage/coverage.*
|
||||
python -m coverage html -d htmlcov
|
||||
coverage combine coverage/coverage.*
|
||||
coverage html -d htmlcov
|
||||
|
||||
File diff suppressed because it is too large
Load Diff
343
.evergreen/generated_configs/functions.yml
Normal file
343
.evergreen/generated_configs/functions.yml
Normal file
@ -0,0 +1,343 @@
|
||||
functions:
|
||||
# Assume ec2 role
|
||||
assume ec2 role:
|
||||
- command: ec2.assume_role
|
||||
params:
|
||||
role_arn: ${aws_test_secrets_role}
|
||||
duration_seconds: 3600
|
||||
|
||||
# Attach benchmark test results
|
||||
attach benchmark test results:
|
||||
- command: attach.results
|
||||
params:
|
||||
file_location: src/report.json
|
||||
|
||||
# Cleanup
|
||||
cleanup:
|
||||
- command: subprocess.exec
|
||||
params:
|
||||
binary: bash
|
||||
args:
|
||||
- .evergreen/scripts/cleanup.sh
|
||||
working_dir: src
|
||||
type: test
|
||||
|
||||
# Download and merge coverage
|
||||
download and merge coverage:
|
||||
- command: ec2.assume_role
|
||||
params:
|
||||
role_arn: ${assume_role_arn}
|
||||
type: setup
|
||||
- command: subprocess.exec
|
||||
params:
|
||||
binary: bash
|
||||
args:
|
||||
- .evergreen/scripts/download-and-merge-coverage.sh
|
||||
- ${bucket_name}
|
||||
- ${revision}
|
||||
- ${version_id}
|
||||
working_dir: src
|
||||
silent: true
|
||||
include_expansions_in_env:
|
||||
- AWS_ACCESS_KEY_ID
|
||||
- AWS_SECRET_ACCESS_KEY
|
||||
- AWS_SESSION_TOKEN
|
||||
type: test
|
||||
- command: subprocess.exec
|
||||
params:
|
||||
binary: bash
|
||||
args:
|
||||
- .evergreen/combine-coverage.sh
|
||||
working_dir: src
|
||||
type: test
|
||||
- command: subprocess.exec
|
||||
params:
|
||||
binary: bash
|
||||
args:
|
||||
- .evergreen/scripts/upload-coverage-report.sh
|
||||
- ${bucket_name}
|
||||
- ${revision}
|
||||
- ${version_id}
|
||||
working_dir: src
|
||||
silent: true
|
||||
include_expansions_in_env:
|
||||
- AWS_ACCESS_KEY_ID
|
||||
- AWS_SECRET_ACCESS_KEY
|
||||
- AWS_SESSION_TOKEN
|
||||
type: test
|
||||
- command: s3.put
|
||||
params:
|
||||
remote_file: coverage/${revision}/${version_id}/htmlcov/index.html
|
||||
aws_key: ${AWS_ACCESS_KEY_ID}
|
||||
aws_secret: ${AWS_SECRET_ACCESS_KEY}
|
||||
aws_session_token: ${AWS_SESSION_TOKEN}
|
||||
bucket: ${bucket_name}
|
||||
local_file: src/htmlcov/index.html
|
||||
permissions: public-read
|
||||
content_type: text/html
|
||||
display_name: Coverage Report HTML
|
||||
optional: "true"
|
||||
type: setup
|
||||
|
||||
# Fetch source
|
||||
fetch source:
|
||||
- command: git.get_project
|
||||
params:
|
||||
directory: src
|
||||
|
||||
# Run server
|
||||
run server:
|
||||
- command: subprocess.exec
|
||||
params:
|
||||
binary: bash
|
||||
args:
|
||||
- .evergreen/just.sh
|
||||
- run-server
|
||||
- ${TEST_NAME}
|
||||
working_dir: src
|
||||
include_expansions_in_env:
|
||||
- VERSION
|
||||
- TOPOLOGY
|
||||
- AUTH
|
||||
- SSL
|
||||
- ORCHESTRATION_FILE
|
||||
- UV_PYTHON
|
||||
- TOOLCHAIN_VERSION
|
||||
- STORAGE_ENGINE
|
||||
- REQUIRE_API_VERSION
|
||||
- DRIVERS_TOOLS
|
||||
- TEST_CRYPT_SHARED
|
||||
- AUTH_AWS
|
||||
- LOAD_BALANCER
|
||||
- LOCAL_ATLAS
|
||||
- NO_EXT
|
||||
type: test
|
||||
- command: expansions.update
|
||||
params:
|
||||
file: ${DRIVERS_TOOLS}/mo-expansion.yml
|
||||
|
||||
# Run tests
|
||||
run tests:
|
||||
- command: subprocess.exec
|
||||
params:
|
||||
binary: bash
|
||||
args:
|
||||
- .evergreen/just.sh
|
||||
- setup-tests
|
||||
- ${TEST_NAME}
|
||||
- ${SUB_TEST_NAME}
|
||||
working_dir: src
|
||||
include_expansions_in_env:
|
||||
- AUTH
|
||||
- SSL
|
||||
- AWS_ACCESS_KEY_ID
|
||||
- AWS_SECRET_ACCESS_KEY
|
||||
- AWS_SESSION_TOKEN
|
||||
- COVERAGE
|
||||
- UV_PYTHON
|
||||
- LIBMONGOCRYPT_URL
|
||||
- MONGODB_URI
|
||||
- TOOLCHAIN_VERSION
|
||||
- DISABLE_TEST_COMMANDS
|
||||
- GREEN_FRAMEWORK
|
||||
- NO_EXT
|
||||
- COMPRESSORS
|
||||
- MONGODB_API_VERSION
|
||||
- REQUIRE_API_VERSION
|
||||
- DEBUG_LOG
|
||||
- DISABLE_FLAKY
|
||||
- ORCHESTRATION_FILE
|
||||
- OCSP_SERVER_TYPE
|
||||
- VERSION
|
||||
- IS_WIN32
|
||||
- REQUIRE_FIPS
|
||||
- TEST_MIN_DEPS
|
||||
type: test
|
||||
- command: subprocess.exec
|
||||
params:
|
||||
binary: bash
|
||||
args:
|
||||
- .evergreen/just.sh
|
||||
- run-tests
|
||||
working_dir: src
|
||||
type: test
|
||||
|
||||
# Send dashboard data
|
||||
send dashboard data:
|
||||
- command: subprocess.exec
|
||||
params:
|
||||
binary: bash
|
||||
args:
|
||||
- .evergreen/scripts/perf-submission-setup.sh
|
||||
working_dir: src
|
||||
include_expansions_in_env:
|
||||
- requester
|
||||
- revision_order_id
|
||||
- project_id
|
||||
- version_id
|
||||
- build_variant
|
||||
- parsed_order_id
|
||||
- task_name
|
||||
- task_id
|
||||
- execution
|
||||
- is_mainline
|
||||
type: test
|
||||
- command: expansions.update
|
||||
params:
|
||||
file: src/expansion.yml
|
||||
- command: subprocess.exec
|
||||
params:
|
||||
binary: bash
|
||||
args:
|
||||
- .evergreen/scripts/perf-submission.sh
|
||||
working_dir: src
|
||||
include_expansions_in_env:
|
||||
- requester
|
||||
- revision_order_id
|
||||
- project_id
|
||||
- version_id
|
||||
- build_variant
|
||||
- parsed_order_id
|
||||
- task_name
|
||||
- task_id
|
||||
- execution
|
||||
- is_mainline
|
||||
type: test
|
||||
|
||||
# Setup system
|
||||
setup system:
|
||||
- command: subprocess.exec
|
||||
params:
|
||||
binary: bash
|
||||
args:
|
||||
- .evergreen/scripts/setup-system.sh
|
||||
working_dir: src
|
||||
include_expansions_in_env:
|
||||
- is_patch
|
||||
- project
|
||||
- version_id
|
||||
type: test
|
||||
- command: expansions.update
|
||||
params:
|
||||
file: src/expansion.yml
|
||||
|
||||
# Teardown system
|
||||
teardown system:
|
||||
- command: subprocess.exec
|
||||
params:
|
||||
binary: bash
|
||||
args:
|
||||
- .evergreen/just.sh
|
||||
- teardown-tests
|
||||
working_dir: src
|
||||
type: test
|
||||
- command: subprocess.exec
|
||||
params:
|
||||
binary: bash
|
||||
args:
|
||||
- ${DRIVERS_TOOLS}/.evergreen/teardown.sh
|
||||
working_dir: src
|
||||
type: test
|
||||
|
||||
# Test numpy
|
||||
test numpy:
|
||||
- command: subprocess.exec
|
||||
params:
|
||||
binary: bash
|
||||
args:
|
||||
- .evergreen/just.sh
|
||||
- test-numpy
|
||||
working_dir: src
|
||||
include_expansions_in_env:
|
||||
- TOOLCHAIN_VERSION
|
||||
- COVERAGE
|
||||
type: test
|
||||
|
||||
# Upload coverage codecov
|
||||
upload codecov:
|
||||
- command: subprocess.exec
|
||||
params:
|
||||
binary: bash
|
||||
args:
|
||||
- .evergreen/scripts/upload-codecov.sh
|
||||
working_dir: src
|
||||
include_expansions_in_env:
|
||||
- CODECOV_TOKEN
|
||||
- build_variant
|
||||
- task_name
|
||||
- github_commit
|
||||
- github_pr_number
|
||||
- github_pr_head_branch
|
||||
- github_author
|
||||
- requester
|
||||
- branch_name
|
||||
type: test
|
||||
|
||||
# Upload coverage
|
||||
upload coverage:
|
||||
- command: ec2.assume_role
|
||||
params:
|
||||
role_arn: ${assume_role_arn}
|
||||
type: setup
|
||||
- command: s3.put
|
||||
params:
|
||||
remote_file: coverage/${revision}/${version_id}/coverage/coverage.${build_variant}.${task_name}
|
||||
aws_key: ${AWS_ACCESS_KEY_ID}
|
||||
aws_secret: ${AWS_SECRET_ACCESS_KEY}
|
||||
aws_session_token: ${AWS_SESSION_TOKEN}
|
||||
bucket: ${bucket_name}
|
||||
local_file: src/.coverage
|
||||
permissions: public-read
|
||||
content_type: text/html
|
||||
display_name: Raw Coverage Report
|
||||
optional: "true"
|
||||
type: setup
|
||||
|
||||
# Upload mo artifacts
|
||||
upload mo artifacts:
|
||||
- command: ec2.assume_role
|
||||
params:
|
||||
role_arn: ${assume_role_arn}
|
||||
type: setup
|
||||
- command: archive.targz_pack
|
||||
params:
|
||||
target: mongo-coredumps.tgz
|
||||
source_dir: ./
|
||||
include:
|
||||
- ./**.core
|
||||
- ./**.mdmp
|
||||
- command: s3.put
|
||||
params:
|
||||
remote_file: ${build_variant}/${revision}/${version_id}/${build_id}/coredumps/${task_id}-${execution}-mongodb-coredumps.tar.gz
|
||||
aws_key: ${AWS_ACCESS_KEY_ID}
|
||||
aws_secret: ${AWS_SECRET_ACCESS_KEY}
|
||||
aws_session_token: ${AWS_SESSION_TOKEN}
|
||||
bucket: ${bucket_name}
|
||||
local_file: mongo-coredumps.tgz
|
||||
permissions: public-read
|
||||
content_type: ${content_type|application/x-gzip}
|
||||
display_name: Core Dumps - Execution
|
||||
optional: "true"
|
||||
type: setup
|
||||
- command: s3.put
|
||||
params:
|
||||
remote_file: ${build_variant}/${revision}/${version_id}/${build_id}/logs/${task_id}-${execution}-drivers-tools-logs.tar.gz
|
||||
aws_key: ${AWS_ACCESS_KEY_ID}
|
||||
aws_secret: ${AWS_SECRET_ACCESS_KEY}
|
||||
aws_session_token: ${AWS_SESSION_TOKEN}
|
||||
bucket: ${bucket_name}
|
||||
local_file: ${DRIVERS_TOOLS}/.evergreen/test_logs.tar.gz
|
||||
permissions: public-read
|
||||
content_type: ${content_type|application/x-gzip}
|
||||
display_name: drivers-tools-logs.tar.gz
|
||||
optional: "true"
|
||||
type: setup
|
||||
|
||||
# Upload test results
|
||||
upload test results:
|
||||
- command: attach.results
|
||||
params:
|
||||
file_location: ${DRIVERS_TOOLS}/results.json
|
||||
- command: attach.xunit_results
|
||||
params:
|
||||
file: src/xunit-results/TEST-*.xml
|
||||
File diff suppressed because it is too large
Load Diff
File diff suppressed because it is too large
Load Diff
56
.evergreen/remove-unimplemented-tests.sh
Executable file
56
.evergreen/remove-unimplemented-tests.sh
Executable file
@ -0,0 +1,56 @@
|
||||
#!/bin/bash
|
||||
PYMONGO=$(dirname "$(cd "$(dirname "$0")" || exit; pwd)")
|
||||
|
||||
rm $PYMONGO/test/transactions/legacy/errors-client.json # PYTHON-1894
|
||||
rm $PYMONGO/test/connection_monitoring/wait-queue-fairness.json # PYTHON-1873
|
||||
rm $PYMONGO/test/discovery_and_monitoring/unified/pool-clear-application-error.json # PYTHON-4918
|
||||
rm $PYMONGO/test/discovery_and_monitoring/unified/pool-clear-checkout-error.json # PYTHON-4918
|
||||
rm $PYMONGO/test/discovery_and_monitoring/unified/pool-clear-min-pool-size-error.json # PYTHON-4918
|
||||
rm $PYMONGO/test/client-side-encryption/spec/unified/client-bulkWrite-qe.json # PYTHON-4929
|
||||
|
||||
# Python doesn't implement DRIVERS-3064
|
||||
rm $PYMONGO/test/collection_management/listCollections-rawdata.json
|
||||
rm $PYMONGO/test/crud/unified/aggregate-rawdata.json
|
||||
rm $PYMONGO/test/crud/unified/bulkWrite-deleteMany-rawdata.json
|
||||
rm $PYMONGO/test/crud/unified/bulkWrite-deleteOne-rawdata.json
|
||||
rm $PYMONGO/test/crud/unified/bulkWrite-replaceOne-rawdata.json
|
||||
rm $PYMONGO/test/crud/unified/bulkWrite-updateMany-rawdata.json
|
||||
rm $PYMONGO/test/crud/unified/bulkWrite-updateOne-rawdata.json
|
||||
rm $PYMONGO/test/crud/unified/client-bulkWrite-delete-rawdata.json
|
||||
rm $PYMONGO/test/crud/unified/client-bulkWrite-replaceOne-rawdata.json
|
||||
rm $PYMONGO/test/crud/unified/client-bulkWrite-update-rawdata.json
|
||||
rm $PYMONGO/test/crud/unified/count-rawdata.json
|
||||
rm $PYMONGO/test/crud/unified/countDocuments-rawdata.json
|
||||
rm $PYMONGO/test/crud/unified/db-aggregate-rawdata.json
|
||||
rm $PYMONGO/test/crud/unified/deleteMany-rawdata.json
|
||||
rm $PYMONGO/test/crud/unified/deleteOne-rawdata.json
|
||||
rm $PYMONGO/test/crud/unified/distinct-rawdata.json
|
||||
rm $PYMONGO/test/crud/unified/estimatedDocumentCount-rawdata.json
|
||||
rm $PYMONGO/test/crud/unified/find-rawdata.json
|
||||
rm $PYMONGO/test/crud/unified/findOneAndDelete-rawdata.json
|
||||
rm $PYMONGO/test/crud/unified/findOneAndReplace-rawdata.json
|
||||
rm $PYMONGO/test/crud/unified/findOneAndUpdate-rawdata.json
|
||||
rm $PYMONGO/test/crud/unified/insertMany-rawdata.json
|
||||
rm $PYMONGO/test/crud/unified/insertOne-rawdata.json
|
||||
rm $PYMONGO/test/crud/unified/replaceOne-rawdata.json
|
||||
rm $PYMONGO/test/crud/unified/updateMany-rawdata.json
|
||||
rm $PYMONGO/test/crud/unified/updateOne-rawdata.json
|
||||
rm $PYMONGO/test/index_management/index-rawdata.json
|
||||
|
||||
# PyMongo does not support modifyCollection
|
||||
rm $PYMONGO/test/collection_management/modifyCollection-*.json
|
||||
|
||||
# PYTHON-5248 - Remove support for MongoDB 4.0
|
||||
find /$PYMONGO/test -type f -name 'pre-42-*.json' -delete
|
||||
|
||||
# PYTHON-3359 - Remove Database and Collection level timeout override
|
||||
rm $PYMONGO/test/csot/override-collection-timeoutMS.json
|
||||
rm $PYMONGO/test/csot/override-database-timeoutMS.json
|
||||
|
||||
# PYTHON-2943 - Socks5 Proxy Support
|
||||
rm $PYMONGO/test/uri_options/proxy-options.json
|
||||
|
||||
# PYTHON-5517 - Avoid clearing the connection pool when the server connection rate limiter triggers
|
||||
rm $PYMONGO/test/discovery_and_monitoring/unified/backpressure-*.json
|
||||
|
||||
echo "Done removing unimplemented tests"
|
||||
@ -1,6 +1,6 @@
|
||||
#!/bin/bash
|
||||
# exit when any command fails
|
||||
set -e
|
||||
# Resync test files from the specifications repo.
|
||||
set -eu
|
||||
PYMONGO=$(dirname "$(cd "$(dirname "$0")"; pwd)")
|
||||
SPECS=${MDB_SPECS:-~/Work/specifications}
|
||||
|
||||
@ -45,9 +45,12 @@ then
|
||||
fi
|
||||
|
||||
# Ensure the JSON files are up to date.
|
||||
cd $SPECS/source
|
||||
make
|
||||
cd -
|
||||
if ! [ -n "${CI:-}" ]
|
||||
then
|
||||
cd $SPECS/source
|
||||
make
|
||||
cd -
|
||||
fi
|
||||
# cpjson unified-test-format/tests/invalid unified-test-format/invalid
|
||||
# * param1: Path to spec tests dir in specifications repo
|
||||
# * param2: Path to where the corresponding tests live in Python.
|
||||
@ -73,9 +76,6 @@ do
|
||||
auth)
|
||||
cpjson auth/tests/ auth
|
||||
;;
|
||||
atlas-data-lake-testing|data_lake)
|
||||
cpjson atlas-data-lake-testing/tests/ data_lake
|
||||
;;
|
||||
bson-binary-vector|bson_binary_vector)
|
||||
cpjson bson-binary-vector/tests/ bson_binary_vector
|
||||
;;
|
||||
@ -94,6 +94,9 @@ do
|
||||
change-streams|change_streams)
|
||||
cpjson change-streams/tests/ change_streams/
|
||||
;;
|
||||
client-backpressure|client_backpressure)
|
||||
cpjson client-backpressure/tests client-backpressure
|
||||
;;
|
||||
client-side-encryption|csfle|fle)
|
||||
cpjson client-side-encryption/tests/ client-side-encryption/spec
|
||||
cpjson client-side-encryption/corpus/ client-side-encryption/corpus
|
||||
@ -110,7 +113,6 @@ do
|
||||
cmap|CMAP|connection-monitoring-and-pooling)
|
||||
cpjson connection-monitoring-and-pooling/tests/logging connection_logging
|
||||
cpjson connection-monitoring-and-pooling/tests/cmap-format connection_monitoring
|
||||
rm $PYMONGO/test/connection_monitoring/wait-queue-fairness.json # PYTHON-1873
|
||||
;;
|
||||
apm|APM|command-monitoring|command_monitoring)
|
||||
cpjson command-logging-and-monitoring/tests/monitoring command_monitoring
|
||||
@ -131,6 +133,9 @@ do
|
||||
gridfs)
|
||||
cpjson gridfs/tests gridfs
|
||||
;;
|
||||
handshake)
|
||||
cpjson mongodb-handshake/tests handshake
|
||||
;;
|
||||
index|index-management)
|
||||
cpjson index-management/tests index_management
|
||||
;;
|
||||
@ -171,7 +176,7 @@ do
|
||||
;;
|
||||
server-selection|server_selection)
|
||||
cpjson server-selection/tests/ server_selection
|
||||
rm -rf $PYMONGO/test/server_selection/logging
|
||||
rm -rf $PYMONGO/test/server_selection/logging # these tests live in server_selection_logging
|
||||
cpjson server-selection/tests/logging server_selection_logging
|
||||
;;
|
||||
server-selection-logging|server_selection_logging)
|
||||
@ -183,7 +188,6 @@ do
|
||||
transactions|transactions-convenient-api)
|
||||
cpjson transactions/tests/ transactions
|
||||
cpjson transactions-convenient-api/tests/ transactions-convenient-api
|
||||
rm $PYMONGO/test/transactions/legacy/errors-client.json # PYTHON-1894
|
||||
;;
|
||||
unified|unified-test-format)
|
||||
cpjson unified-test-format/tests/ unified-test-format/
|
||||
|
||||
@ -1,12 +0,0 @@
|
||||
#!/bin/bash
|
||||
set -o errexit # Exit the script with error if any of the commands fail
|
||||
HERE=$(dirname ${BASH_SOURCE:-$0})
|
||||
. $DRIVERS_TOOLS/.evergreen/csfle/azurekms/setup-secrets.sh
|
||||
export LIBMONGOCRYPT_URL=https://s3.amazonaws.com/mciuploads/libmongocrypt/debian11/master/latest/libmongocrypt.tar.gz
|
||||
SKIP_SERVERS=1 bash $HERE/setup-encryption.sh
|
||||
PYTHON_BINARY=/opt/mongodbtoolchain/v4/bin/python3 \
|
||||
KEY_NAME="${AZUREKMS_KEYNAME}" \
|
||||
KEY_VAULT_ENDPOINT="${AZUREKMS_KEYVAULTENDPOINT}" \
|
||||
SUCCESS=false TEST_FLE_AZURE_AUTO=1 \
|
||||
$HERE/just.sh test-eg
|
||||
bash $HERE/teardown-encryption.sh
|
||||
@ -1,29 +0,0 @@
|
||||
#!/bin/bash
|
||||
set -o errexit # Exit the script with error if any of the commands fail
|
||||
HERE=$(dirname ${BASH_SOURCE:-$0})
|
||||
source ${DRIVERS_TOOLS}/.evergreen/csfle/azurekms/secrets-export.sh
|
||||
echo "Copying files ... begin"
|
||||
export AZUREKMS_RESOURCEGROUP=${AZUREKMS_RESOURCEGROUP}
|
||||
export AZUREKMS_VMNAME=${AZUREKMS_VMNAME}
|
||||
export AZUREKMS_PRIVATEKEYPATH=/tmp/testazurekms_privatekey
|
||||
export LIBMONGOCRYPT_URL=https://s3.amazonaws.com/mciuploads/libmongocrypt/debian11/master/latest/libmongocrypt.tar.gz
|
||||
SKIP_SERVERS=1 bash $HERE/setup-encryption.sh
|
||||
# Set up the remote files to test.
|
||||
git add .
|
||||
git commit -m "add files" || true
|
||||
git archive -o /tmp/mongo-python-driver.tar HEAD
|
||||
tar -rf /tmp/mongo-python-driver.tar libmongocrypt
|
||||
gzip -f /tmp/mongo-python-driver.tar
|
||||
# shellcheck disable=SC2088
|
||||
AZUREKMS_SRC="/tmp/mongo-python-driver.tar.gz" AZUREKMS_DST="~/" \
|
||||
$DRIVERS_TOOLS/.evergreen/csfle/azurekms/copy-file.sh
|
||||
echo "Copying files ... end"
|
||||
echo "Untarring file ... begin"
|
||||
AZUREKMS_CMD="tar xf mongo-python-driver.tar.gz" \
|
||||
$DRIVERS_TOOLS/.evergreen/csfle/azurekms/run-command.sh
|
||||
echo "Untarring file ... end"
|
||||
echo "Running test ... begin"
|
||||
AZUREKMS_CMD="KEY_NAME=\"$AZUREKMS_KEYNAME\" KEY_VAULT_ENDPOINT=\"$AZUREKMS_KEYVAULTENDPOINT\" SUCCESS=true TEST_FLE_AZURE_AUTO=1 bash ./.evergreen/just.sh test-eg" \
|
||||
$DRIVERS_TOOLS/.evergreen/csfle/azurekms/run-command.sh
|
||||
echo "Running test ... end"
|
||||
bash $HERE/teardown-encryption.sh
|
||||
@ -1,10 +0,0 @@
|
||||
#!/bin/bash
|
||||
set -o errexit # Exit the script with error if any of the commands fail
|
||||
|
||||
export PATH="/opt/python/3.9/bin:${PATH}"
|
||||
python --version
|
||||
pushd ./test/lambda
|
||||
|
||||
. build.sh
|
||||
popd
|
||||
. ${DRIVERS_TOOLS}/.evergreen/aws_lambda/run-deployed-lambda-aws-tests.sh
|
||||
@ -1,27 +0,0 @@
|
||||
#!/bin/bash
|
||||
set -o errexit # Exit the script with error if any of the commands fail
|
||||
HERE=$(dirname ${BASH_SOURCE:-$0})
|
||||
|
||||
source ${DRIVERS_TOOLS}/.evergreen/csfle/gcpkms/secrets-export.sh
|
||||
echo "Copying files ... begin"
|
||||
export GCPKMS_GCLOUD=${GCPKMS_GCLOUD}
|
||||
export GCPKMS_PROJECT=${GCPKMS_PROJECT}
|
||||
export GCPKMS_ZONE=${GCPKMS_ZONE}
|
||||
export GCPKMS_INSTANCENAME=${GCPKMS_INSTANCENAME}
|
||||
export LIBMONGOCRYPT_URL=https://s3.amazonaws.com/mciuploads/libmongocrypt/debian11/master/latest/libmongocrypt.tar.gz
|
||||
SKIP_SERVERS=1 bash $HERE/setup-encryption.sh
|
||||
# Set up the remote files to test.
|
||||
git add .
|
||||
git commit -m "add files" || true
|
||||
git archive -o /tmp/mongo-python-driver.tar HEAD
|
||||
tar -rf /tmp/mongo-python-driver.tar libmongocrypt
|
||||
gzip -f /tmp/mongo-python-driver.tar
|
||||
GCPKMS_SRC=/tmp/mongo-python-driver.tar.gz GCPKMS_DST=$GCPKMS_INSTANCENAME: $DRIVERS_TOOLS/.evergreen/csfle/gcpkms/copy-file.sh
|
||||
echo "Copying files ... end"
|
||||
echo "Untarring file ... begin"
|
||||
GCPKMS_CMD="tar xf mongo-python-driver.tar.gz" $DRIVERS_TOOLS/.evergreen/csfle/gcpkms/run-command.sh
|
||||
echo "Untarring file ... end"
|
||||
echo "Running test ... begin"
|
||||
GCPKMS_CMD="SUCCESS=true TEST_FLE_GCP_AUTO=1 ./.evergreen/just.sh test-eg" $DRIVERS_TOOLS/.evergreen/csfle/gcpkms/run-command.sh
|
||||
echo "Running test ... end"
|
||||
bash $HERE/teardown-encryption.sh
|
||||
@ -1,33 +0,0 @@
|
||||
#!/bin/bash -ex
|
||||
|
||||
set -o errexit # Exit the script with error if any of the commands fail
|
||||
set -x
|
||||
|
||||
. .evergreen/utils.sh
|
||||
|
||||
if [ -z "${PYTHON_BINARY:-}" ]; then
|
||||
PYTHON_BINARY=$(find_python3)
|
||||
fi
|
||||
|
||||
# Use the previous commit if this was not a PR run.
|
||||
if [ "$BASE_SHA" == "$HEAD_SHA" ]; then
|
||||
BASE_SHA=$(git rev-parse HEAD~1)
|
||||
fi
|
||||
|
||||
function get_import_time() {
|
||||
local log_file
|
||||
createvirtualenv "$PYTHON_BINARY" import-venv
|
||||
python -m pip install -q ".[aws,encryption,gssapi,ocsp,snappy,zstd]"
|
||||
# Import once to cache modules
|
||||
python -c "import pymongo"
|
||||
log_file="pymongo-$1.log"
|
||||
python -X importtime -c "import pymongo" 2> $log_file
|
||||
}
|
||||
|
||||
get_import_time $HEAD_SHA
|
||||
git stash || true
|
||||
git checkout $BASE_SHA
|
||||
get_import_time $BASE_SHA
|
||||
git checkout $HEAD_SHA
|
||||
git stash apply || true
|
||||
python tools/compare_import_time.py $HEAD_SHA $BASE_SHA
|
||||
@ -1,7 +1,6 @@
|
||||
#!/bin/bash
|
||||
|
||||
# Don't trace since the URI contains a password that shouldn't show up in the logs
|
||||
set -o errexit # Exit the script with error if any of the commands fail
|
||||
# Script run on an ECS host to test MONGODB-AWS.
|
||||
set -eu
|
||||
|
||||
############################################
|
||||
# Main Program #
|
||||
@ -20,15 +19,14 @@ fi
|
||||
# Now we can safely enable xtrace
|
||||
set -o xtrace
|
||||
|
||||
# Install python with pip.
|
||||
PYTHON_VER="python3.9"
|
||||
# Install a c compiler.
|
||||
apt-get -qq update < /dev/null > /dev/null
|
||||
apt-get -qq install $PYTHON_VER $PYTHON_VER-venv build-essential $PYTHON_VER-dev -y < /dev/null > /dev/null
|
||||
apt-get -q install -y build-essential
|
||||
|
||||
export PYTHON_BINARY=$PYTHON_VER
|
||||
export TEST_AUTH_AWS=1
|
||||
export AUTH="auth"
|
||||
export SET_XTRACE_ON=1
|
||||
cd src
|
||||
rm -rf .venv
|
||||
bash .evergreen/just.sh test-eg
|
||||
rm -f .evergreen/scripts/test-env.sh || true
|
||||
rm -f .evergreen/scripts/env.sh || true
|
||||
bash ./.evergreen/just.sh setup-tests auth_aws ecs-remote
|
||||
bash .evergreen/just.sh run-tests
|
||||
|
||||
@ -1,60 +0,0 @@
|
||||
#!/bin/bash
|
||||
|
||||
set +x # Disable debug trace
|
||||
set -eu
|
||||
|
||||
echo "Running MONGODB-OIDC remote tests"
|
||||
|
||||
OIDC_ENV=${OIDC_ENV:-"test"}
|
||||
|
||||
# Make sure DRIVERS_TOOLS is set.
|
||||
if [ -z "$DRIVERS_TOOLS" ]; then
|
||||
echo "Must specify DRIVERS_TOOLS"
|
||||
exit 1
|
||||
fi
|
||||
|
||||
# Set up the remote files to test.
|
||||
git add .
|
||||
git commit -m "add files" || true
|
||||
export TEST_TAR_FILE=/tmp/mongo-python-driver.tgz
|
||||
git archive -o $TEST_TAR_FILE HEAD
|
||||
|
||||
pushd $DRIVERS_TOOLS
|
||||
|
||||
if [ $OIDC_ENV == "test" ]; then
|
||||
echo "Test OIDC environment does not support remote test!"
|
||||
exit 1
|
||||
|
||||
elif [ $OIDC_ENV == "azure" ]; then
|
||||
export AZUREOIDC_DRIVERS_TAR_FILE=$TEST_TAR_FILE
|
||||
export AZUREOIDC_TEST_CMD="OIDC_ENV=azure ./.evergreen/run-mongodb-oidc-test.sh"
|
||||
bash ./.evergreen/auth_oidc/azure/run-driver-test.sh
|
||||
|
||||
elif [ $OIDC_ENV == "gcp" ]; then
|
||||
export GCPOIDC_DRIVERS_TAR_FILE=$TEST_TAR_FILE
|
||||
export GCPOIDC_TEST_CMD="OIDC_ENV=gcp ./.evergreen/run-mongodb-oidc-test.sh"
|
||||
bash ./.evergreen/auth_oidc/gcp/run-driver-test.sh
|
||||
|
||||
elif [ $OIDC_ENV == "k8s" ]; then
|
||||
# Make sure K8S_VARIANT is set.
|
||||
if [ -z "$K8S_VARIANT" ]; then
|
||||
echo "Must specify K8S_VARIANT"
|
||||
popd
|
||||
exit 1
|
||||
fi
|
||||
|
||||
bash ./.evergreen/auth_oidc/k8s/setup-pod.sh
|
||||
bash ./.evergreen/auth_oidc/k8s/run-self-test.sh
|
||||
export K8S_DRIVERS_TAR_FILE=$TEST_TAR_FILE
|
||||
export K8S_TEST_CMD="OIDC_ENV=k8s ./.evergreen/run-mongodb-oidc-test.sh"
|
||||
source ./.evergreen/auth_oidc/k8s/secrets-export.sh # for MONGODB_URI
|
||||
bash ./.evergreen/auth_oidc/k8s/run-driver-test.sh
|
||||
bash ./.evergreen/auth_oidc/k8s/teardown-pod.sh
|
||||
|
||||
else
|
||||
echo "Unrecognized OIDC_ENV $OIDC_ENV"
|
||||
pod
|
||||
exit 1
|
||||
fi
|
||||
|
||||
popd
|
||||
@ -1,35 +1,17 @@
|
||||
#!/bin/bash
|
||||
|
||||
set +x # Disable debug trace
|
||||
# Script run on a remote host to test MONGODB-OIDC.
|
||||
set -eu
|
||||
|
||||
echo "Running MONGODB-OIDC authentication tests"
|
||||
|
||||
OIDC_ENV=${OIDC_ENV:-"test"}
|
||||
|
||||
if [ $OIDC_ENV == "test" ]; then
|
||||
# Make sure DRIVERS_TOOLS is set.
|
||||
if [ -z "$DRIVERS_TOOLS" ]; then
|
||||
echo "Must specify DRIVERS_TOOLS"
|
||||
exit 1
|
||||
fi
|
||||
source ${DRIVERS_TOOLS}/.evergreen/auth_oidc/secrets-export.sh
|
||||
|
||||
elif [ $OIDC_ENV == "azure" ]; then
|
||||
source ./env.sh
|
||||
|
||||
elif [ $OIDC_ENV == "gcp" ]; then
|
||||
source ./secrets-export.sh
|
||||
|
||||
elif [ $OIDC_ENV == "k8s" ]; then
|
||||
echo "Running oidc on k8s"
|
||||
echo "Running MONGODB-OIDC authentication tests on ${OIDC_ENV}..."
|
||||
|
||||
if [ ${OIDC_ENV} == "k8s" ]; then
|
||||
SUB_TEST_NAME=$K8S_VARIANT-remote
|
||||
else
|
||||
echo "Unrecognized OIDC_ENV $OIDC_ENV"
|
||||
exit 1
|
||||
SUB_TEST_NAME=$OIDC_ENV-remote
|
||||
sudo apt-get install -y python3-dev build-essential
|
||||
fi
|
||||
|
||||
export TEST_AUTH_OIDC=1
|
||||
export COVERAGE=1
|
||||
export AUTH="auth"
|
||||
bash ./.evergreen/just.sh test-eg "${@:1}"
|
||||
bash ./.evergreen/just.sh setup-tests auth_oidc $SUB_TEST_NAME
|
||||
bash ./.evergreen/just.sh run-tests "${@:1}"
|
||||
|
||||
echo "Running MONGODB-OIDC authentication tests on ${OIDC_ENV}... done."
|
||||
|
||||
@ -1,19 +0,0 @@
|
||||
#!/bin/bash
|
||||
|
||||
set -o xtrace
|
||||
set -o errexit
|
||||
|
||||
git clone --depth 1 https://github.com/mongodb/specifications.git
|
||||
pushd specifications/source/benchmarking/data
|
||||
tar xf extended_bson.tgz
|
||||
tar xf parallel.tgz
|
||||
tar xf single_and_multi_document.tgz
|
||||
popd
|
||||
|
||||
export TEST_PATH="${PROJECT_DIRECTORY}/specifications/source/benchmarking/data"
|
||||
export OUTPUT_FILE="${PROJECT_DIRECTORY}/results.json"
|
||||
|
||||
export PYTHON_BINARY=/opt/mongodbtoolchain/v4/bin/python3
|
||||
export PERF_TEST=1
|
||||
|
||||
bash ./.evergreen/just.sh test-eg
|
||||
@ -1,294 +1,45 @@
|
||||
#!/bin/bash
|
||||
set -o errexit # Exit the script with error if any of the commands fail
|
||||
set -o xtrace
|
||||
# Run a test suite that was configured with setup-tests.sh.
|
||||
set -eu
|
||||
|
||||
# Note: It is assumed that you have already set up a virtual environment before running this file.
|
||||
SCRIPT_DIR=$(dirname ${BASH_SOURCE:-$0})
|
||||
SCRIPT_DIR="$( cd -- "$SCRIPT_DIR" > /dev/null 2>&1 && pwd )"
|
||||
ROOT_DIR="$(dirname $SCRIPT_DIR)"
|
||||
|
||||
# Supported/used environment variables:
|
||||
# AUTH Set to enable authentication. Defaults to "noauth"
|
||||
# SSL Set to enable SSL. Defaults to "nossl"
|
||||
# GREEN_FRAMEWORK The green framework to test with, if any.
|
||||
# COVERAGE If non-empty, run the test suite with coverage.
|
||||
# COMPRESSORS If non-empty, install appropriate compressor.
|
||||
# LIBMONGOCRYPT_URL The URL to download libmongocrypt.
|
||||
# TEST_DATA_LAKE If non-empty, run data lake tests.
|
||||
# TEST_ENCRYPTION If non-empty, run encryption tests.
|
||||
# TEST_CRYPT_SHARED If non-empty, install crypt_shared lib.
|
||||
# TEST_SERVERLESS If non-empy, test on serverless.
|
||||
# TEST_LOADBALANCER If non-empy, test load balancing.
|
||||
# TEST_FLE_AZURE_AUTO If non-empy, test auto FLE on Azure
|
||||
# TEST_FLE_GCP_AUTO If non-empy, test auto FLE on GCP
|
||||
# TEST_PYOPENSSL If non-empy, test with PyOpenSSL
|
||||
# TEST_ENTERPRISE_AUTH If non-empty, test with Enterprise Auth
|
||||
# TEST_AUTH_AWS If non-empty, test AWS Auth Mechanism
|
||||
# TEST_AUTH_OIDC If non-empty, test OIDC Auth Mechanism
|
||||
# TEST_PERF If non-empty, run performance tests
|
||||
# TEST_OCSP If non-empty, run OCSP tests
|
||||
# TEST_ATLAS If non-empty, test Atlas connections
|
||||
# TEST_INDEX_MANAGEMENT If non-empty, run index management tests
|
||||
# TEST_ENCRYPTION_PYOPENSSL If non-empy, test encryption with PyOpenSSL
|
||||
PREV_DIR=$(pwd)
|
||||
cd $ROOT_DIR
|
||||
|
||||
AUTH=${AUTH:-noauth}
|
||||
SSL=${SSL:-nossl}
|
||||
TEST_SUITES=${TEST_SUITES:-}
|
||||
TEST_ARGS="${*:1}"
|
||||
|
||||
export PIP_QUIET=1 # Quiet by default
|
||||
export PIP_PREFER_BINARY=1 # Prefer binary dists by default
|
||||
|
||||
set +x
|
||||
PYTHON_IMPL=$(uv run --frozen python -c "import platform; print(platform.python_implementation())")
|
||||
|
||||
# Try to source local Drivers Secrets
|
||||
if [ -f ./secrets-export.sh ]; then
|
||||
echo "Sourcing secrets"
|
||||
source ./secrets-export.sh
|
||||
# Try to source the env file.
|
||||
if [ -f $SCRIPT_DIR/scripts/env.sh ]; then
|
||||
echo "Sourcing env inputs"
|
||||
. $SCRIPT_DIR/scripts/env.sh
|
||||
else
|
||||
echo "Not sourcing secrets"
|
||||
echo "Not sourcing env inputs"
|
||||
fi
|
||||
|
||||
# Start compiling the args we'll pass to uv.
|
||||
# Run in an isolated environment so as not to pollute the base venv.
|
||||
UV_ARGS=("--isolated --frozen --extra test")
|
||||
|
||||
# Ensure C extensions if applicable.
|
||||
if [ -z "${NO_EXT:-}" ] && [ "$PYTHON_IMPL" = "CPython" ]; then
|
||||
uv run --frozen tools/fail_if_no_c.py
|
||||
fi
|
||||
|
||||
if [ "$AUTH" != "noauth" ]; then
|
||||
if [ -n "$TEST_DATA_LAKE" ]; then
|
||||
export DB_USER="mhuser"
|
||||
export DB_PASSWORD="pencil"
|
||||
elif [ -n "$TEST_SERVERLESS" ]; then
|
||||
source "${DRIVERS_TOOLS}"/.evergreen/serverless/secrets-export.sh
|
||||
export DB_USER=$SERVERLESS_ATLAS_USER
|
||||
export DB_PASSWORD=$SERVERLESS_ATLAS_PASSWORD
|
||||
export MONGODB_URI="$SERVERLESS_URI"
|
||||
echo "MONGODB_URI=$MONGODB_URI"
|
||||
export SINGLE_MONGOS_LB_URI=$MONGODB_URI
|
||||
export MULTI_MONGOS_LB_URI=$MONGODB_URI
|
||||
elif [ -n "$TEST_AUTH_OIDC" ]; then
|
||||
export DB_USER=$OIDC_ADMIN_USER
|
||||
export DB_PASSWORD=$OIDC_ADMIN_PWD
|
||||
export DB_IP="$MONGODB_URI"
|
||||
else
|
||||
export DB_USER="bob"
|
||||
export DB_PASSWORD="pwd123"
|
||||
fi
|
||||
echo "Added auth, DB_USER: $DB_USER"
|
||||
fi
|
||||
|
||||
if [ -n "$TEST_ENTERPRISE_AUTH" ]; then
|
||||
UV_ARGS+=("--extra gssapi")
|
||||
if [ "Windows_NT" = "$OS" ]; then
|
||||
echo "Setting GSSAPI_PASS"
|
||||
export GSSAPI_PASS=${SASL_PASS}
|
||||
export GSSAPI_CANONICALIZE="true"
|
||||
else
|
||||
# BUILD-3830
|
||||
touch krb5.conf.empty
|
||||
export KRB5_CONFIG=${PROJECT_DIRECTORY}/.evergreen/krb5.conf.empty
|
||||
|
||||
echo "Writing keytab"
|
||||
echo ${KEYTAB_BASE64} | base64 -d > ${PROJECT_DIRECTORY}/.evergreen/drivers.keytab
|
||||
echo "Running kinit"
|
||||
kinit -k -t ${PROJECT_DIRECTORY}/.evergreen/drivers.keytab -p ${PRINCIPAL}
|
||||
fi
|
||||
echo "Setting GSSAPI variables"
|
||||
export GSSAPI_HOST=${SASL_HOST}
|
||||
export GSSAPI_PORT=${SASL_PORT}
|
||||
export GSSAPI_PRINCIPAL=${PRINCIPAL}
|
||||
|
||||
export TEST_SUITES="auth"
|
||||
fi
|
||||
|
||||
if [ -n "$TEST_LOADBALANCER" ]; then
|
||||
export LOAD_BALANCER=1
|
||||
export SINGLE_MONGOS_LB_URI="${SINGLE_MONGOS_LB_URI:-mongodb://127.0.0.1:8000/?loadBalanced=true}"
|
||||
export MULTI_MONGOS_LB_URI="${MULTI_MONGOS_LB_URI:-mongodb://127.0.0.1:8001/?loadBalanced=true}"
|
||||
export TEST_SUITES="load_balancer"
|
||||
fi
|
||||
|
||||
if [ "$SSL" != "nossl" ]; then
|
||||
export CLIENT_PEM="$DRIVERS_TOOLS/.evergreen/x509gen/client.pem"
|
||||
export CA_PEM="$DRIVERS_TOOLS/.evergreen/x509gen/ca.pem"
|
||||
|
||||
if [ -n "$TEST_LOADBALANCER" ]; then
|
||||
export SINGLE_MONGOS_LB_URI="${SINGLE_MONGOS_LB_URI}&tls=true"
|
||||
export MULTI_MONGOS_LB_URI="${MULTI_MONGOS_LB_URI}&tls=true"
|
||||
fi
|
||||
fi
|
||||
|
||||
if [ "$COMPRESSORS" = "snappy" ]; then
|
||||
UV_ARGS+=("--extra snappy")
|
||||
elif [ "$COMPRESSORS" = "zstd" ]; then
|
||||
UV_ARGS+=("--extra zstandard")
|
||||
fi
|
||||
|
||||
# PyOpenSSL test setup.
|
||||
if [ -n "$TEST_PYOPENSSL" ]; then
|
||||
UV_ARGS+=("--extra ocsp")
|
||||
fi
|
||||
|
||||
if [ -n "$TEST_ENCRYPTION" ] || [ -n "$TEST_FLE_AZURE_AUTO" ] || [ -n "$TEST_FLE_GCP_AUTO" ]; then
|
||||
# Check for libmongocrypt download.
|
||||
if [ ! -d "libmongocrypt" ]; then
|
||||
echo "Run encryption setup first!"
|
||||
exit 1
|
||||
fi
|
||||
|
||||
UV_ARGS+=("--extra encryption")
|
||||
# TODO: Test with 'pip install pymongocrypt'
|
||||
UV_ARGS+=("--group pymongocrypt_source")
|
||||
|
||||
# Use the nocrypto build to avoid dependency issues with older windows/python versions.
|
||||
BASE=$(pwd)/libmongocrypt/nocrypto
|
||||
if [ -f "${BASE}/lib/libmongocrypt.so" ]; then
|
||||
PYMONGOCRYPT_LIB=${BASE}/lib/libmongocrypt.so
|
||||
elif [ -f "${BASE}/lib/libmongocrypt.dylib" ]; then
|
||||
PYMONGOCRYPT_LIB=${BASE}/lib/libmongocrypt.dylib
|
||||
elif [ -f "${BASE}/bin/mongocrypt.dll" ]; then
|
||||
PYMONGOCRYPT_LIB=${BASE}/bin/mongocrypt.dll
|
||||
# libmongocrypt's windows dll is not marked executable.
|
||||
chmod +x $PYMONGOCRYPT_LIB
|
||||
PYMONGOCRYPT_LIB=$(cygpath -m $PYMONGOCRYPT_LIB)
|
||||
elif [ -f "${BASE}/lib64/libmongocrypt.so" ]; then
|
||||
PYMONGOCRYPT_LIB=${BASE}/lib64/libmongocrypt.so
|
||||
else
|
||||
echo "Cannot find libmongocrypt shared object file"
|
||||
exit 1
|
||||
fi
|
||||
export PYMONGOCRYPT_LIB
|
||||
# Ensure pymongocrypt is working properly.
|
||||
# shellcheck disable=SC2048
|
||||
uv run ${UV_ARGS[*]} python -c "import pymongocrypt; print('pymongocrypt version: '+pymongocrypt.__version__)"
|
||||
# shellcheck disable=SC2048
|
||||
uv run ${UV_ARGS[*]} python -c "import pymongocrypt; print('libmongocrypt version: '+pymongocrypt.libmongocrypt_version())"
|
||||
# PATH is updated by configure-env.sh for access to mongocryptd.
|
||||
fi
|
||||
|
||||
if [ -n "$TEST_ENCRYPTION" ]; then
|
||||
if [ -n "$TEST_ENCRYPTION_PYOPENSSL" ]; then
|
||||
UV_ARGS+=("--extra ocsp")
|
||||
fi
|
||||
|
||||
if [ -n "$TEST_CRYPT_SHARED" ]; then
|
||||
CRYPT_SHARED_DIR=`dirname $CRYPT_SHARED_LIB_PATH`
|
||||
echo "using crypt_shared_dir $CRYPT_SHARED_DIR"
|
||||
export DYLD_FALLBACK_LIBRARY_PATH=$CRYPT_SHARED_DIR:$DYLD_FALLBACK_LIBRARY_PATH
|
||||
export LD_LIBRARY_PATH=$CRYPT_SHARED_DIR:$LD_LIBRARY_PATH
|
||||
export PATH=$CRYPT_SHARED_DIR:$PATH
|
||||
fi
|
||||
# Only run the encryption tests.
|
||||
TEST_SUITES="encryption"
|
||||
fi
|
||||
|
||||
if [ -n "$TEST_FLE_AZURE_AUTO" ] || [ -n "$TEST_FLE_GCP_AUTO" ]; then
|
||||
if [[ -z "$SUCCESS" ]]; then
|
||||
echo "Must define SUCCESS"
|
||||
exit 1
|
||||
fi
|
||||
|
||||
if echo "$MONGODB_URI" | grep -q "@"; then
|
||||
echo "MONGODB_URI unexpectedly contains user credentials in FLE test!";
|
||||
exit 1
|
||||
fi
|
||||
TEST_SUITES="csfle"
|
||||
fi
|
||||
|
||||
if [ -n "$TEST_INDEX_MANAGEMENT" ]; then
|
||||
source $DRIVERS_TOOLS/.evergreen/atlas/secrets-export.sh
|
||||
export DB_USER="${DRIVERS_ATLAS_LAMBDA_USER}"
|
||||
set +x
|
||||
export DB_PASSWORD="${DRIVERS_ATLAS_LAMBDA_PASSWORD}"
|
||||
set -x
|
||||
TEST_SUITES="index_management"
|
||||
fi
|
||||
|
||||
if [ -n "$TEST_DATA_LAKE" ] && [ -z "$TEST_ARGS" ]; then
|
||||
TEST_SUITES="data_lake"
|
||||
fi
|
||||
|
||||
if [ -n "$TEST_ATLAS" ]; then
|
||||
TEST_SUITES="atlas"
|
||||
fi
|
||||
|
||||
if [ -n "$TEST_OCSP" ]; then
|
||||
UV_ARGS+=("--extra ocsp")
|
||||
TEST_SUITES="ocsp"
|
||||
fi
|
||||
|
||||
if [ -n "$TEST_AUTH_AWS" ]; then
|
||||
UV_ARGS+=("--extra aws")
|
||||
TEST_SUITES="auth_aws"
|
||||
fi
|
||||
|
||||
if [ -n "$TEST_AUTH_OIDC" ]; then
|
||||
UV_ARGS+=("--extra aws")
|
||||
TEST_SUITES="auth_oidc"
|
||||
fi
|
||||
|
||||
if [ -n "$PERF_TEST" ]; then
|
||||
UV_ARGS+=("--group perf")
|
||||
start_time=$(date +%s)
|
||||
TEST_SUITES="perf"
|
||||
# PYTHON-4769 Run perf_test.py directly otherwise pytest's test collection negatively
|
||||
# affects the benchmark results.
|
||||
TEST_ARGS="test/performance/perf_test.py $TEST_ARGS"
|
||||
fi
|
||||
|
||||
echo "Running $AUTH tests over $SSL with python $(uv python find)"
|
||||
uv run --frozen python -c 'import sys; print(sys.version)'
|
||||
|
||||
|
||||
# Run the tests, and store the results in Evergreen compatible XUnit XML
|
||||
# files in the xunit-results/ directory.
|
||||
|
||||
# Run the tests with coverage if requested and coverage is installed.
|
||||
# Only cover CPython. PyPy reports suspiciously low coverage.
|
||||
if [ -n "$COVERAGE" ] && [ "$PYTHON_IMPL" = "CPython" ]; then
|
||||
# Keep in sync with combine-coverage.sh.
|
||||
# coverage >=5 is needed for relative_files=true.
|
||||
UV_ARGS+=("--group coverage")
|
||||
TEST_ARGS="$TEST_ARGS --cov"
|
||||
fi
|
||||
|
||||
if [ -n "$GREEN_FRAMEWORK" ]; then
|
||||
UV_ARGS+=("--group $GREEN_FRAMEWORK")
|
||||
fi
|
||||
|
||||
# Show the installed packages
|
||||
# shellcheck disable=SC2048
|
||||
PIP_QUIET=0 uv run ${UV_ARGS[*]} --with pip pip list
|
||||
|
||||
if [ -z "$GREEN_FRAMEWORK" ]; then
|
||||
# Use --capture=tee-sys so pytest prints test output inline:
|
||||
# https://docs.pytest.org/en/stable/how-to/capture-stdout-stderr.html
|
||||
PYTEST_ARGS="-v --capture=tee-sys --durations=5 $TEST_ARGS"
|
||||
if [ -n "$TEST_SUITES" ]; then
|
||||
PYTEST_ARGS="-m $TEST_SUITES $PYTEST_ARGS"
|
||||
fi
|
||||
# shellcheck disable=SC2048
|
||||
uv run ${UV_ARGS[*]} pytest $PYTEST_ARGS
|
||||
# Handle test inputs.
|
||||
if [ -f $SCRIPT_DIR/scripts/test-env.sh ]; then
|
||||
echo "Sourcing test inputs"
|
||||
. $SCRIPT_DIR/scripts/test-env.sh
|
||||
else
|
||||
# shellcheck disable=SC2048
|
||||
uv run ${UV_ARGS[*]} green_framework_test.py $GREEN_FRAMEWORK -v $TEST_ARGS
|
||||
echo "Missing test inputs, please run 'just setup-tests'"
|
||||
exit 1
|
||||
fi
|
||||
|
||||
# Handle perf test post actions.
|
||||
if [ -n "$PERF_TEST" ]; then
|
||||
end_time=$(date +%s)
|
||||
elapsed_secs=$((end_time-start_time))
|
||||
cleanup_tests() {
|
||||
# Avoid leaving the lock file in a changed state when we change the resolution type.
|
||||
if [ -n "${TEST_MIN_DEPS:-}" ]; then
|
||||
git checkout uv.lock || true
|
||||
fi
|
||||
cd $PREV_DIR
|
||||
}
|
||||
|
||||
cat results.json
|
||||
trap "cleanup_tests" SIGINT ERR
|
||||
|
||||
echo "{\"failures\": 0, \"results\": [{\"status\": \"pass\", \"exit_code\": 0, \"test_file\": \"BenchMarkTests\", \"start\": $start_time, \"end\": $end_time, \"elapsed\": $elapsed_secs}]}" > report.json
|
||||
# Start the test runner.
|
||||
echo "Running tests with UV_PYTHON=${UV_PYTHON:-}..."
|
||||
echo "UV_ARGS=${UV_ARGS}"
|
||||
uv run ${UV_ARGS} --reinstall-package pymongo .evergreen/scripts/run_tests.py "$@"
|
||||
echo "Running tests with UV_PYTHON=${UV_PYTHON:-}... done."
|
||||
|
||||
cat report.json
|
||||
fi
|
||||
|
||||
# Handle coverage post actions.
|
||||
if [ -n "$COVERAGE" ]; then
|
||||
rm -rf .pytest_cache
|
||||
fi
|
||||
cleanup_tests
|
||||
|
||||
0
.evergreen/scripts/__init__.py
Normal file
0
.evergreen/scripts/__init__.py
Normal file
@ -1,8 +0,0 @@
|
||||
#!/bin/bash
|
||||
|
||||
set -o xtrace
|
||||
mkdir out_dir
|
||||
# shellcheck disable=SC2156
|
||||
find "$MONGO_ORCHESTRATION_HOME" -name \*.log -exec sh -c 'x="{}"; mv $x $PWD/out_dir/$(basename $(dirname $x))_$(basename $x)' \;
|
||||
tar zcvf mongodb-logs.tar.gz -C out_dir/ .
|
||||
rm -rf out_dir
|
||||
@ -1,46 +0,0 @@
|
||||
#!/bin/bash
|
||||
|
||||
set -o xtrace
|
||||
|
||||
# Enable core dumps if enabled on the machine
|
||||
# Copied from https://github.com/mongodb/mongo/blob/master/etc/evergreen.yml
|
||||
if [ -f /proc/self/coredump_filter ]; then
|
||||
# Set the shell process (and its children processes) to dump ELF headers (bit 4),
|
||||
# anonymous shared mappings (bit 1), and anonymous private mappings (bit 0).
|
||||
echo 0x13 >/proc/self/coredump_filter
|
||||
|
||||
if [ -f /sbin/sysctl ]; then
|
||||
# Check that the core pattern is set explicitly on our distro image instead
|
||||
# of being the OS's default value. This ensures that coredump names are consistent
|
||||
# across distros and can be picked up by Evergreen.
|
||||
core_pattern=$(/sbin/sysctl -n "kernel.core_pattern")
|
||||
if [ "$core_pattern" = "dump_%e.%p.core" ]; then
|
||||
echo "Enabling coredumps"
|
||||
ulimit -c unlimited
|
||||
fi
|
||||
fi
|
||||
fi
|
||||
|
||||
if [ "$(uname -s)" = "Darwin" ]; then
|
||||
core_pattern_mac=$(/usr/sbin/sysctl -n "kern.corefile")
|
||||
if [ "$core_pattern_mac" = "dump_%N.%P.core" ]; then
|
||||
echo "Enabling coredumps"
|
||||
ulimit -c unlimited
|
||||
fi
|
||||
fi
|
||||
|
||||
if [ -n "${skip_crypt_shared}" ]; then
|
||||
export SKIP_CRYPT_SHARED=1
|
||||
fi
|
||||
|
||||
MONGODB_VERSION=${VERSION} \
|
||||
TOPOLOGY=${TOPOLOGY} \
|
||||
AUTH=${AUTH:-noauth} \
|
||||
SSL=${SSL:-nossl} \
|
||||
STORAGE_ENGINE=${STORAGE_ENGINE:-} \
|
||||
DISABLE_TEST_COMMANDS=${DISABLE_TEST_COMMANDS:-} \
|
||||
ORCHESTRATION_FILE=${ORCHESTRATION_FILE:-} \
|
||||
REQUIRE_API_VERSION=${REQUIRE_API_VERSION:-} \
|
||||
LOAD_BALANCER=${LOAD_BALANCER:-} \
|
||||
bash ${DRIVERS_TOOLS}/.evergreen/run-orchestration.sh
|
||||
# run-orchestration generates expansion file with the MONGODB_URI for the cluster
|
||||
@ -1,7 +1,41 @@
|
||||
#!/bin/bash
|
||||
# Check for regressions in the import time of pymongo.
|
||||
set -eu
|
||||
|
||||
. .evergreen/scripts/env.sh
|
||||
set -x
|
||||
export BASE_SHA="$1"
|
||||
export HEAD_SHA="$2"
|
||||
bash .evergreen/run-import-time-test.sh
|
||||
HERE=$(dirname ${BASH_SOURCE:-$0})
|
||||
|
||||
source $HERE/env.sh
|
||||
|
||||
pushd $HERE/../.. >/dev/null
|
||||
|
||||
BASE_SHA="$1"
|
||||
HEAD_SHA="$2"
|
||||
|
||||
# Set up the virtual env.
|
||||
. $HERE/setup-dev-env.sh
|
||||
uv venv --seed
|
||||
source .venv/bin/activate
|
||||
|
||||
# Use the previous commit if this was not a PR run.
|
||||
if [ "$BASE_SHA" == "$HEAD_SHA" ]; then
|
||||
BASE_SHA=$(git rev-parse HEAD~1)
|
||||
fi
|
||||
|
||||
function get_import_time() {
|
||||
local log_file
|
||||
python -m pip install -q ".[aws,encryption,gssapi,ocsp,snappy,zstd]"
|
||||
# Import once to cache modules
|
||||
python -c "import pymongo"
|
||||
log_file="pymongo-$1.log"
|
||||
python -X importtime -c "import pymongo" 2> $log_file
|
||||
}
|
||||
|
||||
get_import_time $HEAD_SHA
|
||||
git stash || true
|
||||
git checkout $BASE_SHA
|
||||
get_import_time $BASE_SHA
|
||||
git checkout $HEAD_SHA
|
||||
git stash apply || true
|
||||
python tools/compare_import_time.py $HEAD_SHA $BASE_SHA
|
||||
|
||||
popd >/dev/null
|
||||
|
||||
@ -1,7 +1,14 @@
|
||||
#!/bin/bash
|
||||
# Clean up resources at the end of an evergreen run.
|
||||
set -eu
|
||||
|
||||
if [ -f "$DRIVERS_TOOLS"/.evergreen/csfle/secrets-export.sh ]; then
|
||||
bash .evergreen/teardown-encryption.sh
|
||||
HERE=$(dirname ${BASH_SOURCE:-$0})
|
||||
|
||||
# Try to source the env file.
|
||||
if [ -f $HERE/env.sh ]; then
|
||||
echo "Sourcing env file"
|
||||
source $HERE/env.sh
|
||||
fi
|
||||
|
||||
rm -rf "${DRIVERS_TOOLS}" || true
|
||||
rm -f ./secrets-export.sh || true
|
||||
rm -f $HERE/../../secrets-export.sh || true
|
||||
|
||||
@ -1,5 +1,5 @@
|
||||
#!/bin/bash
|
||||
|
||||
# Configure an evergreen test environment.
|
||||
set -eu
|
||||
|
||||
# Get the current unique version of this checkout
|
||||
@ -16,6 +16,18 @@ DRIVERS_TOOLS="$(dirname $PROJECT_DIRECTORY)/drivers-tools"
|
||||
CARGO_HOME=${CARGO_HOME:-${DRIVERS_TOOLS}/.cargo}
|
||||
UV_TOOL_DIR=$PROJECT_DIRECTORY/.local/uv/tools
|
||||
UV_CACHE_DIR=$PROJECT_DIRECTORY/.local/uv/cache
|
||||
DRIVERS_TOOLS_BINARIES="$DRIVERS_TOOLS/.bin"
|
||||
MONGODB_BINARIES="$DRIVERS_TOOLS/mongodb/bin"
|
||||
|
||||
# On Evergreen jobs, "CI" will be set, and we don't want to write to $HOME.
|
||||
if [ "${CI:-}" == "true" ]; then
|
||||
PYMONGO_BIN_DIR=${DRIVERS_TOOLS_BINARIES:-}
|
||||
# We want to use a path that's already on PATH on spawn hosts.
|
||||
else
|
||||
PYMONGO_BIN_DIR=$HOME/cli_bin
|
||||
fi
|
||||
|
||||
PATH_EXT="$MONGODB_BINARIES:$DRIVERS_TOOLS_BINARIES:$PYMONGO_BIN_DIR:\$PATH"
|
||||
|
||||
# Python has cygwin path problems on Windows. Detect prospective mongo-orchestration home directory
|
||||
if [ "Windows_NT" = "${OS:-}" ]; then # Magic variable in cygwin
|
||||
@ -24,6 +36,9 @@ if [ "Windows_NT" = "${OS:-}" ]; then # Magic variable in cygwin
|
||||
CARGO_HOME=$(cygpath -m $CARGO_HOME)
|
||||
UV_TOOL_DIR=$(cygpath -m "$UV_TOOL_DIR")
|
||||
UV_CACHE_DIR=$(cygpath -m "$UV_CACHE_DIR")
|
||||
DRIVERS_TOOLS_BINARIES=$(cygpath -m "$DRIVERS_TOOLS_BINARIES")
|
||||
MONGODB_BINARIES=$(cygpath -m "$MONGODB_BINARIES")
|
||||
PYMONGO_BIN_DIR=$(cygpath -m "$PYMONGO_BIN_DIR")
|
||||
fi
|
||||
|
||||
SCRIPT_DIR="$PROJECT_DIRECTORY/.evergreen/scripts"
|
||||
@ -36,50 +51,64 @@ fi
|
||||
|
||||
export MONGO_ORCHESTRATION_HOME="$DRIVERS_TOOLS/.evergreen/orchestration"
|
||||
export MONGODB_BINARIES="$DRIVERS_TOOLS/mongodb/bin"
|
||||
export DRIVERS_TOOLS_BINARIES="$DRIVERS_TOOLS/.bin"
|
||||
|
||||
cat <<EOT > "$SCRIPT_DIR"/env.sh
|
||||
export PROJECT_DIRECTORY="$PROJECT_DIRECTORY"
|
||||
export CURRENT_VERSION="$CURRENT_VERSION"
|
||||
export SKIP_LEGACY_SHELL=1
|
||||
export DRIVERS_TOOLS="$DRIVERS_TOOLS"
|
||||
export MONGO_ORCHESTRATION_HOME="$MONGO_ORCHESTRATION_HOME"
|
||||
export MONGODB_BINARIES="$MONGODB_BINARIES"
|
||||
export DRIVERS_TOOLS_BINARIES="$DRIVERS_TOOLS_BINARIES"
|
||||
export PROJECT_DIRECTORY="$PROJECT_DIRECTORY"
|
||||
export SETDEFAULTENCODING="${SETDEFAULTENCODING:-}"
|
||||
export SKIP_CSOT_TESTS="${SKIP_CSOT_TESTS:-}"
|
||||
export MONGODB_STARTED="${MONGODB_STARTED:-}"
|
||||
export DISABLE_TEST_COMMANDS="${DISABLE_TEST_COMMANDS:-}"
|
||||
export GREEN_FRAMEWORK="${GREEN_FRAMEWORK:-}"
|
||||
export NO_EXT="${NO_EXT:-}"
|
||||
export COVERAGE="${COVERAGE:-}"
|
||||
export COMPRESSORS="${COMPRESSORS:-}"
|
||||
export MONGODB_API_VERSION="${MONGODB_API_VERSION:-}"
|
||||
export skip_crypt_shared="${skip_crypt_shared:-}"
|
||||
export STORAGE_ENGINE="${STORAGE_ENGINE:-}"
|
||||
export REQUIRE_API_VERSION="${REQUIRE_API_VERSION:-}"
|
||||
export skip_web_identity_auth_test="${skip_web_identity_auth_test:-}"
|
||||
export skip_ECS_auth_test="${skip_ECS_auth_test:-}"
|
||||
|
||||
export CARGO_HOME="$CARGO_HOME"
|
||||
export TMPDIR="$MONGO_ORCHESTRATION_HOME/db"
|
||||
export UV_TOOL_DIR="$UV_TOOL_DIR"
|
||||
export UV_CACHE_DIR="$UV_CACHE_DIR"
|
||||
export UV_TOOL_BIN_DIR="$DRIVERS_TOOLS_BINARIES"
|
||||
export PATH="$MONGODB_BINARIES:$DRIVERS_TOOLS_BINARIES:$PATH"
|
||||
export PYMONGO_BIN_DIR="$PYMONGO_BIN_DIR"
|
||||
export PATH="$PATH_EXT"
|
||||
# shellcheck disable=SC2154
|
||||
export PROJECT="${project:-mongo-python-driver}"
|
||||
export PIP_QUIET=1
|
||||
EOT
|
||||
|
||||
# Skip CSOT tests on non-linux platforms.
|
||||
if [ "$(uname -s)" != "Linux" ]; then
|
||||
echo "export SKIP_CSOT_TESTS=1" >> $SCRIPT_DIR/env.sh
|
||||
fi
|
||||
# Write the .env file for drivers-tools.
|
||||
rm -rf $DRIVERS_TOOLS
|
||||
BRANCH=master
|
||||
ORG=mongodb-labs
|
||||
git clone --branch $BRANCH https://github.com/$ORG/drivers-evergreen-tools.git $DRIVERS_TOOLS
|
||||
|
||||
cat <<EOT > ${DRIVERS_TOOLS}/.env
|
||||
SKIP_LEGACY_SHELL=1
|
||||
DRIVERS_TOOLS="$DRIVERS_TOOLS"
|
||||
MONGO_ORCHESTRATION_HOME="$MONGO_ORCHESTRATION_HOME"
|
||||
MONGODB_BINARIES="$MONGODB_BINARIES"
|
||||
EOT
|
||||
|
||||
# Add these expansions to make it easier to call out tests scripts from the EVG yaml
|
||||
cat <<EOT > expansion.yml
|
||||
DRIVERS_TOOLS: "$DRIVERS_TOOLS"
|
||||
PROJECT_DIRECTORY: "$PROJECT_DIRECTORY"
|
||||
EOT
|
||||
|
||||
# If the toolchain is available, symlink binaries to the bin dir. This has to be done
|
||||
# after drivers-tools is cloned, since we might be using its binary dir.
|
||||
_bin_path=""
|
||||
if [ "Windows_NT" == "${OS:-}" ]; then
|
||||
_bin_path="/cygdrive/c/Python/Current/Scripts"
|
||||
elif [ "$(uname -s)" == "Darwin" ]; then
|
||||
_bin_path="/Library/Frameworks/Python.Framework/Versions/Current/bin"
|
||||
else
|
||||
_bin_path="/opt/python/Current/bin"
|
||||
fi
|
||||
if [ -d "${_bin_path}" ]; then
|
||||
_suffix=""
|
||||
if [ "Windows_NT" == "${OS:-}" ]; then
|
||||
_suffix=".exe"
|
||||
fi
|
||||
echo "Symlinking binaries from toolchain"
|
||||
mkdir -p $PYMONGO_BIN_DIR
|
||||
ln -s ${_bin_path}/just${_suffix} $PYMONGO_BIN_DIR/just${_suffix}
|
||||
ln -s ${_bin_path}/uv${_suffix} $PYMONGO_BIN_DIR/uv${_suffix}
|
||||
ln -s ${_bin_path}/uvx${_suffix} $PYMONGO_BIN_DIR/uvx${_suffix}
|
||||
fi
|
||||
|
||||
50
.evergreen/scripts/create-spec-pr.sh
Executable file
50
.evergreen/scripts/create-spec-pr.sh
Executable file
@ -0,0 +1,50 @@
|
||||
#!/usr/bin/env bash
|
||||
|
||||
tools="$(realpath -s "../drivers-tools")"
|
||||
pushd $tools/.evergreen/github_app || exit
|
||||
|
||||
owner="mongodb"
|
||||
repo="mongo-python-driver"
|
||||
|
||||
# Bootstrap the app.
|
||||
echo "bootstrapping"
|
||||
source utils.sh
|
||||
bootstrap drivers/comment-bot
|
||||
|
||||
# Run the app.
|
||||
source ./secrets-export.sh
|
||||
|
||||
# Get a github access token for the git checkout.
|
||||
echo "Getting github token..."
|
||||
|
||||
token=$(bash ./get-access-token.sh $repo $owner)
|
||||
if [ -z "${token}" ]; then
|
||||
echo "Failed to get github access token!"
|
||||
popd || exit
|
||||
exit 1
|
||||
fi
|
||||
echo "Getting github token... done."
|
||||
popd || exit
|
||||
|
||||
# Make the git checkout and create a new branch.
|
||||
echo "Creating the git checkout..."
|
||||
branch="spec-resync-"$(date '+%m-%d-%Y')
|
||||
|
||||
git remote set-url origin https://x-access-token:${token}@github.com/$owner/$repo.git
|
||||
git checkout -b $branch "origin/master"
|
||||
git add ./test
|
||||
git commit -am "resyncing specs $(date '+%m-%d-%Y')"
|
||||
echo "Creating the git checkout... done."
|
||||
|
||||
git push origin $branch
|
||||
resp=$(curl -L \
|
||||
-X POST \
|
||||
-H "Accept: application/vnd.github+json" \
|
||||
-H "Authorization: Bearer $token" \
|
||||
-H "X-GitHub-Api-Version: 2022-11-28" \
|
||||
-d "{\"title\":\"[Spec Resync] $(date '+%m-%d-%Y')\",\"body\":\"$(cat "$1")\",\"head\":\"${branch}\",\"base\":\"master\"}" \
|
||||
--url https://api.github.com/repos/$owner/$repo/pulls)
|
||||
echo $resp | jq '.html_url'
|
||||
echo "Creating the PR... done."
|
||||
|
||||
rm -rf $tools
|
||||
@ -1,4 +1,4 @@
|
||||
#!/bin/bash
|
||||
|
||||
# Download all the task coverage files.
|
||||
set -eu
|
||||
aws s3 cp --recursive s3://"$1"/coverage/"$2"/"$3"/coverage/ coverage/
|
||||
|
||||
@ -1,8 +0,0 @@
|
||||
#!/bin/bash
|
||||
|
||||
set +x
|
||||
. src/.evergreen/scripts/env.sh
|
||||
# shellcheck disable=SC2044
|
||||
for filename in $(find $DRIVERS_TOOLS -name \*.json); do
|
||||
perl -p -i -e "s|ABSOLUTE_PATH_REPLACEMENT_TOKEN|$DRIVERS_TOOLS|g" $filename
|
||||
done
|
||||
6
.evergreen/scripts/generate-config.sh
Executable file
6
.evergreen/scripts/generate-config.sh
Executable file
@ -0,0 +1,6 @@
|
||||
#!/bin/bash
|
||||
# Entry point for the generate-config pre-commit hook.
|
||||
|
||||
set -eu
|
||||
|
||||
python .evergreen/scripts/generate_config.py
|
||||
File diff suppressed because it is too large
Load Diff
357
.evergreen/scripts/generate_config_utils.py
Normal file
357
.evergreen/scripts/generate_config_utils.py
Normal file
@ -0,0 +1,357 @@
|
||||
from __future__ import annotations
|
||||
|
||||
from dataclasses import dataclass
|
||||
from inspect import getmembers, isfunction
|
||||
from itertools import cycle, zip_longest
|
||||
from pathlib import Path
|
||||
from typing import Any
|
||||
|
||||
from shrub.v3.evg_build_variant import BuildVariant
|
||||
from shrub.v3.evg_command import (
|
||||
EvgCommandType,
|
||||
ec2_assume_role,
|
||||
s3_put,
|
||||
subprocess_exec,
|
||||
)
|
||||
from shrub.v3.evg_project import EvgProject
|
||||
from shrub.v3.evg_task import EvgTaskRef
|
||||
from shrub.v3.shrub_service import ShrubService
|
||||
|
||||
##############
|
||||
# Globals
|
||||
##############
|
||||
|
||||
ALL_VERSIONS = ["4.2", "4.4", "5.0", "6.0", "7.0", "8.0", "rapid", "latest"]
|
||||
CPYTHONS = ["3.10", "3.11", "3.12", "3.13", "3.14t", "3.14"]
|
||||
PYPYS = ["pypy3.11"]
|
||||
MIN_SUPPORT_VERSIONS = ["3.9", "pypy3.9", "pypy3.10"]
|
||||
ALL_PYTHONS = CPYTHONS + PYPYS
|
||||
MIN_MAX_PYTHON = [CPYTHONS[0], CPYTHONS[-1]]
|
||||
BATCHTIME_WEEK = 10080
|
||||
BATCHTIME_DAY = 1440
|
||||
AUTH_SSLS = [("auth", "ssl"), ("noauth", "ssl"), ("noauth", "nossl")]
|
||||
TOPOLOGIES = ["standalone", "replica_set", "sharded_cluster"]
|
||||
C_EXTS = ["without_ext", "with_ext"]
|
||||
SYNCS = ["sync", "async"]
|
||||
DISPLAY_LOOKUP = dict(
|
||||
ssl=dict(ssl="SSL", nossl="NoSSL"),
|
||||
auth=dict(auth="Auth", noauth="NoAuth"),
|
||||
topology=dict(
|
||||
standalone="Standalone", replica_set="Replica Set", sharded_cluster="Sharded Cluster"
|
||||
),
|
||||
test_suites=dict(default="Sync", default_async="Async"),
|
||||
sync={"sync": "Sync", "async": "Async"},
|
||||
coverage={"1": "cov"},
|
||||
no_ext={"1": "No C"},
|
||||
test_min_deps={"1": "Min Deps"},
|
||||
)
|
||||
HOSTS = dict()
|
||||
|
||||
|
||||
@dataclass
|
||||
class Host:
|
||||
name: str
|
||||
run_on: str
|
||||
display_name: str
|
||||
variables: dict[str, str] | None
|
||||
|
||||
|
||||
# Hosts with toolchains.
|
||||
HOSTS["rhel8"] = Host("rhel8", "rhel87-small", "RHEL8", dict())
|
||||
HOSTS["win64"] = Host("win64", "windows-64-vsMulti-small", "Win64", dict())
|
||||
HOSTS["win-latest"] = Host("win-latest", "windows-2022-latest-small", "WinLatest", dict())
|
||||
HOSTS["win32"] = Host("win32", "windows-64-vsMulti-small", "Win32", dict())
|
||||
HOSTS["macos"] = Host("macos", "macos-14", "macOS", dict())
|
||||
HOSTS["macos-arm64"] = Host("macos-arm64", "macos-14-arm64", "macOS Arm64", dict())
|
||||
HOSTS["ubuntu22"] = Host("ubuntu22", "ubuntu2204-small", "Ubuntu-22", dict())
|
||||
HOSTS["ubuntu24"] = Host("ubuntu24", "ubuntu2404-small", "Ubuntu-24", dict())
|
||||
HOSTS["perf"] = Host("perf", "rhel90-dbx-perf-large", "", dict())
|
||||
HOSTS["debian11"] = Host("debian11", "debian11-small", "Debian11", dict())
|
||||
DEFAULT_HOST = HOSTS["rhel8"]
|
||||
|
||||
# Other hosts
|
||||
OTHER_HOSTS = ["RHEL9-FIPS", "RHEL8-zseries", "RHEL8-POWER8", "RHEL8-arm64", "Amazon2023"]
|
||||
for name, run_on in zip(
|
||||
OTHER_HOSTS,
|
||||
[
|
||||
"rhel92-fips",
|
||||
"rhel8-zseries-small",
|
||||
"rhel8-power-small",
|
||||
"rhel82-arm64-small",
|
||||
"amazon2023-arm64-latest-large-m8g",
|
||||
],
|
||||
):
|
||||
HOSTS[name] = Host(name, run_on, name, dict())
|
||||
|
||||
##############
|
||||
# Helpers
|
||||
##############
|
||||
|
||||
|
||||
def create_variant_generic(
|
||||
tasks: list[str | EvgTaskRef],
|
||||
display_name: str,
|
||||
*,
|
||||
host: Host | str | None = None,
|
||||
default_run_on="rhel87-small",
|
||||
expansions: dict | None = None,
|
||||
**kwargs: Any,
|
||||
) -> BuildVariant:
|
||||
"""Create a build variant for the given inputs."""
|
||||
task_refs = []
|
||||
if isinstance(host, str):
|
||||
host = HOSTS[host]
|
||||
for t in tasks:
|
||||
if isinstance(t, EvgTaskRef):
|
||||
task_refs.append(t)
|
||||
else:
|
||||
task_refs.append(EvgTaskRef(name=t))
|
||||
expansions = expansions and expansions.copy() or dict()
|
||||
if "run_on" in kwargs:
|
||||
run_on = kwargs.pop("run_on")
|
||||
elif host:
|
||||
run_on = [host.run_on]
|
||||
if host.variables:
|
||||
expansions.update(host.variables)
|
||||
else:
|
||||
run_on = [default_run_on]
|
||||
if isinstance(run_on, str):
|
||||
run_on = [run_on]
|
||||
name = display_name.replace(" ", "-").replace("*-", "").lower()
|
||||
return BuildVariant(
|
||||
name=name,
|
||||
display_name=display_name,
|
||||
tasks=task_refs,
|
||||
expansions=expansions or None,
|
||||
run_on=run_on,
|
||||
**kwargs,
|
||||
)
|
||||
|
||||
|
||||
def create_variant(
|
||||
tasks: list[str | EvgTaskRef],
|
||||
display_name: str,
|
||||
*,
|
||||
version: str | None = None,
|
||||
host: Host | str | None = None,
|
||||
expansions: dict | None = None,
|
||||
**kwargs: Any,
|
||||
) -> BuildVariant:
|
||||
expansions = expansions and expansions.copy() or dict()
|
||||
if version:
|
||||
expansions["VERSION"] = version
|
||||
# 8.0+ Windows builds must run on win-latest
|
||||
if (
|
||||
"win64" in display_name.lower()
|
||||
or "win32" in display_name.lower()
|
||||
and version
|
||||
and version >= "8.0"
|
||||
):
|
||||
kwargs["run_on"] = HOSTS["win-latest"].run_on
|
||||
return create_variant_generic(
|
||||
tasks, display_name, version=version, host=host, expansions=expansions, **kwargs
|
||||
)
|
||||
|
||||
|
||||
def get_versions_from(min_version: str) -> list[str]:
|
||||
"""Get all server versions starting from a minimum version."""
|
||||
min_version_float = float(min_version)
|
||||
rapid_latest = ["rapid", "latest"]
|
||||
versions = [v for v in ALL_VERSIONS if v not in rapid_latest]
|
||||
return [v for v in versions if float(v) >= min_version_float] + rapid_latest
|
||||
|
||||
|
||||
def get_versions_until(max_version: str) -> list[str]:
|
||||
"""Get all server version up to a max version."""
|
||||
max_version_float = float(max_version)
|
||||
versions = [v for v in ALL_VERSIONS if v not in ["rapid", "latest"]]
|
||||
versions = [v for v in versions if float(v) <= max_version_float]
|
||||
if not len(versions):
|
||||
raise ValueError(f"No server versions found less <= {max_version}")
|
||||
return versions
|
||||
|
||||
|
||||
def get_common_name(base: str, sep: str, **kwargs) -> str:
|
||||
display_name = base
|
||||
version = kwargs.pop("VERSION", None)
|
||||
version = version or kwargs.pop("version", None)
|
||||
if version:
|
||||
if version not in ["rapid", "latest"]:
|
||||
version = f"v{version}"
|
||||
display_name = f"{display_name}{sep}{version}"
|
||||
for key, value in kwargs.items():
|
||||
name = value
|
||||
if key.lower() in ["python", "toolchain_version"]:
|
||||
if not value.startswith("pypy"):
|
||||
name = f"Python{value}"
|
||||
else:
|
||||
name = f"PyPy{value.replace('pypy', '')}"
|
||||
elif key.lower() in DISPLAY_LOOKUP and value in DISPLAY_LOOKUP[key.lower()]:
|
||||
name = DISPLAY_LOOKUP[key.lower()][value]
|
||||
else:
|
||||
continue
|
||||
display_name = f"{display_name}{sep}{name}"
|
||||
return display_name
|
||||
|
||||
|
||||
def get_variant_name(base: str, host: str | Host | None = None, **kwargs) -> str:
|
||||
"""Get the display name of a variant."""
|
||||
display_name = base
|
||||
if isinstance(host, str):
|
||||
host = HOSTS[host]
|
||||
if host is not None:
|
||||
display_name += f" {host.display_name}"
|
||||
return get_common_name(display_name, " ", **kwargs)
|
||||
|
||||
|
||||
def get_task_name(base: str, **kwargs):
|
||||
return get_common_name(base, "-", **kwargs).replace(" ", "-").lower()
|
||||
|
||||
|
||||
def zip_cycle(*iterables, empty_default=None):
|
||||
"""Get all combinations of the inputs, cycling over the shorter list(s)."""
|
||||
cycles = [cycle(i) for i in iterables]
|
||||
for _ in zip_longest(*iterables):
|
||||
yield tuple(next(i, empty_default) for i in cycles)
|
||||
|
||||
|
||||
def handle_c_ext(c_ext, expansions) -> None:
|
||||
"""Handle c extension option."""
|
||||
if c_ext == C_EXTS[0]:
|
||||
expansions["NO_EXT"] = "1"
|
||||
|
||||
|
||||
def get_standard_auth_ssl(topology):
|
||||
auth = "auth" if topology == "sharded_cluster" else "noauth"
|
||||
ssl = "nossl" if topology == "standalone" else "ssl"
|
||||
return auth, ssl
|
||||
|
||||
|
||||
def get_assume_role(**kwargs):
|
||||
kwargs.setdefault("command_type", EvgCommandType.SETUP)
|
||||
kwargs.setdefault("role_arn", "${assume_role_arn}")
|
||||
return ec2_assume_role(**kwargs)
|
||||
|
||||
|
||||
def get_subprocess_exec(**kwargs):
|
||||
kwargs.setdefault("binary", "bash")
|
||||
kwargs.setdefault("working_dir", "src")
|
||||
kwargs.setdefault("command_type", EvgCommandType.TEST)
|
||||
return subprocess_exec(**kwargs)
|
||||
|
||||
|
||||
def get_s3_put(**kwargs):
|
||||
kwargs["aws_key"] = "${AWS_ACCESS_KEY_ID}"
|
||||
kwargs["aws_secret"] = "${AWS_SECRET_ACCESS_KEY}" # noqa:S105
|
||||
kwargs["aws_session_token"] = "${AWS_SESSION_TOKEN}" # noqa:S105
|
||||
kwargs["bucket"] = "${bucket_name}"
|
||||
kwargs.setdefault("optional", "true")
|
||||
kwargs.setdefault("permissions", "public-read")
|
||||
kwargs.setdefault("content_type", "${content_type|application/x-gzip}")
|
||||
kwargs.setdefault("command_type", EvgCommandType.SETUP)
|
||||
return s3_put(**kwargs)
|
||||
|
||||
|
||||
def generate_yaml(tasks=None, variants=None):
|
||||
"""Generate the yaml for a given set of tasks and variants."""
|
||||
project = EvgProject(tasks=tasks, buildvariants=variants)
|
||||
out = ShrubService.generate_yaml(project)
|
||||
# Dedent by two spaces to match what we use in config.yml
|
||||
lines = [line[2:] for line in out.splitlines()]
|
||||
print("\n".join(lines))
|
||||
|
||||
|
||||
##################
|
||||
# Generate Config
|
||||
##################
|
||||
|
||||
|
||||
def write_variants_to_file(mod):
|
||||
here = Path(__file__).absolute().parent
|
||||
target = here.parent / "generated_configs" / "variants.yml"
|
||||
if target.exists():
|
||||
target.unlink()
|
||||
with target.open("w") as fid:
|
||||
fid.write("buildvariants:\n")
|
||||
|
||||
for name, func in sorted(getmembers(mod, isfunction)):
|
||||
if not name.endswith("_variants"):
|
||||
continue
|
||||
if not name.startswith("create_"):
|
||||
raise ValueError("Variant creators must start with create_")
|
||||
title = name.replace("create_", "").replace("_variants", "").replace("_", " ").capitalize()
|
||||
project = EvgProject(tasks=None, buildvariants=func())
|
||||
out = ShrubService.generate_yaml(project).splitlines()
|
||||
with target.open("a") as fid:
|
||||
fid.write(f" # {title} tests\n")
|
||||
for line in out[1:]:
|
||||
fid.write(f"{line}\n")
|
||||
fid.write("\n")
|
||||
|
||||
# Remove extra trailing newline:
|
||||
data = target.read_text().splitlines()
|
||||
with target.open("w") as fid:
|
||||
for line in data[:-1]:
|
||||
fid.write(f"{line}\n")
|
||||
|
||||
|
||||
def write_tasks_to_file(mod):
|
||||
here = Path(__file__).absolute().parent
|
||||
target = here.parent / "generated_configs" / "tasks.yml"
|
||||
if target.exists():
|
||||
target.unlink()
|
||||
with target.open("w") as fid:
|
||||
fid.write("tasks:\n")
|
||||
|
||||
for name, func in sorted(getmembers(mod, isfunction)):
|
||||
if name.startswith("_") or not name.endswith("_tasks"):
|
||||
continue
|
||||
if not name.startswith("create_"):
|
||||
raise ValueError("Task creators must start with create_")
|
||||
title = name.replace("create_", "").replace("_tasks", "").replace("_", " ").capitalize()
|
||||
project = EvgProject(tasks=func(), buildvariants=None)
|
||||
out = ShrubService.generate_yaml(project).splitlines()
|
||||
with target.open("a") as fid:
|
||||
fid.write(f" # {title} tests\n")
|
||||
for line in out[1:]:
|
||||
fid.write(f"{line}\n")
|
||||
fid.write("\n")
|
||||
|
||||
# Remove extra trailing newline:
|
||||
data = target.read_text().splitlines()
|
||||
with target.open("w") as fid:
|
||||
for line in data[:-1]:
|
||||
fid.write(f"{line}\n")
|
||||
|
||||
|
||||
def write_functions_to_file(mod):
|
||||
here = Path(__file__).absolute().parent
|
||||
target = here.parent / "generated_configs" / "functions.yml"
|
||||
if target.exists():
|
||||
target.unlink()
|
||||
with target.open("w") as fid:
|
||||
fid.write("functions:\n")
|
||||
|
||||
functions = dict()
|
||||
for name, func in sorted(getmembers(mod, isfunction)):
|
||||
if name.startswith("_") or not name.endswith("_func"):
|
||||
continue
|
||||
if not name.startswith("create_"):
|
||||
raise ValueError("Function creators must start with create_")
|
||||
title = name.replace("create_", "").replace("_func", "").replace("_", " ").capitalize()
|
||||
func_name, cmds = func()
|
||||
functions = dict()
|
||||
functions[func_name] = cmds
|
||||
project = EvgProject(functions=functions, tasks=None, buildvariants=None)
|
||||
out = ShrubService.generate_yaml(project).splitlines()
|
||||
with target.open("a") as fid:
|
||||
fid.write(f" # {title}\n")
|
||||
for line in out[1:]:
|
||||
fid.write(f"{line}\n")
|
||||
fid.write("\n")
|
||||
|
||||
# Remove extra trailing newline:
|
||||
data = target.read_text().splitlines()
|
||||
with target.open("w") as fid:
|
||||
for line in data[:-1]:
|
||||
fid.write(f"{line}\n")
|
||||
@ -1,5 +0,0 @@
|
||||
#!/bin/bash
|
||||
|
||||
set +x
|
||||
. src/.evergreen/scripts/env.sh
|
||||
echo '{"results": [{ "status": "FAIL", "test_file": "Build", "log_raw": "No test-results.json found was created" } ]}' >$PROJECT_DIRECTORY/test-results.json
|
||||
@ -1,55 +1,36 @@
|
||||
#!/bin/bash
|
||||
|
||||
# Install the necessary dependencies.
|
||||
set -eu
|
||||
|
||||
# On Evergreen jobs, "CI" will be set, and we don't want to write to $HOME.
|
||||
if [ "${CI:-}" == "true" ]; then
|
||||
_BIN_DIR=${DRIVERS_TOOLS_BINARIES:-}
|
||||
else
|
||||
_BIN_DIR=$HOME/.local/bin
|
||||
HERE=$(dirname ${BASH_SOURCE:-$0})
|
||||
pushd "$(dirname "$(dirname $HERE)")" > /dev/null
|
||||
|
||||
# Source the env files to pick up common variables.
|
||||
if [ -f $HERE/env.sh ]; then
|
||||
. $HERE/env.sh
|
||||
fi
|
||||
|
||||
|
||||
# Helper function to pip install a dependency using a temporary python env.
|
||||
function _pip_install() {
|
||||
_HERE=$(dirname ${BASH_SOURCE:-$0})
|
||||
. $_HERE/../utils.sh
|
||||
_VENV_PATH=$(mktemp -d)
|
||||
echo "Installing $2 using pip..."
|
||||
createvirtualenv "$(find_python3)" $_VENV_PATH
|
||||
python -m pip install $1
|
||||
ln -s "$(which $2)" $_BIN_DIR/$2
|
||||
echo "Installing $2 using pip... done."
|
||||
}
|
||||
|
||||
|
||||
# Ensure just is installed.
|
||||
if ! command -v just 2>/dev/null; then
|
||||
# On most systems we can install directly.
|
||||
_TARGET=""
|
||||
if [ "Windows_NT" = "${OS:-}" ]; then
|
||||
_TARGET="--target x86_64-pc-windows-msvc"
|
||||
fi
|
||||
echo "Installing just..."
|
||||
mkdir -p "$_BIN_DIR" 2>/dev/null || true
|
||||
curl --proto '=https' --tlsv1.2 -sSf https://just.systems/install.sh | bash -s -- $_TARGET --to "$_BIN_DIR" || {
|
||||
_pip_install rust-just just
|
||||
}
|
||||
if ! command -v just 2>/dev/null; then
|
||||
export PATH="$PATH:$_BIN_DIR"
|
||||
fi
|
||||
echo "Installing just... done."
|
||||
# Set up the default bin directory.
|
||||
if [ -z "${PYMONGO_BIN_DIR:-}" ]; then
|
||||
PYMONGO_BIN_DIR="$HOME/.local/bin"
|
||||
fi
|
||||
|
||||
# Install uv.
|
||||
if ! command -v uv 2>/dev/null; then
|
||||
# Ensure uv is installed.
|
||||
if ! command -v uv &>/dev/null; then
|
||||
_BIN_DIR=$PYMONGO_BIN_DIR
|
||||
mkdir -p ${_BIN_DIR}
|
||||
echo "Installing uv..."
|
||||
# On most systems we can install directly.
|
||||
curl -LsSf https://astral.sh/uv/install.sh | env UV_INSTALL_DIR="$_BIN_DIR" INSTALLER_NO_MODIFY_PATH=1 sh || {
|
||||
_pip_install uv uv
|
||||
}
|
||||
if ! command -v uv 2>/dev/null; then
|
||||
export PATH="$PATH:$_BIN_DIR"
|
||||
curl -LsSf https://astral.sh/uv/install.sh | env UV_INSTALL_DIR="$_BIN_DIR" INSTALLER_NO_MODIFY_PATH=1 sh
|
||||
if [ "Windows_NT" = "${OS:-}" ]; then
|
||||
chmod +x "$(cygpath -u $_BIN_DIR)/uv.exe"
|
||||
fi
|
||||
export PATH="$PYMONGO_BIN_DIR:$PATH"
|
||||
echo "Installing uv... done."
|
||||
fi
|
||||
|
||||
# Ensure just is installed.
|
||||
if ! command -v just &>/dev/null; then
|
||||
uv tool install rust-just
|
||||
fi
|
||||
|
||||
popd > /dev/null
|
||||
|
||||
144
.evergreen/scripts/kms_tester.py
Normal file
144
.evergreen/scripts/kms_tester.py
Normal file
@ -0,0 +1,144 @@
|
||||
from __future__ import annotations
|
||||
|
||||
import os
|
||||
|
||||
from utils import (
|
||||
DRIVERS_TOOLS,
|
||||
LOGGER,
|
||||
TMP_DRIVER_FILE,
|
||||
create_archive,
|
||||
read_env,
|
||||
run_command,
|
||||
write_env,
|
||||
)
|
||||
|
||||
DIRS = dict(
|
||||
gcp=f"{DRIVERS_TOOLS}/.evergreen/csfle/gcpkms",
|
||||
azure=f"{DRIVERS_TOOLS}/.evergreen/csfle/azurekms",
|
||||
)
|
||||
|
||||
|
||||
def _setup_azure_vm(base_env: dict[str, str]) -> None:
|
||||
LOGGER.info("Setting up Azure VM...")
|
||||
azure_dir = DIRS["azure"]
|
||||
env = base_env.copy()
|
||||
env["AZUREKMS_SRC"] = TMP_DRIVER_FILE
|
||||
env["AZUREKMS_DST"] = "~/"
|
||||
run_command(f"{azure_dir}/copy-file.sh", env=env)
|
||||
|
||||
env = base_env.copy()
|
||||
env["AZUREKMS_CMD"] = "tar xf mongo-python-driver.tgz"
|
||||
run_command(f"{azure_dir}/run-command.sh", env=env)
|
||||
|
||||
env["AZUREKMS_CMD"] = "sudo apt-get install -y python3-dev build-essential"
|
||||
run_command(f"{azure_dir}/run-command.sh", env=env)
|
||||
|
||||
env["AZUREKMS_CMD"] = "bash .evergreen/just.sh setup-tests kms azure-remote"
|
||||
run_command(f"{azure_dir}/run-command.sh", env=env)
|
||||
LOGGER.info("Setting up Azure VM... done.")
|
||||
|
||||
|
||||
def _setup_gcp_vm(base_env: dict[str, str]) -> None:
|
||||
LOGGER.info("Setting up GCP VM...")
|
||||
gcp_dir = DIRS["gcp"]
|
||||
env = base_env.copy()
|
||||
env["GCPKMS_SRC"] = TMP_DRIVER_FILE
|
||||
env["GCPKMS_DST"] = f"{env['GCPKMS_INSTANCENAME']}:"
|
||||
run_command(f"{gcp_dir}/copy-file.sh", env=env)
|
||||
|
||||
env = base_env.copy()
|
||||
env["GCPKMS_CMD"] = "tar xf mongo-python-driver.tgz"
|
||||
run_command(f"{gcp_dir}/run-command.sh", env=env)
|
||||
|
||||
env["GCPKMS_CMD"] = "sudo apt-get install -y python3-dev build-essential"
|
||||
run_command(f"{gcp_dir}/run-command.sh", env=env)
|
||||
|
||||
env["GCPKMS_CMD"] = "bash ./.evergreen/just.sh setup-tests kms gcp-remote"
|
||||
run_command(f"{gcp_dir}/run-command.sh", env=env)
|
||||
LOGGER.info("Setting up GCP VM...")
|
||||
|
||||
|
||||
def _load_kms_config(sub_test_target: str) -> dict[str, str]:
|
||||
target_dir = DIRS[sub_test_target]
|
||||
config = read_env(f"{target_dir}/secrets-export.sh")
|
||||
base_env = os.environ.copy()
|
||||
for key, value in config.items():
|
||||
base_env[key] = str(value)
|
||||
return base_env
|
||||
|
||||
|
||||
def setup_kms(sub_test_name: str) -> None:
|
||||
if "-" in sub_test_name:
|
||||
sub_test_target, sub_test_type = sub_test_name.split("-")
|
||||
else:
|
||||
sub_test_target = sub_test_name
|
||||
sub_test_type = ""
|
||||
|
||||
assert sub_test_target in ["azure", "gcp"], sub_test_target
|
||||
assert sub_test_type in ["", "remote", "fail"], sub_test_type
|
||||
success = sub_test_type != "fail"
|
||||
kms_dir = DIRS[sub_test_target]
|
||||
|
||||
if sub_test_target == "azure":
|
||||
write_env("TEST_FLE_AZURE_AUTO")
|
||||
else:
|
||||
write_env("TEST_FLE_GCP_AUTO")
|
||||
|
||||
write_env("SUCCESS", success)
|
||||
|
||||
# For remote tests, there is no further work required.
|
||||
if sub_test_type == "remote":
|
||||
return
|
||||
|
||||
if sub_test_target == "azure":
|
||||
run_command("./setup-secrets.sh", cwd=kms_dir)
|
||||
|
||||
if success:
|
||||
create_archive()
|
||||
if sub_test_target == "azure":
|
||||
os.environ["AZUREKMS_VMNAME_PREFIX"] = "PYTHON_DRIVER"
|
||||
|
||||
# Found using "az vm image list --output table"
|
||||
os.environ[
|
||||
"AZUREKMS_IMAGE"
|
||||
] = "Canonical:0001-com-ubuntu-server-jammy:22_04-lts-gen2:latest"
|
||||
else:
|
||||
os.environ["GCPKMS_IMAGEFAMILY"] = "debian-12"
|
||||
|
||||
run_command("./setup.sh", cwd=kms_dir)
|
||||
base_env = _load_kms_config(sub_test_target)
|
||||
|
||||
if sub_test_target == "azure":
|
||||
_setup_azure_vm(base_env)
|
||||
else:
|
||||
_setup_gcp_vm(base_env)
|
||||
|
||||
if sub_test_target == "azure":
|
||||
config = read_env(f"{kms_dir}/secrets-export.sh")
|
||||
if success:
|
||||
write_env("AZUREKMS_VMNAME", config["AZUREKMS_VMNAME"])
|
||||
|
||||
write_env("KEY_NAME", config["AZUREKMS_KEYNAME"])
|
||||
write_env("KEY_VAULT_ENDPOINT", config["AZUREKMS_KEYVAULTENDPOINT"])
|
||||
|
||||
|
||||
def test_kms_send_to_remote(sub_test_name: str) -> None:
|
||||
env = _load_kms_config(sub_test_name)
|
||||
if sub_test_name == "azure":
|
||||
key_name = os.environ["KEY_NAME"]
|
||||
key_vault_endpoint = os.environ["KEY_VAULT_ENDPOINT"]
|
||||
env[
|
||||
"AZUREKMS_CMD"
|
||||
] = f'KEY_NAME="{key_name}" KEY_VAULT_ENDPOINT="{key_vault_endpoint}" bash ./.evergreen/just.sh run-tests'
|
||||
else:
|
||||
env["GCPKMS_CMD"] = "./.evergreen/just.sh run-tests"
|
||||
cmd = f"{DIRS[sub_test_name]}/run-command.sh"
|
||||
run_command(cmd, env=env)
|
||||
|
||||
|
||||
def teardown_kms(sub_test_name: str) -> None:
|
||||
run_command(f"{DIRS[sub_test_name]}/teardown.sh")
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
setup_kms()
|
||||
@ -1,8 +0,0 @@
|
||||
#!/bin/bash
|
||||
|
||||
set +x
|
||||
. src/.evergreen/scripts/env.sh
|
||||
# shellcheck disable=SC2044
|
||||
for i in $(find "$DRIVERS_TOOLS"/.evergreen "$PROJECT_DIRECTORY"/.evergreen -name \*.sh); do
|
||||
chmod +x "$i"
|
||||
done
|
||||
93
.evergreen/scripts/mod_wsgi_tester.py
Normal file
93
.evergreen/scripts/mod_wsgi_tester.py
Normal file
@ -0,0 +1,93 @@
|
||||
from __future__ import annotations
|
||||
|
||||
import os
|
||||
import sys
|
||||
import time
|
||||
import urllib.error
|
||||
import urllib.request
|
||||
from pathlib import Path
|
||||
from shutil import which
|
||||
|
||||
from utils import LOGGER, ROOT, run_command, write_env
|
||||
|
||||
|
||||
def make_request(url, timeout=10):
|
||||
for _ in range(int(timeout)):
|
||||
try:
|
||||
urllib.request.urlopen(url) # noqa: S310
|
||||
return
|
||||
except urllib.error.HTTPError:
|
||||
pass
|
||||
time.sleep(1)
|
||||
raise TimeoutError(f"Failed to access {url}")
|
||||
|
||||
|
||||
def setup_mod_wsgi(sub_test_name: str) -> None:
|
||||
env = os.environ.copy()
|
||||
if sub_test_name == "embedded":
|
||||
env["MOD_WSGI_CONF"] = "mod_wsgi_test_embedded.conf"
|
||||
elif sub_test_name == "standalone":
|
||||
env["MOD_WSGI_CONF"] = "mod_wsgi_test.conf"
|
||||
else:
|
||||
raise ValueError("mod_wsgi sub test must be either 'standalone' or 'embedded'")
|
||||
write_env("MOD_WSGI_CONF", env["MOD_WSGI_CONF"])
|
||||
apache = which("apache2")
|
||||
if not apache and Path("/usr/lib/apache2/mpm-prefork/apache2").exists():
|
||||
apache = "/usr/lib/apache2/mpm-prefork/apache2"
|
||||
if apache:
|
||||
apache_config = "apache24ubuntu161404.conf"
|
||||
else:
|
||||
apache = which("httpd")
|
||||
if not apache:
|
||||
raise ValueError("Could not find apache2 or httpd")
|
||||
apache_config = "apache22amazon.conf"
|
||||
python_version = ".".join(str(val) for val in sys.version_info[:2])
|
||||
mod_wsgi_version = 4
|
||||
so_file = f"/opt/python/mod_wsgi/python_version/{python_version}/mod_wsgi_version/{mod_wsgi_version}/mod_wsgi.so"
|
||||
write_env("MOD_WSGI_SO", so_file)
|
||||
env["MOD_WSGI_SO"] = so_file
|
||||
env["PYTHONHOME"] = f"/opt/python/{python_version}"
|
||||
env["PROJECT_DIRECTORY"] = project_directory = str(ROOT)
|
||||
write_env("APACHE_BINARY", apache)
|
||||
write_env("APACHE_CONFIG", apache_config)
|
||||
uri1 = f"http://localhost:8080/interpreter1{project_directory}"
|
||||
write_env("TEST_URI1", uri1)
|
||||
uri2 = f"http://localhost:8080/interpreter2{project_directory}"
|
||||
write_env("TEST_URI2", uri2)
|
||||
run_command(f"{apache} -k start -f {ROOT}/test/mod_wsgi_test/{apache_config}", env=env)
|
||||
|
||||
# Wait for the endpoints to be available.
|
||||
try:
|
||||
make_request(uri1, 10)
|
||||
make_request(uri2, 10)
|
||||
except Exception as e:
|
||||
LOGGER.error(Path("error_log").read_text())
|
||||
raise e
|
||||
|
||||
|
||||
def test_mod_wsgi() -> None:
|
||||
sys.path.insert(0, ROOT)
|
||||
from test.mod_wsgi_test.test_client import main, parse_args
|
||||
|
||||
uri1 = os.environ["TEST_URI1"]
|
||||
uri2 = os.environ["TEST_URI2"]
|
||||
args = f"-n 25000 -t 100 parallel {uri1} {uri2}"
|
||||
try:
|
||||
main(*parse_args(args.split()))
|
||||
|
||||
args = f"-n 25000 serial {uri1} {uri2}"
|
||||
main(*parse_args(args.split()))
|
||||
except Exception as e:
|
||||
LOGGER.error(Path("error_log").read_text())
|
||||
raise e
|
||||
|
||||
|
||||
def teardown_mod_wsgi() -> None:
|
||||
apache = os.environ["APACHE_BINARY"]
|
||||
apache_config = os.environ["APACHE_CONFIG"]
|
||||
|
||||
run_command(f"{apache} -k stop -f {ROOT}/test/mod_wsgi_test/{apache_config}")
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
setup_mod_wsgi()
|
||||
111
.evergreen/scripts/oidc_tester.py
Normal file
111
.evergreen/scripts/oidc_tester.py
Normal file
@ -0,0 +1,111 @@
|
||||
from __future__ import annotations
|
||||
|
||||
import os
|
||||
|
||||
from utils import (
|
||||
DRIVERS_TOOLS,
|
||||
TMP_DRIVER_FILE,
|
||||
create_archive,
|
||||
read_env,
|
||||
run_command,
|
||||
write_env,
|
||||
)
|
||||
|
||||
K8S_NAMES = ["aks", "gke", "eks"]
|
||||
K8S_REMOTE_NAMES = [f"{n}-remote" for n in K8S_NAMES]
|
||||
|
||||
|
||||
def _get_target_dir(sub_test_name: str) -> str:
|
||||
if sub_test_name == "default":
|
||||
target_dir = "auth_oidc"
|
||||
elif sub_test_name.startswith("azure"):
|
||||
target_dir = "auth_oidc/azure"
|
||||
elif sub_test_name.startswith("gcp"):
|
||||
target_dir = "auth_oidc/gcp"
|
||||
elif sub_test_name in K8S_NAMES + K8S_REMOTE_NAMES:
|
||||
target_dir = "auth_oidc/k8s"
|
||||
else:
|
||||
raise ValueError(f"Invalid sub test name '{sub_test_name}'")
|
||||
return f"{DRIVERS_TOOLS}/.evergreen/{target_dir}"
|
||||
|
||||
|
||||
def setup_oidc(sub_test_name: str) -> dict[str, str] | None:
|
||||
target_dir = _get_target_dir(sub_test_name)
|
||||
env = os.environ.copy()
|
||||
|
||||
if sub_test_name == "eks" and "AWS_ACCESS_KEY_ID" in os.environ:
|
||||
# Store AWS creds for kubectl access.
|
||||
for key in ["AWS_ACCESS_KEY_ID", "AWS_SECRET_ACCESS_KEY", "AWS_SESSION_TOKEN"]:
|
||||
if key in os.environ:
|
||||
write_env(key, os.environ[key])
|
||||
|
||||
if sub_test_name == "azure":
|
||||
env["AZUREOIDC_VMNAME_PREFIX"] = "PYTHON_DRIVER"
|
||||
if "-remote" not in sub_test_name:
|
||||
if sub_test_name == "azure":
|
||||
# Found using "az vm image list --output table"
|
||||
env["AZUREOIDC_IMAGE"] = "Canonical:0001-com-ubuntu-server-jammy:22_04-lts-gen2:latest"
|
||||
else:
|
||||
env["GCPKMS_IMAGEFAMILY"] = "debian-12"
|
||||
run_command(f"bash {target_dir}/setup.sh", env=env)
|
||||
if sub_test_name in K8S_NAMES:
|
||||
run_command(f"bash {target_dir}/setup-pod.sh {sub_test_name}")
|
||||
run_command(f"bash {target_dir}/run-self-test.sh")
|
||||
return None
|
||||
|
||||
source_file = None
|
||||
if sub_test_name == "default":
|
||||
source_file = f"{target_dir}/secrets-export.sh"
|
||||
elif sub_test_name in ["azure-remote", "gcp-remote"]:
|
||||
source_file = "./secrets-export.sh"
|
||||
if sub_test_name in K8S_REMOTE_NAMES:
|
||||
return os.environ.copy()
|
||||
if source_file is None:
|
||||
return None
|
||||
|
||||
config = read_env(source_file)
|
||||
write_env("MONGODB_URI_SINGLE", config["MONGODB_URI_SINGLE"])
|
||||
write_env("MONGODB_URI", config["MONGODB_URI"])
|
||||
write_env("DB_IP", config["MONGODB_URI"])
|
||||
|
||||
if sub_test_name == "default":
|
||||
write_env("OIDC_TOKEN_FILE", config["OIDC_TOKEN_FILE"])
|
||||
write_env("OIDC_TOKEN_DIR", config["OIDC_TOKEN_DIR"])
|
||||
if "OIDC_DOMAIN" in config:
|
||||
write_env("OIDC_DOMAIN", config["OIDC_DOMAIN"])
|
||||
elif sub_test_name == "azure-remote":
|
||||
write_env("AZUREOIDC_RESOURCE", config["AZUREOIDC_RESOURCE"])
|
||||
elif sub_test_name == "gcp-remote":
|
||||
write_env("GCPOIDC_AUDIENCE", config["GCPOIDC_AUDIENCE"])
|
||||
return config
|
||||
|
||||
|
||||
def test_oidc_send_to_remote(sub_test_name: str) -> None:
|
||||
env = os.environ.copy()
|
||||
target_dir = _get_target_dir(sub_test_name)
|
||||
create_archive()
|
||||
if sub_test_name in ["azure", "gcp"]:
|
||||
upper_name = sub_test_name.upper()
|
||||
env[f"{upper_name}OIDC_DRIVERS_TAR_FILE"] = TMP_DRIVER_FILE
|
||||
env[
|
||||
f"{upper_name}OIDC_TEST_CMD"
|
||||
] = f"OIDC_ENV={sub_test_name} ./.evergreen/run-mongodb-oidc-test.sh"
|
||||
elif sub_test_name in K8S_NAMES:
|
||||
env["K8S_DRIVERS_TAR_FILE"] = TMP_DRIVER_FILE
|
||||
env["K8S_TEST_CMD"] = "OIDC_ENV=k8s ./.evergreen/run-mongodb-oidc-test.sh"
|
||||
run_command(f"bash {target_dir}/run-driver-test.sh", env=env)
|
||||
|
||||
|
||||
def teardown_oidc(sub_test_name: str) -> None:
|
||||
target_dir = _get_target_dir(sub_test_name)
|
||||
# For k8s, make sure an error while tearing down the pod doesn't prevent
|
||||
# the Altas server teardown.
|
||||
error = None
|
||||
if sub_test_name in K8S_NAMES:
|
||||
try:
|
||||
run_command(f"bash {target_dir}/teardown-pod.sh")
|
||||
except Exception as e:
|
||||
error = e
|
||||
run_command(f"bash {target_dir}/teardown.sh")
|
||||
if error:
|
||||
raise error
|
||||
15
.evergreen/scripts/perf-submission-setup.sh
Executable file
15
.evergreen/scripts/perf-submission-setup.sh
Executable file
@ -0,0 +1,15 @@
|
||||
#!/bin/bash
|
||||
# We use the requester expansion to determine whether the data is from a mainline evergreen run or not
|
||||
|
||||
set -eu
|
||||
|
||||
# shellcheck disable=SC2154
|
||||
if [ "${requester}" == "commit" ]; then
|
||||
echo "is_mainline: true" >> expansion.yml
|
||||
else
|
||||
echo "is_mainline: false" >> expansion.yml
|
||||
fi
|
||||
|
||||
# We parse the username out of the order_id as patches append that in and SPS does not need that information
|
||||
# shellcheck disable=SC2154
|
||||
echo "parsed_order_id: $(echo "${revision_order_id}" | awk -F'_' '{print $NF}')" >> expansion.yml
|
||||
25
.evergreen/scripts/perf-submission.sh
Executable file
25
.evergreen/scripts/perf-submission.sh
Executable file
@ -0,0 +1,25 @@
|
||||
#!/bin/bash
|
||||
# We use the requester expansion to determine whether the data is from a mainline evergreen run or not
|
||||
|
||||
set -eu
|
||||
|
||||
# Submit the performance data to the SPS endpoint
|
||||
# shellcheck disable=SC2154
|
||||
response=$(curl -s -w "\nHTTP_STATUS:%{http_code}" -X 'POST' \
|
||||
"https://performance-monitoring-api.corp.mongodb.com/raw_perf_results/cedar_report?project=${project_id}&version=${version_id}&variant=${build_variant}&order=${parsed_order_id}&task_name=${task_name}&task_id=${task_id}&execution=${execution}&mainline=${is_mainline}" \
|
||||
-H 'accept: application/json' \
|
||||
-H 'Content-Type: application/json' \
|
||||
-d @results.json)
|
||||
|
||||
http_status=$(echo "$response" | grep "HTTP_STATUS" | awk -F':' '{print $2}')
|
||||
response_body=$(echo "$response" | sed '/HTTP_STATUS/d')
|
||||
|
||||
# We want to throw an error if the data was not successfully submitted
|
||||
if [ "$http_status" -ne 200 ]; then
|
||||
echo "Error: Received HTTP status $http_status"
|
||||
echo "Response Body: $response_body"
|
||||
exit 1
|
||||
fi
|
||||
|
||||
echo "Response Body: $response_body"
|
||||
echo "HTTP Status: $http_status"
|
||||
@ -1,28 +0,0 @@
|
||||
#!/bin/bash
|
||||
set -eu
|
||||
|
||||
HERE=$(dirname ${BASH_SOURCE:-$0})
|
||||
pushd $HERE
|
||||
. env.sh
|
||||
|
||||
rm -rf $DRIVERS_TOOLS
|
||||
git clone https://github.com/mongodb-labs/drivers-evergreen-tools.git $DRIVERS_TOOLS
|
||||
echo "{ \"releases\": { \"default\": \"$MONGODB_BINARIES\" }}" >$MONGO_ORCHESTRATION_HOME/orchestration.config
|
||||
|
||||
popd
|
||||
|
||||
# Copy PyMongo's test certificates over driver-evergreen-tools'
|
||||
cp ${PROJECT_DIRECTORY}/test/certificates/* ${DRIVERS_TOOLS}/.evergreen/x509gen/
|
||||
|
||||
# Replace MongoOrchestration's client certificate.
|
||||
cp ${PROJECT_DIRECTORY}/test/certificates/client.pem ${MONGO_ORCHESTRATION_HOME}/lib/client.pem
|
||||
|
||||
if [ -w /etc/hosts ]; then
|
||||
SUDO=""
|
||||
else
|
||||
SUDO="sudo"
|
||||
fi
|
||||
|
||||
# Add 'server' and 'hostname_not_in_cert' as a hostnames
|
||||
echo "127.0.0.1 server" | $SUDO tee -a /etc/hosts
|
||||
echo "127.0.0.1 hostname_not_in_cert" | $SUDO tee -a /etc/hosts
|
||||
150
.evergreen/scripts/resync-all-specs.py
Normal file
150
.evergreen/scripts/resync-all-specs.py
Normal file
@ -0,0 +1,150 @@
|
||||
from __future__ import annotations
|
||||
|
||||
import argparse
|
||||
import os
|
||||
import pathlib
|
||||
import subprocess
|
||||
from argparse import Namespace
|
||||
from subprocess import CalledProcessError
|
||||
|
||||
JIRA_FILTER = "https://jira.mongodb.org/issues/?jql=labels%20%3D%20automated-sync%20AND%20status%20!%3D%20Closed"
|
||||
|
||||
|
||||
def resync_specs(directory: pathlib.Path, errored: dict[str, str]) -> None:
|
||||
"""Actually sync the specs"""
|
||||
print("Beginning to sync specs")
|
||||
for spec in os.scandir(directory):
|
||||
if not spec.is_dir():
|
||||
continue
|
||||
|
||||
if spec.name in ["asynchronous"]:
|
||||
continue
|
||||
try:
|
||||
subprocess.run(
|
||||
["bash", "./.evergreen/resync-specs.sh", spec.name], # noqa: S603, S607
|
||||
capture_output=True,
|
||||
text=True,
|
||||
check=True,
|
||||
)
|
||||
except CalledProcessError as exc:
|
||||
errored[spec.name] = exc.stderr
|
||||
print("Done syncing specs")
|
||||
|
||||
|
||||
def apply_patches(errored):
|
||||
print("Beginning to apply patches")
|
||||
subprocess.run(
|
||||
["bash", "./.evergreen/remove-unimplemented-tests.sh"], # noqa: S603, S607
|
||||
check=True,
|
||||
)
|
||||
try:
|
||||
# Avoid shell=True by passing arguments as a list.
|
||||
# Note: glob expansion doesn't work in shell=False, so we use a list of files.
|
||||
patches = [str(p) for p in pathlib.Path("./.evergreen/spec-patch/").glob("*")]
|
||||
if patches:
|
||||
subprocess.run(
|
||||
[ # noqa: S603, S607
|
||||
"git",
|
||||
"apply",
|
||||
"-R",
|
||||
"--allow-empty",
|
||||
"--whitespace=fix",
|
||||
*patches,
|
||||
],
|
||||
check=True,
|
||||
stderr=subprocess.PIPE,
|
||||
)
|
||||
except CalledProcessError as exc:
|
||||
errored["applying patches"] = exc.stderr
|
||||
|
||||
|
||||
def check_new_spec_directories(directory: pathlib.Path) -> list[str]:
|
||||
"""Check to see if there are any directories in the spec repo that don't exist in pymongo/test"""
|
||||
spec_dir = pathlib.Path(os.environ["MDB_SPECS"]) / "source"
|
||||
spec_set = {
|
||||
entry.name.replace("-", "_")
|
||||
for entry in os.scandir(spec_dir)
|
||||
if entry.is_dir()
|
||||
and (pathlib.Path(entry.path) / "tests").is_dir()
|
||||
and len(list(os.scandir(pathlib.Path(entry.path) / "tests"))) > 1
|
||||
}
|
||||
test_set = {entry.name.replace("-", "_") for entry in os.scandir(directory) if entry.is_dir()}
|
||||
known_mappings = {
|
||||
"ocsp_support": "ocsp",
|
||||
"client_side_operations_timeout": "csot",
|
||||
"mongodb_handshake": "handshake",
|
||||
"load_balancers": "load_balancer",
|
||||
"connection_monitoring_and_pooling": "connection_monitoring",
|
||||
"command_logging_and_monitoring": "command_logging",
|
||||
"initial_dns_seedlist_discovery": "srv_seedlist",
|
||||
"server_discovery_and_monitoring": "sdam_monitoring",
|
||||
}
|
||||
|
||||
for k, v in known_mappings.items():
|
||||
if k in spec_set:
|
||||
spec_set.remove(k)
|
||||
spec_set.add(v)
|
||||
return list(spec_set - test_set)
|
||||
|
||||
|
||||
def write_summary(errored: dict[str, str], new: list[str], filename: str | None) -> None:
|
||||
"""Generate the PR description"""
|
||||
pr_body = ""
|
||||
# Avoid shell=True and complex pipes by using Python to process git output
|
||||
process = subprocess.run(
|
||||
["git", "diff", "--name-only"], # noqa: S603, S607
|
||||
capture_output=True,
|
||||
text=True,
|
||||
check=True,
|
||||
)
|
||||
changed_files = process.stdout.strip().splitlines()
|
||||
succeeded_set = set()
|
||||
for f in changed_files:
|
||||
parts = f.split("/")
|
||||
if len(parts) > 1:
|
||||
succeeded_set.add(parts[1])
|
||||
succeeded = sorted(succeeded_set)
|
||||
|
||||
if len(succeeded) > 0:
|
||||
pr_body += "The following specs were changed:\n -"
|
||||
pr_body += "\n -".join(succeeded)
|
||||
pr_body += "\n"
|
||||
if len(errored) > 0:
|
||||
pr_body += "\n\nThe following spec syncs encountered errors:"
|
||||
for k, v in errored.items():
|
||||
pr_body += f"\n -{k}\n```{v}\n```"
|
||||
pr_body += "\n"
|
||||
if len(new) > 0:
|
||||
pr_body += "\n\nThe following directories are in the specification repository and not in our test directory:\n -"
|
||||
pr_body += "\n -".join(new)
|
||||
pr_body += "\n"
|
||||
if pr_body != "":
|
||||
pr_body = f"Jira tickets: {JIRA_FILTER}\n\n" + pr_body
|
||||
if filename is None:
|
||||
print(f"\n{pr_body}")
|
||||
else:
|
||||
with open(filename, "w") as f:
|
||||
# replacements made for proper json
|
||||
f.write(pr_body.replace("\n", "\\n").replace("\t", "\\t"))
|
||||
|
||||
|
||||
def main(args: Namespace):
|
||||
directory = pathlib.Path("./test")
|
||||
errored: dict[str, str] = {}
|
||||
resync_specs(directory, errored)
|
||||
apply_patches(errored)
|
||||
new = check_new_spec_directories(directory)
|
||||
write_summary(errored, new, args.filename)
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
parser = argparse.ArgumentParser(
|
||||
description="Python Script to resync all specs and generate summary for PR."
|
||||
)
|
||||
parser.add_argument(
|
||||
"--filename",
|
||||
help="Name of file for the summary to be written into.",
|
||||
default=None,
|
||||
)
|
||||
args = parser.parse_args()
|
||||
main(args)
|
||||
43
.evergreen/scripts/resync-all-specs.sh
Executable file
43
.evergreen/scripts/resync-all-specs.sh
Executable file
@ -0,0 +1,43 @@
|
||||
#!/usr/bin/env bash
|
||||
# Run spec syncing script and create PR
|
||||
set -eu
|
||||
|
||||
# SETUP
|
||||
SRC_URL="https://github.com/mongodb/specifications.git"
|
||||
# needs to be set for resync-specs.sh
|
||||
SPEC_SRC="$(realpath "../specifications")"
|
||||
SCRIPT="$(realpath "./.evergreen/resync-specs.sh")"
|
||||
|
||||
# Clone the spec repo if the directory does not exist
|
||||
if [[ ! -d $SPEC_SRC ]]; then
|
||||
git clone $SRC_URL $SPEC_SRC
|
||||
if [[ $? -ne 0 ]]; then
|
||||
echo "Error: Failed to clone repository."
|
||||
exit 1
|
||||
fi
|
||||
fi
|
||||
|
||||
# Set environment variable to the cloned spec repo for resync-specs.sh
|
||||
export MDB_SPECS="$SPEC_SRC"
|
||||
|
||||
# Check that resync-specs.sh exists and is executable
|
||||
if [[ ! -x $SCRIPT ]]; then
|
||||
echo "Error: $SCRIPT not found or is not executable."
|
||||
exit 1
|
||||
fi
|
||||
|
||||
PR_DESC="spec_sync.txt"
|
||||
|
||||
# run python script that actually does all the resyncing
|
||||
if ! [ -n "${CI:-}" ]
|
||||
then
|
||||
# we're running locally
|
||||
python3 ./.evergreen/scripts/resync-all-specs.py
|
||||
else
|
||||
/opt/devtools/bin/python3.11 ./.evergreen/scripts/resync-all-specs.py --filename "$PR_DESC"
|
||||
if [[ -f $PR_DESC ]]; then
|
||||
# changes were made -> call scrypt to create PR for us
|
||||
.evergreen/scripts/create-spec-pr.sh "$PR_DESC"
|
||||
rm "$PR_DESC"
|
||||
fi
|
||||
fi
|
||||
@ -1,7 +0,0 @@
|
||||
#!/bin/bash
|
||||
|
||||
# Disable xtrace for security reasons (just in case it was accidentally set).
|
||||
set +x
|
||||
set -o errexit
|
||||
bash "${DRIVERS_TOOLS}"/.evergreen/auth_aws/setup_secrets.sh drivers/atlas_connect
|
||||
TEST_ATLAS=1 bash "${PROJECT_DIRECTORY}"/.evergreen/just.sh test-eg
|
||||
@ -1,15 +0,0 @@
|
||||
#!/bin/bash
|
||||
|
||||
# shellcheck disable=SC2154
|
||||
if [ "${skip_ECS_auth_test}" = "true" ]; then
|
||||
echo "This platform does not support the ECS auth test, skipping..."
|
||||
exit 0
|
||||
fi
|
||||
set -ex
|
||||
cd "$DRIVERS_TOOLS"/.evergreen/auth_aws
|
||||
. ./activate-authawsvenv.sh
|
||||
. aws_setup.sh ecs
|
||||
export MONGODB_BINARIES="$MONGODB_BINARIES"
|
||||
export PROJECT_DIRECTORY="$PROJECT_DIRECTORY"
|
||||
python aws_tester.py ecs
|
||||
cd -
|
||||
@ -1,10 +0,0 @@
|
||||
#!/bin/bash
|
||||
set -x
|
||||
. .evergreen/utils.sh
|
||||
|
||||
. .evergreen/scripts/env.sh
|
||||
createvirtualenv "$PYTHON_BINARY" .venv
|
||||
|
||||
export PYMONGO_C_EXT_MUST_BUILD=1
|
||||
pip install -e ".[test]"
|
||||
pytest -v
|
||||
@ -1,4 +0,0 @@
|
||||
#!/bin/bash
|
||||
|
||||
set -o xtrace
|
||||
PYTHON_BINARY=${PYTHON_BINARY} bash "${PROJECT_DIRECTORY}"/.evergreen/just.sh docs-test
|
||||
@ -1,8 +0,0 @@
|
||||
#!/bin/bash
|
||||
set -eu
|
||||
|
||||
# Disable xtrace for security reasons (just in case it was accidentally set).
|
||||
set +x
|
||||
# Use the default python to bootstrap secrets.
|
||||
bash "${DRIVERS_TOOLS}"/.evergreen/secrets_handling/setup-secrets.sh drivers/enterprise_auth
|
||||
TEST_ENTERPRISE_AUTH=1 AUTH=auth bash "${PROJECT_DIRECTORY}"/.evergreen/just.sh test-eg
|
||||
@ -1,7 +0,0 @@
|
||||
#!/bin/bash
|
||||
|
||||
. .evergreen/scripts/env.sh
|
||||
export PYTHON_BINARY=/opt/mongodbtoolchain/v4/bin/python3
|
||||
export LIBMONGOCRYPT_URL=https://s3.amazonaws.com/mciuploads/libmongocrypt/debian11/master/latest/libmongocrypt.tar.gz
|
||||
SKIP_SERVERS=1 bash ./.evergreen/setup-encryption.sh
|
||||
SUCCESS=false TEST_FLE_GCP_AUTO=1 ./.evergreen/just.sh test-eg
|
||||
@ -1,11 +1,14 @@
|
||||
#!/bin/bash
|
||||
# Get the debug data for an evergreen task.
|
||||
set -eu
|
||||
|
||||
set -o xtrace
|
||||
. ${DRIVERS_TOOLS}/.evergreen/download-mongodb.sh || true
|
||||
. ${DRIVERS_TOOLS}/.evergreen/get-distro.sh || true
|
||||
get_distro || true
|
||||
echo $DISTRO
|
||||
echo $MARCH
|
||||
echo $OS
|
||||
|
||||
set -x
|
||||
uname -a || true
|
||||
ls /etc/*release* || true
|
||||
cc --version || true
|
||||
@ -20,3 +23,4 @@ ls -la /usr/local/Cellar/ || true
|
||||
scan-build --version || true
|
||||
genhtml --version || true
|
||||
valgrind --version || true
|
||||
set +x
|
||||
|
||||
@ -1,3 +0,0 @@
|
||||
#!/bin/bash
|
||||
|
||||
MONGODB_URI=${MONGODB_URI} bash "${DRIVERS_TOOLS}"/.evergreen/run-load-balancer.sh start
|
||||
@ -1,5 +0,0 @@
|
||||
#!/bin/bash
|
||||
|
||||
set -o xtrace
|
||||
export PYTHON_BINARY=${PYTHON_BINARY}
|
||||
bash "${PROJECT_DIRECTORY}"/.evergreen/just.sh test-mockupdb
|
||||
@ -1,52 +0,0 @@
|
||||
#!/bin/bash
|
||||
set -o xtrace
|
||||
set -o errexit
|
||||
|
||||
APACHE=$(command -v apache2 || command -v /usr/lib/apache2/mpm-prefork/apache2) || true
|
||||
if [ -n "$APACHE" ]; then
|
||||
APACHE_CONFIG=apache24ubuntu161404.conf
|
||||
else
|
||||
APACHE=$(command -v httpd) || true
|
||||
if [ -z "$APACHE" ]; then
|
||||
echo "Could not find apache2 binary"
|
||||
exit 1
|
||||
else
|
||||
APACHE_CONFIG=apache22amazon.conf
|
||||
fi
|
||||
fi
|
||||
|
||||
|
||||
PYTHON_VERSION=$(${PYTHON_BINARY} -c "import sys; sys.stdout.write('.'.join(str(val) for val in sys.version_info[:2]))")
|
||||
|
||||
# Ensure the C extensions are installed.
|
||||
${PYTHON_BINARY} -m venv --system-site-packages .venv
|
||||
source .venv/bin/activate
|
||||
pip install -U pip
|
||||
python -m pip install -e .
|
||||
|
||||
export MOD_WSGI_SO=/opt/python/mod_wsgi/python_version/$PYTHON_VERSION/mod_wsgi_version/$MOD_WSGI_VERSION/mod_wsgi.so
|
||||
export PYTHONHOME=/opt/python/$PYTHON_VERSION
|
||||
# If MOD_WSGI_EMBEDDED is set use the default embedded mode behavior instead
|
||||
# of daemon mode (WSGIDaemonProcess).
|
||||
if [ -n "${MOD_WSGI_EMBEDDED:-}" ]; then
|
||||
export MOD_WSGI_CONF=mod_wsgi_test_embedded.conf
|
||||
else
|
||||
export MOD_WSGI_CONF=mod_wsgi_test.conf
|
||||
fi
|
||||
|
||||
cd ..
|
||||
$APACHE -k start -f ${PROJECT_DIRECTORY}/test/mod_wsgi_test/${APACHE_CONFIG}
|
||||
trap '$APACHE -k stop -f ${PROJECT_DIRECTORY}/test/mod_wsgi_test/${APACHE_CONFIG}' EXIT HUP
|
||||
|
||||
wget -t 1 -T 10 -O - "http://localhost:8080/interpreter1${PROJECT_DIRECTORY}" || (cat error_log && exit 1)
|
||||
wget -t 1 -T 10 -O - "http://localhost:8080/interpreter2${PROJECT_DIRECTORY}" || (cat error_log && exit 1)
|
||||
|
||||
python ${PROJECT_DIRECTORY}/test/mod_wsgi_test/test_client.py -n 25000 -t 100 parallel \
|
||||
http://localhost:8080/interpreter1${PROJECT_DIRECTORY} http://localhost:8080/interpreter2${PROJECT_DIRECTORY} || \
|
||||
(tail -n 100 error_log && exit 1)
|
||||
|
||||
python ${PROJECT_DIRECTORY}/test/mod_wsgi_test/test_client.py -n 25000 serial \
|
||||
http://localhost:8080/interpreter1${PROJECT_DIRECTORY} http://localhost:8080/interpreter2${PROJECT_DIRECTORY} || \
|
||||
(tail -n 100 error_log && exit 1)
|
||||
|
||||
rm -rf .venv
|
||||
@ -1,33 +0,0 @@
|
||||
#!/bin/bash
|
||||
|
||||
set -o xtrace
|
||||
set -o errexit # Exit the script with error if any of the commands fail
|
||||
|
||||
############################################
|
||||
# Main Program #
|
||||
############################################
|
||||
|
||||
# Supported/used environment variables:
|
||||
# MONGODB_URI Set the URI, including an optional username/password to use
|
||||
# to connect to the server via MONGODB-AWS authentication
|
||||
# mechanism.
|
||||
# PYTHON_BINARY The Python version to use.
|
||||
|
||||
# shellcheck disable=SC2154
|
||||
if [ "${skip_EC2_auth_test:-}" = "true" ] && { [ "$1" = "ec2" ] || [ "$1" = "web-identity" ]; }; then
|
||||
echo "This platform does not support the EC2 auth test, skipping..."
|
||||
exit 0
|
||||
fi
|
||||
|
||||
echo "Running MONGODB-AWS authentication tests for $1"
|
||||
|
||||
# Handle credentials and environment setup.
|
||||
. "$DRIVERS_TOOLS"/.evergreen/auth_aws/aws_setup.sh "$1"
|
||||
|
||||
# show test output
|
||||
set -x
|
||||
|
||||
export TEST_AUTH_AWS=1
|
||||
export AUTH="auth"
|
||||
export SET_XTRACE_ON=1
|
||||
bash ./.evergreen/just.sh test-eg
|
||||
@ -1,8 +0,0 @@
|
||||
#!/bin/bash
|
||||
|
||||
TEST_OCSP=1 \
|
||||
PYTHON_BINARY="${PYTHON_BINARY}" \
|
||||
CA_FILE="${DRIVERS_TOOLS}/.evergreen/ocsp/${OCSP_ALGORITHM}/ca.pem" \
|
||||
OCSP_TLS_SHOULD_SUCCEED="${OCSP_TLS_SHOULD_SUCCEED}" \
|
||||
bash "${PROJECT_DIRECTORY}"/.evergreen/just.sh test-eg
|
||||
bash "${DRIVERS_TOOLS}"/.evergreen/ocsp/teardown.sh
|
||||
@ -1,4 +0,0 @@
|
||||
#!/bin/bash
|
||||
|
||||
PROJECT_DIRECTORY=${PROJECT_DIRECTORY}
|
||||
bash "${PROJECT_DIRECTORY}"/.evergreen/run-perf-tests.sh
|
||||
13
.evergreen/scripts/run-server.sh
Executable file
13
.evergreen/scripts/run-server.sh
Executable file
@ -0,0 +1,13 @@
|
||||
#!/bin/bash
|
||||
|
||||
set -eu
|
||||
|
||||
HERE=$(dirname ${BASH_SOURCE:-$0})
|
||||
|
||||
# Try to source the env file.
|
||||
if [ -f $HERE/env.sh ]; then
|
||||
echo "Sourcing env file"
|
||||
source $HERE/env.sh
|
||||
fi
|
||||
|
||||
uv run $HERE/run_server.py "$@"
|
||||
@ -1,54 +0,0 @@
|
||||
#!/bin/bash
|
||||
|
||||
# Disable xtrace
|
||||
set +x
|
||||
if [ -n "${MONGODB_STARTED}" ]; then
|
||||
export PYMONGO_MUST_CONNECT=true
|
||||
fi
|
||||
if [ -n "${DISABLE_TEST_COMMANDS}" ]; then
|
||||
export PYMONGO_DISABLE_TEST_COMMANDS=1
|
||||
fi
|
||||
if [ -n "${test_encryption}" ]; then
|
||||
# Disable xtrace (just in case it was accidentally set).
|
||||
set +x
|
||||
bash "${DRIVERS_TOOLS}"/.evergreen/csfle/await-servers.sh
|
||||
export TEST_ENCRYPTION=1
|
||||
if [ -n "${test_encryption_pyopenssl}" ]; then
|
||||
export TEST_ENCRYPTION_PYOPENSSL=1
|
||||
fi
|
||||
fi
|
||||
if [ -n "${test_crypt_shared}" ]; then
|
||||
export TEST_CRYPT_SHARED=1
|
||||
export CRYPT_SHARED_LIB_PATH=${CRYPT_SHARED_LIB_PATH}
|
||||
fi
|
||||
if [ -n "${test_pyopenssl}" ]; then
|
||||
export TEST_PYOPENSSL=1
|
||||
fi
|
||||
if [ -n "${SETDEFAULTENCODING}" ]; then
|
||||
export SETDEFAULTENCODING="${SETDEFAULTENCODING}"
|
||||
fi
|
||||
if [ -n "${test_loadbalancer}" ]; then
|
||||
export TEST_LOADBALANCER=1
|
||||
export SINGLE_MONGOS_LB_URI="${SINGLE_MONGOS_LB_URI}"
|
||||
export MULTI_MONGOS_LB_URI="${MULTI_MONGOS_LB_URI}"
|
||||
fi
|
||||
if [ -n "${test_serverless}" ]; then
|
||||
export TEST_SERVERLESS=1
|
||||
fi
|
||||
if [ -n "${TEST_INDEX_MANAGEMENT:-}" ]; then
|
||||
export TEST_INDEX_MANAGEMENT=1
|
||||
fi
|
||||
if [ -n "${SKIP_CSOT_TESTS}" ]; then
|
||||
export SKIP_CSOT_TESTS=1
|
||||
fi
|
||||
GREEN_FRAMEWORK=${GREEN_FRAMEWORK} \
|
||||
PYTHON_BINARY=${PYTHON_BINARY} \
|
||||
NO_EXT=${NO_EXT} \
|
||||
COVERAGE=${COVERAGE} \
|
||||
COMPRESSORS=${COMPRESSORS} \
|
||||
AUTH=${AUTH} \
|
||||
SSL=${SSL} \
|
||||
TEST_DATA_LAKE=${TEST_DATA_LAKE:-} \
|
||||
TEST_SUITES=${TEST_SUITES:-} \
|
||||
MONGODB_API_VERSION=${MONGODB_API_VERSION} \
|
||||
bash "${PROJECT_DIRECTORY}"/.evergreen/just.sh test-eg
|
||||
@ -1,21 +0,0 @@
|
||||
#!/bin/bash -eu
|
||||
|
||||
# Example use: bash run-with-env.sh run-tests.sh {args...}
|
||||
|
||||
# Parameter expansion to get just the current directory's name
|
||||
if [ "${PWD##*/}" == "src" ]; then
|
||||
. .evergreen/scripts/env.sh
|
||||
if [ -f ".evergreen/scripts/test-env.sh" ]; then
|
||||
. .evergreen/scripts/test-env.sh
|
||||
fi
|
||||
else
|
||||
. src/.evergreen/scripts/env.sh
|
||||
if [ -f "src/.evergreen/scripts/test-env.sh" ]; then
|
||||
. src/.evergreen/scripts/test-env.sh
|
||||
fi
|
||||
fi
|
||||
|
||||
set -eu
|
||||
|
||||
# shellcheck source=/dev/null
|
||||
. "$@"
|
||||
59
.evergreen/scripts/run_server.py
Normal file
59
.evergreen/scripts/run_server.py
Normal file
@ -0,0 +1,59 @@
|
||||
from __future__ import annotations
|
||||
|
||||
import os
|
||||
from typing import Any
|
||||
|
||||
from utils import DRIVERS_TOOLS, ROOT, get_test_options, run_command
|
||||
|
||||
|
||||
def set_env(name: str, value: Any = "1") -> None:
|
||||
os.environ[name] = str(value)
|
||||
|
||||
|
||||
def start_server():
|
||||
opts, extra_opts = get_test_options(
|
||||
"Run a MongoDB server. All given flags will be passed to run-mongodb.sh in DRIVERS_TOOLS.",
|
||||
require_sub_test_name=False,
|
||||
allow_extra_opts=True,
|
||||
)
|
||||
test_name = opts.test_name
|
||||
|
||||
# drivers-evergreen-tools expects the version variable to be named MONGODB_VERSION.
|
||||
if "VERSION" in os.environ:
|
||||
os.environ["MONGODB_VERSION"] = os.environ["VERSION"]
|
||||
|
||||
if test_name == "auth_aws":
|
||||
set_env("AUTH_AWS")
|
||||
|
||||
elif test_name == "load_balancer":
|
||||
set_env("LOAD_BALANCER")
|
||||
|
||||
elif test_name == "search_index":
|
||||
os.environ["TOPOLOGY"] = "replica_set"
|
||||
os.environ["MONGODB_VERSION"] = "7.0"
|
||||
|
||||
if not os.environ.get("TEST_CRYPT_SHARED"):
|
||||
set_env("SKIP_CRYPT_SHARED")
|
||||
|
||||
if opts.ssl:
|
||||
extra_opts.append("--ssl")
|
||||
if test_name != "ocsp":
|
||||
certs = ROOT / "test/certificates"
|
||||
set_env("TLS_CERT_KEY_FILE", certs / "client.pem")
|
||||
set_env("TLS_PEM_KEY_FILE", certs / "server.pem")
|
||||
set_env("TLS_CA_FILE", certs / "ca.pem")
|
||||
|
||||
if opts.auth:
|
||||
extra_opts.append("--auth")
|
||||
|
||||
if opts.verbose:
|
||||
extra_opts.append("-v")
|
||||
elif opts.quiet:
|
||||
extra_opts.append("-q")
|
||||
|
||||
cmd = ["bash", f"{DRIVERS_TOOLS}/.evergreen/run-mongodb.sh", "start", *extra_opts]
|
||||
run_command(cmd, cwd=DRIVERS_TOOLS)
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
start_server()
|
||||
228
.evergreen/scripts/run_tests.py
Normal file
228
.evergreen/scripts/run_tests.py
Normal file
@ -0,0 +1,228 @@
|
||||
from __future__ import annotations
|
||||
|
||||
import json
|
||||
import logging
|
||||
import os
|
||||
import platform
|
||||
import shlex
|
||||
import shutil
|
||||
import subprocess
|
||||
import sys
|
||||
from datetime import datetime
|
||||
from pathlib import Path
|
||||
from shutil import which
|
||||
|
||||
try:
|
||||
import importlib_metadata
|
||||
except ImportError:
|
||||
from importlib import metadata as importlib_metadata
|
||||
|
||||
|
||||
import pytest
|
||||
from utils import DRIVERS_TOOLS, LOGGER, ROOT, run_command
|
||||
|
||||
AUTH = os.environ.get("AUTH", "noauth")
|
||||
SSL = os.environ.get("SSL", "nossl")
|
||||
UV_ARGS = os.environ.get("UV_ARGS", "")
|
||||
TEST_PERF = os.environ.get("TEST_PERF")
|
||||
GREEN_FRAMEWORK = os.environ.get("GREEN_FRAMEWORK")
|
||||
TEST_ARGS = os.environ.get("TEST_ARGS", "").split()
|
||||
TEST_NAME = os.environ.get("TEST_NAME")
|
||||
SUB_TEST_NAME = os.environ.get("SUB_TEST_NAME")
|
||||
|
||||
|
||||
def list_packages():
|
||||
packages = set()
|
||||
for distribution in importlib_metadata.distributions():
|
||||
if distribution.name:
|
||||
packages.add(distribution.name)
|
||||
print("Package Version URL")
|
||||
print("------------------- ----------- ----------------------------------------------------")
|
||||
for name in sorted(packages):
|
||||
distribution = importlib_metadata.distribution(name)
|
||||
url = ""
|
||||
if distribution.origin is not None:
|
||||
url = distribution.origin.url
|
||||
print(f"{name:20s}{distribution.version:12s}{url}")
|
||||
print("------------------- ----------- ----------------------------------------------------\n")
|
||||
|
||||
|
||||
def handle_perf(start_time: datetime):
|
||||
end_time = datetime.now()
|
||||
elapsed_secs = (end_time - start_time).total_seconds()
|
||||
with open("results.json") as fid:
|
||||
results = json.load(fid)
|
||||
LOGGER.info("results.json:\n%s", json.dumps(results, indent=2))
|
||||
|
||||
results = dict(
|
||||
status="PASS",
|
||||
exit_code=0,
|
||||
test_file="BenchMarkTests",
|
||||
start=int(start_time.timestamp()),
|
||||
end=int(end_time.timestamp()),
|
||||
elapsed=elapsed_secs,
|
||||
)
|
||||
report = dict(failures=0, results=[results])
|
||||
LOGGER.info("report.json\n%s", json.dumps(report, indent=2))
|
||||
|
||||
with open("report.json", "w", newline="\n") as fid:
|
||||
json.dump(report, fid)
|
||||
|
||||
|
||||
def handle_green_framework() -> None:
|
||||
if GREEN_FRAMEWORK == "gevent":
|
||||
from gevent import monkey
|
||||
|
||||
monkey.patch_all()
|
||||
|
||||
# Never run async tests with a framework.
|
||||
if len(TEST_ARGS) <= 1:
|
||||
TEST_ARGS.extend(["-m", "not default_async and default"])
|
||||
else:
|
||||
for i in range(len(TEST_ARGS) - 1):
|
||||
if "-m" in TEST_ARGS[i]:
|
||||
TEST_ARGS[i + 1] = f"not default_async and {TEST_ARGS[i + 1]}"
|
||||
|
||||
LOGGER.info(f"Running tests with {GREEN_FRAMEWORK}...")
|
||||
|
||||
|
||||
def handle_c_ext() -> None:
|
||||
if platform.python_implementation() != "CPython":
|
||||
return
|
||||
sys.path.insert(0, str(ROOT / "tools"))
|
||||
from fail_if_no_c import main as fail_if_no_c
|
||||
|
||||
fail_if_no_c()
|
||||
|
||||
|
||||
def handle_pymongocrypt() -> None:
|
||||
import pymongocrypt
|
||||
|
||||
LOGGER.info(f"pymongocrypt version: {pymongocrypt.__version__})")
|
||||
LOGGER.info(f"libmongocrypt version: {pymongocrypt.libmongocrypt_version()})")
|
||||
|
||||
|
||||
def handle_aws_lambda() -> None:
|
||||
env = os.environ.copy()
|
||||
target_dir = ROOT / "test/lambda"
|
||||
env["TEST_LAMBDA_DIRECTORY"] = str(target_dir)
|
||||
env.setdefault("AWS_REGION", "us-east-1")
|
||||
dirs = ["pymongo", "gridfs", "bson"]
|
||||
# Remove the original .so files.
|
||||
for dname in dirs:
|
||||
so_paths = [f"{f.parent.name}/{f.name}" for f in (ROOT / dname).glob("*.so")]
|
||||
for so_path in list(so_paths):
|
||||
Path(so_path).unlink()
|
||||
# Build the c extensions.
|
||||
docker = which("docker") or which("podman")
|
||||
if not docker:
|
||||
raise ValueError("Could not find docker!")
|
||||
image = "quay.io/pypa/manylinux2014_x86_64:latest"
|
||||
run_command(
|
||||
f'{docker} run --rm -v "{ROOT}:/src" --platform linux/amd64 {image} /src/test/lambda/build_internal.sh'
|
||||
)
|
||||
for dname in dirs:
|
||||
target = ROOT / "test/lambda/mongodb" / dname
|
||||
shutil.rmtree(target, ignore_errors=True)
|
||||
shutil.copytree(ROOT / dname, target)
|
||||
# Remove the new so files from the ROOT directory.
|
||||
for dname in dirs:
|
||||
so_paths = [f"{f.parent.name}/{f.name}" for f in (ROOT / dname).glob("*.so")]
|
||||
for so_path in list(so_paths):
|
||||
Path(so_path).unlink()
|
||||
|
||||
script_name = "run-deployed-lambda-aws-tests.sh"
|
||||
run_command(f"bash {DRIVERS_TOOLS}/.evergreen/aws_lambda/{script_name}", env=env)
|
||||
|
||||
|
||||
def run() -> None:
|
||||
# Add diagnostic for python version.
|
||||
print("Running with python", sys.version)
|
||||
|
||||
# List the installed packages.
|
||||
list_packages()
|
||||
|
||||
# Handle green framework first so they can patch modules.
|
||||
if GREEN_FRAMEWORK:
|
||||
handle_green_framework()
|
||||
|
||||
# Ensure C extensions if applicable.
|
||||
if not os.environ.get("NO_EXT"):
|
||||
handle_c_ext()
|
||||
|
||||
if os.environ.get("PYMONGOCRYPT_LIB"):
|
||||
handle_pymongocrypt()
|
||||
|
||||
LOGGER.info(f"Test setup:\n{AUTH=}\n{SSL=}\n{UV_ARGS=}\n{TEST_ARGS=}")
|
||||
|
||||
# Record the start time for a perf test.
|
||||
if TEST_PERF:
|
||||
start_time = datetime.now()
|
||||
|
||||
# Run mod_wsgi tests using the helper.
|
||||
if TEST_NAME == "mod_wsgi":
|
||||
from mod_wsgi_tester import test_mod_wsgi
|
||||
|
||||
test_mod_wsgi()
|
||||
return
|
||||
|
||||
# Send kms tests to run remotely.
|
||||
if TEST_NAME == "kms" and SUB_TEST_NAME in ["azure", "gcp"]:
|
||||
from kms_tester import test_kms_send_to_remote
|
||||
|
||||
test_kms_send_to_remote(SUB_TEST_NAME)
|
||||
return
|
||||
|
||||
# Handle doctests.
|
||||
if TEST_NAME == "doctest":
|
||||
from sphinx.cmd.build import main
|
||||
|
||||
result = main("-E -b doctest doc ./doc/_build/doctest".split())
|
||||
sys.exit(result)
|
||||
|
||||
# Send ecs tests to run remotely.
|
||||
if TEST_NAME == "auth_aws" and SUB_TEST_NAME == "ecs":
|
||||
run_command(f"{DRIVERS_TOOLS}/.evergreen/auth_aws/aws_setup.sh ecs")
|
||||
return
|
||||
|
||||
# Send OIDC tests to run remotely.
|
||||
if (
|
||||
TEST_NAME == "auth_oidc"
|
||||
and SUB_TEST_NAME != "default"
|
||||
and not SUB_TEST_NAME.endswith("-remote")
|
||||
):
|
||||
from oidc_tester import test_oidc_send_to_remote
|
||||
|
||||
test_oidc_send_to_remote(SUB_TEST_NAME)
|
||||
return
|
||||
|
||||
# Run deployed aws lambda tests.
|
||||
if TEST_NAME == "aws_lambda":
|
||||
handle_aws_lambda()
|
||||
return
|
||||
|
||||
if os.environ.get("DEBUG_LOG"):
|
||||
TEST_ARGS.extend(f"-o log_cli_level={logging.DEBUG}".split())
|
||||
|
||||
if os.environ.get("COVERAGE"):
|
||||
binary = sys.executable.replace(os.sep, "/")
|
||||
cmd = f"{binary} -m coverage run -m pytest {' '.join(TEST_ARGS)} {' '.join(sys.argv[1:])}"
|
||||
result = subprocess.run(shlex.split(cmd), check=False) # noqa: S603
|
||||
cmd = f"{binary} -m coverage report"
|
||||
subprocess.run(shlex.split(cmd), check=False) # noqa: S603
|
||||
if result.returncode != 0:
|
||||
print(result.stderr)
|
||||
sys.exit(result.returncode)
|
||||
|
||||
# Run local tests.
|
||||
ret = pytest.main(TEST_ARGS + sys.argv[1:])
|
||||
if ret != 0:
|
||||
sys.exit(ret)
|
||||
|
||||
# Handle perf test post actions.
|
||||
if TEST_PERF:
|
||||
handle_perf(start_time)
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
run()
|
||||
@ -1,42 +1,58 @@
|
||||
#!/bin/bash
|
||||
|
||||
# Set up development environment.
|
||||
set -eu
|
||||
|
||||
HERE=$(dirname ${BASH_SOURCE:-$0})
|
||||
pushd "$(dirname "$(dirname $HERE)")" > /dev/null
|
||||
HERE="$( cd -- "$HERE" > /dev/null 2>&1 && pwd )"
|
||||
ROOT=$(dirname "$(dirname $HERE)")
|
||||
|
||||
# Source the env file to pick up common variables.
|
||||
# Source the env files to pick up common variables.
|
||||
if [ -f $HERE/env.sh ]; then
|
||||
source $HERE/env.sh
|
||||
. $HERE/env.sh
|
||||
fi
|
||||
|
||||
# Get variables defined in test-env.sh.
|
||||
if [ -f $HERE/test-env.sh ]; then
|
||||
. $HERE/test-env.sh
|
||||
fi
|
||||
|
||||
# Ensure dependencies are installed.
|
||||
. $HERE/install-dependencies.sh
|
||||
bash $HERE/install-dependencies.sh
|
||||
|
||||
# Handle the value for UV_PYTHON.
|
||||
. $HERE/setup-uv-python.sh
|
||||
|
||||
# Set the location of the python bin dir.
|
||||
if [ "Windows_NT" = "${OS:-}" ]; then
|
||||
BIN_DIR=.venv/Scripts
|
||||
else
|
||||
BIN_DIR=.venv/bin
|
||||
fi
|
||||
|
||||
# Ensure there is a python venv.
|
||||
if [ ! -d $BIN_DIR ]; then
|
||||
. .evergreen/utils.sh
|
||||
|
||||
if [ -z "${PYTHON_BINARY:-}" ]; then
|
||||
PYTHON_BINARY=$(find_python3)
|
||||
# Only run the next part if not running on CI.
|
||||
if [ -z "${CI:-}" ]; then
|
||||
# Add the default install path to the path if needed.
|
||||
if [ -z "${PYMONGO_BIN_DIR:-}" ]; then
|
||||
export PATH="$PATH:$HOME/.local/bin"
|
||||
fi
|
||||
export UV_PYTHON=${PYTHON_BINARY}
|
||||
echo "export UV_PYTHON=$UV_PYTHON" >> $HERE/env.sh
|
||||
echo "Using python $UV_PYTHON"
|
||||
fi
|
||||
uv sync --frozen
|
||||
uv run --frozen --with pip pip install -e .
|
||||
echo "Setting up python environment... done."
|
||||
|
||||
# Ensure there is a pre-commit hook if there is a git checkout.
|
||||
if [ -d .git ] && [ ! -f .git/hooks/pre-commit ]; then
|
||||
uv run --frozen pre-commit install
|
||||
# Set up venv, making sure c extensions build unless disabled.
|
||||
if [ -z "${NO_EXT:-}" ]; then
|
||||
export PYMONGO_C_EXT_MUST_BUILD=1
|
||||
fi
|
||||
|
||||
(
|
||||
cd $ROOT && uv sync
|
||||
)
|
||||
|
||||
# Set up build utilities on Windows spawn hosts.
|
||||
if [ -f $HOME/.visualStudioEnv.sh ]; then
|
||||
set +u
|
||||
SSH_TTY=1 source $HOME/.visualStudioEnv.sh
|
||||
set -u
|
||||
fi
|
||||
|
||||
# Only set up pre-commit if we are in a git checkout.
|
||||
if [ -f $HERE/.git ]; then
|
||||
if ! command -v pre-commit &>/dev/null; then
|
||||
uv tool install pre-commit
|
||||
fi
|
||||
|
||||
if [ ! -f .git/hooks/pre-commit ]; then
|
||||
uvx pre-commit install
|
||||
fi
|
||||
fi
|
||||
fi
|
||||
|
||||
@ -1,5 +0,0 @@
|
||||
#!/bin/bash
|
||||
|
||||
if [ -n "${test_encryption}" ]; then
|
||||
bash .evergreen/setup-encryption.sh
|
||||
fi
|
||||
@ -1,5 +1,5 @@
|
||||
#!/bin/bash
|
||||
|
||||
# Set up the system on an evergreen host.
|
||||
set -eu
|
||||
|
||||
HERE=$(dirname ${BASH_SOURCE:-$0})
|
||||
@ -7,8 +7,49 @@ pushd "$(dirname "$(dirname $HERE)")"
|
||||
echo "Setting up system..."
|
||||
bash .evergreen/scripts/configure-env.sh
|
||||
source .evergreen/scripts/env.sh
|
||||
bash .evergreen/scripts/prepare-resources.sh
|
||||
bash $DRIVERS_TOOLS/.evergreen/setup.sh
|
||||
bash .evergreen/scripts/install-dependencies.sh
|
||||
popd
|
||||
|
||||
# Run spawn host-specific tasks.
|
||||
if [ -z "${CI:-}" ]; then
|
||||
bash $HERE/setup-dev-env.sh
|
||||
fi
|
||||
|
||||
# Enable core dumps if enabled on the machine
|
||||
# Copied from https://github.com/mongodb/mongo/blob/master/etc/evergreen.yml
|
||||
if [ -f /proc/self/coredump_filter ]; then
|
||||
# Set the shell process (and its children processes) to dump ELF headers (bit 4),
|
||||
# anonymous shared mappings (bit 1), and anonymous private mappings (bit 0).
|
||||
echo 0x13 >/proc/self/coredump_filter
|
||||
|
||||
if [ -f /sbin/sysctl ]; then
|
||||
# Check that the core pattern is set explicitly on our distro image instead
|
||||
# of being the OS's default value. This ensures that coredump names are consistent
|
||||
# across distros and can be picked up by Evergreen.
|
||||
core_pattern=$(/sbin/sysctl -n "kernel.core_pattern")
|
||||
if [ "$core_pattern" = "dump_%e.%p.core" ]; then
|
||||
echo "Enabling coredumps"
|
||||
ulimit -c unlimited
|
||||
fi
|
||||
fi
|
||||
fi
|
||||
|
||||
if [ "$(uname -s)" = "Darwin" ]; then
|
||||
core_pattern_mac=$(/usr/sbin/sysctl -n "kern.corefile")
|
||||
if [ "$core_pattern_mac" = "dump_%N.%P.core" ]; then
|
||||
echo "Enabling coredumps"
|
||||
ulimit -c unlimited
|
||||
fi
|
||||
fi
|
||||
|
||||
if [ -w /etc/hosts ]; then
|
||||
SUDO=""
|
||||
else
|
||||
SUDO="sudo"
|
||||
fi
|
||||
|
||||
# Add 'server' and 'hostname_not_in_cert' as a hostnames
|
||||
echo "127.0.0.1 server" | $SUDO tee -a /etc/hosts
|
||||
echo "127.0.0.1 hostname_not_in_cert" | $SUDO tee -a /etc/hosts
|
||||
|
||||
echo "Setting up system... done."
|
||||
|
||||
@ -1,27 +1,26 @@
|
||||
#!/bin/bash -eux
|
||||
#!/bin/bash
|
||||
# Set up the test environment, including secrets and services.
|
||||
set -eu
|
||||
|
||||
PROJECT_DIRECTORY="$(pwd)"
|
||||
SCRIPT_DIR="$PROJECT_DIRECTORY/.evergreen/scripts"
|
||||
# Supported/used environment variables:
|
||||
# AUTH Set to enable authentication. Defaults to "noauth"
|
||||
# SSL Set to enable SSL. Defaults to "nossl"
|
||||
# GREEN_FRAMEWORK The green framework to test with, if any.
|
||||
# COVERAGE If non-empty, run the test suite with coverage.
|
||||
# COMPRESSORS If non-empty, install appropriate compressor.
|
||||
# LIBMONGOCRYPT_URL The URL to download libmongocrypt.
|
||||
# TEST_CRYPT_SHARED If non-empty, install crypt_shared lib.
|
||||
# MONGODB_API_VERSION The mongodb api version to use in tests.
|
||||
# MONGODB_URI If non-empty, use as the MONGODB_URI in tests.
|
||||
# USE_ACTIVE_VENV If non-empty, use the active virtual environment.
|
||||
|
||||
if [ -f "$SCRIPT_DIR/test-env.sh" ]; then
|
||||
echo "Reading $SCRIPT_DIR/test-env.sh file"
|
||||
. "$SCRIPT_DIR/test-env.sh"
|
||||
exit 0
|
||||
SCRIPT_DIR=$(dirname ${BASH_SOURCE:-$0})
|
||||
|
||||
# Try to source the env file.
|
||||
if [ -f $SCRIPT_DIR/env.sh ]; then
|
||||
source $SCRIPT_DIR/env.sh
|
||||
fi
|
||||
|
||||
cat <<EOT > "$SCRIPT_DIR"/test-env.sh
|
||||
export test_encryption="${test_encryption:-}"
|
||||
export test_encryption_pyopenssl="${test_encryption_pyopenssl:-}"
|
||||
export test_crypt_shared="${test_crypt_shared:-}"
|
||||
export test_pyopenssl="${test_pyopenssl:-}"
|
||||
export test_loadbalancer="${test_loadbalancer:-}"
|
||||
export test_serverless="${test_serverless:-}"
|
||||
export TEST_INDEX_MANAGEMENT="${TEST_INDEX_MANAGEMENT:-}"
|
||||
export TEST_DATA_LAKE="${TEST_DATA_LAKE:-}"
|
||||
export ORCHESTRATION_FILE="${ORCHESTRATION_FILE:-}"
|
||||
export AUTH="${AUTH:-noauth}"
|
||||
export SSL="${SSL:-nossl}"
|
||||
export PYTHON_BINARY="${PYTHON_BINARY:-}"
|
||||
EOT
|
||||
|
||||
chmod +x "$SCRIPT_DIR"/test-env.sh
|
||||
echo "Setting up tests with args \"$*\"..."
|
||||
uv run ${USE_ACTIVE_VENV:+--active} "$SCRIPT_DIR/setup_tests.py" "$@"
|
||||
echo "Setting up tests with args \"$*\"... done."
|
||||
|
||||
53
.evergreen/scripts/setup-uv-python.sh
Executable file
53
.evergreen/scripts/setup-uv-python.sh
Executable file
@ -0,0 +1,53 @@
|
||||
#!/bin/bash
|
||||
# Set up the UV_PYTHON variable.
|
||||
set -eu
|
||||
|
||||
HERE=$(dirname ${BASH_SOURCE:-$0})
|
||||
HERE="$( cd -- "$HERE" > /dev/null 2>&1 && pwd )"
|
||||
|
||||
# Use min supported version by default.
|
||||
_python="3.10"
|
||||
|
||||
# Source the env files to pick up common variables.
|
||||
if [ -f $HERE/env.sh ]; then
|
||||
. $HERE/env.sh
|
||||
fi
|
||||
|
||||
# Get variables defined in test-env.sh.
|
||||
if [ -f $HERE/test-env.sh ]; then
|
||||
. $HERE/test-env.sh
|
||||
fi
|
||||
|
||||
if [ -z "${UV_PYTHON:-}" ]; then
|
||||
set -x
|
||||
# Translate a TOOLCHAIN_VERSION to UV_PYTHON.
|
||||
if [ -n "${TOOLCHAIN_VERSION:-}" ]; then
|
||||
_python=$TOOLCHAIN_VERSION
|
||||
if [ "$(uname -s)" = "Darwin" ]; then
|
||||
if [[ "$_python" == *"t"* ]]; then
|
||||
binary_name="python3t"
|
||||
framework_dir="PythonT"
|
||||
else
|
||||
binary_name="python3"
|
||||
framework_dir="Python"
|
||||
fi
|
||||
_python=$(echo "$_python" | sed 's/t//g')
|
||||
_python="/Library/Frameworks/$framework_dir.Framework/Versions/$_python/bin/$binary_name"
|
||||
elif [ "Windows_NT" = "${OS:-}" ]; then
|
||||
_python=$(echo $_python | cut -d. -f1,2 | sed 's/\.//g; s/t//g')
|
||||
if [[ "$TOOLCHAIN_VERSION" == *"t"* ]]; then
|
||||
_exe="python${TOOLCHAIN_VERSION}.exe"
|
||||
else
|
||||
_exe="python.exe"
|
||||
fi
|
||||
if [ -n "${IS_WIN32:-}" ]; then
|
||||
_python="C:/python/32/Python${_python}/${_exe}"
|
||||
else
|
||||
_python="C:/python/Python${_python}/${_exe}"
|
||||
fi
|
||||
elif [ -d "/opt/python/$_python/bin" ]; then
|
||||
_python="/opt/python/$_python/bin/python3"
|
||||
fi
|
||||
fi
|
||||
export UV_PYTHON="$_python"
|
||||
fi
|
||||
491
.evergreen/scripts/setup_tests.py
Normal file
491
.evergreen/scripts/setup_tests.py
Normal file
@ -0,0 +1,491 @@
|
||||
from __future__ import annotations
|
||||
|
||||
import base64
|
||||
import os
|
||||
import platform
|
||||
import shutil
|
||||
import stat
|
||||
from pathlib import Path
|
||||
from urllib import request
|
||||
|
||||
from utils import (
|
||||
DRIVERS_TOOLS,
|
||||
ENV_FILE,
|
||||
HERE,
|
||||
LOGGER,
|
||||
PLATFORM,
|
||||
ROOT,
|
||||
TEST_SUITE_MAP,
|
||||
Distro,
|
||||
get_test_options,
|
||||
read_env,
|
||||
run_command,
|
||||
write_env,
|
||||
)
|
||||
|
||||
# Passthrough environment variables.
|
||||
PASS_THROUGH_ENV = [
|
||||
"GREEN_FRAMEWORK",
|
||||
"NO_EXT",
|
||||
"MONGODB_API_VERSION",
|
||||
"DEBUG_LOG",
|
||||
"UV_PYTHON",
|
||||
"REQUIRE_FIPS",
|
||||
"IS_WIN32",
|
||||
]
|
||||
|
||||
# Map the test name to test extra.
|
||||
EXTRAS_MAP = {
|
||||
"auth_aws": "aws",
|
||||
"auth_oidc": "aws",
|
||||
"encryption": "encryption",
|
||||
"enterprise_auth": "gssapi",
|
||||
"kms": "encryption",
|
||||
"ocsp": "ocsp",
|
||||
"pyopenssl": "ocsp",
|
||||
}
|
||||
|
||||
|
||||
# Map the test name to test group.
|
||||
GROUP_MAP = dict(mockupdb="mockupdb", perf="perf")
|
||||
|
||||
# The python version used for perf tests.
|
||||
PERF_PYTHON_VERSION = "3.10.11"
|
||||
|
||||
|
||||
def is_set(var: str) -> bool:
|
||||
value = os.environ.get(var, "")
|
||||
return len(value.strip()) > 0
|
||||
|
||||
|
||||
def get_distro() -> Distro:
|
||||
name = ""
|
||||
version_id = ""
|
||||
arch = platform.machine()
|
||||
with open("/etc/os-release") as fid:
|
||||
for line in fid.readlines():
|
||||
line = line.replace('"', "") # noqa: PLW2901
|
||||
if line.startswith("NAME="):
|
||||
_, _, name = line.strip().partition("=")
|
||||
if line.startswith("VERSION_ID="):
|
||||
_, _, version_id = line.strip().partition("=")
|
||||
return Distro(name=name, version_id=version_id, arch=arch)
|
||||
|
||||
|
||||
def setup_libmongocrypt():
|
||||
target = ""
|
||||
if PLATFORM == "windows":
|
||||
# PYTHON-2808 Ensure this machine has the CA cert for google KMS.
|
||||
if is_set("TEST_FLE_GCP_AUTO"):
|
||||
run_command('powershell.exe "Invoke-WebRequest -URI https://oauth2.googleapis.com/"')
|
||||
target = "windows-test"
|
||||
|
||||
elif PLATFORM == "darwin":
|
||||
target = "macos"
|
||||
|
||||
else:
|
||||
distro = get_distro()
|
||||
if distro.name.startswith("Debian"):
|
||||
target = f"debian{distro.version_id}"
|
||||
elif distro.name.startswith("Ubuntu"):
|
||||
if distro.version_id == "20.04":
|
||||
target = "debian11"
|
||||
elif distro.version_id == "22.04":
|
||||
target = "debian12"
|
||||
elif distro.version_id == "24.04":
|
||||
target = "debian13"
|
||||
elif distro.name.startswith("Red Hat"):
|
||||
if distro.version_id.startswith("7"):
|
||||
target = "rhel-70-64-bit"
|
||||
elif distro.version_id.startswith("8"):
|
||||
if distro.arch == "aarch64":
|
||||
target = "rhel-82-arm64"
|
||||
else:
|
||||
target = "rhel-80-64-bit"
|
||||
|
||||
if not is_set("LIBMONGOCRYPT_URL"):
|
||||
if not target:
|
||||
raise ValueError("Cannot find libmongocrypt target for current platform!")
|
||||
url = f"https://s3.amazonaws.com/mciuploads/libmongocrypt/{target}/master/latest/libmongocrypt.tar.gz"
|
||||
else:
|
||||
url = os.environ["LIBMONGOCRYPT_URL"]
|
||||
|
||||
shutil.rmtree(HERE / "libmongocrypt", ignore_errors=True)
|
||||
|
||||
LOGGER.info(f"Fetching {url}...")
|
||||
with request.urlopen(request.Request(url), timeout=15.0) as response: # noqa: S310
|
||||
if response.status == 200:
|
||||
with Path("libmongocrypt.tar.gz").open("wb") as f:
|
||||
f.write(response.read())
|
||||
Path("libmongocrypt").mkdir()
|
||||
run_command("tar -xzf libmongocrypt.tar.gz -C libmongocrypt")
|
||||
LOGGER.info(f"Fetching {url}... done.")
|
||||
|
||||
run_command("ls -la libmongocrypt")
|
||||
run_command("ls -la libmongocrypt/nocrypto")
|
||||
|
||||
if PLATFORM == "windows":
|
||||
# libmongocrypt's windows dll is not marked executable.
|
||||
run_command("chmod +x libmongocrypt/nocrypto/bin/mongocrypt.dll")
|
||||
|
||||
|
||||
def load_config_from_file(path: str | Path) -> dict[str, str]:
|
||||
config = read_env(path)
|
||||
for key, value in config.items():
|
||||
write_env(key, value)
|
||||
return config
|
||||
|
||||
|
||||
def get_secrets(name: str) -> dict[str, str]:
|
||||
secrets_dir = Path(f"{DRIVERS_TOOLS}/.evergreen/secrets_handling")
|
||||
run_command(f"bash {secrets_dir.as_posix()}/setup-secrets.sh {name}", cwd=secrets_dir)
|
||||
return load_config_from_file(secrets_dir / "secrets-export.sh")
|
||||
|
||||
|
||||
def handle_test_env() -> None:
|
||||
opts, _ = get_test_options("Set up the test environment and services.")
|
||||
test_name = opts.test_name
|
||||
sub_test_name = opts.sub_test_name
|
||||
AUTH = "auth" if opts.auth else "noauth"
|
||||
SSL = "ssl" if opts.ssl else "nossl"
|
||||
TEST_ARGS = ""
|
||||
|
||||
# Start compiling the args we'll pass to uv.
|
||||
UV_ARGS = ["--extra test --no-group dev"]
|
||||
|
||||
# If USE_ACTIVE_VENV is set, add --active to UV_ARGS so run-tests.sh uses the active venv.
|
||||
if is_set("USE_ACTIVE_VENV"):
|
||||
UV_ARGS.append("--active")
|
||||
|
||||
test_title = test_name
|
||||
if sub_test_name:
|
||||
test_title += f" {sub_test_name}"
|
||||
|
||||
# Create the test env file with the initial set of values.
|
||||
with ENV_FILE.open("w", newline="\n") as fid:
|
||||
fid.write("#!/usr/bin/env bash\n")
|
||||
fid.write("set +x\n")
|
||||
ENV_FILE.chmod(ENV_FILE.stat().st_mode | stat.S_IEXEC)
|
||||
|
||||
write_env("PIP_QUIET") # Quiet by default.
|
||||
write_env("PIP_PREFER_BINARY") # Prefer binary dists by default.
|
||||
|
||||
# Set an environment variable for the test name and sub test name.
|
||||
write_env(f"TEST_{test_name.upper()}")
|
||||
write_env("TEST_NAME", test_name)
|
||||
write_env("SUB_TEST_NAME", sub_test_name)
|
||||
|
||||
# Handle pass through env vars.
|
||||
for var in PASS_THROUGH_ENV:
|
||||
if is_set(var) or getattr(opts, var.lower(), ""):
|
||||
write_env(var, os.environ.get(var, getattr(opts, var.lower(), "")))
|
||||
|
||||
if extra := EXTRAS_MAP.get(test_name, ""):
|
||||
UV_ARGS.append(f"--extra {extra}")
|
||||
|
||||
if group := GROUP_MAP.get(test_name, ""):
|
||||
UV_ARGS.append(f"--group {group}")
|
||||
|
||||
if opts.test_min_deps:
|
||||
UV_ARGS.append("--resolution=lowest-direct")
|
||||
|
||||
if test_name == "auth_oidc":
|
||||
from oidc_tester import setup_oidc
|
||||
|
||||
config = setup_oidc(sub_test_name)
|
||||
if not config:
|
||||
AUTH = "noauth"
|
||||
|
||||
if test_name in ["aws_lambda", "search_index"]:
|
||||
env = os.environ.copy()
|
||||
env["MONGODB_VERSION"] = "7.0"
|
||||
env["LAMBDA_STACK_NAME"] = "dbx-python-lambda"
|
||||
write_env("LAMBDA_STACK_NAME", env["LAMBDA_STACK_NAME"])
|
||||
run_command(
|
||||
f"bash {DRIVERS_TOOLS}/.evergreen/atlas/setup-atlas-cluster.sh",
|
||||
env=env,
|
||||
cwd=DRIVERS_TOOLS,
|
||||
)
|
||||
|
||||
if test_name == "search_index":
|
||||
AUTH = "auth"
|
||||
|
||||
if test_name == "ocsp":
|
||||
SSL = "ssl"
|
||||
|
||||
write_env("AUTH", AUTH)
|
||||
write_env("SSL", SSL)
|
||||
LOGGER.info(f"Setting up '{test_title}' with {AUTH=} and {SSL=}...")
|
||||
|
||||
if test_name == "aws_lambda":
|
||||
UV_ARGS.append("--group pip")
|
||||
# Store AWS creds if they were given.
|
||||
if "AWS_ACCESS_KEY_ID" in os.environ:
|
||||
for key in ["AWS_ACCESS_KEY_ID", "AWS_SECRET_ACCESS_KEY", "AWS_SESSION_TOKEN"]:
|
||||
if key in os.environ:
|
||||
write_env(key, os.environ[key])
|
||||
|
||||
if AUTH != "noauth":
|
||||
if test_name == "auth_oidc":
|
||||
DB_USER = config["OIDC_ADMIN_USER"]
|
||||
DB_PASSWORD = config["OIDC_ADMIN_PWD"]
|
||||
elif test_name == "search_index":
|
||||
config = read_env(f"{DRIVERS_TOOLS}/.evergreen/atlas/secrets-export.sh")
|
||||
DB_USER = config["DRIVERS_ATLAS_LAMBDA_USER"]
|
||||
DB_PASSWORD = config["DRIVERS_ATLAS_LAMBDA_PASSWORD"]
|
||||
write_env("MONGODB_URI", config["MONGODB_URI"])
|
||||
else:
|
||||
DB_USER = "bob"
|
||||
DB_PASSWORD = "pwd123" # noqa: S105
|
||||
write_env("DB_USER", DB_USER)
|
||||
write_env("DB_PASSWORD", DB_PASSWORD)
|
||||
LOGGER.info("Added auth, DB_USER: %s", DB_USER)
|
||||
|
||||
if is_set("MONGODB_URI"):
|
||||
write_env("PYMONGO_MUST_CONNECT", "true")
|
||||
|
||||
if opts.disable_test_commands:
|
||||
write_env("PYMONGO_DISABLE_TEST_COMMANDS", "1")
|
||||
|
||||
if test_name == "enterprise_auth":
|
||||
config = get_secrets("drivers/enterprise_auth")
|
||||
if PLATFORM == "windows":
|
||||
LOGGER.info("Setting GSSAPI_PASS")
|
||||
write_env("GSSAPI_PASS", config["SASL_PASS"])
|
||||
write_env("GSSAPI_CANONICALIZE", "true")
|
||||
else:
|
||||
# BUILD-3830
|
||||
krb_conf = ROOT / ".evergreen/krb5.conf.empty"
|
||||
krb_conf.touch()
|
||||
write_env("KRB5_CONFIG", krb_conf)
|
||||
LOGGER.info("Writing keytab")
|
||||
keytab = base64.b64decode(config["KEYTAB_BASE64"])
|
||||
keytab_file = ROOT / ".evergreen/drivers.keytab"
|
||||
with keytab_file.open("wb") as fid:
|
||||
fid.write(keytab)
|
||||
principal = config["PRINCIPAL"]
|
||||
LOGGER.info("Running kinit")
|
||||
os.environ["KRB5_CONFIG"] = str(krb_conf)
|
||||
cmd = f"kinit -k -t {keytab_file} -p {principal}"
|
||||
run_command(cmd)
|
||||
|
||||
LOGGER.info("Setting GSSAPI variables")
|
||||
write_env("GSSAPI_HOST", config["SASL_HOST"])
|
||||
write_env("GSSAPI_PORT", config["SASL_PORT"])
|
||||
write_env("GSSAPI_PRINCIPAL", config["PRINCIPAL"])
|
||||
|
||||
if test_name == "doctest":
|
||||
UV_ARGS.append("--extra docs")
|
||||
|
||||
if test_name == "load_balancer":
|
||||
SINGLE_MONGOS_LB_URI = os.environ.get(
|
||||
"SINGLE_MONGOS_LB_URI", "mongodb://127.0.0.1:8000/?loadBalanced=true"
|
||||
)
|
||||
MULTI_MONGOS_LB_URI = os.environ.get(
|
||||
"MULTI_MONGOS_LB_URI", "mongodb://127.0.0.1:8001/?loadBalanced=true"
|
||||
)
|
||||
if SSL != "nossl":
|
||||
SINGLE_MONGOS_LB_URI += "&tls=true"
|
||||
MULTI_MONGOS_LB_URI += "&tls=true"
|
||||
write_env("SINGLE_MONGOS_LB_URI", SINGLE_MONGOS_LB_URI)
|
||||
write_env("MULTI_MONGOS_LB_URI", MULTI_MONGOS_LB_URI)
|
||||
if not DRIVERS_TOOLS:
|
||||
raise RuntimeError("Missing DRIVERS_TOOLS")
|
||||
cmd = f'bash "{DRIVERS_TOOLS}/.evergreen/run-load-balancer.sh" start'
|
||||
run_command(cmd)
|
||||
|
||||
if test_name == "mod_wsgi":
|
||||
from mod_wsgi_tester import setup_mod_wsgi
|
||||
|
||||
setup_mod_wsgi(sub_test_name)
|
||||
|
||||
if test_name == "ocsp":
|
||||
if sub_test_name:
|
||||
os.environ["OCSP_SERVER_TYPE"] = sub_test_name
|
||||
for name in ["OCSP_SERVER_TYPE", "ORCHESTRATION_FILE"]:
|
||||
if name not in os.environ:
|
||||
raise ValueError(f"Please set {name}")
|
||||
|
||||
server_type = os.environ["OCSP_SERVER_TYPE"]
|
||||
orch_file = os.environ["ORCHESTRATION_FILE"]
|
||||
ocsp_algo = orch_file.split("-")[0]
|
||||
if server_type == "no-responder":
|
||||
tls_should_succeed = "false" if "mustStaple-disableStapling" in orch_file else "true"
|
||||
else:
|
||||
tls_should_succeed = "true" if "valid" in server_type else "false"
|
||||
|
||||
write_env("OCSP_TLS_SHOULD_SUCCEED", tls_should_succeed)
|
||||
write_env("CA_FILE", f"{DRIVERS_TOOLS}/.evergreen/ocsp/{ocsp_algo}/ca.pem")
|
||||
|
||||
if server_type != "no-responder":
|
||||
env = os.environ.copy()
|
||||
env["SERVER_TYPE"] = server_type
|
||||
env["OCSP_ALGORITHM"] = ocsp_algo
|
||||
run_command(f"bash {DRIVERS_TOOLS}/.evergreen/ocsp/setup.sh", env=env)
|
||||
|
||||
# The mock OCSP responder MUST BE started before the mongod as the mongod expects that
|
||||
# a responder will be available upon startup.
|
||||
version = os.environ.get("VERSION", "latest")
|
||||
cmd = [
|
||||
"bash",
|
||||
f"{DRIVERS_TOOLS}/.evergreen/run-mongodb.sh",
|
||||
"start",
|
||||
"--ssl",
|
||||
"--version",
|
||||
version,
|
||||
]
|
||||
if opts.verbose:
|
||||
cmd.append("-v")
|
||||
elif opts.quiet:
|
||||
cmd.append("-q")
|
||||
run_command(cmd, cwd=DRIVERS_TOOLS)
|
||||
|
||||
if SSL != "nossl":
|
||||
if not DRIVERS_TOOLS:
|
||||
raise RuntimeError("Missing DRIVERS_TOOLS")
|
||||
write_env("CLIENT_PEM", f"{DRIVERS_TOOLS}/.evergreen/x509gen/client.pem")
|
||||
write_env("CA_PEM", f"{DRIVERS_TOOLS}/.evergreen/x509gen/ca.pem")
|
||||
|
||||
compressors = os.environ.get("COMPRESSORS") or opts.compressor
|
||||
if compressors == "snappy":
|
||||
UV_ARGS.append("--extra snappy")
|
||||
elif compressors == "zstd":
|
||||
UV_ARGS.append("--extra zstd")
|
||||
|
||||
if test_name in ["encryption", "kms"]:
|
||||
# Check for libmongocrypt download.
|
||||
if not (ROOT / "libmongocrypt").exists():
|
||||
setup_libmongocrypt()
|
||||
|
||||
if not opts.test_min_deps:
|
||||
UV_ARGS.append(
|
||||
"--with pymongocrypt@git+https://github.com/mongodb/libmongocrypt@master#subdirectory=bindings/python"
|
||||
)
|
||||
|
||||
# Use the nocrypto build to avoid dependency issues with older windows/python versions.
|
||||
BASE = ROOT / "libmongocrypt/nocrypto"
|
||||
if PLATFORM == "linux":
|
||||
if (BASE / "lib/libmongocrypt.so").exists():
|
||||
PYMONGOCRYPT_LIB = BASE / "lib/libmongocrypt.so"
|
||||
else:
|
||||
PYMONGOCRYPT_LIB = BASE / "lib64/libmongocrypt.so"
|
||||
elif PLATFORM == "darwin":
|
||||
PYMONGOCRYPT_LIB = BASE / "lib/libmongocrypt.dylib"
|
||||
else:
|
||||
PYMONGOCRYPT_LIB = BASE / "bin/mongocrypt.dll"
|
||||
if not PYMONGOCRYPT_LIB.exists():
|
||||
raise RuntimeError("Cannot find libmongocrypt shared object file")
|
||||
write_env("PYMONGOCRYPT_LIB", PYMONGOCRYPT_LIB.as_posix())
|
||||
# PATH is updated by configure-env.sh for access to mongocryptd.
|
||||
|
||||
if test_name == "encryption":
|
||||
if not DRIVERS_TOOLS:
|
||||
raise RuntimeError("Missing DRIVERS_TOOLS")
|
||||
csfle_dir = Path(f"{DRIVERS_TOOLS}/.evergreen/csfle")
|
||||
run_command(f"bash {csfle_dir.as_posix()}/setup-secrets.sh", cwd=csfle_dir)
|
||||
load_config_from_file(csfle_dir / "secrets-export.sh")
|
||||
run_command(f"bash {csfle_dir.as_posix()}/start-servers.sh")
|
||||
|
||||
if sub_test_name == "pyopenssl":
|
||||
UV_ARGS.append("--extra ocsp")
|
||||
|
||||
if opts.crypt_shared:
|
||||
config = read_env(f"{DRIVERS_TOOLS}/mo-expansion.sh")
|
||||
CRYPT_SHARED_DIR = Path(config["CRYPT_SHARED_LIB_PATH"]).parent.as_posix()
|
||||
LOGGER.info("Using crypt_shared_dir %s", CRYPT_SHARED_DIR)
|
||||
if PLATFORM == "windows":
|
||||
write_env("PATH", f"{CRYPT_SHARED_DIR}:$PATH")
|
||||
else:
|
||||
write_env(
|
||||
"DYLD_FALLBACK_LIBRARY_PATH",
|
||||
f"{CRYPT_SHARED_DIR}:${{DYLD_FALLBACK_LIBRARY_PATH:-}}",
|
||||
)
|
||||
write_env("LD_LIBRARY_PATH", f"{CRYPT_SHARED_DIR}:${{LD_LIBRARY_PATH:-}}")
|
||||
|
||||
if test_name == "kms":
|
||||
from kms_tester import setup_kms
|
||||
|
||||
setup_kms(sub_test_name)
|
||||
|
||||
if test_name == "auth_aws" and sub_test_name != "ecs-remote":
|
||||
auth_aws_dir = f"{DRIVERS_TOOLS}/.evergreen/auth_aws"
|
||||
if "AWS_ROLE_SESSION_NAME" in os.environ:
|
||||
write_env("AWS_ROLE_SESSION_NAME")
|
||||
if sub_test_name != "ecs":
|
||||
aws_setup = f"{auth_aws_dir}/aws_setup.sh"
|
||||
run_command(f"bash {aws_setup} {sub_test_name}")
|
||||
creds = read_env(f"{auth_aws_dir}/test-env.sh")
|
||||
for name, value in creds.items():
|
||||
write_env(name, value)
|
||||
else:
|
||||
run_command(f"bash {auth_aws_dir}/setup-secrets.sh")
|
||||
|
||||
if test_name == "atlas_connect":
|
||||
secrets = get_secrets("drivers/atlas_connect")
|
||||
|
||||
# Write file with Atlas X509 client certificate:
|
||||
decoded = base64.b64decode(secrets["ATLAS_X509_DEV_CERT_BASE64"]).decode("utf8")
|
||||
cert_file = ROOT / ".evergreen/atlas_x509_dev_client_certificate.pem"
|
||||
with cert_file.open("w") as file:
|
||||
file.write(decoded)
|
||||
write_env(
|
||||
"ATLAS_X509_DEV_WITH_CERT",
|
||||
secrets["ATLAS_X509_DEV"] + "&tlsCertificateKeyFile=" + str(cert_file),
|
||||
)
|
||||
|
||||
# We do not want the default client_context to be initialized.
|
||||
write_env("DISABLE_CONTEXT")
|
||||
|
||||
if test_name == "numpy":
|
||||
UV_ARGS.append("--with numpy")
|
||||
|
||||
if test_name == "perf":
|
||||
data_dir = ROOT / "specifications/source/benchmarking/data"
|
||||
if not data_dir.exists():
|
||||
run_command("git clone --depth 1 https://github.com/mongodb/specifications.git")
|
||||
run_command("tar xf extended_bson.tgz", cwd=data_dir)
|
||||
run_command("tar xf parallel.tgz", cwd=data_dir)
|
||||
run_command("tar xf single_and_multi_document.tgz", cwd=data_dir)
|
||||
write_env("TEST_PATH", str(data_dir))
|
||||
write_env("OUTPUT_FILE", str(ROOT / "results.json"))
|
||||
# Overwrite the UV_PYTHON from the env.sh file.
|
||||
write_env("UV_PYTHON", "")
|
||||
|
||||
UV_ARGS.append(f"--python={PERF_PYTHON_VERSION}")
|
||||
|
||||
# PYTHON-4769 Run perf_test.py directly otherwise pytest's test collection negatively
|
||||
# affects the benchmark results.
|
||||
if sub_test_name == "sync":
|
||||
TEST_ARGS = f"test/performance/perf_test.py {TEST_ARGS}"
|
||||
else:
|
||||
TEST_ARGS = f"test/performance/async_perf_test.py {TEST_ARGS}"
|
||||
|
||||
# Add coverage if requested.
|
||||
# Only cover CPython. PyPy reports suspiciously low coverage.
|
||||
if opts.cov and platform.python_implementation() == "CPython":
|
||||
# Keep in sync with combine-coverage.sh.
|
||||
# coverage >=5 is needed for relative_files=true.
|
||||
UV_ARGS.append("--group coverage")
|
||||
write_env("COVERAGE")
|
||||
|
||||
if opts.green_framework:
|
||||
framework = opts.green_framework or os.environ["GREEN_FRAMEWORK"]
|
||||
UV_ARGS.append(f"--group {framework}")
|
||||
if framework == "gevent" and opts.test_min_deps:
|
||||
# PYTHON-5729. This can be removed when the min supported gevent is moved to 25.9.1.
|
||||
UV_ARGS.append('--with "setuptools==81.0"')
|
||||
|
||||
else:
|
||||
TEST_ARGS = f"-v --durations=5 {TEST_ARGS}"
|
||||
TEST_SUITE = TEST_SUITE_MAP.get(test_name)
|
||||
if TEST_SUITE:
|
||||
TEST_ARGS = f"-m {TEST_SUITE} {TEST_ARGS}"
|
||||
|
||||
write_env("TEST_ARGS", TEST_ARGS)
|
||||
write_env("UV_ARGS", " ".join(UV_ARGS))
|
||||
|
||||
LOGGER.info(f"Setting up test '{test_title}' with {AUTH=} and {SSL=}... done.")
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
handle_test_env()
|
||||
@ -1,5 +0,0 @@
|
||||
#!/bin/bash
|
||||
|
||||
cd "${DRIVERS_TOOLS}"/.evergreen || exit
|
||||
DRIVERS_TOOLS=${DRIVERS_TOOLS}
|
||||
bash "${DRIVERS_TOOLS}"/.evergreen/run-load-balancer.sh stop
|
||||
14
.evergreen/scripts/stop-server.sh
Executable file
14
.evergreen/scripts/stop-server.sh
Executable file
@ -0,0 +1,14 @@
|
||||
#!/bin/bash
|
||||
# Stop a server that was started using run-mongodb.sh in DRIVERS_TOOLS.
|
||||
set -eu
|
||||
|
||||
HERE=$(dirname ${BASH_SOURCE:-$0})
|
||||
HERE="$( cd -- "$HERE" > /dev/null 2>&1 && pwd )"
|
||||
|
||||
# Try to source the env file.
|
||||
if [ -f $HERE/env.sh ]; then
|
||||
echo "Sourcing env file"
|
||||
source $HERE/env.sh
|
||||
fi
|
||||
|
||||
bash ${DRIVERS_TOOLS}/.evergreen/run-mongodb.sh stop
|
||||
24
.evergreen/scripts/teardown-tests.sh
Executable file
24
.evergreen/scripts/teardown-tests.sh
Executable file
@ -0,0 +1,24 @@
|
||||
#!/bin/bash
|
||||
# Tear down any services that were used by tests.
|
||||
set -eu
|
||||
|
||||
SCRIPT_DIR=$(dirname ${BASH_SOURCE:-$0})
|
||||
|
||||
# Try to source the env file.
|
||||
if [ -f $SCRIPT_DIR/env.sh ]; then
|
||||
echo "Sourcing env inputs"
|
||||
. $SCRIPT_DIR/env.sh
|
||||
else
|
||||
echo "Not sourcing env inputs"
|
||||
fi
|
||||
|
||||
# Handle test inputs.
|
||||
if [ -f $SCRIPT_DIR/test-env.sh ]; then
|
||||
echo "Sourcing test inputs"
|
||||
. $SCRIPT_DIR/test-env.sh
|
||||
else
|
||||
echo "Missing test inputs, please run 'just setup-tests'"
|
||||
fi
|
||||
|
||||
# Teardown the test runner.
|
||||
uv run $SCRIPT_DIR/teardown_tests.py
|
||||
64
.evergreen/scripts/teardown_tests.py
Normal file
64
.evergreen/scripts/teardown_tests.py
Normal file
@ -0,0 +1,64 @@
|
||||
from __future__ import annotations
|
||||
|
||||
import os
|
||||
import shutil
|
||||
import sys
|
||||
from pathlib import Path
|
||||
|
||||
from utils import DRIVERS_TOOLS, LOGGER, ROOT, run_command
|
||||
|
||||
TEST_NAME = os.environ.get("TEST_NAME", "unconfigured")
|
||||
SUB_TEST_NAME = os.environ.get("SUB_TEST_NAME")
|
||||
|
||||
LOGGER.info(f"Tearing down tests of type '{TEST_NAME}'...")
|
||||
|
||||
# Shut down csfle servers if applicable.
|
||||
if TEST_NAME == "encryption":
|
||||
run_command(f"bash {DRIVERS_TOOLS}/.evergreen/csfle/stop-servers.sh")
|
||||
|
||||
# Shut down load balancer if applicable.
|
||||
elif TEST_NAME == "load-balancer":
|
||||
run_command(f"bash {DRIVERS_TOOLS}/.evergreen/run-load-balancer.sh stop")
|
||||
|
||||
# Tear down kms VM if applicable.
|
||||
elif TEST_NAME == "kms" and SUB_TEST_NAME in ["azure", "gcp"]:
|
||||
from kms_tester import teardown_kms
|
||||
|
||||
teardown_kms(SUB_TEST_NAME)
|
||||
|
||||
# Tear down OIDC if applicable.
|
||||
elif TEST_NAME == "auth_oidc":
|
||||
from oidc_tester import teardown_oidc
|
||||
|
||||
teardown_oidc(SUB_TEST_NAME)
|
||||
|
||||
# Tear down ocsp if applicable.
|
||||
elif TEST_NAME == "ocsp":
|
||||
run_command(f"bash {DRIVERS_TOOLS}/.evergreen/ocsp/teardown.sh")
|
||||
|
||||
# Tear down atlas cluster if applicable.
|
||||
if TEST_NAME in ["aws_lambda", "search_index"]:
|
||||
run_command(f"bash {DRIVERS_TOOLS}/.evergreen/atlas/teardown-atlas-cluster.sh")
|
||||
|
||||
# Tear down auth_aws if applicable.
|
||||
# We do not run web-identity hosts on macos, because the hosts lack permissions,
|
||||
# so there is no reason to run the teardown, which would error with a 401.
|
||||
elif TEST_NAME == "auth_aws" and sys.platform != "darwin":
|
||||
run_command(f"bash {DRIVERS_TOOLS}/.evergreen/auth_aws/teardown.sh")
|
||||
|
||||
# Tear down perf if applicable.
|
||||
elif TEST_NAME == "perf":
|
||||
shutil.rmtree(ROOT / "specifications", ignore_errors=True)
|
||||
Path(os.environ["OUTPUT_FILE"]).unlink(missing_ok=True)
|
||||
|
||||
# Tear down mog_wsgi if applicable.
|
||||
elif TEST_NAME == "mod_wsgi":
|
||||
from mod_wsgi_tester import teardown_mod_wsgi
|
||||
|
||||
teardown_mod_wsgi()
|
||||
|
||||
# Tear down coverage if applicable.
|
||||
if os.environ.get("COVERAGE"):
|
||||
shutil.rmtree(".pytest_cache", ignore_errors=True)
|
||||
|
||||
LOGGER.info(f"Tearing down tests of type '{TEST_NAME}'... done.")
|
||||
57
.evergreen/scripts/upload-codecov.sh
Executable file
57
.evergreen/scripts/upload-codecov.sh
Executable file
@ -0,0 +1,57 @@
|
||||
#!/bin/bash
|
||||
# shellcheck disable=SC2154
|
||||
# Upload a coverate report to codecov.
|
||||
set -eu
|
||||
|
||||
HERE=$(dirname ${BASH_SOURCE:-$0})
|
||||
ROOT=$(dirname "$(dirname $HERE)")
|
||||
|
||||
pushd $ROOT > /dev/null
|
||||
export FNAME=coverage.xml
|
||||
REQUESTER=${requester:-}
|
||||
|
||||
if [ ! -f ".coverage" ]; then
|
||||
echo "There are no coverage results, not running codecov"
|
||||
exit 0
|
||||
fi
|
||||
|
||||
if [[ "${REQUESTER}" == "github_pr" || "${REQUESTER}" == "commit" ]]; then
|
||||
echo "Uploading codecov for $REQUESTER..."
|
||||
else
|
||||
echo "Error: requester must be 'github_pr' or 'commit', got '${REQUESTER}'" >&2
|
||||
exit 1
|
||||
fi
|
||||
|
||||
printf 'sha: %s\n' "$github_commit"
|
||||
printf 'flag: %s-%s\n' "$build_variant" "$task_name"
|
||||
printf 'file: %s\n' "$FNAME"
|
||||
uv tool run --with "coverage[toml]" coverage xml
|
||||
|
||||
codecov_args=(
|
||||
upload-process
|
||||
--report-type coverage
|
||||
--disable-search
|
||||
--fail-on-error
|
||||
--git-service github
|
||||
--token "${CODECOV_TOKEN}"
|
||||
--sha "${github_commit}"
|
||||
--flag "${build_variant}-${task_name}"
|
||||
--file "${FNAME}"
|
||||
)
|
||||
|
||||
if [ -n "${github_pr_number:-}" ]; then
|
||||
printf 'branch: %s:%s\n' "$github_author" "$github_pr_head_branch"
|
||||
printf 'pr: %s\n' "$github_pr_number"
|
||||
uv tool run --from codecov-cli codecovcli \
|
||||
"${codecov_args[@]}" \
|
||||
--pr "${github_pr_number}" \
|
||||
--branch "${github_author}:${github_pr_head_branch}"
|
||||
else
|
||||
printf 'branch: %s\n' "$branch_name"
|
||||
uv tool run --from codecov-cli codecovcli \
|
||||
"${codecov_args[@]}" \
|
||||
--branch "${branch_name}"
|
||||
fi
|
||||
echo "Uploading codecov for $REQUESTER... done."
|
||||
|
||||
popd > /dev/null
|
||||
@ -1,3 +1,4 @@
|
||||
#!/bin/bash
|
||||
|
||||
# Upload a coverate report to s3.
|
||||
set -eu
|
||||
aws s3 cp htmlcov/ s3://"$1"/coverage/"$2"/"$3"/htmlcov/ --recursive --acl public-read --region us-east-1
|
||||
|
||||
228
.evergreen/scripts/utils.py
Normal file
228
.evergreen/scripts/utils.py
Normal file
@ -0,0 +1,228 @@
|
||||
from __future__ import annotations
|
||||
|
||||
import argparse
|
||||
import dataclasses
|
||||
import logging
|
||||
import os
|
||||
import shlex
|
||||
import subprocess
|
||||
import sys
|
||||
from pathlib import Path
|
||||
from typing import Any
|
||||
|
||||
HERE = Path(__file__).absolute().parent
|
||||
ROOT = HERE.parent.parent
|
||||
DRIVERS_TOOLS = os.environ.get("DRIVERS_TOOLS", "").replace(os.sep, "/")
|
||||
TMP_DRIVER_FILE = "/tmp/mongo-python-driver.tgz" # noqa: S108
|
||||
|
||||
LOGGER = logging.getLogger("test")
|
||||
logging.basicConfig(level=logging.INFO, format="%(levelname)-8s %(message)s")
|
||||
ENV_FILE = HERE / "test-env.sh"
|
||||
PLATFORM = "windows" if os.name == "nt" else sys.platform.lower()
|
||||
|
||||
|
||||
@dataclasses.dataclass
|
||||
class Distro:
|
||||
name: str
|
||||
version_id: str
|
||||
arch: str
|
||||
|
||||
|
||||
# Map the test name to a test suite.
|
||||
TEST_SUITE_MAP = {
|
||||
"atlas_connect": "atlas_connect",
|
||||
"auth_aws": "auth_aws",
|
||||
"auth_oidc": "auth_oidc",
|
||||
"default": "",
|
||||
"default_async": "default_async",
|
||||
"default_sync": "default",
|
||||
"encryption": "encryption",
|
||||
"enterprise_auth": "auth",
|
||||
"search_index": "search_index",
|
||||
"kms": "kms",
|
||||
"load_balancer": "load_balancer",
|
||||
"mockupdb": "mockupdb",
|
||||
"ocsp": "ocsp",
|
||||
"perf": "perf",
|
||||
"numpy": "",
|
||||
}
|
||||
|
||||
# Tests that require a sub test suite.
|
||||
SUB_TEST_REQUIRED = ["auth_aws", "auth_oidc", "kms", "mod_wsgi", "perf"]
|
||||
|
||||
EXTRA_TESTS = ["mod_wsgi", "aws_lambda", "doctest"]
|
||||
|
||||
# Tests that do not use run-mongodb directly.
|
||||
NO_RUN_ORCHESTRATION = [
|
||||
"auth_oidc",
|
||||
"atlas_connect",
|
||||
"aws_lambda",
|
||||
"mockupdb",
|
||||
"ocsp",
|
||||
]
|
||||
|
||||
# Mapping of env variables to options
|
||||
OPTION_TO_ENV_VAR = {"cov": "COVERAGE", "crypt_shared": "TEST_CRYPT_SHARED"}
|
||||
|
||||
|
||||
def get_test_options(
|
||||
description, require_sub_test_name=True, allow_extra_opts=False
|
||||
) -> tuple[argparse.Namespace, list[str]]:
|
||||
parser = argparse.ArgumentParser(
|
||||
description=description, formatter_class=argparse.RawDescriptionHelpFormatter
|
||||
)
|
||||
if require_sub_test_name:
|
||||
parser.add_argument(
|
||||
"test_name",
|
||||
choices=sorted(list(TEST_SUITE_MAP) + EXTRA_TESTS),
|
||||
nargs="?",
|
||||
default="default",
|
||||
help="The optional name of the test suite to set up, typically the same name as a pytest marker.",
|
||||
)
|
||||
parser.add_argument(
|
||||
"sub_test_name", nargs="?", help="The optional sub test name, for example 'azure'."
|
||||
)
|
||||
else:
|
||||
parser.add_argument(
|
||||
"test_name",
|
||||
choices=set(list(TEST_SUITE_MAP) + EXTRA_TESTS) - set(NO_RUN_ORCHESTRATION),
|
||||
nargs="?",
|
||||
default="default",
|
||||
help="The optional name of the test suite to be run, which informs the server configuration.",
|
||||
)
|
||||
parser.add_argument(
|
||||
"--verbose", "-v", action="store_true", help="Whether to log at the DEBUG level."
|
||||
)
|
||||
parser.add_argument(
|
||||
"--quiet", "-q", action="store_true", help="Whether to log at the WARNING level."
|
||||
)
|
||||
parser.add_argument("--auth", action="store_true", help="Whether to add authentication.")
|
||||
parser.add_argument("--ssl", action="store_true", help="Whether to add TLS configuration.")
|
||||
parser.add_argument(
|
||||
"--test-min-deps", action="store_true", help="Test against minimum dependency versions"
|
||||
)
|
||||
|
||||
# Add the test modifiers.
|
||||
if require_sub_test_name:
|
||||
parser.add_argument(
|
||||
"--debug-log", action="store_true", help="Enable pymongo standard logging."
|
||||
)
|
||||
parser.add_argument("--cov", action="store_true", help="Add test coverage.")
|
||||
parser.add_argument(
|
||||
"--green-framework",
|
||||
nargs=1,
|
||||
choices=["gevent"],
|
||||
help="Optional green framework to test against.",
|
||||
)
|
||||
parser.add_argument(
|
||||
"--compressor",
|
||||
nargs=1,
|
||||
choices=["zlib", "zstd", "snappy"],
|
||||
help="Optional compression algorithm.",
|
||||
)
|
||||
parser.add_argument("--crypt-shared", action="store_true", help="Test with crypt_shared.")
|
||||
parser.add_argument("--no-ext", action="store_true", help="Run without c extensions.")
|
||||
parser.add_argument(
|
||||
"--mongodb-api-version", choices=["1"], help="MongoDB stable API version to use."
|
||||
)
|
||||
parser.add_argument(
|
||||
"--disable-test-commands", action="store_true", help="Disable test commands."
|
||||
)
|
||||
|
||||
# Get the options.
|
||||
if not allow_extra_opts:
|
||||
opts, extra_opts = parser.parse_args(), []
|
||||
else:
|
||||
opts, extra_opts = parser.parse_known_args()
|
||||
|
||||
# Convert list inputs to strings.
|
||||
for name in vars(opts):
|
||||
value = getattr(opts, name)
|
||||
if isinstance(value, list):
|
||||
setattr(opts, name, value[0])
|
||||
|
||||
# Handle validation and environment variable overrides.
|
||||
test_name = opts.test_name
|
||||
sub_test_name = opts.sub_test_name if require_sub_test_name else ""
|
||||
if require_sub_test_name and test_name in SUB_TEST_REQUIRED and not sub_test_name:
|
||||
raise ValueError(f"Test '{test_name}' requires a sub_test_name")
|
||||
handle_env_overrides(parser, opts)
|
||||
if "auth" in test_name:
|
||||
opts.auth = True
|
||||
# 'auth_aws ecs' shouldn't have extra auth set.
|
||||
if test_name == "auth_aws" and sub_test_name == "ecs":
|
||||
opts.auth = False
|
||||
if opts.verbose:
|
||||
LOGGER.setLevel(logging.DEBUG)
|
||||
elif opts.quiet:
|
||||
LOGGER.setLevel(logging.WARNING)
|
||||
return opts, extra_opts
|
||||
|
||||
|
||||
def handle_env_overrides(parser: argparse.ArgumentParser, opts: argparse.Namespace) -> None:
|
||||
# Get the options, and then allow environment variable overrides.
|
||||
for key in vars(opts):
|
||||
if key in OPTION_TO_ENV_VAR:
|
||||
env_var = OPTION_TO_ENV_VAR[key]
|
||||
else:
|
||||
env_var = key.upper()
|
||||
if env_var in os.environ:
|
||||
if parser.get_default(key) != getattr(opts, key):
|
||||
LOGGER.info("Overriding env var '%s' with cli option", env_var)
|
||||
elif env_var == "AUTH":
|
||||
opts.auth = os.environ.get("AUTH") == "auth"
|
||||
elif env_var == "SSL":
|
||||
ssl_opt = os.environ.get("SSL", "")
|
||||
opts.ssl = ssl_opt and ssl_opt.lower() != "nossl"
|
||||
elif isinstance(getattr(opts, key), bool):
|
||||
if os.environ[env_var]:
|
||||
setattr(opts, key, True)
|
||||
else:
|
||||
setattr(opts, key, os.environ[env_var])
|
||||
|
||||
|
||||
def read_env(path: Path | str) -> dict[str, str]:
|
||||
config = dict()
|
||||
with Path(path).open() as fid:
|
||||
for line in fid.readlines():
|
||||
if "=" not in line:
|
||||
continue
|
||||
name, _, value = line.strip().partition("=")
|
||||
if value.startswith(('"', "'")):
|
||||
value = value[1:-1]
|
||||
name = name.replace("export ", "")
|
||||
config[name] = value
|
||||
return config
|
||||
|
||||
|
||||
def write_env(name: str, value: Any = "1") -> None:
|
||||
with ENV_FILE.open("a", newline="\n") as fid:
|
||||
# Remove any existing quote chars.
|
||||
value = str(value).replace('"', "")
|
||||
fid.write(f'export {name}="{value}"\n')
|
||||
|
||||
|
||||
def run_command(cmd: str | list[str], **kwargs: Any) -> None:
|
||||
if isinstance(cmd, list):
|
||||
cmd = " ".join(cmd)
|
||||
LOGGER.info("Running command '%s'...", cmd)
|
||||
kwargs.setdefault("check", True)
|
||||
# Prevent overriding the python used by other tools.
|
||||
env = kwargs.pop("env", os.environ).copy()
|
||||
if "UV_PYTHON" in env:
|
||||
del env["UV_PYTHON"]
|
||||
kwargs["env"] = env
|
||||
try:
|
||||
subprocess.run(shlex.split(cmd), **kwargs) # noqa: PLW1510, S603
|
||||
except subprocess.CalledProcessError as e:
|
||||
LOGGER.error(e.output)
|
||||
LOGGER.error(str(e))
|
||||
sys.exit(e.returncode)
|
||||
LOGGER.info("Running command '%s'... done.", cmd)
|
||||
|
||||
|
||||
def create_archive() -> str:
|
||||
run_command("git add .", cwd=ROOT)
|
||||
run_command('git commit --no-verify -m "add files"', check=False, cwd=ROOT)
|
||||
run_command(f"git archive -o {TMP_DRIVER_FILE} HEAD", cwd=ROOT)
|
||||
return TMP_DRIVER_FILE
|
||||
@ -1,11 +0,0 @@
|
||||
#!/bin/bash
|
||||
|
||||
set +x
|
||||
. src/.evergreen/scripts/env.sh
|
||||
# shellcheck disable=SC2044
|
||||
for i in $(find "$DRIVERS_TOOLS"/.evergreen "$PROJECT_DIRECTORY"/.evergreen -name \*.sh); do
|
||||
< "$i" tr -d '\r' >"$i".new
|
||||
mv "$i".new "$i"
|
||||
done
|
||||
# Copy client certificate because symlinks do not work on Windows.
|
||||
cp "$DRIVERS_TOOLS"/.evergreen/x509gen/client.pem "$MONGO_ORCHESTRATION_HOME"/lib/client.pem
|
||||
@ -1,60 +0,0 @@
|
||||
#!/bin/bash
|
||||
set -o errexit # Exit the script with error if any of the commands fail
|
||||
set -o xtrace
|
||||
|
||||
if [ -z "${DRIVERS_TOOLS}" ]; then
|
||||
echo "Missing environment variable DRIVERS_TOOLS"
|
||||
exit 1
|
||||
fi
|
||||
|
||||
TARGET=""
|
||||
|
||||
if [ "Windows_NT" = "${OS:-''}" ]; then # Magic variable in cygwin
|
||||
# PYTHON-2808 Ensure this machine has the CA cert for google KMS.
|
||||
powershell.exe "Invoke-WebRequest -URI https://oauth2.googleapis.com/" > /dev/null || true
|
||||
TARGET="windows-test"
|
||||
fi
|
||||
|
||||
if [ "$(uname -s)" = "Darwin" ]; then
|
||||
TARGET="macos"
|
||||
fi
|
||||
|
||||
if [ "$(uname -s)" = "Linux" ]; then
|
||||
rhel_ver=$(awk -F'=' '/VERSION_ID/{ gsub(/"/,""); print $2}' /etc/os-release)
|
||||
arch=$(uname -m)
|
||||
echo "RHEL $rhel_ver $arch"
|
||||
if [[ $rhel_ver =~ 7 ]]; then
|
||||
TARGET="rhel-70-64-bit"
|
||||
elif [[ $rhel_ver =~ 8 ]]; then
|
||||
if [ "$arch" = "x86_64" ]; then
|
||||
TARGET="rhel-80-64-bit"
|
||||
elif [ "$arch" = "arm" ]; then
|
||||
TARGET="rhel-82-arm64"
|
||||
fi
|
||||
fi
|
||||
fi
|
||||
|
||||
if [ -z "$LIBMONGOCRYPT_URL" ] && [ -n "$TARGET" ]; then
|
||||
LIBMONGOCRYPT_URL="https://s3.amazonaws.com/mciuploads/libmongocrypt/$TARGET/master/latest/libmongocrypt.tar.gz"
|
||||
fi
|
||||
|
||||
if [ -z "$LIBMONGOCRYPT_URL" ]; then
|
||||
echo "Cannot test client side encryption without LIBMONGOCRYPT_URL!"
|
||||
exit 1
|
||||
fi
|
||||
rm -rf libmongocrypt libmongocrypt.tar.gz
|
||||
echo "Fetching $LIBMONGOCRYPT_URL..."
|
||||
curl -O "$LIBMONGOCRYPT_URL"
|
||||
echo "Fetching $LIBMONGOCRYPT_URL...done"
|
||||
mkdir libmongocrypt
|
||||
tar xzf libmongocrypt.tar.gz -C ./libmongocrypt
|
||||
ls -la libmongocrypt
|
||||
ls -la libmongocrypt/nocrypto
|
||||
|
||||
if [ -z "${SKIP_SERVERS:-}" ]; then
|
||||
PYTHON_BINARY_OLD=${PYTHON_BINARY}
|
||||
export PYTHON_BINARY=""
|
||||
bash "${DRIVERS_TOOLS}"/.evergreen/csfle/setup-secrets.sh
|
||||
export PYTHON_BINARY=$PYTHON_BINARY_OLD
|
||||
bash "${DRIVERS_TOOLS}"/.evergreen/csfle/start-servers.sh
|
||||
fi
|
||||
@ -1,5 +1,5 @@
|
||||
#!/bin/bash
|
||||
|
||||
# Set up a remote evergreen spawn host.
|
||||
set -eu
|
||||
|
||||
if [ -z "$1" ]
|
||||
@ -15,5 +15,4 @@ echo "Copying files to $target..."
|
||||
rsync -az -e ssh --exclude '.git' --filter=':- .gitignore' -r . $target:$remote_dir
|
||||
echo "Copying files to $target... done"
|
||||
|
||||
ssh $target $remote_dir/.evergreen/scripts/setup-system.sh
|
||||
ssh $target "cd $remote_dir && PYTHON_BINARY=${PYTHON_BINARY:-} just install"
|
||||
ssh $target "$remote_dir/.evergreen/scripts/setup-system.sh"
|
||||
|
||||
24
.evergreen/spec-patch/PYTHON-4918.patch
Normal file
24
.evergreen/spec-patch/PYTHON-4918.patch
Normal file
@ -0,0 +1,24 @@
|
||||
diff --git a/test/connection_monitoring/pool-create-min-size-error.json b/test/connection_monitoring/pool-create-min-size-error.json
|
||||
index 1c744b85..509b2a23 100644
|
||||
--- a/test/connection_monitoring/pool-create-min-size-error.json
|
||||
+++ b/test/connection_monitoring/pool-create-min-size-error.json
|
||||
@@ -49,15 +49,15 @@
|
||||
"type": "ConnectionCreated",
|
||||
"address": 42
|
||||
},
|
||||
+ {
|
||||
+ "type": "ConnectionPoolCleared",
|
||||
+ "address": 42
|
||||
+ },
|
||||
{
|
||||
"type": "ConnectionClosed",
|
||||
"address": 42,
|
||||
"connectionId": 42,
|
||||
"reason": "error"
|
||||
- },
|
||||
- {
|
||||
- "type": "ConnectionPoolCleared",
|
||||
- "address": 42
|
||||
}
|
||||
],
|
||||
"ignore": [
|
||||
440
.evergreen/spec-patch/PYTHON-5052.patch
Normal file
440
.evergreen/spec-patch/PYTHON-5052.patch
Normal file
@ -0,0 +1,440 @@
|
||||
diff --git a/test/unified-test-format/invalid/entity-client-observeTracingMessages-additionalProperties.json b/test/unified-test-format/invalid/entity-client-observeTracingMessages-additionalProperties.json
|
||||
new file mode 100644
|
||||
index 00000000..aa8046d2
|
||||
--- /dev/null
|
||||
+++ b/test/unified-test-format/invalid/entity-client-observeTracingMessages-additionalProperties.json
|
||||
@@ -0,0 +1,20 @@
|
||||
+{
|
||||
+ "description": "entity-client-observeTracingMessages-additionalProperties",
|
||||
+ "schemaVersion": "1.26",
|
||||
+ "createEntities": [
|
||||
+ {
|
||||
+ "client": {
|
||||
+ "id": "client0",
|
||||
+ "observeTracingMessages": {
|
||||
+ "foo": "bar"
|
||||
+ }
|
||||
+ }
|
||||
+ }
|
||||
+ ],
|
||||
+ "tests": [
|
||||
+ {
|
||||
+ "description": "observeTracingMessages must not have additional properties'",
|
||||
+ "operations": []
|
||||
+ }
|
||||
+ ]
|
||||
+}
|
||||
diff --git a/test/unified-test-format/invalid/entity-client-observeTracingMessages-additionalPropertyType.json b/test/unified-test-format/invalid/entity-client-observeTracingMessages-additionalPropertyType.json
|
||||
new file mode 100644
|
||||
index 00000000..0b3a65f5
|
||||
--- /dev/null
|
||||
+++ b/test/unified-test-format/invalid/entity-client-observeTracingMessages-additionalPropertyType.json
|
||||
@@ -0,0 +1,20 @@
|
||||
+{
|
||||
+ "description": "entity-client-observeTracingMessages-additionalPropertyType",
|
||||
+ "schemaVersion": "1.26",
|
||||
+ "createEntities": [
|
||||
+ {
|
||||
+ "client": {
|
||||
+ "id": "client0",
|
||||
+ "observeTracingMessages": {
|
||||
+ "enableCommandPayload": 0
|
||||
+ }
|
||||
+ }
|
||||
+ }
|
||||
+ ],
|
||||
+ "tests": [
|
||||
+ {
|
||||
+ "description": "observeTracingMessages enableCommandPayload must be boolean",
|
||||
+ "operations": []
|
||||
+ }
|
||||
+ ]
|
||||
+}
|
||||
diff --git a/test/unified-test-format/invalid/entity-client-observeTracingMessages-type.json b/test/unified-test-format/invalid/entity-client-observeTracingMessages-type.json
|
||||
new file mode 100644
|
||||
index 00000000..de3ef39a
|
||||
--- /dev/null
|
||||
+++ b/test/unified-test-format/invalid/entity-client-observeTracingMessages-type.json
|
||||
@@ -0,0 +1,18 @@
|
||||
+{
|
||||
+ "description": "entity-client-observeTracingMessages-type",
|
||||
+ "schemaVersion": "1.26",
|
||||
+ "createEntities": [
|
||||
+ {
|
||||
+ "client": {
|
||||
+ "id": "client0",
|
||||
+ "observeTracingMessages": "foo"
|
||||
+ }
|
||||
+ }
|
||||
+ ],
|
||||
+ "tests": [
|
||||
+ {
|
||||
+ "description": "observeTracingMessages must be an object",
|
||||
+ "operations": []
|
||||
+ }
|
||||
+ ]
|
||||
+}
|
||||
diff --git a/test/unified-test-format/invalid/expectedTracingSpans-additionalProperties.json b/test/unified-test-format/invalid/expectedTracingSpans-additionalProperties.json
|
||||
new file mode 100644
|
||||
index 00000000..5947a286
|
||||
--- /dev/null
|
||||
+++ b/test/unified-test-format/invalid/expectedTracingSpans-additionalProperties.json
|
||||
@@ -0,0 +1,30 @@
|
||||
+{
|
||||
+ "description": "expectedTracingSpans-additionalProperties",
|
||||
+ "schemaVersion": "1.26",
|
||||
+ "createEntities": [
|
||||
+ {
|
||||
+ "client": {
|
||||
+ "id": "client0"
|
||||
+ }
|
||||
+ }
|
||||
+ ],
|
||||
+ "tests": [
|
||||
+ {
|
||||
+ "description": "additional property foo not allowed in expectTracingMessages",
|
||||
+ "operations": [],
|
||||
+ "expectTracingMessages": {
|
||||
+ "client": "client0",
|
||||
+ "ignoreExtraSpans": false,
|
||||
+ "spans": [
|
||||
+ {
|
||||
+ "name": "command",
|
||||
+ "tags": {
|
||||
+ "db.system": "mongodb"
|
||||
+ }
|
||||
+ }
|
||||
+ ],
|
||||
+ "foo": 0
|
||||
+ }
|
||||
+ }
|
||||
+ ]
|
||||
+}
|
||||
diff --git a/test/unified-test-format/invalid/expectedTracingSpans-clientType.json b/test/unified-test-format/invalid/expectedTracingSpans-clientType.json
|
||||
new file mode 100644
|
||||
index 00000000..2fe7faea
|
||||
--- /dev/null
|
||||
+++ b/test/unified-test-format/invalid/expectedTracingSpans-clientType.json
|
||||
@@ -0,0 +1,28 @@
|
||||
+{
|
||||
+ "description": "expectedTracingSpans-clientType",
|
||||
+ "schemaVersion": "1.26",
|
||||
+ "createEntities": [
|
||||
+ {
|
||||
+ "client": {
|
||||
+ "id": "client0"
|
||||
+ }
|
||||
+ }
|
||||
+ ],
|
||||
+ "tests": [
|
||||
+ {
|
||||
+ "description": "client type must be string",
|
||||
+ "operations": [],
|
||||
+ "expectTracingMessages": {
|
||||
+ "client": 0,
|
||||
+ "spans": [
|
||||
+ {
|
||||
+ "name": "command",
|
||||
+ "tags": {
|
||||
+ "db.system": "mongodb"
|
||||
+ }
|
||||
+ }
|
||||
+ ]
|
||||
+ }
|
||||
+ }
|
||||
+ ]
|
||||
+}
|
||||
diff --git a/test/unified-test-format/invalid/expectedTracingSpans-emptyNestedSpan.json b/test/unified-test-format/invalid/expectedTracingSpans-emptyNestedSpan.json
|
||||
new file mode 100644
|
||||
index 00000000..8a98d5ba
|
||||
--- /dev/null
|
||||
+++ b/test/unified-test-format/invalid/expectedTracingSpans-emptyNestedSpan.json
|
||||
@@ -0,0 +1,29 @@
|
||||
+{
|
||||
+ "description": "expectedTracingSpans-emptyNestedSpan",
|
||||
+ "schemaVersion": "1.26",
|
||||
+ "createEntities": [
|
||||
+ {
|
||||
+ "client": {
|
||||
+ "id": "client0"
|
||||
+ }
|
||||
+ }
|
||||
+ ],
|
||||
+ "tests": [
|
||||
+ {
|
||||
+ "description": "nested spans must not have fewer than 1 items'",
|
||||
+ "operations": [],
|
||||
+ "expectTracingMessages": {
|
||||
+ "client": "client0",
|
||||
+ "spans": [
|
||||
+ {
|
||||
+ "name": "command",
|
||||
+ "tags": {
|
||||
+ "db.system": "mongodb"
|
||||
+ },
|
||||
+ "nested": []
|
||||
+ }
|
||||
+ ]
|
||||
+ }
|
||||
+ }
|
||||
+ ]
|
||||
+}
|
||||
diff --git a/test/unified-test-format/invalid/expectedTracingSpans-invalidNestedSpan.json b/test/unified-test-format/invalid/expectedTracingSpans-invalidNestedSpan.json
|
||||
new file mode 100644
|
||||
index 00000000..79a86744
|
||||
--- /dev/null
|
||||
+++ b/test/unified-test-format/invalid/expectedTracingSpans-invalidNestedSpan.json
|
||||
@@ -0,0 +1,31 @@
|
||||
+{
|
||||
+ "description": "expectedTracingSpans-invalidNestedSpan",
|
||||
+ "schemaVersion": "1.26",
|
||||
+ "createEntities": [
|
||||
+ {
|
||||
+ "client": {
|
||||
+ "id": "client0"
|
||||
+ }
|
||||
+ }
|
||||
+ ],
|
||||
+ "tests": [
|
||||
+ {
|
||||
+ "description": "nested span must have required property name",
|
||||
+ "operations": [],
|
||||
+ "expectTracingMessages": {
|
||||
+ "client": "client0",
|
||||
+ "spans": [
|
||||
+ {
|
||||
+ "name": "command",
|
||||
+ "tags": {
|
||||
+ "db.system": "mongodb"
|
||||
+ },
|
||||
+ "nested": [
|
||||
+ {}
|
||||
+ ]
|
||||
+ }
|
||||
+ ]
|
||||
+ }
|
||||
+ }
|
||||
+ ]
|
||||
+}
|
||||
diff --git a/test/unified-test-format/invalid/expectedTracingSpans-missingPropertyClient.json b/test/unified-test-format/invalid/expectedTracingSpans-missingPropertyClient.json
|
||||
new file mode 100644
|
||||
index 00000000..2fb1cd5b
|
||||
--- /dev/null
|
||||
+++ b/test/unified-test-format/invalid/expectedTracingSpans-missingPropertyClient.json
|
||||
@@ -0,0 +1,27 @@
|
||||
+{
|
||||
+ "description": "expectedTracingSpans-missingPropertyClient",
|
||||
+ "schemaVersion": "1.26",
|
||||
+ "createEntities": [
|
||||
+ {
|
||||
+ "client": {
|
||||
+ "id": "client0"
|
||||
+ }
|
||||
+ }
|
||||
+ ],
|
||||
+ "tests": [
|
||||
+ {
|
||||
+ "description": "missing required property client",
|
||||
+ "operations": [],
|
||||
+ "expectTracingMessages": {
|
||||
+ "spans": [
|
||||
+ {
|
||||
+ "name": "command",
|
||||
+ "tags": {
|
||||
+ "db.system": "mongodb"
|
||||
+ }
|
||||
+ }
|
||||
+ ]
|
||||
+ }
|
||||
+ }
|
||||
+ ]
|
||||
+}
|
||||
diff --git a/test/unified-test-format/invalid/expectedTracingSpans-missingPropertySpans.json b/test/unified-test-format/invalid/expectedTracingSpans-missingPropertySpans.json
|
||||
new file mode 100644
|
||||
index 00000000..acd10307
|
||||
--- /dev/null
|
||||
+++ b/test/unified-test-format/invalid/expectedTracingSpans-missingPropertySpans.json
|
||||
@@ -0,0 +1,20 @@
|
||||
+{
|
||||
+ "description": "expectedTracingSpans-missingPropertySpans",
|
||||
+ "schemaVersion": "1.26",
|
||||
+ "createEntities": [
|
||||
+ {
|
||||
+ "client": {
|
||||
+ "id": "client0"
|
||||
+ }
|
||||
+ }
|
||||
+ ],
|
||||
+ "tests": [
|
||||
+ {
|
||||
+ "description": "missing required property spans",
|
||||
+ "operations": [],
|
||||
+ "expectTracingMessages": {
|
||||
+ "client": "client0"
|
||||
+ }
|
||||
+ }
|
||||
+ ]
|
||||
+}
|
||||
diff --git a/test/unified-test-format/invalid/expectedTracingSpans-spanMalformedAdditionalProperties.json b/test/unified-test-format/invalid/expectedTracingSpans-spanMalformedAdditionalProperties.json
|
||||
new file mode 100644
|
||||
index 00000000..17299f86
|
||||
--- /dev/null
|
||||
+++ b/test/unified-test-format/invalid/expectedTracingSpans-spanMalformedAdditionalProperties.json
|
||||
@@ -0,0 +1,28 @@
|
||||
+{
|
||||
+ "description": "expectedTracingSpans-spanMalformedAdditionalProperties",
|
||||
+ "schemaVersion": "1.26",
|
||||
+ "createEntities": [
|
||||
+ {
|
||||
+ "client": {
|
||||
+ "id": "client0"
|
||||
+ }
|
||||
+ }
|
||||
+ ],
|
||||
+ "tests": [
|
||||
+ {
|
||||
+ "description": "Span must not have additional properties",
|
||||
+ "operations": [],
|
||||
+ "expectTracingMessages": {
|
||||
+ "client": "client0",
|
||||
+ "spans": [
|
||||
+ {
|
||||
+ "name": "foo",
|
||||
+ "tags": {},
|
||||
+ "nested": [],
|
||||
+ "foo": "bar"
|
||||
+ }
|
||||
+ ]
|
||||
+ }
|
||||
+ }
|
||||
+ ]
|
||||
+}
|
||||
diff --git a/test/unified-test-format/invalid/expectedTracingSpans-spanMalformedMissingName.json b/test/unified-test-format/invalid/expectedTracingSpans-spanMalformedMissingName.json
|
||||
new file mode 100644
|
||||
index 00000000..0257cd9b
|
||||
--- /dev/null
|
||||
+++ b/test/unified-test-format/invalid/expectedTracingSpans-spanMalformedMissingName.json
|
||||
@@ -0,0 +1,27 @@
|
||||
+{
|
||||
+ "description": "expectedTracingSpans-spanMalformedMissingName",
|
||||
+ "schemaVersion": "1.26",
|
||||
+ "createEntities": [
|
||||
+ {
|
||||
+ "client": {
|
||||
+ "id": "client0"
|
||||
+ }
|
||||
+ }
|
||||
+ ],
|
||||
+ "tests": [
|
||||
+ {
|
||||
+ "description": "missing required span name",
|
||||
+ "operations": [],
|
||||
+ "expectTracingMessages": {
|
||||
+ "client": "client0",
|
||||
+ "spans": [
|
||||
+ {
|
||||
+ "tags": {
|
||||
+ "db.system": "mongodb"
|
||||
+ }
|
||||
+ }
|
||||
+ ]
|
||||
+ }
|
||||
+ }
|
||||
+ ]
|
||||
+}
|
||||
diff --git a/test/unified-test-format/invalid/expectedTracingSpans-spanMalformedMissingTags.json b/test/unified-test-format/invalid/expectedTracingSpans-spanMalformedMissingTags.json
|
||||
new file mode 100644
|
||||
index 00000000..a09ca31c
|
||||
--- /dev/null
|
||||
+++ b/test/unified-test-format/invalid/expectedTracingSpans-spanMalformedMissingTags.json
|
||||
@@ -0,0 +1,25 @@
|
||||
+{
|
||||
+ "description": "expectedTracingSpans-spanMalformedMissingTags",
|
||||
+ "schemaVersion": "1.26",
|
||||
+ "createEntities": [
|
||||
+ {
|
||||
+ "client": {
|
||||
+ "id": "client0"
|
||||
+ }
|
||||
+ }
|
||||
+ ],
|
||||
+ "tests": [
|
||||
+ {
|
||||
+ "description": "missing required span tags",
|
||||
+ "operations": [],
|
||||
+ "expectTracingMessages": {
|
||||
+ "client": "client0",
|
||||
+ "spans": [
|
||||
+ {
|
||||
+ "name": "foo"
|
||||
+ }
|
||||
+ ]
|
||||
+ }
|
||||
+ }
|
||||
+ ]
|
||||
+}
|
||||
diff --git a/test/unified-test-format/invalid/expectedTracingSpans-spanMalformedNestedMustBeArray.json b/test/unified-test-format/invalid/expectedTracingSpans-spanMalformedNestedMustBeArray.json
|
||||
new file mode 100644
|
||||
index 00000000..ccff0410
|
||||
--- /dev/null
|
||||
+++ b/test/unified-test-format/invalid/expectedTracingSpans-spanMalformedNestedMustBeArray.json
|
||||
@@ -0,0 +1,27 @@
|
||||
+{
|
||||
+ "description": "expectedTracingSpans-spanMalformedNestedMustBeArray",
|
||||
+ "schemaVersion": "1.26",
|
||||
+ "createEntities": [
|
||||
+ {
|
||||
+ "client": {
|
||||
+ "id": "client0"
|
||||
+ }
|
||||
+ }
|
||||
+ ],
|
||||
+ "tests": [
|
||||
+ {
|
||||
+ "description": "nested spans must be an array",
|
||||
+ "operations": [],
|
||||
+ "expectTracingMessages": {
|
||||
+ "client": "client0",
|
||||
+ "spans": [
|
||||
+ {
|
||||
+ "name": "foo",
|
||||
+ "tags": {},
|
||||
+ "nested": {}
|
||||
+ }
|
||||
+ ]
|
||||
+ }
|
||||
+ }
|
||||
+ ]
|
||||
+}
|
||||
diff --git a/test/unified-test-format/invalid/expectedTracingSpans-spanMalformedTagsMustBeObject.json b/test/unified-test-format/invalid/expectedTracingSpans-spanMalformedTagsMustBeObject.json
|
||||
new file mode 100644
|
||||
index 00000000..72af1c29
|
||||
--- /dev/null
|
||||
+++ b/test/unified-test-format/invalid/expectedTracingSpans-spanMalformedTagsMustBeObject.json
|
||||
@@ -0,0 +1,26 @@
|
||||
+{
|
||||
+ "description": "expectedTracingSpans-spanMalformedNestedMustBeObject",
|
||||
+ "schemaVersion": "1.26",
|
||||
+ "createEntities": [
|
||||
+ {
|
||||
+ "client": {
|
||||
+ "id": "client0"
|
||||
+ }
|
||||
+ }
|
||||
+ ],
|
||||
+ "tests": [
|
||||
+ {
|
||||
+ "description": "span tags must be an object",
|
||||
+ "operations": [],
|
||||
+ "expectTracingMessages": {
|
||||
+ "client": "client0",
|
||||
+ "spans": [
|
||||
+ {
|
||||
+ "name": "foo",
|
||||
+ "tags": []
|
||||
+ }
|
||||
+ ]
|
||||
+ }
|
||||
+ }
|
||||
+ ]
|
||||
+}
|
||||
26
.evergreen/spec-patch/PYTHON-5445.patch
Normal file
26
.evergreen/spec-patch/PYTHON-5445.patch
Normal file
@ -0,0 +1,26 @@
|
||||
diff --git a/test/auth/legacy/connection-string.json b/test/auth/legacy/connection-string.json
|
||||
index 3a099c813..8982b61d5 100644
|
||||
--- a/test/auth/legacy/connection-string.json
|
||||
+++ b/test/auth/legacy/connection-string.json
|
||||
@@ -440,6 +440,21 @@
|
||||
}
|
||||
}
|
||||
},
|
||||
+ {
|
||||
+ "description": "should throw an exception if username provided (MONGODB-AWS)",
|
||||
+ "uri": "mongodb://user@localhost.com/?authMechanism=MONGODB-AWS",
|
||||
+ "valid": false
|
||||
+ },
|
||||
+ {
|
||||
+ "description": "should throw an exception if username and password provided (MONGODB-AWS)",
|
||||
+ "uri": "mongodb://user:pass@localhost.com/?authMechanism=MONGODB-AWS",
|
||||
+ "valid": false
|
||||
+ },
|
||||
+ {
|
||||
+ "description": "should throw an exception if AWS_SESSION_TOKEN provided (MONGODB-AWS)",
|
||||
+ "uri": "mongodb://localhost/?authMechanism=MONGODB-AWS&authMechanismProperties=AWS_SESSION_TOKEN:token",
|
||||
+ "valid": false
|
||||
+ },
|
||||
{
|
||||
"description": "should recognise the mechanism with test environment (MONGODB-OIDC)",
|
||||
"uri": "mongodb://localhost/?authMechanism=MONGODB-OIDC&authMechanismProperties=ENVIRONMENT:test",
|
||||
50
.evergreen/spec-patch/PYTHON-5493.patch
Normal file
50
.evergreen/spec-patch/PYTHON-5493.patch
Normal file
@ -0,0 +1,50 @@
|
||||
diff --git a/test/connection_logging/connection-logging.json b/test/connection_logging/connection-logging.json
|
||||
index 5799e834..72103b3c 100644
|
||||
--- a/test/connection_logging/connection-logging.json
|
||||
+++ b/test/connection_logging/connection-logging.json
|
||||
@@ -446,6 +446,22 @@
|
||||
}
|
||||
}
|
||||
},
|
||||
+ {
|
||||
+ "level": "debug",
|
||||
+ "component": "connection",
|
||||
+ "data": {
|
||||
+ "message": "Connection pool cleared",
|
||||
+ "serverHost": {
|
||||
+ "$$type": "string"
|
||||
+ },
|
||||
+ "serverPort": {
|
||||
+ "$$type": [
|
||||
+ "int",
|
||||
+ "long"
|
||||
+ ]
|
||||
+ }
|
||||
+ }
|
||||
+ },
|
||||
{
|
||||
"level": "debug",
|
||||
"component": "connection",
|
||||
@@ -498,22 +514,6 @@
|
||||
]
|
||||
}
|
||||
}
|
||||
- },
|
||||
- {
|
||||
- "level": "debug",
|
||||
- "component": "connection",
|
||||
- "data": {
|
||||
- "message": "Connection pool cleared",
|
||||
- "serverHost": {
|
||||
- "$$type": "string"
|
||||
- },
|
||||
- "serverPort": {
|
||||
- "$$type": [
|
||||
- "int",
|
||||
- "long"
|
||||
- ]
|
||||
- }
|
||||
- }
|
||||
}
|
||||
]
|
||||
}
|
||||
815
.evergreen/spec-patch/PYTHON-5559.patch
Normal file
815
.evergreen/spec-patch/PYTHON-5559.patch
Normal file
@ -0,0 +1,815 @@
|
||||
diff --git a/test/sessions/snapshot-sessions.json b/test/sessions/snapshot-sessions.json
|
||||
index 260f8b6f4..8f806ea75 100644
|
||||
--- a/test/sessions/snapshot-sessions.json
|
||||
+++ b/test/sessions/snapshot-sessions.json
|
||||
@@ -988,6 +988,810 @@
|
||||
}
|
||||
}
|
||||
]
|
||||
+ },
|
||||
+ {
|
||||
+ "description": "Find operation with snapshot and snapshot time",
|
||||
+ "operations": [
|
||||
+ {
|
||||
+ "name": "find",
|
||||
+ "object": "collection0",
|
||||
+ "arguments": {
|
||||
+ "session": "session0",
|
||||
+ "filter": {}
|
||||
+ },
|
||||
+ "expectResult": [
|
||||
+ {
|
||||
+ "_id": 1,
|
||||
+ "x": 11
|
||||
+ },
|
||||
+ {
|
||||
+ "_id": 2,
|
||||
+ "x": 11
|
||||
+ }
|
||||
+ ]
|
||||
+ },
|
||||
+ {
|
||||
+ "name": "getSnapshotTime",
|
||||
+ "object": "session0",
|
||||
+ "saveResultAsEntity": "savedSnapshotTime"
|
||||
+ },
|
||||
+ {
|
||||
+ "name": "insertOne",
|
||||
+ "object": "collection0",
|
||||
+ "arguments": {
|
||||
+ "document": {
|
||||
+ "_id": 3,
|
||||
+ "x": 33
|
||||
+ }
|
||||
+ }
|
||||
+ },
|
||||
+ {
|
||||
+ "name": "createEntities",
|
||||
+ "object": "testRunner",
|
||||
+ "arguments": {
|
||||
+ "entities": [
|
||||
+ {
|
||||
+ "session": {
|
||||
+ "id": "session2",
|
||||
+ "client": "client0",
|
||||
+ "sessionOptions": {
|
||||
+ "snapshot": true,
|
||||
+ "snapshotTime": "savedSnapshotTime"
|
||||
+ }
|
||||
+ }
|
||||
+ }
|
||||
+ ]
|
||||
+ }
|
||||
+ },
|
||||
+ {
|
||||
+ "name": "find",
|
||||
+ "object": "collection0",
|
||||
+ "arguments": {
|
||||
+ "session": "session2",
|
||||
+ "filter": {}
|
||||
+ },
|
||||
+ "expectResult": [
|
||||
+ {
|
||||
+ "_id": 1,
|
||||
+ "x": 11
|
||||
+ },
|
||||
+ {
|
||||
+ "_id": 2,
|
||||
+ "x": 11
|
||||
+ }
|
||||
+ ]
|
||||
+ },
|
||||
+ {
|
||||
+ "name": "find",
|
||||
+ "object": "collection0",
|
||||
+ "arguments": {
|
||||
+ "session": "session2",
|
||||
+ "filter": {}
|
||||
+ },
|
||||
+ "expectResult": [
|
||||
+ {
|
||||
+ "_id": 1,
|
||||
+ "x": 11
|
||||
+ },
|
||||
+ {
|
||||
+ "_id": 2,
|
||||
+ "x": 11
|
||||
+ }
|
||||
+ ]
|
||||
+ },
|
||||
+ {
|
||||
+ "name": "find",
|
||||
+ "object": "collection0",
|
||||
+ "arguments": {
|
||||
+ "filter": {}
|
||||
+ },
|
||||
+ "expectResult": [
|
||||
+ {
|
||||
+ "_id": 1,
|
||||
+ "x": 11
|
||||
+ },
|
||||
+ {
|
||||
+ "_id": 2,
|
||||
+ "x": 11
|
||||
+ },
|
||||
+ {
|
||||
+ "_id": 3,
|
||||
+ "x": 33
|
||||
+ }
|
||||
+ ]
|
||||
+ }
|
||||
+ ],
|
||||
+ "expectEvents": [
|
||||
+ {
|
||||
+ "client": "client0",
|
||||
+ "events": [
|
||||
+ {
|
||||
+ "commandStartedEvent": {
|
||||
+ "command": {
|
||||
+ "find": "collection0",
|
||||
+ "readConcern": {
|
||||
+ "level": "snapshot",
|
||||
+ "atClusterTime": {
|
||||
+ "$$exists": false
|
||||
+ }
|
||||
+ }
|
||||
+ },
|
||||
+ "databaseName": "database0"
|
||||
+ }
|
||||
+ },
|
||||
+ {
|
||||
+ "commandStartedEvent": {
|
||||
+ "command": {
|
||||
+ "find": "collection0",
|
||||
+ "readConcern": {
|
||||
+ "level": "snapshot",
|
||||
+ "atClusterTime": {
|
||||
+ "$$matchesEntity": "savedSnapshotTime"
|
||||
+ }
|
||||
+ }
|
||||
+ },
|
||||
+ "databaseName": "database0"
|
||||
+ }
|
||||
+ },
|
||||
+ {
|
||||
+ "commandStartedEvent": {
|
||||
+ "command": {
|
||||
+ "find": "collection0",
|
||||
+ "readConcern": {
|
||||
+ "level": "snapshot",
|
||||
+ "atClusterTime": {
|
||||
+ "$$matchesEntity": "savedSnapshotTime"
|
||||
+ }
|
||||
+ }
|
||||
+ },
|
||||
+ "databaseName": "database0"
|
||||
+ }
|
||||
+ },
|
||||
+ {
|
||||
+ "commandStartedEvent": {
|
||||
+ "command": {
|
||||
+ "find": "collection0",
|
||||
+ "readConcern": {
|
||||
+ "$$exists": false
|
||||
+ }
|
||||
+ },
|
||||
+ "databaseName": "database0"
|
||||
+ }
|
||||
+ }
|
||||
+ ]
|
||||
+ }
|
||||
+ ]
|
||||
+ },
|
||||
+ {
|
||||
+ "description": "Distinct operation with snapshot and snapshot time",
|
||||
+ "operations": [
|
||||
+ {
|
||||
+ "name": "distinct",
|
||||
+ "object": "collection0",
|
||||
+ "arguments": {
|
||||
+ "session": "session0",
|
||||
+ "filter": {},
|
||||
+ "fieldName": "x"
|
||||
+ },
|
||||
+ "expectResult": [
|
||||
+ 11
|
||||
+ ]
|
||||
+ },
|
||||
+ {
|
||||
+ "name": "getSnapshotTime",
|
||||
+ "object": "session0",
|
||||
+ "saveResultAsEntity": "savedSnapshotTime"
|
||||
+ },
|
||||
+ {
|
||||
+ "name": "insertOne",
|
||||
+ "object": "collection0",
|
||||
+ "arguments": {
|
||||
+ "document": {
|
||||
+ "_id": 3,
|
||||
+ "x": 33
|
||||
+ }
|
||||
+ }
|
||||
+ },
|
||||
+ {
|
||||
+ "name": "createEntities",
|
||||
+ "object": "testRunner",
|
||||
+ "arguments": {
|
||||
+ "entities": [
|
||||
+ {
|
||||
+ "session": {
|
||||
+ "id": "session2",
|
||||
+ "client": "client0",
|
||||
+ "sessionOptions": {
|
||||
+ "snapshot": true,
|
||||
+ "snapshotTime": "savedSnapshotTime"
|
||||
+ }
|
||||
+ }
|
||||
+ }
|
||||
+ ]
|
||||
+ }
|
||||
+ },
|
||||
+ {
|
||||
+ "name": "distinct",
|
||||
+ "object": "collection0",
|
||||
+ "arguments": {
|
||||
+ "session": "session2",
|
||||
+ "filter": {},
|
||||
+ "fieldName": "x"
|
||||
+ },
|
||||
+ "expectResult": [
|
||||
+ 11
|
||||
+ ]
|
||||
+ },
|
||||
+ {
|
||||
+ "name": "distinct",
|
||||
+ "object": "collection0",
|
||||
+ "arguments": {
|
||||
+ "session": "session2",
|
||||
+ "filter": {},
|
||||
+ "fieldName": "x"
|
||||
+ },
|
||||
+ "expectResult": [
|
||||
+ 11
|
||||
+ ]
|
||||
+ },
|
||||
+ {
|
||||
+ "name": "distinct",
|
||||
+ "object": "collection0",
|
||||
+ "arguments": {
|
||||
+ "filter": {},
|
||||
+ "fieldName": "x"
|
||||
+ },
|
||||
+ "expectResult": [
|
||||
+ 11,
|
||||
+ 33
|
||||
+ ]
|
||||
+ }
|
||||
+ ],
|
||||
+ "expectEvents": [
|
||||
+ {
|
||||
+ "client": "client0",
|
||||
+ "events": [
|
||||
+ {
|
||||
+ "commandStartedEvent": {
|
||||
+ "command": {
|
||||
+ "distinct": "collection0",
|
||||
+ "readConcern": {
|
||||
+ "level": "snapshot",
|
||||
+ "atClusterTime": {
|
||||
+ "$$exists": false
|
||||
+ }
|
||||
+ }
|
||||
+ },
|
||||
+ "databaseName": "database0"
|
||||
+ }
|
||||
+ },
|
||||
+ {
|
||||
+ "commandStartedEvent": {
|
||||
+ "command": {
|
||||
+ "distinct": "collection0",
|
||||
+ "readConcern": {
|
||||
+ "level": "snapshot",
|
||||
+ "atClusterTime": {
|
||||
+ "$$matchesEntity": "savedSnapshotTime"
|
||||
+ }
|
||||
+ }
|
||||
+ },
|
||||
+ "databaseName": "database0"
|
||||
+ }
|
||||
+ },
|
||||
+ {
|
||||
+ "commandStartedEvent": {
|
||||
+ "command": {
|
||||
+ "distinct": "collection0",
|
||||
+ "readConcern": {
|
||||
+ "level": "snapshot",
|
||||
+ "atClusterTime": {
|
||||
+ "$$matchesEntity": "savedSnapshotTime"
|
||||
+ }
|
||||
+ }
|
||||
+ },
|
||||
+ "databaseName": "database0"
|
||||
+ }
|
||||
+ },
|
||||
+ {
|
||||
+ "commandStartedEvent": {
|
||||
+ "command": {
|
||||
+ "distinct": "collection0",
|
||||
+ "readConcern": {
|
||||
+ "$$exists": false
|
||||
+ }
|
||||
+ },
|
||||
+ "databaseName": "database0"
|
||||
+ }
|
||||
+ }
|
||||
+ ]
|
||||
+ }
|
||||
+ ]
|
||||
+ },
|
||||
+ {
|
||||
+ "description": "Aggregate operation with snapshot and snapshot time",
|
||||
+ "operations": [
|
||||
+ {
|
||||
+ "name": "aggregate",
|
||||
+ "object": "collection0",
|
||||
+ "arguments": {
|
||||
+ "session": "session0",
|
||||
+ "pipeline": [
|
||||
+ {
|
||||
+ "$match": {
|
||||
+ "_id": 1
|
||||
+ }
|
||||
+ }
|
||||
+ ]
|
||||
+ },
|
||||
+ "expectResult": [
|
||||
+ {
|
||||
+ "_id": 1,
|
||||
+ "x": 11
|
||||
+ }
|
||||
+ ]
|
||||
+ },
|
||||
+ {
|
||||
+ "name": "getSnapshotTime",
|
||||
+ "object": "session0",
|
||||
+ "saveResultAsEntity": "savedSnapshotTime"
|
||||
+ },
|
||||
+ {
|
||||
+ "name": "findOneAndUpdate",
|
||||
+ "object": "collection0",
|
||||
+ "arguments": {
|
||||
+ "filter": {
|
||||
+ "_id": 1
|
||||
+ },
|
||||
+ "update": {
|
||||
+ "$inc": {
|
||||
+ "x": 1
|
||||
+ }
|
||||
+ },
|
||||
+ "returnDocument": "After"
|
||||
+ },
|
||||
+ "expectResult": {
|
||||
+ "_id": 1,
|
||||
+ "x": 12
|
||||
+ }
|
||||
+ },
|
||||
+ {
|
||||
+ "name": "createEntities",
|
||||
+ "object": "testRunner",
|
||||
+ "arguments": {
|
||||
+ "entities": [
|
||||
+ {
|
||||
+ "session": {
|
||||
+ "id": "session2",
|
||||
+ "client": "client0",
|
||||
+ "sessionOptions": {
|
||||
+ "snapshot": true,
|
||||
+ "snapshotTime": "savedSnapshotTime"
|
||||
+ }
|
||||
+ }
|
||||
+ }
|
||||
+ ]
|
||||
+ }
|
||||
+ },
|
||||
+ {
|
||||
+ "name": "aggregate",
|
||||
+ "object": "collection0",
|
||||
+ "arguments": {
|
||||
+ "session": "session2",
|
||||
+ "pipeline": [
|
||||
+ {
|
||||
+ "$match": {
|
||||
+ "_id": 1
|
||||
+ }
|
||||
+ }
|
||||
+ ]
|
||||
+ },
|
||||
+ "expectResult": [
|
||||
+ {
|
||||
+ "_id": 1,
|
||||
+ "x": 11
|
||||
+ }
|
||||
+ ]
|
||||
+ },
|
||||
+ {
|
||||
+ "name": "aggregate",
|
||||
+ "object": "collection0",
|
||||
+ "arguments": {
|
||||
+ "session": "session2",
|
||||
+ "pipeline": [
|
||||
+ {
|
||||
+ "$match": {
|
||||
+ "_id": 1
|
||||
+ }
|
||||
+ }
|
||||
+ ]
|
||||
+ },
|
||||
+ "expectResult": [
|
||||
+ {
|
||||
+ "_id": 1,
|
||||
+ "x": 11
|
||||
+ }
|
||||
+ ]
|
||||
+ },
|
||||
+ {
|
||||
+ "name": "aggregate",
|
||||
+ "object": "collection0",
|
||||
+ "arguments": {
|
||||
+ "pipeline": [
|
||||
+ {
|
||||
+ "$match": {
|
||||
+ "_id": 1
|
||||
+ }
|
||||
+ }
|
||||
+ ]
|
||||
+ },
|
||||
+ "expectResult": [
|
||||
+ {
|
||||
+ "_id": 1,
|
||||
+ "x": 12
|
||||
+ }
|
||||
+ ]
|
||||
+ }
|
||||
+ ],
|
||||
+ "expectEvents": [
|
||||
+ {
|
||||
+ "client": "client0",
|
||||
+ "events": [
|
||||
+ {
|
||||
+ "commandStartedEvent": {
|
||||
+ "command": {
|
||||
+ "aggregate": "collection0",
|
||||
+ "readConcern": {
|
||||
+ "level": "snapshot",
|
||||
+ "atClusterTime": {
|
||||
+ "$$exists": false
|
||||
+ }
|
||||
+ }
|
||||
+ },
|
||||
+ "databaseName": "database0"
|
||||
+ }
|
||||
+ },
|
||||
+ {
|
||||
+ "commandStartedEvent": {
|
||||
+ "command": {
|
||||
+ "aggregate": "collection0",
|
||||
+ "readConcern": {
|
||||
+ "level": "snapshot",
|
||||
+ "atClusterTime": {
|
||||
+ "$$matchesEntity": "savedSnapshotTime"
|
||||
+ }
|
||||
+ }
|
||||
+ },
|
||||
+ "databaseName": "database0"
|
||||
+ }
|
||||
+ },
|
||||
+ {
|
||||
+ "commandStartedEvent": {
|
||||
+ "command": {
|
||||
+ "aggregate": "collection0",
|
||||
+ "readConcern": {
|
||||
+ "level": "snapshot",
|
||||
+ "atClusterTime": {
|
||||
+ "$$matchesEntity": "savedSnapshotTime"
|
||||
+ }
|
||||
+ }
|
||||
+ },
|
||||
+ "databaseName": "database0"
|
||||
+ }
|
||||
+ },
|
||||
+ {
|
||||
+ "commandStartedEvent": {
|
||||
+ "command": {
|
||||
+ "aggregate": "collection0",
|
||||
+ "readConcern": {
|
||||
+ "$$exists": false
|
||||
+ }
|
||||
+ },
|
||||
+ "databaseName": "database0"
|
||||
+ }
|
||||
+ }
|
||||
+ ]
|
||||
+ }
|
||||
+ ]
|
||||
+ },
|
||||
+ {
|
||||
+ "description": "countDocuments operation with snapshot and snapshot time",
|
||||
+ "operations": [
|
||||
+ {
|
||||
+ "name": "countDocuments",
|
||||
+ "object": "collection0",
|
||||
+ "arguments": {
|
||||
+ "session": "session0",
|
||||
+ "filter": {}
|
||||
+ },
|
||||
+ "expectResult": 2
|
||||
+ },
|
||||
+ {
|
||||
+ "name": "getSnapshotTime",
|
||||
+ "object": "session0",
|
||||
+ "saveResultAsEntity": "savedSnapshotTime"
|
||||
+ },
|
||||
+ {
|
||||
+ "name": "insertOne",
|
||||
+ "object": "collection0",
|
||||
+ "arguments": {
|
||||
+ "document": {
|
||||
+ "_id": 3,
|
||||
+ "x": 33
|
||||
+ }
|
||||
+ }
|
||||
+ },
|
||||
+ {
|
||||
+ "name": "createEntities",
|
||||
+ "object": "testRunner",
|
||||
+ "arguments": {
|
||||
+ "entities": [
|
||||
+ {
|
||||
+ "session": {
|
||||
+ "id": "session2",
|
||||
+ "client": "client0",
|
||||
+ "sessionOptions": {
|
||||
+ "snapshot": true,
|
||||
+ "snapshotTime": "savedSnapshotTime"
|
||||
+ }
|
||||
+ }
|
||||
+ }
|
||||
+ ]
|
||||
+ }
|
||||
+ },
|
||||
+ {
|
||||
+ "name": "countDocuments",
|
||||
+ "object": "collection0",
|
||||
+ "arguments": {
|
||||
+ "session": "session2",
|
||||
+ "filter": {}
|
||||
+ },
|
||||
+ "expectResult": 2
|
||||
+ },
|
||||
+ {
|
||||
+ "name": "countDocuments",
|
||||
+ "object": "collection0",
|
||||
+ "arguments": {
|
||||
+ "session": "session2",
|
||||
+ "filter": {}
|
||||
+ },
|
||||
+ "expectResult": 2
|
||||
+ },
|
||||
+ {
|
||||
+ "name": "countDocuments",
|
||||
+ "object": "collection0",
|
||||
+ "arguments": {
|
||||
+ "filter": {}
|
||||
+ },
|
||||
+ "expectResult": 3
|
||||
+ }
|
||||
+ ],
|
||||
+ "expectEvents": [
|
||||
+ {
|
||||
+ "client": "client0",
|
||||
+ "events": [
|
||||
+ {
|
||||
+ "commandStartedEvent": {
|
||||
+ "command": {
|
||||
+ "aggregate": "collection0",
|
||||
+ "readConcern": {
|
||||
+ "level": "snapshot",
|
||||
+ "atClusterTime": {
|
||||
+ "$$exists": false
|
||||
+ }
|
||||
+ }
|
||||
+ },
|
||||
+ "databaseName": "database0"
|
||||
+ }
|
||||
+ },
|
||||
+ {
|
||||
+ "commandStartedEvent": {
|
||||
+ "command": {
|
||||
+ "aggregate": "collection0",
|
||||
+ "readConcern": {
|
||||
+ "level": "snapshot",
|
||||
+ "atClusterTime": {
|
||||
+ "$$matchesEntity": "savedSnapshotTime"
|
||||
+ }
|
||||
+ }
|
||||
+ },
|
||||
+ "databaseName": "database0"
|
||||
+ }
|
||||
+ },
|
||||
+ {
|
||||
+ "commandStartedEvent": {
|
||||
+ "command": {
|
||||
+ "aggregate": "collection0",
|
||||
+ "readConcern": {
|
||||
+ "level": "snapshot",
|
||||
+ "atClusterTime": {
|
||||
+ "$$matchesEntity": "savedSnapshotTime"
|
||||
+ }
|
||||
+ }
|
||||
+ },
|
||||
+ "databaseName": "database0"
|
||||
+ }
|
||||
+ },
|
||||
+ {
|
||||
+ "commandStartedEvent": {
|
||||
+ "command": {
|
||||
+ "aggregate": "collection0",
|
||||
+ "readConcern": {
|
||||
+ "$$exists": false
|
||||
+ }
|
||||
+ },
|
||||
+ "databaseName": "database0"
|
||||
+ }
|
||||
+ }
|
||||
+ ]
|
||||
+ }
|
||||
+ ]
|
||||
+ },
|
||||
+ {
|
||||
+ "description": "Mixed operation with snapshot and snapshotTime",
|
||||
+ "operations": [
|
||||
+ {
|
||||
+ "name": "find",
|
||||
+ "object": "collection0",
|
||||
+ "arguments": {
|
||||
+ "session": "session0",
|
||||
+ "filter": {
|
||||
+ "_id": 1
|
||||
+ }
|
||||
+ },
|
||||
+ "expectResult": [
|
||||
+ {
|
||||
+ "_id": 1,
|
||||
+ "x": 11
|
||||
+ }
|
||||
+ ]
|
||||
+ },
|
||||
+ {
|
||||
+ "name": "getSnapshotTime",
|
||||
+ "object": "session0",
|
||||
+ "saveResultAsEntity": "savedSnapshotTime"
|
||||
+ },
|
||||
+ {
|
||||
+ "name": "findOneAndUpdate",
|
||||
+ "object": "collection0",
|
||||
+ "arguments": {
|
||||
+ "filter": {
|
||||
+ "_id": 1
|
||||
+ },
|
||||
+ "update": {
|
||||
+ "$inc": {
|
||||
+ "x": 1
|
||||
+ }
|
||||
+ },
|
||||
+ "returnDocument": "After"
|
||||
+ },
|
||||
+ "expectResult": {
|
||||
+ "_id": 1,
|
||||
+ "x": 12
|
||||
+ }
|
||||
+ },
|
||||
+ {
|
||||
+ "name": "createEntities",
|
||||
+ "object": "testRunner",
|
||||
+ "arguments": {
|
||||
+ "entities": [
|
||||
+ {
|
||||
+ "session": {
|
||||
+ "id": "session2",
|
||||
+ "client": "client0",
|
||||
+ "sessionOptions": {
|
||||
+ "snapshot": true,
|
||||
+ "snapshotTime": "savedSnapshotTime"
|
||||
+ }
|
||||
+ }
|
||||
+ }
|
||||
+ ]
|
||||
+ }
|
||||
+ },
|
||||
+ {
|
||||
+ "name": "find",
|
||||
+ "object": "collection0",
|
||||
+ "arguments": {
|
||||
+ "filter": {
|
||||
+ "_id": 1
|
||||
+ }
|
||||
+ },
|
||||
+ "expectResult": [
|
||||
+ {
|
||||
+ "_id": 1,
|
||||
+ "x": 12
|
||||
+ }
|
||||
+ ]
|
||||
+ },
|
||||
+ {
|
||||
+ "name": "aggregate",
|
||||
+ "object": "collection0",
|
||||
+ "arguments": {
|
||||
+ "pipeline": [
|
||||
+ {
|
||||
+ "$match": {
|
||||
+ "_id": 1
|
||||
+ }
|
||||
+ }
|
||||
+ ],
|
||||
+ "session": "session2"
|
||||
+ },
|
||||
+ "expectResult": [
|
||||
+ {
|
||||
+ "_id": 1,
|
||||
+ "x": 11
|
||||
+ }
|
||||
+ ]
|
||||
+ },
|
||||
+ {
|
||||
+ "name": "distinct",
|
||||
+ "object": "collection0",
|
||||
+ "arguments": {
|
||||
+ "fieldName": "x",
|
||||
+ "filter": {},
|
||||
+ "session": "session2"
|
||||
+ },
|
||||
+ "expectResult": [
|
||||
+ 11
|
||||
+ ]
|
||||
+ }
|
||||
+ ],
|
||||
+ "expectEvents": [
|
||||
+ {
|
||||
+ "client": "client0",
|
||||
+ "events": [
|
||||
+ {
|
||||
+ "commandStartedEvent": {
|
||||
+ "command": {
|
||||
+ "find": "collection0",
|
||||
+ "readConcern": {
|
||||
+ "level": "snapshot",
|
||||
+ "atClusterTime": {
|
||||
+ "$$exists": false
|
||||
+ }
|
||||
+ }
|
||||
+ }
|
||||
+ }
|
||||
+ },
|
||||
+ {
|
||||
+ "commandStartedEvent": {
|
||||
+ "command": {
|
||||
+ "find": "collection0",
|
||||
+ "readConcern": {
|
||||
+ "$$exists": false
|
||||
+ }
|
||||
+ }
|
||||
+ }
|
||||
+ },
|
||||
+ {
|
||||
+ "commandStartedEvent": {
|
||||
+ "command": {
|
||||
+ "aggregate": "collection0",
|
||||
+ "readConcern": {
|
||||
+ "level": "snapshot",
|
||||
+ "atClusterTime": {
|
||||
+ "$$matchesEntity": "savedSnapshotTime"
|
||||
+ }
|
||||
+ }
|
||||
+ }
|
||||
+ }
|
||||
+ },
|
||||
+ {
|
||||
+ "commandStartedEvent": {
|
||||
+ "command": {
|
||||
+ "distinct": "collection0",
|
||||
+ "readConcern": {
|
||||
+ "level": "snapshot",
|
||||
+ "atClusterTime": {
|
||||
+ "$$matchesEntity": "savedSnapshotTime"
|
||||
+ }
|
||||
+ }
|
||||
+ }
|
||||
+ }
|
||||
+ }
|
||||
+ ]
|
||||
+ }
|
||||
+ ]
|
||||
}
|
||||
]
|
||||
}
|
||||
460
.evergreen/spec-patch/PYTHON-5759.patch
Normal file
460
.evergreen/spec-patch/PYTHON-5759.patch
Normal file
@ -0,0 +1,460 @@
|
||||
diff --git a/test/client-side-encryption/spec/unified/accessToken-azure.json b/test/client-side-encryption/spec/unified/accessToken-azure.json
|
||||
new file mode 100644
|
||||
index 00000000..510d8795
|
||||
--- /dev/null
|
||||
+++ b/test/client-side-encryption/spec/unified/accessToken-azure.json
|
||||
@@ -0,0 +1,186 @@
|
||||
+{
|
||||
+ "description": "accessToken-azure",
|
||||
+ "schemaVersion": "1.28",
|
||||
+ "runOnRequirements": [
|
||||
+ {
|
||||
+ "minServerVersion": "4.1.10",
|
||||
+ "csfle": {
|
||||
+ "minLibmongocryptVersion": "1.6.0"
|
||||
+ }
|
||||
+ }
|
||||
+ ],
|
||||
+ "createEntities": [
|
||||
+ {
|
||||
+ "client": {
|
||||
+ "id": "client",
|
||||
+ "autoEncryptOpts": {
|
||||
+ "keyVaultNamespace": "keyvault.datakeys",
|
||||
+ "kmsProviders": {
|
||||
+ "azure": {
|
||||
+ "accessToken": {
|
||||
+ "$$placeholder": 1
|
||||
+ }
|
||||
+ }
|
||||
+ }
|
||||
+ }
|
||||
+ }
|
||||
+ },
|
||||
+ {
|
||||
+ "database": {
|
||||
+ "id": "db",
|
||||
+ "client": "client",
|
||||
+ "databaseName": "db"
|
||||
+ }
|
||||
+ },
|
||||
+ {
|
||||
+ "collection": {
|
||||
+ "id": "coll",
|
||||
+ "database": "db",
|
||||
+ "collectionName": "coll"
|
||||
+ }
|
||||
+ },
|
||||
+ {
|
||||
+ "clientEncryption": {
|
||||
+ "id": "clientEncryption",
|
||||
+ "clientEncryptionOpts": {
|
||||
+ "keyVaultClient": "client",
|
||||
+ "keyVaultNamespace": "keyvault.datakeys",
|
||||
+ "kmsProviders": {
|
||||
+ "azure": {
|
||||
+ "accessToken": {
|
||||
+ "$$placeholder": 1
|
||||
+ }
|
||||
+ }
|
||||
+ }
|
||||
+ }
|
||||
+ }
|
||||
+ }
|
||||
+ ],
|
||||
+ "initialData": [
|
||||
+ {
|
||||
+ "databaseName": "db",
|
||||
+ "collectionName": "coll",
|
||||
+ "documents": [],
|
||||
+ "createOptions": {
|
||||
+ "validator": {
|
||||
+ "$jsonSchema": {
|
||||
+ "properties": {
|
||||
+ "secret": {
|
||||
+ "encrypt": {
|
||||
+ "keyId": [
|
||||
+ {
|
||||
+ "$binary": {
|
||||
+ "base64": "AZURE+AAAAAAAAAAAAAAAA==",
|
||||
+ "subType": "04"
|
||||
+ }
|
||||
+ }
|
||||
+ ],
|
||||
+ "bsonType": "string",
|
||||
+ "algorithm": "AEAD_AES_256_CBC_HMAC_SHA_512-Deterministic"
|
||||
+ }
|
||||
+ }
|
||||
+ },
|
||||
+ "bsonType": "object"
|
||||
+ }
|
||||
+ }
|
||||
+ }
|
||||
+ },
|
||||
+ {
|
||||
+ "databaseName": "keyvault",
|
||||
+ "collectionName": "datakeys",
|
||||
+ "documents": [
|
||||
+ {
|
||||
+ "_id": {
|
||||
+ "$binary": {
|
||||
+ "base64": "AZURE+AAAAAAAAAAAAAAAA==",
|
||||
+ "subType": "04"
|
||||
+ }
|
||||
+ },
|
||||
+ "keyAltNames": [
|
||||
+ "my-key"
|
||||
+ ],
|
||||
+ "keyMaterial": {
|
||||
+ "$binary": {
|
||||
+ "base64": "n+HWZ0ZSVOYA3cvQgP7inN4JSXfOH85IngmeQxRpQHjCCcqT3IFqEWNlrsVHiz3AELimHhX4HKqOLWMUeSIT6emUDDoQX9BAv8DR1+E1w4nGs/NyEneac78EYFkK3JysrFDOgl2ypCCTKAypkn9CkAx1if4cfgQE93LW4kczcyHdGiH36CIxrCDGv1UzAvERN5Qa47DVwsM6a+hWsF2AAAJVnF0wYLLJU07TuRHdMrrphPWXZsFgyV+lRqJ7DDpReKNO8nMPLV/mHqHBHGPGQiRdb9NoJo8CvokGz4+KE8oLwzKf6V24dtwZmRkrsDV4iOhvROAzz+Euo1ypSkL3mw==",
|
||||
+ "subType": "00"
|
||||
+ }
|
||||
+ },
|
||||
+ "creationDate": {
|
||||
+ "$date": {
|
||||
+ "$numberLong": "1552949630483"
|
||||
+ }
|
||||
+ },
|
||||
+ "updateDate": {
|
||||
+ "$date": {
|
||||
+ "$numberLong": "1552949630483"
|
||||
+ }
|
||||
+ },
|
||||
+ "status": {
|
||||
+ "$numberInt": "0"
|
||||
+ },
|
||||
+ "masterKey": {
|
||||
+ "provider": "azure",
|
||||
+ "keyVaultEndpoint": "key-vault-csfle.vault.azure.net",
|
||||
+ "keyName": "key-name-csfle"
|
||||
+ }
|
||||
+ }
|
||||
+ ]
|
||||
+ }
|
||||
+ ],
|
||||
+ "tests": [
|
||||
+ {
|
||||
+ "description": "Auto encrypt using access token Azure credentials",
|
||||
+ "operations": [
|
||||
+ {
|
||||
+ "name": "insertOne",
|
||||
+ "arguments": {
|
||||
+ "document": {
|
||||
+ "_id": 1,
|
||||
+ "secret": "string0"
|
||||
+ }
|
||||
+ },
|
||||
+ "object": "coll"
|
||||
+ }
|
||||
+ ],
|
||||
+ "outcome": [
|
||||
+ {
|
||||
+ "documents": [
|
||||
+ {
|
||||
+ "_id": 1,
|
||||
+ "secret": {
|
||||
+ "$binary": {
|
||||
+ "base64": "AQGVERPgAAAAAAAAAAAAAAAC5DbBSwPwfSlBrDtRuglvNvCXD1KzDuCKY2P+4bRFtHDjpTOE2XuytPAUaAbXf1orsPq59PVZmsbTZbt2CB8qaQ==",
|
||||
+ "subType": "06"
|
||||
+ }
|
||||
+ }
|
||||
+ }
|
||||
+ ],
|
||||
+ "collectionName": "coll",
|
||||
+ "databaseName": "db"
|
||||
+ }
|
||||
+ ]
|
||||
+ },
|
||||
+ {
|
||||
+ "description": "Explicit encrypt using access token Azure credentials",
|
||||
+ "operations": [
|
||||
+ {
|
||||
+ "name": "encrypt",
|
||||
+ "object": "clientEncryption",
|
||||
+ "arguments": {
|
||||
+ "value": "string0",
|
||||
+ "opts": {
|
||||
+ "keyAltName": "my-key",
|
||||
+ "algorithm": "AEAD_AES_256_CBC_HMAC_SHA_512-Deterministic"
|
||||
+ }
|
||||
+ },
|
||||
+ "expectResult": {
|
||||
+ "$binary": {
|
||||
+ "base64": "AQGVERPgAAAAAAAAAAAAAAAC5DbBSwPwfSlBrDtRuglvNvCXD1KzDuCKY2P+4bRFtHDjpTOE2XuytPAUaAbXf1orsPq59PVZmsbTZbt2CB8qaQ==",
|
||||
+ "subType": "06"
|
||||
+ }
|
||||
+ }
|
||||
+ }
|
||||
+ ]
|
||||
+ }
|
||||
+ ]
|
||||
+}
|
||||
diff --git a/test/client-side-encryption/spec/unified/accessToken-gcp.json b/test/client-side-encryption/spec/unified/accessToken-gcp.json
|
||||
new file mode 100644
|
||||
index 00000000..f5cf8914
|
||||
--- /dev/null
|
||||
+++ b/test/client-side-encryption/spec/unified/accessToken-gcp.json
|
||||
@@ -0,0 +1,188 @@
|
||||
+{
|
||||
+ "description": "accessToken-gcp",
|
||||
+ "schemaVersion": "1.28",
|
||||
+ "runOnRequirements": [
|
||||
+ {
|
||||
+ "minServerVersion": "4.1.10",
|
||||
+ "csfle": {
|
||||
+ "minLibmongocryptVersion": "1.6.0"
|
||||
+ }
|
||||
+ }
|
||||
+ ],
|
||||
+ "createEntities": [
|
||||
+ {
|
||||
+ "client": {
|
||||
+ "id": "client",
|
||||
+ "autoEncryptOpts": {
|
||||
+ "keyVaultNamespace": "keyvault.datakeys",
|
||||
+ "kmsProviders": {
|
||||
+ "gcp": {
|
||||
+ "accessToken": {
|
||||
+ "$$placeholder": 1
|
||||
+ }
|
||||
+ }
|
||||
+ }
|
||||
+ }
|
||||
+ }
|
||||
+ },
|
||||
+ {
|
||||
+ "database": {
|
||||
+ "id": "db",
|
||||
+ "client": "client",
|
||||
+ "databaseName": "db"
|
||||
+ }
|
||||
+ },
|
||||
+ {
|
||||
+ "collection": {
|
||||
+ "id": "coll",
|
||||
+ "database": "db",
|
||||
+ "collectionName": "coll"
|
||||
+ }
|
||||
+ },
|
||||
+ {
|
||||
+ "clientEncryption": {
|
||||
+ "id": "clientEncryption",
|
||||
+ "clientEncryptionOpts": {
|
||||
+ "keyVaultClient": "client",
|
||||
+ "keyVaultNamespace": "keyvault.datakeys",
|
||||
+ "kmsProviders": {
|
||||
+ "gcp": {
|
||||
+ "accessToken": {
|
||||
+ "$$placeholder": 1
|
||||
+ }
|
||||
+ }
|
||||
+ }
|
||||
+ }
|
||||
+ }
|
||||
+ }
|
||||
+ ],
|
||||
+ "initialData": [
|
||||
+ {
|
||||
+ "databaseName": "db",
|
||||
+ "collectionName": "coll",
|
||||
+ "documents": [],
|
||||
+ "createOptions": {
|
||||
+ "validator": {
|
||||
+ "$jsonSchema": {
|
||||
+ "properties": {
|
||||
+ "secret": {
|
||||
+ "encrypt": {
|
||||
+ "keyId": [
|
||||
+ {
|
||||
+ "$binary": {
|
||||
+ "base64": "GCP+AAAAAAAAAAAAAAAAAA==",
|
||||
+ "subType": "04"
|
||||
+ }
|
||||
+ }
|
||||
+ ],
|
||||
+ "bsonType": "string",
|
||||
+ "algorithm": "AEAD_AES_256_CBC_HMAC_SHA_512-Deterministic"
|
||||
+ }
|
||||
+ }
|
||||
+ },
|
||||
+ "bsonType": "object"
|
||||
+ }
|
||||
+ }
|
||||
+ }
|
||||
+ },
|
||||
+ {
|
||||
+ "databaseName": "keyvault",
|
||||
+ "collectionName": "datakeys",
|
||||
+ "documents": [
|
||||
+ {
|
||||
+ "_id": {
|
||||
+ "$binary": {
|
||||
+ "base64": "GCP+AAAAAAAAAAAAAAAAAA==",
|
||||
+ "subType": "04"
|
||||
+ }
|
||||
+ },
|
||||
+ "keyAltNames": [
|
||||
+ "my-key"
|
||||
+ ],
|
||||
+ "keyMaterial": {
|
||||
+ "$binary": {
|
||||
+ "base64": "CiQAIgLj0WyktnB4dfYHo5SLZ41K4ASQrjJUaSzl5vvVH0G12G0SiQEAjlV8XPlbnHDEDFbdTO4QIe8ER2/172U1ouLazG0ysDtFFIlSvWX5ZnZUrRMmp/R2aJkzLXEt/zf8Mn4Lfm+itnjgo5R9K4pmPNvvPKNZX5C16lrPT+aA+rd+zXFSmlMg3i5jnxvTdLHhg3G7Q/Uv1ZIJskKt95bzLoe0tUVzRWMYXLIEcohnQg==",
|
||||
+ "subType": "00"
|
||||
+ }
|
||||
+ },
|
||||
+ "creationDate": {
|
||||
+ "$date": {
|
||||
+ "$numberLong": "1552949630483"
|
||||
+ }
|
||||
+ },
|
||||
+ "updateDate": {
|
||||
+ "$date": {
|
||||
+ "$numberLong": "1552949630483"
|
||||
+ }
|
||||
+ },
|
||||
+ "status": {
|
||||
+ "$numberInt": "0"
|
||||
+ },
|
||||
+ "masterKey": {
|
||||
+ "provider": "gcp",
|
||||
+ "projectId": "devprod-drivers",
|
||||
+ "location": "global",
|
||||
+ "keyRing": "key-ring-csfle",
|
||||
+ "keyName": "key-name-csfle"
|
||||
+ }
|
||||
+ }
|
||||
+ ]
|
||||
+ }
|
||||
+ ],
|
||||
+ "tests": [
|
||||
+ {
|
||||
+ "description": "Auto encrypt using access token GCP credentials",
|
||||
+ "operations": [
|
||||
+ {
|
||||
+ "name": "insertOne",
|
||||
+ "arguments": {
|
||||
+ "document": {
|
||||
+ "_id": 1,
|
||||
+ "secret": "string0"
|
||||
+ }
|
||||
+ },
|
||||
+ "object": "coll"
|
||||
+ }
|
||||
+ ],
|
||||
+ "outcome": [
|
||||
+ {
|
||||
+ "documents": [
|
||||
+ {
|
||||
+ "_id": 1,
|
||||
+ "secret": {
|
||||
+ "$binary": {
|
||||
+ "base64": "ARgj/gAAAAAAAAAAAAAAAAACwFd+Y5Ojw45GUXNvbcIpN9YkRdoHDHkR4kssdn0tIMKlDQOLFkWFY9X07IRlXsxPD8DcTiKnl6XINK28vhcGlg==",
|
||||
+ "subType": "06"
|
||||
+ }
|
||||
+ }
|
||||
+ }
|
||||
+ ],
|
||||
+ "collectionName": "coll",
|
||||
+ "databaseName": "db"
|
||||
+ }
|
||||
+ ]
|
||||
+ },
|
||||
+ {
|
||||
+ "description": "Explicit encrypt using access token GCP credentials",
|
||||
+ "operations": [
|
||||
+ {
|
||||
+ "name": "encrypt",
|
||||
+ "object": "clientEncryption",
|
||||
+ "arguments": {
|
||||
+ "value": "string0",
|
||||
+ "opts": {
|
||||
+ "keyAltName": "my-key",
|
||||
+ "algorithm": "AEAD_AES_256_CBC_HMAC_SHA_512-Deterministic"
|
||||
+ }
|
||||
+ },
|
||||
+ "expectResult": {
|
||||
+ "$binary": {
|
||||
+ "base64": "ARgj/gAAAAAAAAAAAAAAAAACwFd+Y5Ojw45GUXNvbcIpN9YkRdoHDHkR4kssdn0tIMKlDQOLFkWFY9X07IRlXsxPD8DcTiKnl6XINK28vhcGlg==",
|
||||
+ "subType": "06"
|
||||
+ }
|
||||
+ }
|
||||
+ }
|
||||
+ ]
|
||||
+ }
|
||||
+ ]
|
||||
+}
|
||||
diff --git a/test/unified-test-format/invalid/clientEncryptionOpts-kmsProviders-azure-accessToken-type.json b/test/unified-test-format/invalid/clientEncryptionOpts-kmsProviders-azure-accessToken-type.json
|
||||
new file mode 100644
|
||||
index 00000000..8fe5c150
|
||||
--- /dev/null
|
||||
+++ b/test/unified-test-format/invalid/clientEncryptionOpts-kmsProviders-azure-accessToken-type.json
|
||||
@@ -0,0 +1,31 @@
|
||||
+{
|
||||
+ "description": "clientEncryptionOpts-kmsProviders-azure-accessToken-type",
|
||||
+ "schemaVersion": "1.28",
|
||||
+ "createEntities": [
|
||||
+ {
|
||||
+ "client": {
|
||||
+ "id": "client0"
|
||||
+ }
|
||||
+ },
|
||||
+ {
|
||||
+ "clientEncryption": {
|
||||
+ "id": "clientEncryption0",
|
||||
+ "clientEncryptionOpts": {
|
||||
+ "keyVaultClient": "client0",
|
||||
+ "keyVaultNamespace": "keyvault.datakeys",
|
||||
+ "kmsProviders": {
|
||||
+ "azure": {
|
||||
+ "accessToken": 0
|
||||
+ }
|
||||
+ }
|
||||
+ }
|
||||
+ }
|
||||
+ }
|
||||
+ ],
|
||||
+ "tests": [
|
||||
+ {
|
||||
+ "description": "",
|
||||
+ "operations": []
|
||||
+ }
|
||||
+ ]
|
||||
+}
|
||||
diff --git a/test/unified-test-format/invalid/clientEncryptionOpts-kmsProviders-gcp-accessToken-type.json b/test/unified-test-format/invalid/clientEncryptionOpts-kmsProviders-gcp-accessToken-type.json
|
||||
new file mode 100644
|
||||
index 00000000..2284e26c
|
||||
--- /dev/null
|
||||
+++ b/test/unified-test-format/invalid/clientEncryptionOpts-kmsProviders-gcp-accessToken-type.json
|
||||
@@ -0,0 +1,31 @@
|
||||
+{
|
||||
+ "description": "clientEncryptionOpts-kmsProviders-gcp-accessToken-type",
|
||||
+ "schemaVersion": "1.28",
|
||||
+ "createEntities": [
|
||||
+ {
|
||||
+ "client": {
|
||||
+ "id": "client0"
|
||||
+ }
|
||||
+ },
|
||||
+ {
|
||||
+ "clientEncryption": {
|
||||
+ "id": "clientEncryption0",
|
||||
+ "clientEncryptionOpts": {
|
||||
+ "keyVaultClient": "client0",
|
||||
+ "keyVaultNamespace": "keyvault.datakeys",
|
||||
+ "kmsProviders": {
|
||||
+ "gcp": {
|
||||
+ "accessToken": 0
|
||||
+ }
|
||||
+ }
|
||||
+ }
|
||||
+ }
|
||||
+ }
|
||||
+ ],
|
||||
+ "tests": [
|
||||
+ {
|
||||
+ "description": "",
|
||||
+ "operations": []
|
||||
+ }
|
||||
+ ]
|
||||
+}
|
||||
@ -1,4 +1,6 @@
|
||||
#!/bin/bash
|
||||
# Synchronize local files to a remote Evergreen spawn host.
|
||||
set -eu
|
||||
|
||||
if [ -z "$1" ]
|
||||
then
|
||||
@ -7,9 +9,12 @@ fi
|
||||
|
||||
target=$1
|
||||
user=${target%@*}
|
||||
remote_dir=/home/$user/mongo-python-driver
|
||||
|
||||
echo "Copying files to $target..."
|
||||
rsync -az -e ssh --exclude '.git' --filter=':- .gitignore' -r . $target:$remote_dir
|
||||
echo "Copying files to $target... done."
|
||||
echo "Syncing files to $target..."
|
||||
rsync -haz -e ssh --exclude '.git' --filter=':- .gitignore' -r . $target:/home/$user/mongo-python-driver
|
||||
# shellcheck disable=SC2034
|
||||
fswatch -o . | while read f; do rsync -hazv -e ssh --exclude '.git' --filter=':- .gitignore' -r . $target:/home/$user/mongo-python-driver; done
|
||||
echo "Syncing files to $target... done."
|
||||
|
||||
@ -1,10 +0,0 @@
|
||||
#!/bin/bash
|
||||
set -o errexit # Exit the script with error if any of the commands fail
|
||||
set -o xtrace
|
||||
|
||||
if [ -z "${DRIVERS_TOOLS}" ]; then
|
||||
echo "Missing environment variable DRIVERS_TOOLS"
|
||||
fi
|
||||
|
||||
bash ${DRIVERS_TOOLS}/.evergreen/csfle/stop-servers.sh
|
||||
rm -rf libmongocrypt/ libmongocrypt.tar.gz mongocryptd.pid
|
||||
@ -1,117 +0,0 @@
|
||||
#!/bin/bash
|
||||
|
||||
set -eu
|
||||
|
||||
find_python3() {
|
||||
PYTHON=""
|
||||
# Add a fallback system python3 if it is available and Python 3.9+.
|
||||
if is_python_39 "$(command -v python3)"; then
|
||||
PYTHON="$(command -v python3)"
|
||||
fi
|
||||
# Find a suitable toolchain version, if available.
|
||||
if [ "$(uname -s)" = "Darwin" ]; then
|
||||
# macos 11.00
|
||||
if [ -d "/Library/Frameworks/Python.Framework/Versions/3.10" ]; then
|
||||
PYTHON="/Library/Frameworks/Python.Framework/Versions/3.10/bin/python3"
|
||||
# macos 10.14
|
||||
elif [ -d "/Library/Frameworks/Python.Framework/Versions/3.9" ]; then
|
||||
PYTHON="/Library/Frameworks/Python.Framework/Versions/3.9/bin/python3"
|
||||
fi
|
||||
elif [ "Windows_NT" = "${OS:-}" ]; then # Magic variable in cygwin
|
||||
PYTHON="C:/python/Python39/python.exe"
|
||||
else
|
||||
# Prefer our own toolchain, fall back to mongodb toolchain if it has Python 3.9+.
|
||||
if [ -f "/opt/python/3.9/bin/python3" ]; then
|
||||
PYTHON="/opt/python/3.9/bin/python3"
|
||||
elif is_python_39 "$(command -v /opt/mongodbtoolchain/v4/bin/python3)"; then
|
||||
PYTHON="/opt/mongodbtoolchain/v4/bin/python3"
|
||||
elif is_python_39 "$(command -v /opt/mongodbtoolchain/v3/bin/python3)"; then
|
||||
PYTHON="/opt/mongodbtoolchain/v3/bin/python3"
|
||||
fi
|
||||
fi
|
||||
if [ -z "$PYTHON" ]; then
|
||||
echo "Cannot test without python3.9+ installed!"
|
||||
exit 1
|
||||
fi
|
||||
echo "$PYTHON"
|
||||
}
|
||||
|
||||
# Usage:
|
||||
# createvirtualenv /path/to/python /output/path/for/venv
|
||||
# * param1: Python binary to use for the virtualenv
|
||||
# * param2: Path to the virtualenv to create
|
||||
createvirtualenv () {
|
||||
PYTHON=$1
|
||||
VENVPATH=$2
|
||||
|
||||
# Prefer venv
|
||||
VENV="$PYTHON -m venv"
|
||||
if [ "$(uname -s)" = "Darwin" ]; then
|
||||
VIRTUALENV="$PYTHON -m virtualenv"
|
||||
else
|
||||
VIRTUALENV=$(command -v virtualenv 2>/dev/null || echo "$PYTHON -m virtualenv")
|
||||
VIRTUALENV="$VIRTUALENV -p $PYTHON"
|
||||
fi
|
||||
if ! $VENV $VENVPATH 2>/dev/null; then
|
||||
# Workaround for bug in older versions of virtualenv.
|
||||
$VIRTUALENV $VENVPATH 2>/dev/null || $VIRTUALENV $VENVPATH
|
||||
fi
|
||||
if [ "Windows_NT" = "${OS:-}" ]; then
|
||||
# Workaround https://bugs.python.org/issue32451:
|
||||
# mongovenv/Scripts/activate: line 3: $'\r': command not found
|
||||
dos2unix $VENVPATH/Scripts/activate || true
|
||||
. $VENVPATH/Scripts/activate
|
||||
else
|
||||
. $VENVPATH/bin/activate
|
||||
fi
|
||||
|
||||
export PIP_QUIET=1
|
||||
python -m pip install --upgrade pip
|
||||
}
|
||||
|
||||
# Usage:
|
||||
# testinstall /path/to/python /path/to/.whl ["no-virtualenv"]
|
||||
# * param1: Python binary to test
|
||||
# * param2: Path to the wheel to install
|
||||
# * param3 (optional): If set to a non-empty string, don't create a virtualenv. Used in manylinux containers.
|
||||
testinstall () {
|
||||
PYTHON=$1
|
||||
RELEASE=$2
|
||||
NO_VIRTUALENV=$3
|
||||
PYTHON_IMPL=$(python -c "import platform; print(platform.python_implementation())")
|
||||
|
||||
if [ -z "$NO_VIRTUALENV" ]; then
|
||||
createvirtualenv $PYTHON venvtestinstall
|
||||
PYTHON=python
|
||||
fi
|
||||
|
||||
$PYTHON -m pip install --upgrade $RELEASE
|
||||
cd tools
|
||||
|
||||
if [ "$PYTHON_IMPL" = "CPython" ]; then
|
||||
$PYTHON fail_if_no_c.py
|
||||
fi
|
||||
|
||||
$PYTHON -m pip uninstall -y pymongo
|
||||
cd ..
|
||||
|
||||
if [ -z "$NO_VIRTUALENV" ]; then
|
||||
deactivate
|
||||
rm -rf venvtestinstall
|
||||
fi
|
||||
}
|
||||
|
||||
# Function that returns success if the provided Python binary is version 3.9 or later
|
||||
# Usage:
|
||||
# is_python_39 /path/to/python
|
||||
# * param1: Python binary
|
||||
is_python_39() {
|
||||
if [ -z "$1" ]; then
|
||||
return 1
|
||||
elif $1 -c "import sys; exit(sys.version_info[:2] < (3, 9))"; then
|
||||
# runs when sys.version_info[:2] >= (3, 9)
|
||||
return 0
|
||||
else
|
||||
return 1
|
||||
fi
|
||||
}
|
||||
1
.github/CODEOWNERS
vendored
Normal file
1
.github/CODEOWNERS
vendored
Normal file
@ -0,0 +1 @@
|
||||
* @mongodb/dbx-python
|
||||
44
.github/copilot-instructions.md
vendored
Normal file
44
.github/copilot-instructions.md
vendored
Normal file
@ -0,0 +1,44 @@
|
||||
When reviewing code, focus on:
|
||||
|
||||
## Security Critical Issues
|
||||
- Check for hardcoded secrets, API keys, or credentials.
|
||||
- Check for instances of potential method call injection, dynamic code execution, symbol injection or other code injection vulnerabilities.
|
||||
|
||||
## Performance Red Flags
|
||||
- Spot inefficient loops and algorithmic issues.
|
||||
- Check for memory leaks and resource cleanup.
|
||||
|
||||
## Code Quality Essentials
|
||||
- Methods should be focused and appropriately sized. If a method is doing too much, suggest refactorings to split it up.
|
||||
- Use clear, descriptive naming conventions.
|
||||
- Avoid encapsulation violations and ensure proper separation of concerns.
|
||||
- All public classes, modules, and methods should have clear documentation in Sphinx format.
|
||||
|
||||
## PyMongo-specific Concerns
|
||||
- Do not review files within `pymongo/synchronous` or files in `test/` that also have a file of the same name in `test/asynchronous` unless the reviewed changes include a `_IS_SYNC` statement. PyMongo generates these files from `pymongo/asynchronous` and `test/asynchronous` using `tools/synchro.py`.
|
||||
- All asynchronous functions must not call any blocking I/O.
|
||||
|
||||
## Review Style
|
||||
- Be specific and actionable in feedback.
|
||||
- Explain the "why" behind recommendations.
|
||||
- Acknowledge good patterns when you see them.
|
||||
- Ask clarifying questions when code intent is unclear.
|
||||
|
||||
Always prioritize security vulnerabilities and performance issues that could impact users.
|
||||
|
||||
Always suggest changes to improve readability and testability. For example, this suggestion seeks to make the code more readable, reusable, and testable:
|
||||
|
||||
```python
|
||||
# Instead of:
|
||||
if user.email and "@" in user.email and len(user.email) > 5:
|
||||
submit_button.enabled = True
|
||||
else:
|
||||
submit_button.enabled = False
|
||||
|
||||
# Consider:
|
||||
def valid_email(email):
|
||||
return email and "@" in email and len(email) > 5
|
||||
|
||||
|
||||
submit_button.enabled = valid_email(user.email)
|
||||
```
|
||||
2
.github/dependabot.yml
vendored
2
.github/dependabot.yml
vendored
@ -5,6 +5,8 @@ updates:
|
||||
directory: "/"
|
||||
schedule:
|
||||
interval: "weekly"
|
||||
cooldown:
|
||||
default-days: 7
|
||||
groups:
|
||||
actions:
|
||||
patterns:
|
||||
|
||||
33
.github/pull_request_template.md
vendored
Normal file
33
.github/pull_request_template.md
vendored
Normal file
@ -0,0 +1,33 @@
|
||||
<!-- Thanks for contributing! -->
|
||||
<!-- Please ensure that the title of the PR is in the following form:
|
||||
[JIRA TICKET]: Issue Title
|
||||
|
||||
If you are an external contributor and there is no JIRA ticket associated with your change, then use your best judgement
|
||||
for the PR title. A MongoDB employee will create a JIRA ticket and edit the name and links as appropriate.
|
||||
|
||||
Note on AI Contributions:
|
||||
We only accept pull requests that are authored and submitted by human contributors who fully understand the changes they are proposing.
|
||||
All contributions must be written and understood by human contributors. Please read about our policy in our contributing guide.
|
||||
-->
|
||||
[JIRA TICKET]
|
||||
|
||||
## Changes in this PR
|
||||
<!-- What changes did you make to the code? What new APIs (public or private) were added, removed, or edited to generate
|
||||
the desired outcome explained in the above summary? -->
|
||||
|
||||
## Test Plan
|
||||
<!-- How did you test the code? If you added unit tests, you can say that. If you didn’t introduce unit tests, explain why.
|
||||
All code should be tested in some way – so please list what your validation strategy was. -->
|
||||
|
||||
## Checklist
|
||||
<!-- Do not delete the items provided on this checklist. -->
|
||||
|
||||
### Checklist for Author
|
||||
- [ ] Did you update the changelog (if necessary)?
|
||||
- [ ] Is there test coverage?
|
||||
- [ ] Is any followup work tracked in a JIRA ticket? If so, add link(s).
|
||||
|
||||
### Checklist for Reviewer
|
||||
- [ ] Does the title of the PR reference a JIRA Ticket?
|
||||
- [ ] Do you fully understand the implementation? (Would you be comfortable explaining how this code works to someone else?)
|
||||
- [ ] Is all relevant documentation (README or docstring) updated?
|
||||
9
.github/workflows/codeql.yml
vendored
9
.github/workflows/codeql.yml
vendored
@ -38,15 +38,15 @@ jobs:
|
||||
build-mode: none
|
||||
steps:
|
||||
- name: Checkout repository
|
||||
uses: actions/checkout@v4
|
||||
uses: actions/checkout@v6
|
||||
with:
|
||||
ref: ${{ inputs.ref }}
|
||||
persist-credentials: false
|
||||
- uses: actions/setup-python@v5
|
||||
- uses: actions/setup-python@v6
|
||||
|
||||
# Initializes the CodeQL tools for scanning.
|
||||
- name: Initialize CodeQL
|
||||
uses: github/codeql-action/init@v3
|
||||
uses: github/codeql-action/init@5d4e8d1aca955e8d8589aabd499c5cae939e33c7 # v4
|
||||
with:
|
||||
languages: ${{ matrix.language }}
|
||||
build-mode: ${{ matrix.build-mode }}
|
||||
@ -54,7 +54,6 @@ jobs:
|
||||
queries: security-extended
|
||||
config: |
|
||||
paths-ignore:
|
||||
- '.github/**'
|
||||
- 'doc/**'
|
||||
- 'tools/**'
|
||||
- 'test/**'
|
||||
@ -64,6 +63,6 @@ jobs:
|
||||
pip install -e .
|
||||
|
||||
- name: Perform CodeQL Analysis
|
||||
uses: github/codeql-action/analyze@v3
|
||||
uses: github/codeql-action/analyze@5d4e8d1aca955e8d8589aabd499c5cae939e33c7 # v4
|
||||
with:
|
||||
category: "/language:${{matrix.language}}"
|
||||
|
||||
8
.github/workflows/create-release-branch.yml
vendored
8
.github/workflows/create-release-branch.yml
vendored
@ -33,17 +33,19 @@ jobs:
|
||||
outputs:
|
||||
version: ${{ steps.pre-publish.outputs.version }}
|
||||
steps:
|
||||
- uses: mongodb-labs/drivers-github-tools/secure-checkout@v2
|
||||
- uses: mongodb-labs/drivers-github-tools/secure-checkout@v3
|
||||
with:
|
||||
app_id: ${{ vars.APP_ID }}
|
||||
private_key: ${{ secrets.APP_PRIVATE_KEY }}
|
||||
- uses: mongodb-labs/drivers-github-tools/setup@v2
|
||||
- uses: mongodb-labs/drivers-github-tools/setup@v3
|
||||
with:
|
||||
aws_role_arn: ${{ secrets.AWS_ROLE_ARN }}
|
||||
aws_region_name: ${{ vars.AWS_REGION_NAME }}
|
||||
aws_secret_id: ${{ secrets.AWS_SECRET_ID }}
|
||||
artifactory_username: ${{ vars.ARTIFACTORY_USERNAME }}
|
||||
- uses: mongodb-labs/drivers-github-tools/create-branch@v2
|
||||
- name: Get hatch
|
||||
run: pip install hatch
|
||||
- uses: mongodb-labs/drivers-github-tools/create-branch@v3
|
||||
id: create-branch
|
||||
with:
|
||||
branch_name: ${{ inputs.branch_name }}
|
||||
|
||||
44
.github/workflows/dist.yml
vendored
44
.github/workflows/dist.yml
vendored
@ -39,28 +39,29 @@ jobs:
|
||||
- [ubuntu-latest, "manylinux_ppc64le", "cp3*-manylinux_ppc64le"]
|
||||
- [ubuntu-latest, "manylinux_s390x", "cp3*-manylinux_s390x"]
|
||||
- [ubuntu-latest, "manylinux_i686", "cp3*-manylinux_i686"]
|
||||
- [windows-2019, "win_amd6", "cp3*-win_amd64"]
|
||||
- [windows-2019, "win32", "cp3*-win32"]
|
||||
- [windows-2022, "win_amd6", "cp3*-win_amd64"]
|
||||
- [windows-2022, "win32", "cp3*-win32"]
|
||||
- [windows-11-arm, "win_arm64", "cp3*-win_arm64"]
|
||||
- [macos-14, "macos", "cp*-macosx_*"]
|
||||
|
||||
steps:
|
||||
- name: Checkout pymongo
|
||||
uses: actions/checkout@v4
|
||||
uses: actions/checkout@v6
|
||||
with:
|
||||
fetch-depth: 0
|
||||
persist-credentials: false
|
||||
ref: ${{ inputs.ref }}
|
||||
|
||||
- uses: actions/setup-python@v5
|
||||
- uses: actions/setup-python@v6
|
||||
with:
|
||||
cache: 'pip'
|
||||
python-version: 3.9
|
||||
python-version: 3.11
|
||||
cache-dependency-path: 'pyproject.toml'
|
||||
allow-prereleases: true
|
||||
|
||||
- name: Set up QEMU
|
||||
if: runner.os == 'Linux'
|
||||
uses: docker/setup-qemu-action@v3
|
||||
uses: docker/setup-qemu-action@ce360397dd3f832beb865e1373c09c0e9f86d70a # v4.0.0
|
||||
with:
|
||||
# setup-qemu-action by default uses `tonistiigi/binfmt:latest` image,
|
||||
# which is out of date. This causes seg faults during build.
|
||||
@ -69,24 +70,16 @@ jobs:
|
||||
platforms: all
|
||||
|
||||
- name: Install cibuildwheel
|
||||
# Note: the default manylinux is manylinux2014
|
||||
# Note: the default manylinux is manylinux_2_28
|
||||
run: |
|
||||
python -m pip install -U pip
|
||||
python -m pip install "cibuildwheel>=2.20,<3"
|
||||
python -m pip install "cibuildwheel>=3.2.0,<4"
|
||||
|
||||
- name: Build wheels
|
||||
env:
|
||||
CIBW_BUILD: ${{ matrix.buildplat[2] }}
|
||||
run: python -m cibuildwheel --output-dir wheelhouse
|
||||
|
||||
- name: Build manylinux1 wheels
|
||||
if: ${{ matrix.buildplat[1] == 'manylinux_x86_64' || matrix.buildplat[1] == 'manylinux_i686' }}
|
||||
env:
|
||||
CIBW_MANYLINUX_X86_64_IMAGE: manylinux1
|
||||
CIBW_MANYLINUX_I686_IMAGE: manylinux1
|
||||
CIBW_BUILD: "cp39-${{ matrix.buildplat[1] }} cp39-${{ matrix.buildplat[1] }}"
|
||||
run: python -m cibuildwheel --output-dir wheelhouse
|
||||
|
||||
- name: Assert all versions in wheelhouse
|
||||
if: ${{ ! startsWith(matrix.buildplat[1], 'macos') }}
|
||||
run: |
|
||||
@ -95,10 +88,11 @@ jobs:
|
||||
ls wheelhouse/*cp311*.whl
|
||||
ls wheelhouse/*cp312*.whl
|
||||
ls wheelhouse/*cp313*.whl
|
||||
ls wheelhouse/*cp314*.whl
|
||||
# Free-threading builds:
|
||||
ls wheelhouse/*cp313t*.whl
|
||||
ls wheelhouse/*cp314t*.whl
|
||||
|
||||
- uses: actions/upload-artifact@v4
|
||||
- uses: actions/upload-artifact@v7
|
||||
with:
|
||||
name: wheel-${{ matrix.buildplat[1] }}
|
||||
path: ./wheelhouse/*.whl
|
||||
@ -106,18 +100,18 @@ jobs:
|
||||
|
||||
make_sdist:
|
||||
name: Make SDist
|
||||
runs-on: macos-13
|
||||
runs-on: macos-latest
|
||||
steps:
|
||||
- uses: actions/checkout@v4
|
||||
- uses: actions/checkout@v6
|
||||
with:
|
||||
fetch-depth: 0
|
||||
persist-credentials: false
|
||||
ref: ${{ inputs.ref }}
|
||||
|
||||
- uses: actions/setup-python@v5
|
||||
- uses: actions/setup-python@v6
|
||||
with:
|
||||
# Build sdist on lowest supported Python
|
||||
python-version: '3.9'
|
||||
python-version: "3.9"
|
||||
|
||||
- name: Build SDist
|
||||
run: |
|
||||
@ -131,7 +125,7 @@ jobs:
|
||||
cd ..
|
||||
python -c "from pymongo import has_c; assert has_c()"
|
||||
|
||||
- uses: actions/upload-artifact@v4
|
||||
- uses: actions/upload-artifact@v7
|
||||
with:
|
||||
name: "sdist"
|
||||
path: ./dist/*.tar.gz
|
||||
@ -142,13 +136,13 @@ jobs:
|
||||
name: Download Wheels
|
||||
steps:
|
||||
- name: Download all workflow run artifacts
|
||||
uses: actions/download-artifact@v4
|
||||
uses: actions/download-artifact@v8
|
||||
- name: Flatten directory
|
||||
working-directory: .
|
||||
run: |
|
||||
find . -mindepth 2 -type f -exec mv {} . \;
|
||||
find . -type d -empty -delete
|
||||
- uses: actions/upload-artifact@v4
|
||||
- uses: actions/upload-artifact@v7
|
||||
with:
|
||||
name: all-dist-${{ github.run_id }}
|
||||
path: "./*"
|
||||
|
||||
23
.github/workflows/pull_request_template.md
vendored
23
.github/workflows/pull_request_template.md
vendored
@ -1,23 +0,0 @@
|
||||
# [JIRA Ticket ID](Link to Ticket)
|
||||
<!-- Please provide explicit URL link to the corresponding JIRA ticket. -->
|
||||
|
||||
# Summary
|
||||
<!-- Please provide a high level overview of what changes have been made. -->
|
||||
|
||||
# Changes in this PR
|
||||
<!-- Highlight any high level architecture changes if the summary doesn't already cover the scope. -->
|
||||
|
||||
# Test Plan
|
||||
<!-- Talk through any unit tests added, and if this is a bug fix, please add repro steps in the event the fix needs to be verified. -->
|
||||
|
||||
# Screenshots (Optional)
|
||||
<!-- Add a before and after picture to indicate changes. -->
|
||||
|
||||
# Callouts or Follow-up items (Optional)
|
||||
<!-- Any additional info not already specified in the PR including but not limited to:
|
||||
|
||||
1. Potential stakeholders
|
||||
2. Slack threads etc.
|
||||
3. Implementation details that need additional oversight
|
||||
4. Callouts on future tactics
|
||||
-->
|
||||
31
.github/workflows/release-python.yml
vendored
31
.github/workflows/release-python.yml
vendored
@ -3,12 +3,8 @@ name: Release
|
||||
on:
|
||||
workflow_dispatch:
|
||||
inputs:
|
||||
version:
|
||||
description: "The new version to set"
|
||||
required: true
|
||||
following_version:
|
||||
description: "The post (dev) version to set"
|
||||
required: true
|
||||
dry_run:
|
||||
description: "Dry Run?"
|
||||
default: false
|
||||
@ -20,14 +16,12 @@ env:
|
||||
# Changes per repo
|
||||
PRODUCT_NAME: PyMongo
|
||||
# Changes per branch
|
||||
SILK_ASSET_GROUP: mongodb-python-driver-4.11
|
||||
EVERGREEN_PROJECT: mongo-python-driver-release
|
||||
EVERGREEN_PROJECT: mongo-python-driver
|
||||
# Constant
|
||||
# inputs will be empty on a scheduled run. so, we only set dry_run
|
||||
# to 'false' when the input is set to 'false'.
|
||||
DRY_RUN: ${{ ! contains(inputs.dry_run, 'false') }}
|
||||
FOLLOWING_VERSION: ${{ inputs.following_version || '' }}
|
||||
VERSION: ${{ inputs.version || '10.10.10.10' }}
|
||||
|
||||
defaults:
|
||||
run:
|
||||
@ -44,20 +38,18 @@ jobs:
|
||||
outputs:
|
||||
version: ${{ steps.pre-publish.outputs.version }}
|
||||
steps:
|
||||
- uses: mongodb-labs/drivers-github-tools/secure-checkout@v2
|
||||
- uses: mongodb-labs/drivers-github-tools/secure-checkout@v3
|
||||
with:
|
||||
app_id: ${{ vars.APP_ID }}
|
||||
private_key: ${{ secrets.APP_PRIVATE_KEY }}
|
||||
- uses: mongodb-labs/drivers-github-tools/setup@v2
|
||||
- uses: mongodb-labs/drivers-github-tools/setup@v3
|
||||
with:
|
||||
aws_role_arn: ${{ secrets.AWS_ROLE_ARN }}
|
||||
aws_region_name: ${{ vars.AWS_REGION_NAME }}
|
||||
aws_secret_id: ${{ secrets.AWS_SECRET_ID }}
|
||||
artifactory_username: ${{ vars.ARTIFACTORY_USERNAME }}
|
||||
- uses: mongodb-labs/drivers-github-tools/python/pre-publish@v2
|
||||
- uses: mongodb-labs/drivers-github-tools/python/pre-publish@v3
|
||||
id: pre-publish
|
||||
with:
|
||||
version: ${{ env.VERSION }}
|
||||
dry_run: ${{ env.DRY_RUN }}
|
||||
|
||||
build-dist:
|
||||
@ -83,19 +75,19 @@ jobs:
|
||||
id-token: write
|
||||
steps:
|
||||
- name: Download all the dists
|
||||
uses: actions/download-artifact@v4
|
||||
uses: actions/download-artifact@v8
|
||||
with:
|
||||
name: all-dist-${{ github.run_id }}
|
||||
path: dist/
|
||||
- name: Publish package distributions to TestPyPI
|
||||
uses: pypa/gh-action-pypi-publish@release/v1
|
||||
uses: pypa/gh-action-pypi-publish@ed0c53931b1dc9bd32cbe73a98c7f6766f8a527e # release/v1
|
||||
with:
|
||||
repository-url: https://test.pypi.org/legacy/
|
||||
skip-existing: true
|
||||
attestations: ${{ env.DRY_RUN }}
|
||||
- name: Publish package distributions to PyPI
|
||||
if: startsWith(env.DRY_RUN, 'false')
|
||||
uses: pypa/gh-action-pypi-publish@release/v1
|
||||
uses: pypa/gh-action-pypi-publish@ed0c53931b1dc9bd32cbe73a98c7f6766f8a527e # release/v1
|
||||
|
||||
post-publish:
|
||||
needs: [publish]
|
||||
@ -107,22 +99,19 @@ jobs:
|
||||
attestations: write
|
||||
security-events: write
|
||||
steps:
|
||||
- uses: mongodb-labs/drivers-github-tools/secure-checkout@v2
|
||||
- uses: mongodb-labs/drivers-github-tools/secure-checkout@v3
|
||||
with:
|
||||
app_id: ${{ vars.APP_ID }}
|
||||
private_key: ${{ secrets.APP_PRIVATE_KEY }}
|
||||
- uses: mongodb-labs/drivers-github-tools/setup@v2
|
||||
- uses: mongodb-labs/drivers-github-tools/setup@v3
|
||||
with:
|
||||
aws_role_arn: ${{ secrets.AWS_ROLE_ARN }}
|
||||
aws_region_name: ${{ vars.AWS_REGION_NAME }}
|
||||
aws_secret_id: ${{ secrets.AWS_SECRET_ID }}
|
||||
artifactory_username: ${{ vars.ARTIFACTORY_USERNAME }}
|
||||
- uses: mongodb-labs/drivers-github-tools/python/post-publish@v2
|
||||
- uses: mongodb-labs/drivers-github-tools/python/post-publish@v3
|
||||
with:
|
||||
version: ${{ env.VERSION }}
|
||||
following_version: ${{ env.FOLLOWING_VERSION }}
|
||||
product_name: ${{ env.PRODUCT_NAME }}
|
||||
silk_asset_group: ${{ env.SILK_ASSET_GROUP }}
|
||||
evergreen_project: ${{ env.EVERGREEN_PROJECT }}
|
||||
token: ${{ github.token }}
|
||||
dry_run: ${{ env.DRY_RUN }}
|
||||
|
||||
104
.github/workflows/sbom.yml
vendored
Normal file
104
.github/workflows/sbom.yml
vendored
Normal file
@ -0,0 +1,104 @@
|
||||
name: Generate SBOM
|
||||
|
||||
# This workflow uses cyclonedx-py and publishes an sbom.json artifact.
|
||||
# It runs on manual trigger or when package files change on main branch,
|
||||
# and creates a PR with the updated SBOM.
|
||||
# Internal documentation: go/sbom-scope
|
||||
|
||||
on:
|
||||
workflow_dispatch: {}
|
||||
push:
|
||||
branches: ['master']
|
||||
paths:
|
||||
- 'requirements.txt'
|
||||
- 'requirements/**.txt'
|
||||
- '!requirements/docs.txt'
|
||||
- '!requirements/test.txt'
|
||||
|
||||
permissions:
|
||||
contents: write
|
||||
pull-requests: write
|
||||
|
||||
jobs:
|
||||
sbom:
|
||||
name: Generate SBOM and Create PR
|
||||
runs-on: ubuntu-latest
|
||||
concurrency:
|
||||
group: sbom-${{ github.ref }}
|
||||
cancel-in-progress: false
|
||||
|
||||
steps:
|
||||
- name: Checkout repository
|
||||
uses: actions/checkout@v6
|
||||
with:
|
||||
persist-credentials: false
|
||||
|
||||
- name: Set up Python
|
||||
uses: actions/setup-python@v6
|
||||
with:
|
||||
python-version: "3.10"
|
||||
|
||||
- name: Generate SBOM
|
||||
run: |
|
||||
python -m venv .venv
|
||||
source .venv/bin/activate
|
||||
python tools/generate_sbom_requirements.py
|
||||
pip install -r sbom-requirements.txt
|
||||
pip install .
|
||||
pip uninstall -y pip setuptools
|
||||
deactivate
|
||||
python -m venv .venv-sbom
|
||||
source .venv-sbom/bin/activate
|
||||
pip install cyclonedx-bom==7.2.1
|
||||
cyclonedx-py environment --spec-version 1.5 --output-format JSON --output-file sbom.json .venv
|
||||
# Add PURL for pymongo (local package doesn't get PURL automatically)
|
||||
jq '(.components[] | select(.name == "pymongo" and .purl == null)) |= (. + {purl: ("pkg:pypi/pymongo@" + .version)})' sbom.json > sbom.tmp.json && mv sbom.tmp.json sbom.json
|
||||
|
||||
- name: Download CycloneDX CLI
|
||||
run: |
|
||||
curl -L -s -o /tmp/cyclonedx "https://github.com/CycloneDX/cyclonedx-cli/releases/download/v0.29.1/cyclonedx-linux-x64"
|
||||
chmod +x /tmp/cyclonedx
|
||||
|
||||
- name: Validate SBOM
|
||||
run: /tmp/cyclonedx validate --input-file sbom.json --fail-on-errors
|
||||
|
||||
- name: Cleanup
|
||||
if: always()
|
||||
run: rm -rf .venv .venv-sbom sbom-requirements.txt
|
||||
|
||||
- name: Upload SBOM artifact
|
||||
uses: actions/upload-artifact@v7
|
||||
with:
|
||||
name: sbom
|
||||
path: sbom.json
|
||||
if-no-files-found: error
|
||||
|
||||
- name: Create Pull Request
|
||||
uses: peter-evans/create-pull-request@c0f553fe549906ede9cf27b5156039d195d2ece0 # v8
|
||||
with:
|
||||
token: ${{ secrets.GITHUB_TOKEN }}
|
||||
commit-message: 'chore: Update SBOM after dependency changes'
|
||||
branch: auto-update-sbom-${{ github.run_id }}
|
||||
delete-branch: true
|
||||
title: 'Automation: Update SBOM'
|
||||
body: |
|
||||
## Automated SBOM Update
|
||||
|
||||
This PR was automatically generated because dependency manifest files changed.
|
||||
|
||||
### Changes
|
||||
- Updated `sbom.json` to reflect current dependencies
|
||||
|
||||
### Verification
|
||||
The SBOM was generated using cyclonedx-py v7.2.1 with the current Python environment.
|
||||
|
||||
### Triggered by
|
||||
- Commit: ${{ github.sha }}
|
||||
- Workflow run: ${{ github.run_id }}
|
||||
|
||||
---
|
||||
_This PR was created automatically by the [SBOM workflow](${{ github.server_url }}/${{ github.repository }}/actions/runs/${{ github.run_id }})_
|
||||
labels: |
|
||||
sbom
|
||||
automated
|
||||
dependencies
|
||||
246
.github/workflows/test-python.yml
vendored
246
.github/workflows/test-python.yml
vendored
@ -14,21 +14,24 @@ defaults:
|
||||
run:
|
||||
shell: bash -eux {0}
|
||||
|
||||
permissions:
|
||||
contents: read
|
||||
|
||||
jobs:
|
||||
|
||||
static:
|
||||
runs-on: ubuntu-latest
|
||||
steps:
|
||||
- uses: actions/checkout@v4
|
||||
- uses: actions/checkout@v6
|
||||
with:
|
||||
persist-credentials: false
|
||||
- uses: actions/setup-python@v5
|
||||
- name: Install uv
|
||||
uses: astral-sh/setup-uv@37802adc94f370d6bfd71619e3f0bf239e1f3b78 # v7
|
||||
with:
|
||||
python-version: "3.9"
|
||||
cache: 'pip'
|
||||
cache-dependency-path: 'pyproject.toml'
|
||||
enable-cache: true
|
||||
python-version: "3.10"
|
||||
- name: Install just
|
||||
uses: extractions/setup-just@v2
|
||||
run: uv tool install rust-just
|
||||
- name: Install Python dependencies
|
||||
run: |
|
||||
just install
|
||||
@ -50,141 +53,143 @@ jobs:
|
||||
cppcheck pymongo
|
||||
|
||||
build:
|
||||
# supercharge/mongodb-github-action requires containers so we don't test other platforms
|
||||
runs-on: ${{ matrix.os }}
|
||||
strategy:
|
||||
fail-fast: false
|
||||
matrix:
|
||||
# Tests currently only pass on ubuntu on GitHub Actions.
|
||||
os: [ubuntu-latest]
|
||||
python-version: ["3.9", "pypy-3.10", "3.13", "3.13t"]
|
||||
python-version: ["3.10", "pypy-3.11", "3.13t"]
|
||||
mongodb-version: ["8.0"]
|
||||
|
||||
name: CPython ${{ matrix.python-version }}-${{ matrix.os }}
|
||||
steps:
|
||||
- uses: actions/checkout@v4
|
||||
- uses: actions/checkout@v6
|
||||
with:
|
||||
persist-credentials: false
|
||||
- if: ${{ matrix.python-version == '3.13t' }}
|
||||
name: Setup free-threaded Python
|
||||
uses: deadsnakes/action@v3.2.0
|
||||
with:
|
||||
python-version: 3.13
|
||||
nogil: true
|
||||
- if: ${{ matrix.python-version != '3.13t' }}
|
||||
name: Setup Python
|
||||
uses: actions/setup-python@v5
|
||||
- name: Install uv
|
||||
uses: astral-sh/setup-uv@37802adc94f370d6bfd71619e3f0bf239e1f3b78 # v7
|
||||
with:
|
||||
enable-cache: true
|
||||
python-version: ${{ matrix.python-version }}
|
||||
cache: 'pip'
|
||||
cache-dependency-path: 'pyproject.toml'
|
||||
allow-prereleases: true
|
||||
- name: Install just
|
||||
uses: extractions/setup-just@v2
|
||||
- name: Install dependencies
|
||||
run: |
|
||||
if [[ "${{ matrix.python-version }}" == "3.13t" ]]; then
|
||||
# Just can't be installed on 3.13t, use pytest directly.
|
||||
pip install .
|
||||
pip install -r requirements/test.txt
|
||||
else
|
||||
just install
|
||||
fi
|
||||
- name: Start MongoDB
|
||||
uses: supercharge/mongodb-github-action@1.12.0
|
||||
- id: setup-mongodb
|
||||
uses: mongodb-labs/drivers-evergreen-tools@master
|
||||
with:
|
||||
mongodb-version: 6.0
|
||||
version: "${{ matrix.mongodb-version }}"
|
||||
- name: Run tests
|
||||
run: |
|
||||
if [[ "${{ matrix.python-version }}" == "3.13t" ]]; then
|
||||
pytest -v --durations=5 --maxfail=10
|
||||
else
|
||||
just test
|
||||
fi
|
||||
run: uv run --extra test pytest -v
|
||||
|
||||
coverage:
|
||||
# This enables a coverage report for a given PR, which will be augmented by
|
||||
# the combined codecov report uploaded in Evergreen.
|
||||
runs-on: ubuntu-latest
|
||||
|
||||
name: Coverage
|
||||
steps:
|
||||
- uses: actions/checkout@v6
|
||||
with:
|
||||
persist-credentials: false
|
||||
- name: Install uv
|
||||
uses: astral-sh/setup-uv@37802adc94f370d6bfd71619e3f0bf239e1f3b78 # v7
|
||||
with:
|
||||
enable-cache: true
|
||||
python-version: "3.10"
|
||||
- id: setup-mongodb
|
||||
uses: mongodb-labs/drivers-evergreen-tools@master
|
||||
with:
|
||||
version: "8.0"
|
||||
- name: Install just
|
||||
run: uv tool install rust-just
|
||||
- name: Setup tests
|
||||
run: COVERAGE=1 just setup-tests
|
||||
- name: Run tests
|
||||
run: just run-tests
|
||||
- name: Generate xml report
|
||||
run: uv tool run --with "coverage[toml]" coverage xml
|
||||
- name: Upload test results to Codecov
|
||||
uses: codecov/codecov-action@671740ac38dd9b0130fbe1cec585b89eea48d3de # v5
|
||||
with:
|
||||
token: ${{ secrets.CODECOV_TOKEN }}
|
||||
doctest:
|
||||
runs-on: ubuntu-latest
|
||||
name: DocTest
|
||||
steps:
|
||||
- uses: actions/checkout@v4
|
||||
- uses: actions/checkout@v6
|
||||
with:
|
||||
persist-credentials: false
|
||||
- name: Setup Python
|
||||
uses: actions/setup-python@v5
|
||||
- name: Install uv
|
||||
uses: astral-sh/setup-uv@37802adc94f370d6bfd71619e3f0bf239e1f3b78 # v7
|
||||
with:
|
||||
python-version: "3.9"
|
||||
cache: 'pip'
|
||||
cache-dependency-path: 'pyproject.toml'
|
||||
enable-cache: true
|
||||
python-version: "3.10"
|
||||
- name: Install just
|
||||
uses: extractions/setup-just@v2
|
||||
- name: Start MongoDB
|
||||
uses: supercharge/mongodb-github-action@1.12.0
|
||||
run: uv tool install rust-just
|
||||
- id: setup-mongodb
|
||||
uses: mongodb-labs/drivers-evergreen-tools@master
|
||||
with:
|
||||
mongodb-version: '8.0.0-rc4'
|
||||
version: "8.0"
|
||||
- name: Install dependencies
|
||||
run: |
|
||||
just install
|
||||
run: just install
|
||||
- name: Run tests
|
||||
run: |
|
||||
just docs-test
|
||||
run: |
|
||||
just setup-tests doctest
|
||||
just run-tests
|
||||
|
||||
docs:
|
||||
name: Docs Checks
|
||||
runs-on: ubuntu-latest
|
||||
steps:
|
||||
- uses: actions/checkout@v4
|
||||
- uses: actions/checkout@v6
|
||||
with:
|
||||
persist-credentials: false
|
||||
- uses: actions/setup-python@v5
|
||||
- name: Install uv
|
||||
uses: astral-sh/setup-uv@37802adc94f370d6bfd71619e3f0bf239e1f3b78 # v7
|
||||
with:
|
||||
cache: 'pip'
|
||||
cache-dependency-path: 'pyproject.toml'
|
||||
# Build docs on lowest supported Python for furo
|
||||
python-version: '3.9'
|
||||
enable-cache: true
|
||||
python-version: "3.10"
|
||||
- name: Install just
|
||||
uses: extractions/setup-just@v2
|
||||
run: uv tool install rust-just
|
||||
- name: Install dependencies
|
||||
run: |
|
||||
just install
|
||||
run: just install
|
||||
- name: Build docs
|
||||
run: |
|
||||
just docs
|
||||
run: just docs
|
||||
|
||||
linkcheck:
|
||||
name: Link Check
|
||||
runs-on: ubuntu-latest
|
||||
steps:
|
||||
- uses: actions/checkout@v4
|
||||
- uses: actions/checkout@v6
|
||||
with:
|
||||
persist-credentials: false
|
||||
- uses: actions/setup-python@v5
|
||||
- name: Install uv
|
||||
uses: astral-sh/setup-uv@37802adc94f370d6bfd71619e3f0bf239e1f3b78 # v7
|
||||
with:
|
||||
cache: 'pip'
|
||||
cache-dependency-path: 'pyproject.toml'
|
||||
# Build docs on lowest supported Python for furo
|
||||
python-version: '3.9'
|
||||
enable-cache: true
|
||||
python-version: "3.10"
|
||||
- name: Install just
|
||||
uses: extractions/setup-just@v2
|
||||
run: uv tool install rust-just
|
||||
- name: Install dependencies
|
||||
run: |
|
||||
just install
|
||||
run: just install
|
||||
- name: Build docs
|
||||
run: |
|
||||
just docs-linkcheck
|
||||
run: just docs-linkcheck
|
||||
|
||||
typing:
|
||||
name: Typing Tests
|
||||
runs-on: ubuntu-latest
|
||||
strategy:
|
||||
matrix:
|
||||
python: ["3.9", "3.11"]
|
||||
python: ["3.10", "3.11"]
|
||||
steps:
|
||||
- uses: actions/checkout@v4
|
||||
- uses: actions/checkout@v6
|
||||
with:
|
||||
persist-credentials: false
|
||||
- uses: actions/setup-python@v5
|
||||
- name: Install uv
|
||||
uses: astral-sh/setup-uv@37802adc94f370d6bfd71619e3f0bf239e1f3b78 # v7
|
||||
with:
|
||||
enable-cache: true
|
||||
python-version: "${{matrix.python}}"
|
||||
cache: 'pip'
|
||||
cache-dependency-path: 'pyproject.toml'
|
||||
- name: Install just
|
||||
uses: extractions/setup-just@v2
|
||||
run: uv tool install rust-just
|
||||
- name: Install dependencies
|
||||
run: |
|
||||
just install
|
||||
@ -192,25 +197,55 @@ jobs:
|
||||
run: |
|
||||
just typing
|
||||
|
||||
integration_tests:
|
||||
runs-on: ubuntu-latest
|
||||
name: Integration Tests
|
||||
steps:
|
||||
- uses: actions/checkout@v6
|
||||
with:
|
||||
persist-credentials: false
|
||||
- name: Install uv
|
||||
uses: astral-sh/setup-uv@37802adc94f370d6bfd71619e3f0bf239e1f3b78 # v7
|
||||
with:
|
||||
enable-cache: true
|
||||
python-version: "3.10"
|
||||
- name: Install just
|
||||
run: uv tool install rust-just
|
||||
- name: Install dependencies
|
||||
run: |
|
||||
just install
|
||||
- id: setup-mongodb
|
||||
uses: mongodb-labs/drivers-evergreen-tools@master
|
||||
- name: Run tests
|
||||
run: |
|
||||
just integration-tests
|
||||
- id: setup-mongodb-ssl
|
||||
uses: mongodb-labs/drivers-evergreen-tools@master
|
||||
with:
|
||||
ssl: true
|
||||
- name: Run tests
|
||||
run: |
|
||||
just integration-tests
|
||||
|
||||
make_sdist:
|
||||
runs-on: ubuntu-latest
|
||||
name: "Make an sdist"
|
||||
steps:
|
||||
- uses: actions/checkout@v4
|
||||
- uses: actions/checkout@v6
|
||||
with:
|
||||
persist-credentials: false
|
||||
- uses: actions/setup-python@v5
|
||||
- uses: actions/setup-python@v6
|
||||
with:
|
||||
cache: 'pip'
|
||||
cache-dependency-path: 'pyproject.toml'
|
||||
# Build sdist on lowest supported Python
|
||||
python-version: '3.9'
|
||||
python-version: "3.9"
|
||||
- name: Build SDist
|
||||
shell: bash
|
||||
run: |
|
||||
pip install build
|
||||
python -m build --sdist
|
||||
- uses: actions/upload-artifact@v4
|
||||
- uses: actions/upload-artifact@v7
|
||||
with:
|
||||
name: "sdist"
|
||||
path: dist/*.tar.gz
|
||||
@ -222,7 +257,9 @@ jobs:
|
||||
timeout-minutes: 20
|
||||
steps:
|
||||
- name: Download sdist
|
||||
uses: actions/download-artifact@v4
|
||||
uses: actions/download-artifact@v8
|
||||
with:
|
||||
path: sdist/
|
||||
- name: Unpack SDist
|
||||
shell: bash
|
||||
run: |
|
||||
@ -231,14 +268,14 @@ jobs:
|
||||
mkdir test
|
||||
tar --strip-components=1 -zxf *.tar.gz -C ./test
|
||||
ls test
|
||||
- uses: actions/setup-python@v5
|
||||
- uses: actions/setup-python@v6
|
||||
with:
|
||||
cache: 'pip'
|
||||
cache-dependency-path: 'sdist/test/pyproject.toml'
|
||||
# Test sdist on lowest supported Python
|
||||
python-version: '3.9'
|
||||
- name: Start MongoDB
|
||||
uses: supercharge/mongodb-github-action@1.12.0
|
||||
python-version: "3.9"
|
||||
- id: setup-mongodb
|
||||
uses: mongodb-labs/drivers-evergreen-tools@master
|
||||
- name: Run connect test from sdist
|
||||
shell: bash
|
||||
run: |
|
||||
@ -247,3 +284,28 @@ jobs:
|
||||
which python
|
||||
pip install -e ".[test]"
|
||||
PYMONGO_MUST_CONNECT=1 pytest -v -k client_context
|
||||
|
||||
test_minimum:
|
||||
permissions:
|
||||
contents: read
|
||||
runs-on: ubuntu-latest
|
||||
name: Test minimum dependencies and Python
|
||||
steps:
|
||||
- uses: actions/checkout@v6
|
||||
with:
|
||||
persist-credentials: false
|
||||
- name: Install uv
|
||||
uses: astral-sh/setup-uv@37802adc94f370d6bfd71619e3f0bf239e1f3b78 # v7
|
||||
with:
|
||||
python-version: "3.9"
|
||||
- id: setup-mongodb
|
||||
uses: mongodb-labs/drivers-evergreen-tools@master
|
||||
with:
|
||||
version: "8.0"
|
||||
- name: Run tests
|
||||
shell: bash
|
||||
run: |
|
||||
uv venv
|
||||
source .venv/bin/activate
|
||||
uv pip install -e ".[test]" --resolution=lowest-direct --force-reinstall
|
||||
pytest -v test/test_srv_polling.py test/test_dns.py test/asynchronous/test_srv_polling.py test/asynchronous/test_dns.py
|
||||
|
||||
15
.github/workflows/zizmor.yml
vendored
15
.github/workflows/zizmor.yml
vendored
@ -14,19 +14,8 @@ jobs:
|
||||
security-events: write
|
||||
steps:
|
||||
- name: Checkout repository
|
||||
uses: actions/checkout@v4
|
||||
uses: actions/checkout@v6
|
||||
with:
|
||||
persist-credentials: false
|
||||
- name: Setup Rust
|
||||
uses: actions-rust-lang/setup-rust-toolchain@v1
|
||||
- name: Get zizmor
|
||||
run: cargo install zizmor
|
||||
- name: Run zizmor 🌈
|
||||
run: zizmor --format sarif . > results.sarif
|
||||
env:
|
||||
GH_TOKEN: ${{ secrets.GITHUB_TOKEN }}
|
||||
- name: Upload SARIF file
|
||||
uses: github/codeql-action/upload-sarif@v3
|
||||
with:
|
||||
sarif_file: results.sarif
|
||||
category: zizmor
|
||||
uses: zizmorcore/zizmor-action@71321a20a9ded102f6e9ce5718a2fcec2c4f70d8 # v0.5.2
|
||||
|
||||
7
.github/zizmor.yml
vendored
Normal file
7
.github/zizmor.yml
vendored
Normal file
@ -0,0 +1,7 @@
|
||||
rules:
|
||||
unpinned-uses:
|
||||
config:
|
||||
policies:
|
||||
actions/*: ref-pin
|
||||
mongodb-labs/drivers-github-tools/*: ref-pin
|
||||
mongodb-labs/drivers-evergreen-tools: ref-pin
|
||||
9
.gitignore
vendored
9
.gitignore
vendored
@ -18,6 +18,7 @@ mongocryptd.pid
|
||||
.idea/
|
||||
.vscode/
|
||||
.nova/
|
||||
.temp/
|
||||
venv/
|
||||
secrets-export.sh
|
||||
libmongocrypt.tar.gz
|
||||
@ -26,14 +27,20 @@ libmongocrypt/
|
||||
expansion.yml
|
||||
*expansions.yml
|
||||
.evergreen/scripts/env.sh
|
||||
.evergreen/scripts/test-env.sh
|
||||
specifications/
|
||||
results.json
|
||||
.evergreen/atlas_x509_dev_client_certificate.pem
|
||||
|
||||
# Lambda temp files
|
||||
test/lambda/.aws-sam
|
||||
test/lambda/env.json
|
||||
test/lambda/mongodb/pymongo/*
|
||||
test/lambda/mongodb/gridfs/*
|
||||
test/lambda/mongodb/bson/*
|
||||
test/lambda/*.json
|
||||
|
||||
# test results and logs
|
||||
xunit-results/
|
||||
coverage.xml
|
||||
server.log
|
||||
.coverage
|
||||
|
||||
@ -6,6 +6,7 @@ repos:
|
||||
- id: check-added-large-files
|
||||
- id: check-case-conflict
|
||||
- id: check-toml
|
||||
- id: check-json
|
||||
- id: check-yaml
|
||||
exclude: template.yaml
|
||||
- id: debug-statements
|
||||
@ -17,6 +18,14 @@ repos:
|
||||
exclude: .patch
|
||||
exclude_types: [json]
|
||||
|
||||
- repo: https://github.com/astral-sh/ruff-pre-commit
|
||||
# Ruff version.
|
||||
rev: v0.1.3
|
||||
hooks:
|
||||
- id: ruff
|
||||
args: ["--fix", "--show-fixes"]
|
||||
- id: ruff-format
|
||||
|
||||
- repo: local
|
||||
hooks:
|
||||
- id: synchro
|
||||
@ -29,14 +38,6 @@ repos:
|
||||
- ruff==0.1.3
|
||||
- unasync
|
||||
|
||||
- repo: https://github.com/astral-sh/ruff-pre-commit
|
||||
# Ruff version.
|
||||
rev: v0.1.3
|
||||
hooks:
|
||||
- id: ruff
|
||||
args: ["--fix", "--show-fixes"]
|
||||
- id: ruff-format
|
||||
|
||||
- repo: https://github.com/adamchainz/blacken-docs
|
||||
rev: "1.16.0"
|
||||
hooks:
|
||||
@ -115,3 +116,20 @@ repos:
|
||||
(?x)(
|
||||
.evergreen/retry-with-backoff.sh
|
||||
)
|
||||
- id: generate-config
|
||||
name: generate-config
|
||||
entry: .evergreen/scripts/generate-config.sh
|
||||
language: python
|
||||
require_serial: true
|
||||
additional_dependencies: ["shrub.py>=3.10.0", "pyyaml>=6.0.2"]
|
||||
|
||||
- id: uv-lock
|
||||
name: uv-lock
|
||||
entry: uv lock
|
||||
language: python
|
||||
require_serial: true
|
||||
files: ^(uv\.lock|pyproject\.toml|requirements.txt|requirements/.*\.txt)$
|
||||
pass_filenames: false
|
||||
fail_fast: true
|
||||
additional_dependencies:
|
||||
- "uv>=0.8.4"
|
||||
|
||||
423
CONTRIBUTING.md
423
CONTRIBUTING.md
@ -16,7 +16,7 @@ be of interest or that has already been addressed.
|
||||
|
||||
## Supported Interpreters
|
||||
|
||||
PyMongo supports CPython 3.9+ and PyPy3.10+. Language features not
|
||||
PyMongo supports CPython 3.9+ and PyPy3.9+. Language features not
|
||||
supported by all interpreters can not be used.
|
||||
|
||||
## Style Guide
|
||||
@ -85,49 +85,53 @@ likelihood for getting review sooner shoots up.
|
||||
- `versionadded:: 3.11`
|
||||
- `versionchanged:: 3.5`
|
||||
|
||||
**Pull Request Template Breakdown**
|
||||
### AI-Generated Contributions Policy
|
||||
|
||||
- **Github PR Title**
|
||||
#### Our Stance
|
||||
|
||||
- The PR Title format should always be
|
||||
`[JIRA-ID] : Jira Title or Blurb Summary`.
|
||||
We only accept pull requests that are authored and submitted by human contributors who fully understand the changes they are proposing. Pull requests that are not clearly owned and understood by a human contributor may be closed. **All contributions must be submitted, reviewed, and understood by human contributors.**
|
||||
|
||||
- **JIRA LINK**
|
||||
##### Why This Policy Exists
|
||||
|
||||
- Convenient link to the associated JIRA ticket.
|
||||
At MongoDB, we understand the power and prevalence of AI tools in software development. With that being said, many MongoDB libraries are foundational tools used in production systems worldwide. The nature of these libraries requires:
|
||||
|
||||
- **Summary**
|
||||
- **Deep domain expertise**: MongoDB's wire protocol, BSON specification, connection pooling, authentication mechanisms, and concurrency patterns require an understanding that AI alone cannot substantiate.
|
||||
|
||||
- Small blurb on why this is needed. The JIRA task should have
|
||||
the more in-depth description, but this should still, at a
|
||||
high level, give anyone looking an understanding of why the
|
||||
PR has been checked in.
|
||||
- **Long-term maintainability**: Contributors need to be able to explain *why* code is written a certain way, explain design decisions, and be available to iterate on their contributions.
|
||||
|
||||
- **Changes in this PR**
|
||||
- **Security responsibility**: Authentication, credential handling, and TLS implementation cannot be left to probabilistic code generation.
|
||||
|
||||
- The explicit code changes that this PR is introducing. This
|
||||
should be more specific than just the task name. (Unless the
|
||||
task name is very clear).
|
||||
##### What This Means for Contributors
|
||||
|
||||
- **Test Plan**
|
||||
**Required:**
|
||||
|
||||
- Everything needs a test description. Describe what you did
|
||||
to validate your changes actually worked; if you did
|
||||
nothing, then document you did not test it. Aim to make
|
||||
these steps reproducible by other engineers, specifically
|
||||
with your primary reviewer in mind.
|
||||
- Full understanding of every line of code you submit
|
||||
- Ability to explain and defend your implementation choices
|
||||
- Willingness to iterate and maintain your contributions
|
||||
|
||||
- **Screenshots**
|
||||
**Encouraged:**
|
||||
|
||||
- Any images that provide more context to the PR. Usually,
|
||||
these just coincide with the test plan.
|
||||
- Using AI assistants as learning tools to understand concepts
|
||||
- IDE autocomplete features that suggest standard patterns
|
||||
- AI help for brainstorming approaches (but write the code yourself)
|
||||
- Writing code using AI tools, reviewing each line and revising code as necessary.
|
||||
|
||||
- **Callouts or follow-up items**
|
||||
**Not allowed:**
|
||||
|
||||
- This is a good place for identifying "to-dos" that you've
|
||||
placed in the code (Must have an accompanying JIRA Ticket).
|
||||
- Potential bugs that you are unsure how to test in the code.
|
||||
- Opinions you want to receive about your code.
|
||||
- Submitting PRs generated solely by AI tools
|
||||
- Copy-pasting AI-generated code without full understanding
|
||||
|
||||
##### Disclosure
|
||||
|
||||
If you used AI assistance in any way during your contribution, please disclose what the AI assistant was used for in your PR description. We would love to know what tools developers have found useful in iterating in their day to day.
|
||||
|
||||
##### Questions?
|
||||
|
||||
If you're unsure whether your contribution complies with this policy, please ask for guidance within the scope of the PR and clarify any uncertainty. We're happy to guide contributors toward successful contributions.
|
||||
|
||||
---
|
||||
|
||||
*This policy helps us maintain the reliability, security, and trustworthiness that production applications depend on. Thank you for understanding and for contributing thoughtfully to PyMongo.*
|
||||
|
||||
## Running Linters
|
||||
|
||||
@ -178,7 +182,7 @@ documentation including narrative docs, and the [Sphinx docstring format](https:
|
||||
You can build the documentation locally by running:
|
||||
|
||||
```bash
|
||||
just docs-build
|
||||
just docs
|
||||
```
|
||||
|
||||
When updating docs, it can be helpful to run the live docs server as:
|
||||
@ -190,41 +194,259 @@ just docs-serve
|
||||
Browse to the link provided, and then as you make changes to docstrings or narrative docs,
|
||||
the pages will re-render and the browser will automatically refresh.
|
||||
|
||||
|
||||
## Running Tests Locally
|
||||
|
||||
- Ensure you have started the appropriate Mongo Server(s).
|
||||
- Run `just install` to set a local virtual environment, or you can manually
|
||||
create a virtual environment and run `pytest` directly. If you want to use a specific
|
||||
version of Python, remove the `.venv` folder and set `PYTHON_BINARY` before running `just install`.
|
||||
version of Python, set `UV_PYTHON` before running `just install`.
|
||||
- Ensure you have started the appropriate Mongo Server(s). You can run `just run-server` with optional args
|
||||
to set up the server. All given options will be passed to
|
||||
[`run-mongodb.sh`](https://github.com/mongodb-labs/drivers-evergreen-tools/blob/master/.evergreen/run-mongodb.sh). Run `$DRIVERS_TOOLS/.evergreen/run-mongodb.sh start -h`
|
||||
for a full list of options.
|
||||
- Run `just test` or `pytest` to run all of the tests.
|
||||
- Append `test/<mod_name>.py::<class_name>::<test_name>` to run
|
||||
specific tests. You can omit the `<test_name>` to test a full class
|
||||
and the `<class_name>` to test a full module. For example:
|
||||
`just test test/test_change_stream.py::TestUnifiedChangeStreamsErrors::test_change_stream_errors_on_ElectionInProgress`.
|
||||
- Use the `-k` argument to select tests by pattern.
|
||||
- Run `just test-coverage` to run tests with coverage and display a report. After running tests with coverage, use `just coverage-html` to generate an HTML report in `htmlcov/index.html`.
|
||||
|
||||
## Running Load Balancer Tests Locally
|
||||
|
||||
- Install `haproxy` (available as `brew install haproxy` on macOS).
|
||||
- Clone `drivers-evergreen-tools`:
|
||||
`git clone git@github.com:mongodb-labs/drivers-evergreen-tools.git`.
|
||||
- Start the servers using
|
||||
`LOAD_BALANCER=true TOPOLOGY=sharded_cluster AUTH=noauth SSL=nossl MONGODB_VERSION=6.0 DRIVERS_TOOLS=$PWD/drivers-evergreen-tools MONGO_ORCHESTRATION_HOME=$PWD/drivers-evergreen-tools/.evergreen/orchestration $PWD/drivers-evergreen-tools/.evergreen/run-orchestration.sh`.
|
||||
- Start the load balancer using:
|
||||
`MONGODB_URI='mongodb://localhost:27017,localhost:27018/' $PWD/drivers-evergreen-tools/.evergreen/run-load-balancer.sh start`.
|
||||
- Run the tests from the `pymongo` checkout directory using:
|
||||
`TEST_LOADBALANCER=1 just test-eg`.
|
||||
## Running tests that require secrets, services, or other configuration
|
||||
|
||||
### Prerequisites
|
||||
|
||||
## Running Encryption Tests Locally
|
||||
- Clone `drivers-evergreen-tools`:
|
||||
`git clone git@github.com:mongodb-labs/drivers-evergreen-tools.git`.
|
||||
- Run `export DRIVERS_TOOLS=$PWD/drivers-evergreen-tools`
|
||||
- Run `AWS_PROFILE=<profile> just setup-encryption` after setting up your AWS profile with `aws configure sso`.
|
||||
- Run the tests with `TEST_ENCRYPTION=1 just test-eg`.
|
||||
- When done, run `just teardown-encryption` to clean up.
|
||||
`git clone git@github.com:mongodb-labs/drivers-evergreen-tools.git`.
|
||||
- Run `export DRIVERS_TOOLS=$PWD/drivers-evergreen-tools`. This can be put into a `.bashrc` file
|
||||
for convenience.
|
||||
- Some tests require access to [Drivers test secrets](https://github.com/mongodb-labs/drivers-evergreen-tools/tree/master/.evergreen/secrets_handling#secrets-handling).
|
||||
|
||||
## Re-sync Spec Tests
|
||||
### Usage
|
||||
|
||||
- Run `just run-server` with optional args to set up the server.
|
||||
- Run `just setup-tests` with optional args to set up the test environment, secrets, etc.
|
||||
See `just setup-tests -h` for a full list of available options.
|
||||
- Run `just run-tests` to run the tests in an appropriate Python environment.
|
||||
- When done, run `just teardown-tests` to clean up and `just stop-server` to stop the server.
|
||||
|
||||
### SSL tests
|
||||
|
||||
- Run `just run-server --ssl` to start the server with TLS enabled.
|
||||
- Run `just setup-tests --ssl`.
|
||||
- Run `just run-tests`.
|
||||
|
||||
Note: for general testing purposes with an TLS-enabled server, you can use the following (this should ONLY be used
|
||||
for local testing):
|
||||
|
||||
```python
|
||||
from pymongo import MongoClient
|
||||
|
||||
client = MongoClient(
|
||||
"mongodb://localhost:27017?tls=true&tlsAllowInvalidCertificates=true"
|
||||
)
|
||||
```
|
||||
|
||||
If you want to use the actual certificate file then set `tlsCertificateKeyFile` to the local path
|
||||
to `<repo_roo>/test/certificates/client.pem` and `tlsCAFile` to the local path to `<repo_roo>/test/certificates/ca.pem`.
|
||||
|
||||
### Encryption tests
|
||||
|
||||
- Run `just run-server` to start the server.
|
||||
- Run `just setup-tests encryption`.
|
||||
- Run the tests with `just run-tests`.
|
||||
|
||||
To test with `encryption` and `PyOpenSSL`, use `just setup-tests encryption pyopenssl`.
|
||||
|
||||
### PyOpenSSL tests
|
||||
|
||||
- Run `just run-server` to start the server.
|
||||
- Run `just setup-tests default_sync pyopenssl`.
|
||||
- Run the tests with `just run-tests`.
|
||||
|
||||
Note: `PyOpenSSL` is not used in async tests, but you can use `just setup-tests default_async pyopenssl`
|
||||
to verify that PyMongo falls back to the standard library `OpenSSL`.
|
||||
|
||||
### Load balancer tests
|
||||
|
||||
- Install `haproxy` (available as `brew install haproxy` on macOS).
|
||||
- Start the server with `just run-server load_balancer`.
|
||||
- Set up the test with `just setup-tests load_balancer`.
|
||||
- Run the tests with `just run-tests`.
|
||||
|
||||
### AWS auth tests
|
||||
|
||||
- Run `just run-server auth_aws` to start the server.
|
||||
- Run `just setup-tests auth_aws <aws-test-type>` to set up the AWS test.
|
||||
- Run the tests with `just run-tests`.
|
||||
|
||||
### OIDC auth tests
|
||||
|
||||
- Run `just setup-tests auth_oidc <oidc-test-type>` to set up the OIDC test.
|
||||
- Run the tests with `just run-tests`.
|
||||
|
||||
The supported types are [`default`, `azure`, `gcp`, `eks`, `aks`, and `gke`].
|
||||
For the `eks` test, you will need to set up access to the `drivers-test-secrets-role`, see the [Wiki](https://wiki.corp.mongodb.com/spaces/DRIVERS/pages/239737385/Using+AWS+Secrets+Manager+to+Store+Testing+Secrets).
|
||||
|
||||
### KMS tests
|
||||
|
||||
For KMS tests that are run locally, and expected to fail, in this case using `azure`:
|
||||
|
||||
- Run `just run-server`.
|
||||
- Run `just setup-tests kms azure-fail`.
|
||||
- Run `just run-tests`.
|
||||
|
||||
For KMS tests that run remotely and are expected to pass, in this case using `gcp`:
|
||||
|
||||
- Run `just setup-tests kms gcp`.
|
||||
- Run `just run-tests`.
|
||||
|
||||
### Enterprise Auth tests
|
||||
|
||||
Note: these tests can only be run from an Evergreen host.
|
||||
|
||||
- Run `just run-server enterprise_auth`.
|
||||
- Run `just setup-tests enterprise_auth`.
|
||||
- Run `just run-tests`.
|
||||
|
||||
### Atlas Connect tests
|
||||
|
||||
- Run `just setup-tests atlas_connect`.
|
||||
- Run `just run-tests`.
|
||||
|
||||
### Search Index tests
|
||||
|
||||
- Run `just run-server search_index`.
|
||||
- Run `just setup-tests search_index`.
|
||||
- Run `just run-tests`.
|
||||
|
||||
### MockupDB tests
|
||||
|
||||
- Run `just setup-tests mockupdb`.
|
||||
- Run `just run-tests`.
|
||||
|
||||
### Doc tests
|
||||
|
||||
The doc tests require a running server.
|
||||
|
||||
- Run `just run-server`.
|
||||
- Run `just setup-tests doctest`.
|
||||
- Run `just run-tests`.
|
||||
|
||||
### Free-threaded Python Tests
|
||||
|
||||
In the evergreen builds, the tests are configured to use the free-threaded python from the toolchain.
|
||||
Locally you can run:
|
||||
|
||||
- Run `just run-server`.
|
||||
- Run `just setup-tests`.
|
||||
- Run `UV_PYTHON=3.14t just run-tests`.
|
||||
|
||||
### AWS Lambda tests
|
||||
|
||||
You will need to set up access to the `drivers-test-secrets-role`, see the [Wiki](https://wiki.corp.mongodb.com/spaces/DRIVERS/pages/239737385/Using+AWS+Secrets+Manager+to+Store+Testing+Secrets).
|
||||
|
||||
- Run `just setup-tests aws_lambda`.
|
||||
- Run `just run-tests`.
|
||||
|
||||
### mod_wsgi tests
|
||||
|
||||
Note: these tests can only be run from an Evergreen Linux host that has the Python toolchain.
|
||||
|
||||
- Run `just run-server`.
|
||||
- Run `just setup-tests mod_wsgi <mode>`.
|
||||
- Run `just run-tests`.
|
||||
|
||||
The `mode` can be `standalone` or `embedded`. For the `replica_set` version of the tests, use
|
||||
`TOPOLOGY=replica_set just run-server`.
|
||||
|
||||
### OCSP tests
|
||||
|
||||
- Export the orchestration file, e.g. `export ORCHESTRATION_FILE=rsa-basic-tls-ocsp-disableStapling.json`.
|
||||
This corresponds to a config file in `$DRIVERS_TOOLS/.evergreen/orchestration/configs/servers`.
|
||||
MongoDB servers on MacOS and Windows do not staple OCSP responses and only support RSA.
|
||||
NOTE: because the mock ocsp responder MUST be started prior to the server starting, the ocsp tests start the server
|
||||
as part of `setup-tests`.
|
||||
|
||||
- Run `just setup-tests ocsp <sub test>` (options are "valid", "revoked", "valid-delegate", "revoked-delegate").
|
||||
- Run `just run-tests`
|
||||
|
||||
If you are running one of the `no-responder` tests, omit the `run-server` step.
|
||||
|
||||
### Perf Tests
|
||||
|
||||
- Start the appropriate server, e.g. `just run-server --version=v8.0-perf --ssl`.
|
||||
- Set up the tests with `sync` or `async`: `just setup-tests perf sync`.
|
||||
- Run the tests: `just run-tests`.
|
||||
|
||||
## Enable Debug Logs
|
||||
|
||||
- Use `-o log_cli_level="DEBUG" -o log_cli=1` with `just test` or `pytest` to output all debug logs to the terminal. **Warning**: This will output a huge amount of logs.
|
||||
- Add `log_cli=1` and `log_cli_level="DEBUG"` to the `tool.pytest.ini_options` section in `pyproject.toml` to enable debug logs in this manner by default on your machine.
|
||||
- Set `DEBUG_LOG=1` and run `just setup-tests`, `just-test`, or `pytest` to enable debug logs only for failed tests.
|
||||
- Finally, you can use `just setup-tests --debug-log`.
|
||||
- For evergreen patch builds, you can use `evergreen patch --param DEBUG_LOG=1` to enable debug logs for failed tests in the patch.
|
||||
|
||||
## Testing minimum dependencies
|
||||
|
||||
To run any of the test suites with minimum supported dependencies, pass `--test-min-deps` to
|
||||
`just setup-tests`.
|
||||
|
||||
## Testing time-dependent operations
|
||||
|
||||
- `test.utils_shared.delay` - One can trigger an arbitrarily long-running operation on the server using this delay utility
|
||||
in combination with a `$where` operation. Use this to test behaviors around timeouts or signals.
|
||||
|
||||
## Adding a new test suite
|
||||
|
||||
- If adding new tests files that should only be run for that test suite, add a pytest marker to the file and add
|
||||
to the list of pytest markers in `pyproject.toml`. Then add the test suite to the `TEST_SUITE_MAP` in `.evergreen/scripts/utils.py`. If for some reason it is not a pytest-runnable test, add it to the list of `EXTRA_TESTS` instead.
|
||||
- If the test uses Atlas or otherwise doesn't use `run-mongodb.sh`, add it to the `NO_RUN_ORCHESTRATION` list in
|
||||
`.evergreen/scripts/utils.py`.
|
||||
- If there is something special required to run the local server or there is an extra flag that should always be set
|
||||
like `AUTH`, add that logic to `.evergreen/scripts/run_server.py`.
|
||||
- The bulk of the logic will typically be in `.evergreen/scripts/setup_tests.py`. This is where you should fetch secrets and make them available using `write_env`, start services, and write other env vars needed using `write_env`.
|
||||
- If there are any special test considerations, including not running `pytest` at all, handle it in `.evergreen/scripts/run_tests.py`.
|
||||
- If there are any services or atlas clusters to teardown, handle them in `.evergreen/scripts/teardown_tests.py`.
|
||||
- Add functions to generate the test variant(s) and task(s) to the `.evergreen/scripts/generate_config.py`.
|
||||
- There are some considerations about the Python version used in the test:
|
||||
- If a specific version of Python is needed in a task that is running on variants with a toolchain, use
|
||||
``TOOLCHAIN_VERSION`` (e.g. `TOOLCHAIN_VERSION=3.10`). The actual path lookup needs to be done on the host, since
|
||||
tasks are host-agnostic.
|
||||
- If a specific Python binary is needed (for example on the FIPS host), set `UV_PYTHON=/path/to/python`.
|
||||
- If a specific Python version is needed and the toolchain will not be available, use `UV_PYTHON` (e.g. `UV_PYTHON=3.11`).
|
||||
- The default if neither ``TOOLCHAIN_VERSION`` or ``UV_PYTHON`` is set is to use UV to install the minimum
|
||||
supported version of Python and use that. This ensures a consistent behavior across host types that do not
|
||||
have the Python toolchain (e.g. Azure VMs), by having a known version of Python with the build headers (`Python.h`)
|
||||
needed to build the C extensions.
|
||||
- Regenerate the test variants and tasks using `pre-commit run --all-files generate-config`.
|
||||
- Make sure to add instructions for running the test suite to `CONTRIBUTING.md`.
|
||||
|
||||
## Handling flaky tests
|
||||
|
||||
We have a custom `flaky` decorator in [test/asynchronous/utils.py](test/asynchronous/utils.py) that can be used for
|
||||
tests that are `flaky`. By default the decorator only applies when not running on CPython on Linux, since other
|
||||
runtimes tend to have more variation. When using the `flaky` decorator, open a corresponding ticket and
|
||||
a use the ticket number as the "reason" parameter to the decorator, e.g. `@flaky(reason="PYTHON-1234")`.
|
||||
When running tests locally (not in CI), the `flaky` decorator will be disabled unless `ENABLE_FLAKY` is set.
|
||||
To disable the `flaky` decorator in CI, you can use `evergreen patch --param DISABLE_FLAKY=1`.
|
||||
|
||||
## Integration Tests
|
||||
|
||||
The `integration_tests` directory has a set of scripts that verify the usage of PyMongo with downstream packages or frameworks. See the [README](./integration_tests/README.md) for more information.
|
||||
|
||||
To run the tests, use `just integration_tests`.
|
||||
|
||||
The tests should be able to run with and without SSL enabled.
|
||||
|
||||
## Specification Tests
|
||||
|
||||
The MongoDB [specifications repository](https://github.com/mongodb/specifications)
|
||||
holds in progress and completed specifications for features of MongoDB, drivers,
|
||||
and associated products. PyMongo supports the [Unified Test Format](https://jira.mongodb.org/browse/DRIVERS-709)
|
||||
for running specification tests to confirm PyMongo behaves as expected.
|
||||
|
||||
### Resynchronizing the Specification Tests
|
||||
|
||||
If you would like to re-sync the copy of the specification tests in the
|
||||
PyMongo repository with that which is inside the [specifications
|
||||
@ -245,25 +467,114 @@ update in PyMongo. This is primarily helpful if you are implementing a
|
||||
new feature in PyMongo that has spec tests already implemented, or if
|
||||
you are attempting to validate new spec tests in PyMongo.
|
||||
|
||||
### Automated Specification Test Resyncing
|
||||
The (`/.evergreen/scripts/resync-all-specs.sh`) script
|
||||
automatically runs once a week to resync all the specs with the [specifications repo](https://github.com/mongodb/specifications).
|
||||
A PR will be generated by mongodb-drivers-pr-bot containing any changes picked up by this resync.
|
||||
The PR description will display the name(s) of the updated specs along
|
||||
with any errors that occurred.
|
||||
|
||||
Spec test changes associated with a behavioral change or bugfix that has yet to be implemented in PyMongo
|
||||
must be added to a patch file in `/.evergreen/spec-patch`. Each patch
|
||||
file must be named after the associated PYTHON ticket and contain the
|
||||
test differences between PyMongo's current tests and the specification.
|
||||
All changes listed in these patch files will be *undone* by the script and won't
|
||||
be applied to PyMongo's tests.
|
||||
|
||||
When a new test file or folder is added to the spec repo before the associated code changes are implemented, that test's path must be added to `.evergreen/remove-unimplemented-tests.sh` along with a comment indicating the associated PYTHON ticket for those changes.
|
||||
|
||||
Any PR that implements a PYTHON ticket documented in a patch file or within `.evergreen/remove-unimplemented-tests.sh` must also remove the associated patch file or entry in `remove-unimplemented-tests.sh`.
|
||||
|
||||
#### Adding to a patch file
|
||||
To add to or create a patch file, run `git diff` to show the desired changes to undo and copy the
|
||||
results into the patch file.
|
||||
|
||||
For example: the imaginary, unimplemented PYTHON-1234 ticket has associated spec test changes. To add those changes to `PYTHON-1234.patch`), do the following:
|
||||
```bash
|
||||
git diff HEAD~1 path/to/file >> .evergreen/spec-patch/PYTHON-1234.patch
|
||||
```
|
||||
|
||||
#### Running Locally
|
||||
Both `resync-all-specs.sh` and `resync-all-specs.py` can be run locally (and won't generate a PR).
|
||||
```bash
|
||||
./.evergreen/scripts/resync-all-specs.sh
|
||||
python3 ./.evergreen/scripts/resync-all-specs.py
|
||||
```
|
||||
|
||||
## Making a Release
|
||||
|
||||
Follow the [Python Driver Release Process Wiki](https://wiki.corp.mongodb.com/display/DRIVERS/Python+Driver+Release+Process).
|
||||
|
||||
## Asyncio considerations
|
||||
## Project Structure and Asyncio Considerations
|
||||
|
||||
PyMongo adds asyncio capability by modifying the source files in `*/asynchronous` to `*/synchronous` using
|
||||
[unasync](https://github.com/python-trio/unasync/) and some custom transforms.
|
||||
This section describes the layout of the `pymongo/` package.
|
||||
|
||||
Where possible, edit the code in `*/asynchronous/*.py` and not the synchronous files.
|
||||
You can run `pre-commit run --all-files synchro` before running tests if you are testing synchronous code.
|
||||
Within `pymongo/`, the code is further divided into the `pymongo/asynchronous` and `pymongo/synchronous` subdirectories.
|
||||
Files in `pymongo/synchronous` are generated from `pymongo/asynchronous` using the `synchro` pre-commit hook, which uses [unasync](https://github.com/python-trio/unasync/) and some custom transforms.
|
||||
|
||||
As a result, **all modifications** within `pymongo` must be made in either the top-level `pymongo` directory when they have to exhibit differing behavior between sync and async contexts or the `pymongo/asynchronous` directory, not `pymongo/synchronous`.
|
||||
Any changes made directly to files in the `pymongo/synchronous` directory will be overwritten by the `synchro` hook when it is run, which happens automatically on commit.
|
||||
|
||||
Some top-level files (e.g. `pymongo/collection.py`) are re-export files for existing import compatibility and should not be modified directly.
|
||||
The other top-level files (e.g. `pymongo/network_layer.py`, `pymongo/pool_shared.py`) contain either shared code used in both the asynchronous and synchronous APIs, or code that is very different between the two APIs and therefore cannot be generated from the async version using `synchro`.
|
||||
|
||||
Run `pre-commit run --all-files synchro` before running tests to generate the latest version of the synchronous code.
|
||||
|
||||
To prevent the `synchro` hook from accidentally overwriting code, it first checks to see whether a sync version
|
||||
of a file is changing and not its async counterpart, and will fail.
|
||||
In the unlikely scenario that you want to override this behavior, first export `OVERRIDE_SYNCHRO_CHECK=1`.
|
||||
|
||||
Sometimes, the `synchro` hook will fail and introduce changes many previously unmodified files. This is due to static
|
||||
Python errors, such as missing imports, incorrect syntax, or other fatal typos. To resolve these issues,
|
||||
run `pre-commit run --all-files --hook-stage manual ruff` and fix all reported errors before running the `synchro`
|
||||
hook again.
|
||||
|
||||
## Converting a test to async
|
||||
|
||||
The `tools/convert_test_to_async.py` script takes in an existing synchronous test file and outputs a
|
||||
partially-converted asynchronous version of the same name to the `test/asynchronous` directory.
|
||||
Use this generated file as a starting point for the completed conversion.
|
||||
|
||||
The script is used like so: `python tools/convert_test_to_async.py [test_file.py]`
|
||||
|
||||
## CPU profiling
|
||||
|
||||
To profile a test script and generate a flame graph, follow these steps:
|
||||
|
||||
1. Install `py-spy` if you haven't already:
|
||||
```bash
|
||||
pip install py-spy
|
||||
```
|
||||
2. Inside your test script, perform any required setup and then loop over the code you want to profile for improved sampling.
|
||||
3. Run `py-spy record -o <output.svg> -r <sample_rate=100> -- python <path/to/script>` to generate a `.svg` file containing the flame graph.
|
||||
(Note: on macOS you will need to run this command using `sudo` to allow `py-spy` to attach to the Python process.)
|
||||
4. If you need to include native code (for example the C extensions), profiling should be done on a Linux system, as macOS and Windows do not support the `--native` option of `py-spy`.
|
||||
Creating an ubuntu Evergreen spawn host and using `scp` to copy the flamegraph `.svg` file back to your local machine is the best way to do this.
|
||||
5. You can then view the flamegraph using an SVG viewer like a browser.
|
||||
|
||||
## Memory profiling
|
||||
|
||||
To test for a memory leak or any memory-related issues, the current best tool is [memray](https://bloomberg.github.io/memray/overview.html).
|
||||
In order to include code from our C extensions, it must be run in native mode, on Linux.
|
||||
To do so, either spin up an Ubuntu docker container or an Ubuntu Evergreen spawn host.
|
||||
|
||||
From the spawn host or Ubuntu image, do the following:
|
||||
|
||||
1. Install `memray` if you haven't already:
|
||||
```bash
|
||||
pip install memray
|
||||
```
|
||||
2. Inside your test script, perform any required setup and then loop over the code you want to profile for improved sampling.
|
||||
3. Run memray with the script under test with the `--native` flag, e.g. `python -m memray run --native -o test.bin <path/to/script>`.
|
||||
4. Generate the flamegraph with `python -m memray flamegraph -o test.html test.bin`.
|
||||
See the [docs](https://bloomberg.github.io/memray/flamegraph.html) for more options.
|
||||
5. Then, from the host computer, use either scp or docker cp to copy the flamegraph, e.g. `scp ubuntu@ec2-3-82-52-49.compute-1.amazonaws.com:/home/ubuntu/test.html .`.
|
||||
6. You can then view the flamegraph html in a browser.
|
||||
|
||||
## Dependabot updates
|
||||
|
||||
Dependabot will raise PRs at most once per week, grouped by GitHub Actions updates and Python requirement
|
||||
file updates. We have a pre-commit hook that will update the `uv.lock` file when requirements change.
|
||||
To update the lock file on a failing PR, you can use a method like `gh pr checkout <pr number>`, then run
|
||||
`just lint uv-lock` to update the lock file, and then push the changes. If a typing dependency has changed,
|
||||
also run `just typing` and handle any new findings.
|
||||
|
||||
Some files were not shown because too many files have changed in this diff Show More
Loading…
Reference in New Issue
Block a user