Skip to content

Commit 4d835dd

Browse files
committed
delate aimd and csld related codes
1 parent dce7e2d commit 4d835dd

File tree

7 files changed

+9
-878
lines changed

7 files changed

+9
-878
lines changed

atomate/utils/tests/test_database.py

+5-5
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,7 @@
66

77
import boto3
88
from maggma.stores import MemoryStore
9-
from moto import mock_s3
9+
from moto import mock_aws
1010

1111
__author__ = "Jimmy Shen <[email protected]>"
1212

@@ -37,7 +37,7 @@ def tearDownClass(cls):
3737
cls.testdb.connection.drop_database(cls.testdb.db_name)
3838

3939
def test_s3_valid(self):
40-
with mock_s3():
40+
with mock_aws():
4141
conn = boto3.resource("s3", region_name="us-east-1")
4242
conn.create_bucket(Bucket="test_bucket")
4343
index_store = MemoryStore()
@@ -50,7 +50,7 @@ def test_s3_valid(self):
5050
self.assertEqual(res["data"], "111111111110111111")
5151

5252
def test_s3_not_valid(self):
53-
with mock_s3():
53+
with mock_aws():
5454
conn = boto3.resource("s3", region_name="us-east-1")
5555
conn.create_bucket(Bucket="test_bucket_2")
5656
index_store = MemoryStore()
@@ -61,7 +61,7 @@ def test_s3_not_valid(self):
6161
store.connect()
6262

6363
def test_maggma_store_names(self):
64-
with mock_s3():
64+
with mock_aws():
6565
conn = boto3.resource("s3", region_name="us-east-1")
6666
conn.create_bucket(Bucket="test_bucket")
6767
index_store = MemoryStore()
@@ -84,7 +84,7 @@ def test_uri(self):
8484
calc_db.collection.insert_one({"task_id": "mp-1", "data": "12345"})
8585
self.assertEqual(calc_db.collection.find_one()["data"], "12345")
8686

87-
with mock_s3():
87+
with mock_aws():
8888
conn = boto3.resource("s3", region_name="us-east-1")
8989
conn.create_bucket(Bucket="test_bucket")
9090
uri_db = TestToDb.from_db_file(db_dir + "/db_aws_uri.json")

0 commit comments

Comments
 (0)