|
23 | 23 | import mock |
24 | 24 | import pytest |
25 | 25 | from botocore import stub |
26 | | -from moto import mock_dynamodb2, mock_kinesis, mock_sts |
| 26 | +from moto import mock_dynamodb, mock_kinesis, mock_sts |
27 | 27 | from shared_util import custom_boto_config |
28 | 28 |
|
29 | 29 |
|
@@ -186,7 +186,7 @@ def lambda_event_bus_event(): |
186 | 186 |
|
187 | 187 |
|
188 | 188 | @mock_sts |
189 | | -@mock_dynamodb2 |
| 189 | +@mock_dynamodb |
190 | 190 | def test_invoke_lambda_for_ddb_config(get_event_bus_stubber): |
191 | 191 | lambda_event = create_cw_schedule_event() |
192 | 192 | create_ddb_table_for_US_en() |
@@ -215,7 +215,7 @@ def test_invoke_lambda_for_ddb_config(get_event_bus_stubber): |
215 | 215 |
|
216 | 216 |
|
217 | 217 | @mock_sts |
218 | | -@mock_dynamodb2 |
| 218 | +@mock_dynamodb |
219 | 219 | @mock.patch.dict(os.environ, {"CONFIG_PARAM": '{"country":"US", "language":"en"}'}) |
220 | 220 | @mock.patch.dict(os.environ, {"SEARCH_QUERY": "fakequery"}) |
221 | 221 | def test_invoke_lambda_for_json_str(get_event_bus_stubber): |
@@ -245,7 +245,7 @@ def test_invoke_lambda_for_json_str(get_event_bus_stubber): |
245 | 245 |
|
246 | 246 |
|
247 | 247 | @mock_sts |
248 | | -@mock_dynamodb2 |
| 248 | +@mock_dynamodb |
249 | 249 | def test_invoke_lambda_for_ddb_config_with_failed_count(get_event_bus_stubber): |
250 | 250 | lambda_event = create_cw_schedule_event() |
251 | 251 | create_ddb_table_for_US_en() |
@@ -274,7 +274,7 @@ def test_invoke_lambda_for_ddb_config_with_failed_count(get_event_bus_stubber): |
274 | 274 |
|
275 | 275 |
|
276 | 276 | @mock_sts |
277 | | -@mock_dynamodb2 |
| 277 | +@mock_dynamodb |
278 | 278 | @mock_kinesis |
279 | 279 | def test_invoke_process_config_handler(): |
280 | 280 | lambda_event = create_event_bus_consumer_event() |
|
0 commit comments