def get_hook(self):
"""Create and return AWSDataSyncHook.
:return AWSDataSyncHook: An AWSDataSyncHook instance.
"""
if not self.hook:
self.hook = AWSDataSyncHook(aws_conn_id=self.aws_conn_id, )
return self.hook
def get_hook(self):
"""Create and return AWSDataSyncHook.
:return AWSDataSyncHook: An AWSDataSyncHook instance.
"""
if not self.hook:
self.hook = AWSDataSyncHook(
aws_conn_id=self.aws_conn_id,
wait_interval_seconds=self.wait_interval_seconds)
return self.hook
def setUp(self):
self.client = boto3.client("datasync", region_name="us-east-1")
self.hook = AWSDataSyncHook(aws_conn_id="aws_default", wait_interval_seconds=0)
# Create default locations and tasks
self.source_location_arn = self.client.create_location_smb(
ServerHostname=self.source_server_hostname,
Subdirectory=self.source_subdirectory,
User="",
Password="",
AgentArns=["stuff"],
)["LocationArn"]
self.destination_location_arn = self.client.create_location_s3(
S3BucketArn=f"arn:aws:s3:::{self.destination_bucket_name}",
Subdirectory=self.destination_bucket_dir,
S3Config={"BucketAccessRoleArn": "role"},
)["LocationArn"]
self.task_arn = self.client.create_task(
SourceLocationArn=self.source_location_arn,
DestinationLocationArn=self.destination_location_arn,
)["TaskArn"]
def setUp(self):
self.client = boto3.client("datasync", region_name="us-east-1")
self.hook = AWSDataSyncHook(aws_conn_id='aws_default', wait_interval_seconds=0)
# Create default locations and tasks
self.source_location_arn = self.client.create_location_smb(
ServerHostname=self.source_server_hostname,
Subdirectory=self.source_subdirectory,
User='',
Password='',
AgentArns=['stuff']
)['LocationArn']
self.destination_location_arn = self.client.create_location_s3(
S3BucketArn='arn:aws:s3:::{0}'.format(
self.destination_bucket_name),
Subdirectory=self.destination_bucket_dir,
S3Config={'BucketAccessRoleArn': 'role'}
)['LocationArn']
self.task_arn = self.client.create_task(
SourceLocationArn=self.source_location_arn,
DestinationLocationArn=self.destination_location_arn
)['TaskArn']
def test_get_conn(self):
hook = AWSDataSyncHook(aws_conn_id='aws_default')
self.assertIsNotNone(hook.get_conn())
def test_get_conn(self):
hook = AWSDataSyncHook(aws_conn_id="aws_default")
assert hook.get_conn() is not None