| Left: | ||
| Right: |
| LEFT | RIGHT |
|---|---|
| 1 # This file is part of the Adblock Plus web scripts, | 1 # This file is part of the Adblock Plus web scripts, |
| 2 # Copyright (C) 2006-2016 Eyeo GmbH | 2 # Copyright (C) 2006-2017 eyeo GmbH |
| 3 # | 3 # |
| 4 # Adblock Plus is free software: you can redistribute it and/or modify | 4 # Adblock Plus is free software: you can redistribute it and/or modify |
| 5 # it under the terms of the GNU General Public License version 3 as | 5 # it under the terms of the GNU General Public License version 3 as |
| 6 # published by the Free Software Foundation. | 6 # published by the Free Software Foundation. |
| 7 # | 7 # |
| 8 # Adblock Plus is distributed in the hope that it will be useful, | 8 # Adblock Plus is distributed in the hope that it will be useful, |
| 9 # but WITHOUT ANY WARRANTY; without even the implied warranty of | 9 # but WITHOUT ANY WARRANTY; without even the implied warranty of |
| 10 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the | 10 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
| 11 # GNU General Public License for more details. | 11 # GNU General Public License for more details. |
| 12 # | 12 # |
| (...skipping 41 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... | |
| 54 | 54 |
| 55 | 55 |
| 56 # We make this a fixture instead of a constant so we can modify it in each | 56 # We make this a fixture instead of a constant so we can modify it in each |
| 57 # test as needed without affecting other tests. | 57 # test as needed without affecting other tests. |
| 58 @pytest.fixture | 58 @pytest.fixture |
| 59 def form_data(): | 59 def form_data(): |
| 60 return { | 60 return { |
| 61 'email': 'john_doe@gmail.com', | 61 'email': 'john_doe@gmail.com', |
| 62 'mandatory': 'john_doe@gmail.com', | 62 'mandatory': 'john_doe@gmail.com', |
| 63 'non_mandatory_message': 'Once upon a time\nthere lived a king.', | 63 'non_mandatory_message': 'Once upon a time\nthere lived a king.', |
| 64 'non_mandatory_email': 'test@test.com' | 64 'non_mandatory_email': 'test@test.com', |
|
Vasily Kuznetsov
2017/03/21 14:00:34
I usually put the comma at the end of last item in
Jon Sonesen
2017/03/22 14:28:20
Done.
| |
| 65 } | 65 } |
| 66 | 66 |
| 67 | 67 |
| 68 @pytest.fixture | 68 @pytest.fixture |
| 69 def response_for(form_handler): | 69 def response_for(form_handler): |
| 70 """ Registers two intercepts, returns responses for them based on bool """ | 70 """ Registers two intercepts, returns responses for them based on bool """ |
| 71 urllib_intercept.install_opener() | 71 urllib_intercept.install_opener() |
| 72 add_wsgi_intercept(HOST, LOG_PORT, lambda: form_handler[0]) | 72 add_wsgi_intercept(HOST, LOG_PORT, lambda: form_handler[0]) |
| 73 add_wsgi_intercept(HOST, NO_LOG_PORT, lambda: form_handler[1]) | 73 add_wsgi_intercept(HOST, NO_LOG_PORT, lambda: form_handler[1]) |
| 74 | 74 |
| 75 def response_for(data, log=False): | 75 def response_for(data, log=False): |
| 76 url = 'http://{}:{}'.format(HOST, NO_LOG_PORT) | |
| 77 if log: | 76 if log: |
|
Vasily Kuznetsov
2017/03/21 14:00:34
I think this would be more readable as:
if log:
Jon Sonesen
2017/03/22 14:28:20
Done.
| |
| 78 url = 'http://{}:{}'.format(HOST, LOG_PORT) | 77 url = 'http://{}:{}'.format(HOST, LOG_PORT) |
| 78 else: | |
| 79 url = 'http://{}:{}'.format(HOST, NO_LOG_PORT) | |
| 79 if data is None: | 80 if data is None: |
| 80 response = urlopen(url) | 81 response = urlopen(url) |
| 81 else: | 82 else: |
| 82 response = urlopen(url, urlencode(data)) | 83 response = urlopen(url, urlencode(data)) |
| 83 return response.code, response.read() | 84 return response.code, response.read() |
| 84 | 85 |
| 85 yield response_for | 86 yield response_for |
| 86 remove_wsgi_intercept() | 87 remove_wsgi_intercept() |
| 87 | 88 |
| 88 | 89 |
| 89 def test_form_handler_email_errors(form_config): | 90 @pytest.fixture |
| 90 tmp_config = form_config | 91 def sm_mock(mocker): |
|
Vasily Kuznetsov
2017/03/21 14:00:34
What is the purpose of this assignment? Why don't
Jon Sonesen
2017/03/22 14:28:20
whoops. Done
| |
| 91 del tmp_config['url'].value | 92 return mocker.patch('sitescripts.formmail.web.formmail2.sendMail') |
| 92 with pytest.raises(Exception) as error: | |
| 93 formmail2.make_handler('test', tmp_config)[1] | |
| 94 assert error.value.message == 'No URL configured for form handler: test' | |
| 95 | 93 |
| 96 | 94 |
| 97 def test_form_handler_field_errors(form_config): | 95 @pytest.mark.parametrize('key,message', [ |
| 98 tmp_config = form_config | 96 ('url', 'No URL configured for form handler: test'), |
| 99 tmp_config['fields'] = {} | 97 ('fields', 'No fields configured for form handler: test'), |
| 98 ('template', 'No template configured for form handler: test'), | |
| 99 ]) | |
| 100 def test_config_errors(key, message, form_config): | |
| 101 del form_config[key] | |
| 100 with pytest.raises(Exception) as error: | 102 with pytest.raises(Exception) as error: |
| 101 formmail2.make_handler('test', tmp_config)[1] | 103 formmail2.make_handler('test', form_config)[1] |
| 102 assert error.value.message == 'No fields configured for form handler: test' | 104 assert error.value.message == message |
| 103 | 105 |
| 104 del tmp_config['fields'] | 106 |
| 107 @pytest.mark.parametrize('field,message', [ | |
| 108 (('new_field', 'foo'), 'Unexpected field/fields: new_field'), | |
| 109 (('mandatory', ''), 'No mandatory entered'), | |
| 110 (('non_mandatory_email', 'asfaf'), 'Invalid email'), | |
| 111 (('email', 'asfaf'), 'You failed the email validation'), | |
| 112 (('email', ''), 'You failed the email test'), | |
| 113 ]) | |
| 114 def test_http_errs(field, message, response_for, form_data, sm_mock): | |
| 115 key, value = field | |
| 116 form_data[key] = value | |
| 117 with pytest.raises(HTTPError) as error: | |
| 118 response_for(form_data) | |
| 119 assert error.value.read() == message | |
| 120 | |
| 121 | |
| 122 @pytest.mark.parametrize('field,expected', [ | |
| 123 (('non_mandatory_message', '\xc3\xb6'), (200, '')), | |
| 124 (('non_mandatory_message', ''), (200, '')), | |
| 125 ]) | |
| 126 def test_success(field, expected, log_path, response_for, form_data, sm_mock): | |
| 127 key, value = field | |
| 128 form_data[key] = value | |
| 129 assert response_for(form_data, log=False) == expected | |
| 130 assert sm_mock.call_count == 1 | |
| 131 | |
| 132 params = sm_mock.call_args[0][1]['fields'] | |
| 133 assert set(params.keys()) == set(form_data.keys()) | |
| 134 for key, value in form_data.items(): | |
| 135 assert params[key] == value.decode('utf8') | |
| 136 | |
| 137 assert response_for(form_data, log=True) == expected | |
| 138 assert sm_mock.call_count == 2 | |
| 139 | |
| 140 assert response_for(form_data, log=True) == expected | |
| 141 assert sm_mock.call_count == 3 | |
| 142 | |
| 143 with open(log_path) as log_file: | |
| 144 reader = DictReader(log_file) | |
| 145 row = reader.next() | |
| 146 # rows should not be equal because the time field | |
| 147 # is added by the logging function. | |
| 148 assert row != reader.next() | |
| 149 | |
| 150 | |
| 151 def test_config_field_errors(form_config): | |
| 152 form_config['fields'] = {} | |
| 105 with pytest.raises(Exception) as error: | 153 with pytest.raises(Exception) as error: |
| 106 formmail2.make_handler('test', tmp_config)[1] | 154 formmail2.make_handler('test', form_config)[1] |
| 107 assert error.value.message == 'No fields configured for form handler: test' | 155 assert error.value.message == 'No fields configured for form handler: test' |
| 108 | 156 |
| 109 | 157 |
| 110 def test_form_handler_template_errors(form_config): | 158 def test_config_template_errors(form_config): |
| 111 tmp_config = form_config | 159 form_config['template'].value = 'no' |
| 112 tmp_config['template'].value = 'no' | |
| 113 with pytest.raises(Exception) as error: | 160 with pytest.raises(Exception) as error: |
| 114 formmail2.make_handler('test', tmp_config)[1] | 161 formmail2.make_handler('test', form_config)[1] |
| 115 assert error.typename == 'TemplateNotFound' | 162 assert error.value.message == 'Template not found at: no' |
| 116 | |
| 117 del tmp_config['template'].value | |
| 118 with pytest.raises(Exception) as error: | |
| 119 formmail2.make_handler('test', tmp_config)[1] | |
| 120 assert error.value.message == ('No template configured for form handler' | |
| 121 ': test') | |
| 122 del tmp_config['template'] | |
| 123 with pytest.raises(Exception) as error: | |
| 124 formmail2.make_handler('test', tmp_config)[1] | |
| 125 assert error.value.message == ('No template configured for form handler' | |
| 126 ': test') | |
| 127 | 163 |
| 128 | 164 |
| 129 def test_config_parse(form_config): | 165 def test_config_parse(form_config): |
| 130 assert form_config['url'].value == 'test/apply/submit' | 166 assert form_config['url'].value == 'test/apply/submit' |
| 131 assert form_config['fields']['email'].value == 'mandatory, email' | 167 assert form_config['fields']['email'].value == 'mandatory, email' |
| 132 | 168 |
| 133 | 169 |
| 134 def test_sendmail_fail(log_path, response_for, form_data, mocker): | 170 def test_sendmail_fail(log_path, response_for, form_data, sm_mock): |
| 135 sm_mock = mocker.patch('sitescripts.formmail.web.formmail2.sendMail') | |
| 136 sm_mock.side_effect = Exception('Sendmail Fail') | 171 sm_mock.side_effect = Exception('Sendmail Fail') |
| 137 with pytest.raises(HTTPError) as error: | 172 with pytest.raises(HTTPError): |
| 138 response_for(form_data, log=True) | 173 response_for(form_data, log=True) |
| 139 assert error.typename == 'HTTPError' | |
| 140 | 174 |
| 141 with open(log_path) as log_file: | 175 with open(log_path) as log_file: |
| 142 row = DictReader(log_file).next() | 176 row = DictReader(log_file).next() |
| 143 assert 'time' in row | 177 assert row != form_data |
| 144 | |
| 145 | |
| 146 @pytest.mark.parametrize('log, res', | |
| 147 [(True, (200, '')), (False, (200, ''))]) | |
| 148 def test_utf8_success(log, res, log_path, response_for, form_data, mocker): | |
| 149 """ DictWriter does not accpet utf-8, call log handler """ | |
|
Vasily Kuznetsov
2017/03/21 14:00:34
typo in "accpet".
Jon Sonesen
2017/03/22 14:28:19
Done.
| |
| 150 form_data['non_mandatory_message'] = '\xc3\xb6' | |
| 151 sm_mock = mocker.patch('sitescripts.formmail.web.formmail2.sendMail') | |
| 152 assert response_for(form_data, log) == res | |
| 153 assert sm_mock.call_count == 1 | |
| 154 params = sm_mock.call_args[0][1]['fields'] | |
| 155 assert set(params.keys()) == set(form_data.keys()) | |
| 156 for key, value in form_data.items(): | |
| 157 assert params[key] == value.decode('utf8') | |
| 158 | |
| 159 | |
| 160 def test_success(response_for, form_data, mocker): | |
| 161 sm_mock = mocker.patch('sitescripts.formmail.web.formmail2.sendMail') | |
|
Vasily Kuznetsov
2017/03/21 14:00:34
This and some following lines seem to be duplicate
Jon Sonesen
2017/03/22 14:28:20
The thing is that some of the test's have differen
| |
| 162 assert response_for(form_data) == (200, '') | |
| 163 assert sm_mock.call_count == 1 | |
| 164 params = sm_mock.call_args[0][1]['fields'] | |
| 165 assert set(params.keys()) == set(form_data.keys()) | |
| 166 for key, value in form_data.items(): | |
| 167 assert params[key] == value | |
| 168 | |
| 169 | |
| 170 def test_log_success(log_path, response_for, form_data, mocker): | |
| 171 sm_mock = mocker.patch('sitescripts.formmail.web.formmail2.sendMail') | |
| 172 assert response_for(form_data, log=True) == (200, '') | |
| 173 assert sm_mock.call_count == 1 | |
| 174 params = sm_mock.call_args[0][1]['fields'] | |
| 175 assert set(params.keys()) == set(form_data.keys()) | |
| 176 for key, value in form_data.items(): | |
| 177 assert params[key] == value | |
| 178 with open(log_path) as log_file: | |
| 179 row = DictReader(log_file).next() | |
| 180 assert 'time' in row | |
| 181 | |
| 182 | |
| 183 def test_log_append_success(log_path, response_for, form_data, mocker): | |
| 184 sm_mock = mocker.patch('sitescripts.formmail.web.formmail2.sendMail') | |
| 185 assert response_for(form_data, log=True) == (200, '') | |
| 186 form_data['non_mandatory_message'] = '' | |
| 187 assert response_for(form_data, log=True) == (200, '') | |
| 188 assert sm_mock.call_count == 2 | |
| 189 params = sm_mock.call_args[0][1]['fields'] | |
| 190 assert set(params.keys()) == set(form_data.keys()) | |
| 191 for key, value in form_data.items(): | |
| 192 assert params[key] == value | |
| 193 with open(log_path) as log_file: | |
| 194 reader = DictReader(log_file) | |
| 195 row = reader.next() | |
| 196 assert row != reader.next() | |
| 197 | |
| 198 | |
| 199 def test_non_mandatory_no_msg(response_for, form_data, mocker): | |
| 200 mocker.patch('sitescripts.formmail.web.formmail2.sendMail') | |
| 201 form_data['non_mandatory_message'] = '' | |
| 202 assert response_for(form_data) == (200, '') | |
| 203 | |
| 204 | |
| 205 def test_invalid_email_cstm_msg(response_for, form_data, mocker, form_config): | |
| 206 mocker.patch('sitescripts.formmail.web.formmail2.sendMail') | |
| 207 form_data['email'] = 'bademail' | |
| 208 with pytest.raises(HTTPError) as error: | |
| 209 response_for(form_data) | |
| 210 assert error.value.read() == 'You failed the email validation' | |
| 211 | |
| 212 | |
| 213 def test_valid_nan_mandatory_email(response_for, form_data, mocker): | |
|
Vasily Kuznetsov
2017/03/21 14:00:34
...non_mandatory...? :)
Jon Sonesen
2017/03/22 14:28:20
Done.
| |
| 214 mocker.patch('sitescripts.formmail.web.formmail2.sendMail') | |
| 215 form_data['non_mandatory_email'] = 'asfaf' | |
| 216 with pytest.raises(HTTPError) as error: | |
| 217 response_for(form_data) | |
| 218 assert error.value.read() == 'Invalid email' | |
| 219 | |
| 220 del form_data['non_mandatory_email'] | |
| 221 assert response_for(form_data) == (200, '') | |
| 222 | |
| 223 | |
| 224 def test_mandatory_fail_dflt_msg(response_for, form_data, mocker): | |
| 225 mocker.patch('sitescripts.formmail.web.formmail2.sendMail') | |
| 226 del form_data['mandatory'] | |
| 227 with pytest.raises(HTTPError) as error: | |
| 228 response_for(form_data) | |
| 229 assert error.value.read() == 'No mandatory entered' | |
| 230 | |
| 231 | |
| 232 def test_field_err(form_config, form_data, log_path): | |
| 233 """ Submits a form that does not have the dame fields as previous submission s | |
|
Vasily Kuznetsov
2017/03/21 14:00:35
typo in "dame"?
Jon Sonesen
2017/03/22 14:28:20
Done.
| |
| 234 | |
|
Vasily Kuznetsov
2017/03/21 14:00:34
Here you're continuing the same sentence in the li
Jon Sonesen
2017/03/22 14:28:20
Done.
| |
| 235 that have the same form name, asserts that proper message is returned and | |
| 236 the row was properly written | |
| 237 """ | |
| 238 formmail2.log_formdata(form_data, log_path) | |
| 239 del(form_config['fields']['email']) | |
| 240 del(form_data['email']) | |
| 241 try: | |
| 242 formmail2.log_formdata(form_data, log_path) | |
| 243 except Exception as e: | |
| 244 assert e.message == ('Field names have changed, error log ' | |
| 245 'written to {}_error').format(log_path) | |
| 246 | |
| 247 with open(log_path+'_error') as error_log: | |
| 248 assert DictReader(error_log).next() == form_data | |
| 249 | 178 |
| 250 | 179 |
| 251 def test_append_field_err(form_config, form_data, log_path): | 180 def test_append_field_err(form_config, form_data, log_path): |
| 252 """ Submits two identical forms that do not match the previous fields | 181 """ Checks that error logs are correctly written and appended |
| 253 | 182 |
| 254 found in the log file, triggering two rows to be added to the error | 183 Submits three forms, the second two have different fields to the first |
| 255 log and asserting the proper message is returned and that the rows | 184 and should be added to the same log file as each other, and be identical |
| 256 were written as expected | |
| 257 """ | 185 """ |
| 258 formmail2.log_formdata(form_data, log_path) | 186 formmail2.log_formdata(form_data, log_path) |
| 259 del(form_config['fields']['email']) | 187 del form_data['email'] |
| 260 del(form_data['email']) | 188 |
| 261 try: | 189 # submit two forms with fields that dont match the config |
| 190 # this should append the second form to the error log file | |
| 191 with pytest.raises(Exception): | |
| 262 formmail2.log_formdata(form_data, log_path) | 192 formmail2.log_formdata(form_data, log_path) |
| 263 except Exception: | 193 with pytest.raises(Exception): |
| 264 pass | |
| 265 try: | |
| 266 formmail2.log_formdata(form_data, log_path) | 194 formmail2.log_formdata(form_data, log_path) |
| 267 except Exception as e: | |
| 268 assert e.message == ('Field names have changed, error log' | |
| 269 ' appended to {}_error').format(log_path) | |
| 270 | 195 |
| 271 with open(log_path+'_error') as error_log: | 196 with open(log_path + '_error') as error_log: |
| 272 reader = DictReader(error_log) | 197 reader = DictReader(error_log) |
| 273 # two identical rows should be in the error log | |
| 274 assert reader.next() == form_data | 198 assert reader.next() == form_data |
| 275 assert reader.next() == form_data | 199 assert reader.next() == form_data |
| LEFT | RIGHT |