test(data/model): test action logging
This commit is contained in:
parent
6916d82e0d
commit
625bd66b42
3 changed files with 69 additions and 3 deletions
39
data/model/test/test_log.py
Normal file
39
data/model/test/test_log.py
Normal file
|
@ -0,0 +1,39 @@
|
|||
import pytest
|
||||
|
||||
from data.model.log import log_action
|
||||
from peewee import PeeweeException
|
||||
from mock import Mock
|
||||
|
||||
@pytest.mark.parametrize( 'unlogged_pulls_ok,action_kind,db_exception,throws', [
|
||||
(False, 'pull_repo', None, False),
|
||||
(False, 'push_repo', None, False),
|
||||
(False, 'pull_repo', PeeweeException, True),
|
||||
(False, 'push_repo', PeeweeException, True),
|
||||
(True, 'pull_repo', PeeweeException, False),
|
||||
(True, 'push_repo', PeeweeException, True),
|
||||
(True, 'pull_repo', Exception, True)
|
||||
])
|
||||
|
||||
def test_log_action(unlogged_pulls_ok, action_kind, db_exception, throws, app_config, logentry):
|
||||
log_args = {
|
||||
'performer': Mock(id='TEST_PERFORMER_ID'),
|
||||
'repository': Mock(id='TEST_REPO'),
|
||||
'ip': 'TEST_IP',
|
||||
'metadata': {'test_key': 'test_value'},
|
||||
'timestamp': 'TEST_TIMESTAMP'
|
||||
}
|
||||
app_config['SERVICE_LOG_ACCOUNT_ID'] = 'TEST_ACCOUNT_ID'
|
||||
|
||||
app_config['ALLOW_PULLS_WITHOUT_STRICT_LOGGING'] = unlogged_pulls_ok
|
||||
|
||||
logentry.create.side_effect = db_exception
|
||||
if throws:
|
||||
with pytest.raises(db_exception):
|
||||
log_action(action_kind, None, **log_args)
|
||||
else:
|
||||
log_action(action_kind, None, **log_args)
|
||||
|
||||
logentry.create.assert_called_once_with(kind=action_kind+'_kind', account='TEST_ACCOUNT_ID',
|
||||
performer='TEST_PERFORMER_ID', repository='TEST_REPO',
|
||||
ip='TEST_IP', metadata_json='{"test_key": "test_value"}',
|
||||
datetime='TEST_TIMESTAMP')
|
Reference in a new issue