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)