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']
}
}
