Commit b26654e4 authored by Nicolas Wavrant's avatar Nicolas Wavrant

erp5_antivirus_clammit: create documents as a simple user

and not as manager, as some scripts in the workflow have proxy roles, so
it would be nice to test them too.
parent 34b7099c
...@@ -67,21 +67,37 @@ class TestClammitAntivirus(ERP5TypeTestCase): ...@@ -67,21 +67,37 @@ class TestClammitAntivirus(ERP5TypeTestCase):
@mock.patch("requests.request") @mock.patch("requests.request")
def test_analyse_safe_document(self, requests_request_mock): def test_analyse_safe_document(self, requests_request_mock):
requests_request_mock.side_effect = self.getResponseMock(self._SANE_HTTP_STATUS_CODE) requests_request_mock.side_effect = self.getResponseMock(self._SANE_HTTP_STATUS_CODE)
# Create a file as user
uf = self.getPortal().acl_users
user = "someuser"
uf._doAddUser(user, "", ["Author"], [])
self.loginByUserName(user)
document_value = self.portal.document_module.newContent( document_value = self.portal.document_module.newContent(
portal_type="File", portal_type="File",
data="hello_world", data="hello_world",
) )
document_value.setSuspect() document_value.setSuspect()
self.login()
self.tic() self.tic()
self.assertEqual(document_value.getScanState(), "safe") self.assertEqual(document_value.getScanState(), "safe")
@mock.patch("requests.request") @mock.patch("requests.request")
def test_analysed_infected_document(self, requests_request_mock): def test_analysed_infected_document(self, requests_request_mock):
requests_request_mock.side_effect = self.getResponseMock(self._INFECTED_HTTP_STATUS_CODE) requests_request_mock.side_effect = self.getResponseMock(self._INFECTED_HTTP_STATUS_CODE)
# Create a file as user
uf = self.getPortal().acl_users
user = "someuser"
uf._doAddUser(user, "", ["Author"], [])
self.loginByUserName(user)
document_value = self.portal.document_module.newContent( document_value = self.portal.document_module.newContent(
portal_type="File", portal_type="File",
data=b'X5O!P%@AP[4\\PZX54(P^)7CC)7}$EICAR-STANDARD-ANTIVIRUS-TEST-FILE!$H+H*', data=b'X5O!P%@AP[4\\PZX54(P^)7CC)7}$EICAR-STANDARD-ANTIVIRUS-TEST-FILE!$H+H*',
) )
document_value.setSuspect() document_value.setSuspect()
self.login()
self.tic() self.tic()
self.assertEqual(document_value.getScanState(), "infected") self.assertEqual(document_value.getScanState(), "infected")
\ No newline at end of file
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment