replace deprecated mock_s3 with mock_aws

spacetelescope · Apr 2, 2024 · a55306f · a55306f
1 parent dad66ab
commit a55306f
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/tests/skopes/hst/cal/test_hst_cal_predict.py b/tests/skopes/hst/cal/test_hst_cal_predict.py
@@ -1,6 +1,6 @@
 import os
 from pytest import mark
-from moto import mock_s3
+from moto import mock_aws
 from spacekit.skopes.hst.cal.predict import local_handler, lambda_handler
 import boto3
 
@@ -27,7 +27,7 @@ def put_moto_s3_object(dataset, bucketname):
 @mark.cal
 @mark.predict
 @mark.parametrize("pipeline", [("asn")])
-@mock_s3
+@mock_aws
 def test_local_predict_handler(hst_cal_predict_visits, pipeline):
     bucketname = "spacekit_bucket"
     s3 = boto3.resource("s3", region_name="us-east-1")
@@ -43,7 +43,7 @@ def test_local_predict_handler(hst_cal_predict_visits, pipeline):
 @mark.cal
 @mark.predict
 @mark.parametrize("pipeline", [("asn")])
-@mock_s3
+@mock_aws
 def test_local_handler_multiple(hst_cal_predict_visits, pipeline):
     bucketname = "spacekit_bucket"
     s3 = boto3.resource("s3", region_name="us-east-1")
@@ -62,7 +62,7 @@ def test_local_handler_multiple(hst_cal_predict_visits, pipeline):
 @mark.cal
 @mark.predict
 @mark.parametrize("pipeline", [("asn")])
-@mock_s3
+@mock_aws
 def test_lambda_handler(hst_cal_predict_visits, pipeline):
     bucketname = "spacekit_bucket"
     s3 = boto3.resource("s3", region_name="us-east-1")