def test_media_type(self) -> None: file_id = 'BETTY01' file_path = Path('~') sut = File(file_id, file_path) self.assertIsNone(sut.media_type) media_type = MediaType('text/plain') sut.media_type = media_type self.assertEquals(media_type, sut.media_type)
def _load_object(loader: _Loader, element: ElementTree.Element, gramps_tree_directory_path: Path): file_handle = element.get('handle') file_id = element.get('id') file_element = _xpath1(element, './ns:file') file_path = gramps_tree_directory_path / file_element.get('src') file = File(file_id, file_path) file.media_type = MediaType(file_element.get('mime')) description = file_element.get('description') if description: file.description = description _load_attribute_privacy(file, element, 'attribute') loader.add_entity(FlattenedEntity(file, file_handle)) for citation_handle in _load_handles('citationref', element): loader.add_association(File, file_handle, 'citations', Citation, citation_handle) for note_handle in _load_handles('noteref', element): loader.add_association(File, file_handle, 'notes', Note, note_handle)
async def test_file_should_encode_full(self): with NamedTemporaryFile() as f: note = Note('the_note', 'The Note') file = File('the_file', f.name) file.media_type = MediaType('text/plain') file.notes.append(note) Person('the_person').files.append(file) expected = { '$schema': '/schema.json#/definitions/file', 'id': 'the_file', 'mediaType': 'text/plain', 'entities': [ '/en/person/the_person/index.json', ], 'notes': [ '/en/note/the_note/index.json', ], 'links': [ { 'url': '/en/file/the_file/index.json', 'relationship': 'canonical', 'mediaType': 'application/json', }, { 'url': '/nl/file/the_file/index.json', 'relationship': 'alternate', 'locale': 'nl-NL', }, { 'url': '/en/file/the_file/index.html', 'relationship': 'alternate', 'mediaType': 'text/html', }, ], } await self.assert_encodes(expected, file, 'file')