# -*- coding: utf-8 -*- from s3pathlib.aws import context from s3pathlib.core.resolve_s3_client import resolve_s3_client from s3pathlib.tests import run_cov_test from s3pathlib.tests.mock import BaseTest class ResolveS3Client(BaseTest): def _test_resolve_s3_client(self): context.attach_boto_session(boto_ses=self.bsm.boto_ses) assert context.aws_account_id == self.bsm.aws_account_id assert context.aws_region == self.bsm.aws_region s3_client_1 = resolve_s3_client(context, None) s3_client_2 = resolve_s3_client(context, self.bsm) assert id(s3_client_1) != id(s3_client_2) def test(self): self._test_resolve_s3_client() class Test(ResolveS3Client): use_mock = False class TestUseMock(ResolveS3Client): use_mock = True if __name__ == "__main__": run_cov_test(__file__, module="s3pathlib.core.resolve_s3_client", preview=False)