def test_clamav_error():
responses.add(
responses.POST,
"http://ckan.example.com/api/3/action/resource_show",
status=200,
json={
"result": {
"url_type": "upload",
"url": "http://ckan.example.com/foobar.csv",
}
},
)
responses.add(
responses.GET,
"http://ckan.example.com/foobar.csv",
status=200,
body="a,b,c",
)
jobs.update_clamav_definitions = mock.Mock()
jobs.scan_file = mock.Mock(
return_value=mock.MagicMock(returncode=2, stdout=b"oh no"))
with pytest.raises(util.JobError) as excinfo:
jobs.scan("fake-id", test_payload)
assert (
str(excinfo.value) ==
'{"status_code": 2, "description": "oh no", "status_text": "SCAN FAILED"}'
)
def test_failed_resource_show():
responses.add(
responses.POST,
"http://ckan.example.com/api/3/action/resource_show",
status=500,
)
with pytest.raises(util.JobError) as excinfo:
jobs.scan("fake-id", test_payload)
assert "500 Server Error" in str(excinfo.value)
def test_resource_is_not_upload():
responses.add(
responses.POST,
"http://ckan.example.com/api/3/action/resource_show",
status=200,
json={"result": {
"url_type": "datastore",
}},
)
with pytest.raises(util.JobError) as excinfo:
jobs.scan("fake-id", test_payload)
assert (
str(excinfo.value) ==
"Only resources of type 'upload' can be scanned. Received 'datastore'")
def test_completed_scan():
responses.add(
responses.POST,
"http://ckan.example.com/api/3/action/resource_show",
status=200,
json={
"result": {
"url_type": "upload",
"url": "http://ckan.example.com/foobar.csv",
}
},
)
responses.add(
responses.GET,
"http://ckan.example.com/foobar.csv",
status=200,
body="a,b,c",
)
jobs.scan_file = mock.Mock(return_value=mock.MagicMock(
returncode=0, stdout=b"/tmp/tmp37q_kv9u: OK..."))
response = jobs.scan("fake-id", test_payload)
jobs.scan_file.assert_called_once()
assert {
"status_code": 0,
"status_text": "SUCCESSFUL SCAN, FILE CLEAN",
"description": "/tmp/tmp37q_kv9u: OK...",
} == response
def test_invalid_scheme():
responses.add(
responses.POST,
"http://ckan.example.com/api/3/action/resource_show",
status=200,
json={
"result": {
"url_type": "upload",
"url": "git://github.com/okfn/ckan-clamav-service.git",
}
},
)
with pytest.raises(util.JobError) as excinfo:
jobs.scan("fake-id", test_payload)
assert str(
excinfo.value) == "Only http, https, and ftp resources may be fetched."
def test_failed_resource_download():
responses.add(
responses.POST,
"http://ckan.example.com/api/3/action/resource_show",
status=200,
json={
"result": {
"url_type": "upload",
"url": "http://ckan.example.com/foobar.csv",
}
},
)
responses.add(
responses.GET,
"http://ckan.example.com/foobar.csv",
status=500,
)
with pytest.raises(util.JobError) as excinfo:
jobs.scan("fake-id", test_payload)
assert "500 Server Error" in str(excinfo.value)
def test_invalid_payload():
with pytest.raises(util.JobError) as excinfo:
jobs.scan("fake-id", {})
assert str(excinfo.value) == "Metadata missing"
with pytest.raises(util.JobError) as excinfo:
jobs.scan("fake-id", {"metadata": {}})
assert str(excinfo.value) == "No id provided."
with pytest.raises(util.JobError) as excinfo:
jobs.scan("fake-id",
{"metadata": {
"ckan_url": "http://ckan.example.com"
}})
assert str(excinfo.value) == "No id provided."
with pytest.raises(util.JobError) as excinfo:
jobs.scan(
"fake-id",
{"metadata": {
"resource_id": "fake-resource",
}},
)
assert str(excinfo.value) == "No ckan_url provided."
with pytest.raises(util.JobError) as excinfo:
jobs.scan(
"fake-id",
{
"metadata": {
"ckan_url": "http://ckan.example.com",
"resource_id": "fake-resource",
},
},
)
assert str(excinfo.value) == "No CKAN API key provided"