def build_document(self, link, user_id):
"""Build document object"""
document = all_models.Document(
link=link,
title=link,
modified_by_id=user_id,
context=self.row_converter.obj.context,
kind=self.KIND,
)
document.add_admin_role()
return document
def parse_item(self):
lines = [line.rsplit(" ", 1) for line in self.raw_value.splitlines()]
documents = []
for line in lines:
if len(line) != 2:
self.add_warning(errors.WRONG_VALUE, column_name=self.display_name)
continue
title, link = line
documents.append(
all_models.Document(title=title.strip(), link=link.strip()))
return documents
def test_create_invalid_type(self):
"""Test validation kind."""
data = {
"kind": 3,
"title": "test_title",
"link": "test_link",
"owners": [self.gen.create_stub(all_models.Person.query.first())],
}
obj_name = all_models.Document._inflector.table_singular
obj = all_models.Document()
obj_dict = self.gen.obj_to_dict(obj, obj_name)
obj_dict[obj_name].update(data)
resp = self.api.post(all_models.Document, obj_dict)
self.assert400(resp)
self.assertEqual(
'"Invalid value for attribute kind. '
'Expected options are `FILE`, '
'`REFERENCE_URL`"', resp.data)
