def setUp(self) -> None: self.aws_mock = MagicMock() self.client = MagicMock() self.client.put_object = Mock() bucket_policy = MagicMock() bucket_policy.policy = None bucket = Mock() bucket.upload_file = Mock() bucket.objects = Mock() self.bucket = bucket self.upload_file = bucket.upload_file self.download_file = bucket.download_file resource = MagicMock() resource.BucketPolicy = Mock(return_value=bucket_policy) resource.Bucket = Mock(return_value=bucket) session = MagicMock() session.client = Mock(return_value=self.client) session.resource = Mock(return_value=resource) self.aws_mock.is_user = False self.aws_mock.common_session = session self.aws_mock.bucket_name = 'bucket-name' self.aws_mock.new_session.return_value = session
def setUp(self) -> None: self.aws_mock = MagicMock() self.client = MagicMock() self.client.put_object = Mock() bucket_policy = MagicMock() bucket_policy.policy = None bucket = Mock() bucket.upload_file = Mock() self.upload_file = bucket.upload_file resource = MagicMock() resource.BucketPolicy = Mock(return_value=bucket_policy) resource.Bucket = Mock(return_value=bucket) session = MagicMock() session.client = Mock(return_value=self.client) session.resource = Mock(return_value=resource) self.aws_mock.is_user = False self.aws_mock.common_session = session self.aws_mock.bucket_name = 'bucket-name' self.aws_mock.new_session.return_value = session self.path_map = { 'file0': { 'isfile': True, 'isdir': False, 'getsize': 2 }, 'dir1': { 'isfile': False, 'isdir': True, 'listdir': ['file1', 'file2', '.file', '__file', 'dir2'] }, 'dir1/file1': { 'isfile': True, 'isdir': False, 'getsize': 10 }, 'dir1/file2': { 'isfile': True, 'isdir': False, 'getsize': 0 }, 'dir1/dir2/file3': { 'isfile': True, 'isdir': False, 'getsize': 5 }, 'dir1/dir2': { 'isfile': False, 'isdir': True, 'listdir': ['file3'] } }