diff --git a/nhsn/delphi_nhsn/pull.py b/nhsn/delphi_nhsn/pull.py index 31164770e..855c0952f 100644 --- a/nhsn/delphi_nhsn/pull.py +++ b/nhsn/delphi_nhsn/pull.py @@ -38,7 +38,7 @@ def check_last_updated(socrata_token, dataset_id, logger): client = Socrata("data.cdc.gov", socrata_token) response = client.get_metadata(dataset_id) - updated_timestamp = datetime.utcfromtimestamp(int(response["rowsUpdatedAt"])) + updated_timestamp = datetime.utcfromtimestamp(int(response["viewLastModified"])) now = datetime.utcnow() recently_updated_source = (now - updated_timestamp) < timedelta(days=1) diff --git a/nhsn/tests/conftest.py b/nhsn/tests/conftest.py index b321f1236..794372464 100644 --- a/nhsn/tests/conftest.py +++ b/nhsn/tests/conftest.py @@ -72,6 +72,6 @@ def side_effect(*args, **kwargs): else: return [] mock_get.side_effect = side_effect - mock_get_metadata.return_value = {"rowsUpdatedAt": time.time()} + mock_get_metadata.return_value = {"viewLastModified": time.time()} run_module(params) diff --git a/nhsn/tests/test_pull.py b/nhsn/tests/test_pull.py index f8e27868b..ec2de51ef 100644 --- a/nhsn/tests/test_pull.py +++ b/nhsn/tests/test_pull.py @@ -86,7 +86,7 @@ def test_pull_nhsn_data_output(self, mock_create_backup, mock_socrata, dataset, mock_client = MagicMock() mock_socrata.return_value = mock_client mock_client.get.side_effect = [dataset["test_data"],[]] - mock_client.get_metadata.return_value = {"rowsUpdatedAt": now} + mock_client.get_metadata.return_value = {"viewLastModified": now} backup_dir = params["common"]["backup_dir"] test_token = params["indicator"]["socrata_token"] @@ -123,7 +123,7 @@ def test_pull_nhsn_data_backup(self, mock_socrata, dataset, caplog, params): mock_socrata.return_value = mock_client mock_client.get.side_effect = [dataset["test_data"], []] - mock_client.get_metadata.return_value = {"rowsUpdatedAt": now} + mock_client.get_metadata.return_value = {"viewLastModified": now} today = pd.Timestamp.today().strftime("%Y%m%d") backup_dir = params["common"]["backup_dir"] @@ -163,7 +163,7 @@ def test_pull_nhsn_data_backup(self, mock_socrata, dataset, caplog, params): def test_check_last_updated(self, mock_socrata, dataset, updatedAt, caplog): mock_client = MagicMock() mock_socrata.return_value = mock_client - mock_client.get_metadata.return_value = {"rowsUpdatedAt": updatedAt } + mock_client.get_metadata.return_value = {"viewLastModified": updatedAt } logger = get_structured_logger() check_last_updated(mock_client, dataset["id"], logger)