Skip to content

test_migrate_v2_datasets#

integration_settings#

Source code in tests/unit/apps/core/management/test_migrate_v2_datasets.py 
14
15
16
17
18
19
@pytest.fixture(autouse=True)
def integration_settings(settings):
    settings.METAX_V2_HOST = "https://metax-v2-test"
    settings.METAX_V2_USER = "metax-v3-user"
    settings.METAX_V2_PASSWORD = "metax-v3-password"
    return settings

test_migrate_command#

Source code in tests/unit/apps/core/management/test_migrate_v2_datasets.py 
22
23
24
25
26
27
def test_migrate_command(mock_response, reference_data):
    out = StringIO()
    err = StringIO()
    call_command("migrate_v2_datasets", stdout=out, stderr=err, use_env=True, allow_fail=True)
    assert "10 datasets updated succesfully" in out.getvalue().strip()
    assert "Invalid identifier 'invalid', ignoring" in err.getvalue()

test_migrate_command_error#

Source code in tests/unit/apps/core/management/test_migrate_v2_datasets.py 
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
def test_migrate_command_error(mock_response, reference_data):
    out = StringIO()
    err = StringIO()
    call_command(
        "migrate_v2_datasets",
        stdout=out,
        stderr=err,
        use_env=True,
        catalogs=["123"],
        identifiers="123 5423",
    )
    assert (
        err.getvalue().strip()
        == "The --identifiers and --catalogs options are mutually exclusive."
    )

test_migrate_command_file#

Source code in tests/unit/apps/core/management/test_migrate_v2_datasets.py 
47
48
49
50
51
52
53
54
55
56
57
def test_migrate_command_file(mock_response_single_catalog, mock_response, reference_data):
    filepath = os.path.join(
        os.path.dirname(os.path.abspath(__file__)), "legacy_metax_datasets.json"
    )
    out = StringIO()
    err = StringIO()
    call_command("migrate_v2_datasets", stdout=out, stderr=err, use_env=True, file=filepath)
    assert "Invalid identifier 'invalid', ignoring" in err.getvalue()
    output = out.getvalue()
    assert "5 datasets updated" in output
    assert "Ignored invalid legacy data:\n- research_dataset.rights_holder[0]" in output

test_migrate_command_identifier#

Source code in tests/unit/apps/core/management/test_migrate_v2_datasets.py 
60
61
62
63
64
65
66
67
68
69
70
def test_migrate_command_identifier(mock_response_single, reference_data):
    out = StringIO()
    err = StringIO()
    call_command(
        "migrate_v2_datasets",
        stdout=out,
        stderr=err,
        use_env=True,
        identifiers=["c955e904-e3dd-4d7e-99f1-3fed446f96d1"],
    )
    assert "1 datasets updated succesfully" in out.getvalue().strip()

test_migrate_command_identifier_error_in_data#

Source code in tests/unit/apps/core/management/test_migrate_v2_datasets.py 
73
74
75
76
77
78
79
80
81
82
83
def test_migrate_command_identifier_error_in_data(mock_response_single_invalid, reference_data):
    out = StringIO()
    err = StringIO()
    with pytest.raises(ValueError):
        call_command(
            "migrate_v2_datasets",
            stdout=out,
            stderr=err,
            use_env=True,
            identifiers=["c955e904-e3dd-4d7e-99f1-3fed446f96d1"],
        )

test_migrate_command_allow_fail#

Source code in tests/unit/apps/core/management/test_migrate_v2_datasets.py 
86
87
88
89
90
91
92
93
94
95
96
97
98
99
def test_migrate_command_allow_fail(mock_response_single_invalid, reference_data):
    out = StringIO()
    err = StringIO()
    call_command(
        "migrate_v2_datasets",
        stdout=out,
        stderr=err,
        allow_fail=True,
        use_env=True,
        identifiers=["c955e904-e3dd-4d7e-99f1-3fed446f96d1"],
    )

    assert "Value is not a list: porkkana" in err.getvalue()
    assert "1 datasets failed" in out.getvalue().strip()

test_migrate_command_catalog#

Source code in tests/unit/apps/core/management/test_migrate_v2_datasets.py 
102
103
104
105
106
107
108
109
110
111
112
113
114
def test_migrate_command_catalog(mock_response_single_catalog, mock_response, reference_data):
    out = StringIO()
    err = StringIO()
    call_command(
        "migrate_v2_datasets",
        stdout=out,
        stderr=err,
        use_env=True,
        catalogs=["c955e904-e3dd-4d7e-99f1-3fed446f96d1", "unknown_catalog"],
    )
    assert "Migrating catalog: c955e904-e3dd-4d7e-99f1-3fed446f96d1" in out.getvalue()
    assert "Invalid catalog identifier: unknown_catalog" in err.getvalue()
    assert "10 datasets updated" in out.getvalue()

test_migrate_command_catalog_stop_after#

