diff --git a/discord_handler/dev_requirements.txt b/discord_handler/dev_requirements.txt index 4f98922..c5af9c4 100644 --- a/discord_handler/dev_requirements.txt +++ b/discord_handler/dev_requirements.txt @@ -1,3 +1,2 @@ black -moto - +moto ~= 3.1 diff --git a/discord_handler/test_filmbot.py b/discord_handler/test_filmbot.py index 38932de..ff9e171 100644 --- a/discord_handler/test_filmbot.py +++ b/discord_handler/test_filmbot.py @@ -2,7 +2,7 @@ import boto3 from math import factorial from itertools import permutations -from moto import mock_dynamodb2 +from moto import mock_dynamodb from filmbot import ( FilmBot, TABLE_NAME, @@ -105,7 +105,7 @@ def __exit__(self, exc_type, exc_value, exc_tb): class TestFilmBot(unittest.TestCase): - mock_dynamodb2 = mock_dynamodb2() + mock_dynamodb = mock_dynamodb() def setUp(self): """ @@ -115,7 +115,7 @@ def setUp(self): # Set unlimited length for assertEqual diff lengths self.maxDiff = None - self.mock_dynamodb2.start() + self.mock_dynamodb.start() boto3.setup_default_session() self.dynamodb = boto3.resource("dynamodb", region_name=AWS_REGION) self.dynamodb_client = boto3.client("dynamodb", region_name=AWS_REGION) @@ -134,7 +134,7 @@ def tearDown(self): """ Unmock `dynamodb2`. """ - self.mock_dynamodb2.stop() + self.mock_dynamodb.stop() pass def test_get_users(self):