wehub-resource-sync
28 lines
613 B
Python
28 lines
613 B
Python
# Copyright (C) CVAT.ai Corporation
|
|
#
|
|
# SPDX-License-Identifier: MIT
|
|
|
|
from functools import partial
|
|
from pathlib import PurePosixPath
|
|
|
|
import pytest
|
|
|
|
import shared.utils.s3 as s3
|
|
|
|
|
|
@pytest.fixture
|
|
def fxt_uploaded_s3_file(request: pytest.FixtureRequest):
|
|
def upload_file(s3_client: s3.S3Client, *, path: PurePosixPath | str, data: bytes):
|
|
s3_client.create_file(
|
|
data=data,
|
|
filename=str(path),
|
|
)
|
|
request.addfinalizer(
|
|
partial(
|
|
s3_client.remove_file,
|
|
filename=str(path),
|
|
)
|
|
)
|
|
|
|
return upload_file
|