|
30 | 30 | from trove_classifiers import classifiers |
31 | 31 | from webob.multidict import MultiDict |
32 | 32 |
|
| 33 | +import warehouse.constants |
| 34 | + |
33 | 35 | from warehouse.accounts.utils import UserContext |
34 | 36 | from warehouse.admin.flags import AdminFlag, AdminFlagValue |
35 | 37 | from warehouse.classifiers.models import Classifier |
@@ -236,7 +238,9 @@ def test_zipfile_exceeds_compression_threshold(self, tmpdir): |
236 | 238 |
|
237 | 239 | with zipfile.ZipFile(f, "w") as zfp: |
238 | 240 | zfp.writestr("PKG-INFO", b"this is the package info") |
239 | | - zfp.writestr("1.dat", b"0" * 65 * legacy.ONE_MB, zipfile.ZIP_DEFLATED) |
| 241 | + zfp.writestr( |
| 242 | + "1.dat", b"0" * 65 * warehouse.constants.ONE_MIB, zipfile.ZIP_DEFLATED |
| 243 | + ) |
240 | 244 |
|
241 | 245 | assert not legacy._is_valid_dist_file(f, "") |
242 | 246 |
|
@@ -1626,8 +1630,8 @@ def test_upload_fails_with_too_large_project_size_default_limit( |
1626 | 1630 | EmailFactory.create(user=user) |
1627 | 1631 | project = ProjectFactory.create( |
1628 | 1632 | name="foobar", |
1629 | | - upload_limit=legacy.MAX_FILESIZE, |
1630 | | - total_size=legacy.MAX_PROJECT_SIZE - 1, |
| 1633 | + upload_limit=warehouse.constants.MAX_FILESIZE, |
| 1634 | + total_size=warehouse.constants.MAX_PROJECT_SIZE - 1, |
1631 | 1635 | ) |
1632 | 1636 | release = ReleaseFactory.create(project=project, version="1.0") |
1633 | 1637 | RoleFactory.create(user=user, project=project) |
@@ -1673,10 +1677,13 @@ def test_upload_fails_with_too_large_project_size_custom_limit( |
1673 | 1677 | one_megabyte = 1 * 1024 * 1024 |
1674 | 1678 | project = ProjectFactory.create( |
1675 | 1679 | name="foobar", |
1676 | | - upload_limit=legacy.MAX_FILESIZE, |
1677 | | - total_size=legacy.MAX_PROJECT_SIZE, |
1678 | | - total_size_limit=legacy.MAX_PROJECT_SIZE |
1679 | | - + one_megabyte, # Custom Limit for the project |
| 1680 | + upload_limit=warehouse.constants.MAX_FILESIZE, |
| 1681 | + total_size=warehouse.constants.MAX_PROJECT_SIZE, |
| 1682 | + total_size_limit=( |
| 1683 | + warehouse.constants.MAX_PROJECT_SIZE |
| 1684 | + + one_megabyte |
| 1685 | + # Custom Limit for the project |
| 1686 | + ), |
1680 | 1687 | ) |
1681 | 1688 | release = ReleaseFactory.create(project=project, version="1.0") |
1682 | 1689 | RoleFactory.create(user=user, project=project) |
@@ -1726,10 +1733,11 @@ def test_upload_succeeds_custom_project_size_limit( |
1726 | 1733 | one_megabyte = 1 * 1024 * 1024 |
1727 | 1734 | project = ProjectFactory.create( |
1728 | 1735 | name="foobar", |
1729 | | - upload_limit=legacy.MAX_FILESIZE, |
1730 | | - total_size=legacy.MAX_PROJECT_SIZE, |
1731 | | - total_size_limit=legacy.MAX_PROJECT_SIZE |
1732 | | - + (one_megabyte * 60), # Custom Limit for the project |
| 1736 | + upload_limit=warehouse.constants.MAX_FILESIZE, |
| 1737 | + total_size=warehouse.constants.MAX_PROJECT_SIZE, |
| 1738 | + total_size_limit=( |
| 1739 | + warehouse.constants.MAX_PROJECT_SIZE + (one_megabyte * 60) |
| 1740 | + ), # Custom Limit for the project |
1733 | 1741 | ) |
1734 | 1742 | release = ReleaseFactory.create(project=project, version="1.0") |
1735 | 1743 | RoleFactory.create(user=user, project=project) |
@@ -2167,7 +2175,7 @@ def test_upload_fails_with_invalid_filetype( |
2167 | 2175 | }[filetype], |
2168 | 2176 | "content": pretend.stub( |
2169 | 2177 | filename=filename, |
2170 | | - file=io.BytesIO(b"a" * (legacy.MAX_FILESIZE + 1)), |
| 2178 | + file=io.BytesIO(b"a" * (warehouse.constants.MAX_FILESIZE + 1)), |
2171 | 2179 | type="application/tar", |
2172 | 2180 | ), |
2173 | 2181 | } |
@@ -2205,7 +2213,7 @@ def test_upload_fails_with_invalid_extension(self, pyramid_config, db_request): |
2205 | 2213 | "md5_digest": "nope!", |
2206 | 2214 | "content": pretend.stub( |
2207 | 2215 | filename=filename, |
2208 | | - file=io.BytesIO(b"a" * (legacy.MAX_FILESIZE + 1)), |
| 2216 | + file=io.BytesIO(b"a" * (warehouse.constants.MAX_FILESIZE + 1)), |
2209 | 2217 | type="application/tar", |
2210 | 2218 | ), |
2211 | 2219 | } |
@@ -2246,7 +2254,7 @@ def test_upload_fails_with_unsafe_filename( |
2246 | 2254 | "md5_digest": "nope!", |
2247 | 2255 | "content": pretend.stub( |
2248 | 2256 | filename=filename, |
2249 | | - file=io.BytesIO(b"a" * (legacy.MAX_FILESIZE + 1)), |
| 2257 | + file=io.BytesIO(b"a" * (warehouse.constants.MAX_FILESIZE + 1)), |
2250 | 2258 | type="application/tar", |
2251 | 2259 | ), |
2252 | 2260 | } |
@@ -2283,7 +2291,7 @@ def test_upload_fails_with_disallowed_in_filename( |
2283 | 2291 | "md5_digest": "nope!", |
2284 | 2292 | "content": pretend.stub( |
2285 | 2293 | filename=filename, |
2286 | | - file=io.BytesIO(b"a" * (legacy.MAX_FILESIZE + 1)), |
| 2294 | + file=io.BytesIO(b"a" * (warehouse.constants.MAX_FILESIZE + 1)), |
2287 | 2295 | type="application/tar", |
2288 | 2296 | ), |
2289 | 2297 | } |
@@ -2322,7 +2330,7 @@ def test_upload_fails_without_user_permission(self, pyramid_config, db_request): |
2322 | 2330 | "md5_digest": "nope!", |
2323 | 2331 | "content": pretend.stub( |
2324 | 2332 | filename=filename, |
2325 | | - file=io.BytesIO(b"a" * (legacy.MAX_FILESIZE + 1)), |
| 2333 | + file=io.BytesIO(b"a" * (warehouse.constants.MAX_FILESIZE + 1)), |
2326 | 2334 | type="application/tar", |
2327 | 2335 | ), |
2328 | 2336 | } |
@@ -2364,7 +2372,7 @@ def test_upload_fails_without_oidc_publisher_permission( |
2364 | 2372 | "md5_digest": "nope!", |
2365 | 2373 | "content": pretend.stub( |
2366 | 2374 | filename=filename, |
2367 | | - file=io.BytesIO(b"a" * (legacy.MAX_FILESIZE + 1)), |
| 2375 | + file=io.BytesIO(b"a" * (warehouse.constants.MAX_FILESIZE + 1)), |
2368 | 2376 | type="application/tar", |
2369 | 2377 | ), |
2370 | 2378 | } |
|
0 commit comments