def setUp(self):
self.ll = generate_ll()
self.pl = self.ll.pl
self.user = self.pl.create_user('*****@*****.**')
self.task = self.pl.create_task('task')
self.task.id = 1
self.f = MockFileObject('/filename.txt')
def test_task_id_not_sepcified_raises(self):
# given
task = Mock(spec=Task)
task.id = 123
form = {}
files = {'filename': MockFileObject('/filename.exe')}
request = generate_mock_request(form=form, files=files)
admin = self.pl.create_user('admin', is_admin=True)
# expect
self.assertRaises(BadRequest, self.vl.attachment_new, request, admin)
# and
self.r.redirect.assert_not_called()
def test_empty_filename_raises(self):
# given
task = Mock(spec=Task)
task.id = 123
form = {'task_id': task.id, 'description': 'asdf'}
files = {'filename': MockFileObject('')}
request = generate_mock_request(form=form, files=files)
admin = self.pl.create_user('admin', is_admin=True)
# expect
self.assertRaises(BadRequest, self.vl.attachment_new, request, admin)
# and
self.r.redirect.assert_not_called()
def test_description_sets_description(self):
# given
task = Mock(spec=Task)
task.id = 123
form = {'task_id': task.id, 'description': 'asdf'}
f = MockFileObject('/filename.txt')
files = {'filename': f}
request = generate_mock_request(form=form, files=files)
admin = self.pl.create_user('admin', is_admin=True)
# when
result = self.vl.attachment_new(request, admin)
# then
self.ll.create_new_attachment.assert_called_once_with(task.id,
f,
'asdf',
admin,
timestamp=None)
# and
self.r.url_for.assert_called_once_with('view_task', id=task.id)
self.r.redirect.assert_called_once_with(self.r.url_for.return_value)
self.assertIs(self.r.redirect.return_value, result)
def test_imports_data_from_file(self):
# given
src = '''{"tasks":[{
"id": 1,
"summary":"summary"
}]}'''
src_dict = {
"tasks": [{
"id": 1,
"summary": "summary",
}]
}
form = {}
files = {'file': MockFileObject(filename='', content=src)}
request = generate_mock_request(form=form, files=files)
admin = self.pl.create_user('admin', is_admin=True)
# when
result = self.vl.import_(request, admin)
# then
self.ll.do_import_data.assert_called_once_with(src_dict)
# and
self.r.url_for.assert_called_once_with('index')
self.r.redirect.assert_called_once_with(self.r.url_for.return_value)
self.assertIs(self.r.redirect.return_value, result)