Skip to content

test_index_reference_data#

mock_importers#

Source code in tests/unit/apps/refdata/commands/test_index_reference_data.py 
54
55
56
57
58
59
60
61
62
63
64
@pytest.fixture
def mock_importers(mocker):
    return {
        "Finto": mocker.patch("apps.refdata.services.indexer.FintoImporter"),
        "FintoLocation": mocker.patch("apps.refdata.services.indexer.FintoLocationImporter"),
        "LocalJSON": mocker.patch("apps.refdata.services.indexer.LocalJSONImporter"),
        "LocalJSONFileFormatVersion": mocker.patch(
            "apps.refdata.services.indexer.LocalJSONFileFormatVersionImporter"
        ),
        "LocalJSONLicense": mocker.patch("apps.refdata.services.indexer.LocalJSONLicenseImporter"),
    }

test_import_all#

Source code in tests/unit/apps/refdata/commands/test_index_reference_data.py 
 67
 68
 69
 70
 71
 72
 73
 74
 75
 76
 77
 78
 79
 80
 81
 82
 83
 84
 85
 86
 87
 88
 89
 90
 91
 92
 93
 94
 95
 96
 97
 98
 99
100
101
102
103
104
105
106
@override_settings(REFERENCE_DATA_SOURCES=TEST_DATA_SOURCES)
def test_import_all(mock_importers):
    call_command("index_reference_data")
    mock_importers["Finto"].assert_has_calls(
        [
            call(
                model=FieldOfScience,
                source=TEST_DATA_SOURCES["field_of_science"]["source"],
                scheme=None,
            ),
            call(
                model=Theme,
                source=TEST_DATA_SOURCES["theme"]["source"],
                scheme=None,
            ),
        ]
    )
    mock_importers["FintoLocation"].assert_called_once_with(
        model=Location,
        source=TEST_DATA_SOURCES["location"]["source"],
        scheme=None,
    )
    mock_importers["LocalJSON"].assert_called_once_with(
        model=AccessType,
        source=TEST_DATA_SOURCES["access_type"]["source"],
        scheme=TEST_DATA_SOURCES["access_type"]["scheme"],
    )
    mock_importers["LocalJSONLicense"].assert_called_once_with(
        model=License,
        source=TEST_DATA_SOURCES["license"]["source"],
        scheme=TEST_DATA_SOURCES["license"]["scheme"],
    )
    mock_importers["LocalJSONFileFormatVersion"].assert_called_once_with(
        model=FileFormatVersion,
        source=TEST_DATA_SOURCES["file_format_version"]["source"],
        scheme=TEST_DATA_SOURCES["file_format_version"]["scheme"],
    )
    for importer in mock_importers.values():
        # Check .load() is called for an instance of each importer type
        importer.return_value.load.assert_called()

test_import_specific#

Source code in tests/unit/apps/refdata/commands/test_index_reference_data.py 
109
110
111
112
113
114
115
116
117
118
119
120
121
@override_settings(REFERENCE_DATA_SOURCES=TEST_DATA_SOURCES)
def test_import_specific(mock_importers):
    call_command("index_reference_data", "field_of_science", "location")
    mock_importers["Finto"].assert_called_once_with(
        model=FieldOfScience,
        source=TEST_DATA_SOURCES["field_of_science"]["source"],
        scheme=None,
    )
    mock_importers["FintoLocation"].assert_called_once_with(
        model=Location, source=TEST_DATA_SOURCES["location"]["source"], scheme=None
    )
    mock_importers["Finto"].return_value.load.assert_called()
    mock_importers["FintoLocation"].return_value.load.assert_called()

test_import_unknown_type#

Source code in tests/unit/apps/refdata/commands/test_index_reference_data.py 
124
125
126
127
@override_settings(REFERENCE_DATA_SOURCES=TEST_DATA_SOURCES)
def test_import_unknown_type():
    with pytest.raises(CommandError):
        call_command("index_reference_data", "field_of_science", "something")

test_import_duplicate_type#

Source code in tests/unit/apps/refdata/commands/test_index_reference_data.py 
130
131
132
133
@override_settings(REFERENCE_DATA_SOURCES=TEST_DATA_SOURCES)
def test_import_duplicate_type():
    with pytest.raises(CommandError):
        call_command("index_reference_data", "field_of_science", "field_of_science")