fixing s3 tests

base2Services · Nov 20, 2018 · baa7347 · baa7347
1 parent b0d790a
commit baa7347
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/shelvery_tests/s3data_integration_test.py b/shelvery_tests/s3data_integration_test.py
@@ -123,7 +123,7 @@ def test_CreateBackupData(self):
                 # verify the s3 data
                 account_id = ebs_backups_engine.account_id
                 s3path = f"{S3_DATA_PREFIX}/{ebs_backups_engine.get_engine_type()}/{engine_backup.name}.yaml"
-                s3bucket = ShelveryEngine.get_local_bucket_name()
+                s3bucket = ebs_backups_engine.get_local_bucket_name()
                 print(f"Usingbucket {s3bucket}")
                 print(f"Using path {s3path}")
                 bucket = boto3.resource('s3').Bucket(s3bucket)
@@ -163,7 +163,7 @@ def test_CreateSharingInfo(self):
             if backup.entity_id == self.volume['VolumeId']:
                 account_id = ebs_backups_engine.account_id
                 s3path = f"{S3_DATA_PREFIX}/shared/{self.share_with_id}/{ebs_backups_engine.get_engine_type()}/{backup.name}.yaml"
-                s3bucket = ShelveryEngine.get_local_bucket_name()
+                s3bucket = ebs_backups_engine.get_local_bucket_name()
                 bucket = boto3.resource('s3').Bucket(s3bucket)
                 object = bucket.Object(s3path)
                 content = object.get()['Body'].read()
@@ -214,7 +214,7 @@ def test_CleanBackupData(self):
 
                 account_id = ebs_backups_engine.account_id
                 s3path = f"{S3_DATA_PREFIX}/{ebs_backups_engine.get_engine_type()}/removed/{backup.name}.yaml"
-                s3bucket = ShelveryEngine.get_local_bucket_name()
+                s3bucket = ebs_backups_engine.get_local_bucket_name()
                 bucket = boto3.resource('s3').Bucket(s3bucket)
                 object = bucket.Object(s3path)
                 content = object.get()['Body'].read()