Source code in tests/unit/apps/core/management/test_migrate_v2_datasets.py 
117
118
119
120
121
122
123
124
125
126
127
128
129
130
def test_migrate_command_catalog_stop_after(
    mock_response_single_catalog, mock_response, reference_data
):
    out = StringIO()
    err = StringIO()
    call_command(
        "migrate_v2_datasets",
        stdout=out,
        stderr=err,
        use_env=True,
        stop_after=1,
        catalogs=["c955e904-e3dd-4d7e-99f1-3fed446f96d1"],
    )
    assert "1 datasets updated" in out.getvalue()

test_migrate_command_update#

Source code in tests/unit/apps/core/management/test_migrate_v2_datasets.py 
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
def test_migrate_command_update(mock_response_single, reference_data):
    call_command(
        "migrate_v2_datasets",
        identifiers=["c955e904-e3dd-4d7e-99f1-3fed446f96d1"],
        use_env=True,
    )
    out = StringIO()
    err = StringIO()
    call_command(
        "migrate_v2_datasets",
        stdout=out,
        stderr=err,
        update=True,
        identifiers=["c955e904-e3dd-4d7e-99f1-3fed446f96d1"],
    )
    assert err.getvalue() == ""
    output = out.getvalue()
    assert "Processed 1 datasets" in output
    assert "0 datasets updated succesfully" in output

test_migrate_command_update_wrong_api_version#

Source code in tests/unit/apps/core/management/test_migrate_v2_datasets.py 
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
def test_migrate_command_update_wrong_api_version(mock_response_single, reference_data):
    call_command(
        "migrate_v2_datasets",
        identifiers=["c955e904-e3dd-4d7e-99f1-3fed446f96d1"],
        use_env=True,
    )
    out = StringIO()
    err = StringIO()
    LegacyDataset.objects.filter(id="c955e904-e3dd-4d7e-99f1-3fed446f96d1").update(api_version=3)
    call_command(
        "migrate_v2_datasets",
        stdout=out,
        stderr=err,
        update=True,
        force=True,
        identifiers=["c955e904-e3dd-4d7e-99f1-3fed446f96d1"],
    )
    assert err.getvalue() == ""
    output = out.getvalue()
    assert (
        "Dataset 'c955e904-e3dd-4d7e-99f1-3fed446f96d1' has been modified in V3, not updating"
        in output
    )
    assert "Processed 1 datasets" in output
    assert "0 datasets updated succesfully" in output

test_migrate_command_update_wrong_api_version_in_v2#

Source code in tests/unit/apps/core/management/test_migrate_v2_datasets.py 
181
182
183
184
185
186
187
188
189
190
191
def test_migrate_command_update_wrong_api_version_in_v2(
    mock_response_api_version_3, reference_data
):
    out = StringIO()
    call_command(
        "migrate_v2_datasets",
        stdout=out,
        identifiers=["c955e904-e3dd-4d7e-99f1-3fed446f96d1"],
        use_env=True,
    )
    assert "from a later Metax version, ignoring" in out.getvalue()

test_migrate_command_update_draft#

Source code in tests/unit/apps/core/management/test_migrate_v2_datasets.py 
194
195
196
197
198
199
200
201
202
def test_migrate_command_update_draft(mock_response_draft, reference_data):
    out = StringIO()
    call_command(
        "migrate_v2_datasets",
        stdout=out,
        identifiers=["c955e904-e3dd-4d7e-99f1-3fed446f96d1"],
        use_env=True,
    )
    assert "is not published, ignoring" in out.getvalue()

test_migrate_command_update_file#

Source code in tests/unit/apps/core/management/test_migrate_v2_datasets.py 
205
206
207
208
def test_migrate_command_update_file(mock_response_single, reference_data):
    err = StringIO()
    call_command("migrate_v2_datasets", stderr=err, update=True, file="somefile")
    assert "The --file and --update options are mutually exclusive." in err.getvalue().strip()

test_migrate_missing_args#

Source code in tests/unit/apps/core/management/test_migrate_v2_datasets.py 
211
212
213
214
def test_migrate_missing_args():
    err = StringIO()
    call_command("migrate_v2_datasets", stderr=err)
    assert "Metax instance not specified and not using --file or --update." in err.getvalue()

test_migrate_prompt_credentials#

Source code in tests/unit/apps/core/management/test_migrate_v2_datasets.py 
217
218
219
220
221
222
223
224
225
226
227
def test_migrate_prompt_credentials(requests_mock):
    mock = requests_mock.get(url="https://metax-v2-test/rest/v2/datasets", json={"results": []})
    err = StringIO()
    with patch("builtins.input", lambda x: "username"), patch(
        "getpass.getpass", lambda x: "password"
    ):
        call_command("migrate_v2_datasets", stderr=err, use_env=True, prompt_credentials=True)
    assert err.getvalue() == ""
    assert mock.call_count == 2
    auth = mock.last_request.headers["authorization"]
    assert b64decode(auth.replace("Basic ", "")) == b"username:password"