Skip to content

Commit dc0ee8d

Browse files
committed
Mock s3 uploads
1 parent 64f235a commit dc0ee8d

File tree

1 file changed

+17
-2
lines changed

1 file changed

+17
-2
lines changed

tests/test_ci.py

Lines changed: 17 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -6,17 +6,18 @@
66
from docker.models.containers import Container
77
import chromedriver_autoinstaller
88
from docker import DockerClient
9+
from moto import mock_aws
910

1011
from ci.ci import CI, SetEnvs
1112

12-
os.environ["DRY_RUN"] = "true"
13+
os.environ["DRY_RUN"] = "false"
1314
os.environ["IMAGE"] = "linuxserver/test"
1415
os.environ["BASE"] = "alpine"
1516
os.environ["ACCESS_KEY"] = "secret-access-key"
1617
os.environ["SECRET_KEY"] = "secret-key"
1718
os.environ["META_TAG"] = "test-meta-tag"
1819
os.environ["TAGS"] = "amd64-nightly-5.10.1.9109-ls85|arm64v8-nightly-5.10.1.9109-ls85"
19-
os.environ["CI_LOG_LEVEL"] = "DEBUG"
20+
os.environ["CI_LOG_LEVEL"] = "ERROR"
2021
os.environ["NODE_NAME"] = "test-node"
2122
os.environ["SSL"] = "true"
2223
os.environ["PORT"] = "443"
@@ -159,3 +160,17 @@ def test_badge_render(ci:CI):
159160
def test_generate_sbom(ci:CI, syft_mock_container:Mock, sbom_blob:bytes):
160161
sbom = ci.generate_sbom(ci.tags[0])
161162
assert "VERSION" in sbom
163+
164+
def test_create_s3_client(ci:CI):
165+
with mock_aws():
166+
ci.s3_client = ci.create_s3_client()
167+
assert ci.s3_client is not None
168+
169+
def test_upload_file(ci: CI) -> None:
170+
with mock_aws():
171+
# Create the mock S3 client
172+
ci.s3_client = ci.create_s3_client()
173+
# Create the bucket
174+
ci.s3_client.create_bucket(Bucket=ci.bucket)
175+
# Upload a file to the bucket
176+
ci.upload_file("tests/log_blob.log", "log_blob.log", {"ContentType": "text/plain", "ACL": "public-read"})

0 commit comments

Comments
 (0)