def test_get_documents(self): """Test the get_documents method. """ document_file1 = DocumentFile() document_file2 = DocumentFile() document1 = Document() document2 = Document() document3 = Document() project = Project() document_file1.documents = [document1, document2] document_file2.documents = [document3] project.document_files = [document_file1, document_file2] project.save() assert project.get_documents() == [document1, document2, document3]
def test_model_document_file(self): """Test to make sure that DocumentFile is working properly. """ documentfile = DocumentFile() document1 = Document() document2 = Document() project1 = Project() project2 = Project() documentfile.path = "/foo/bar" documentfile.documents = [document1, document2] documentfile.projects = [project1, project2] documentfile.save() assert len(documentfile.documents) == 2 assert len(documentfile.projects) == 2
def test_document_belongs_to(self): """Check if ``belongs_to()`` on ``Document`` is working properly. """ user = User() project = Project() document_file = DocumentFile() document = Document() project.document_files = [document_file] document_file.documents = [document] user.projects = [project] user.save() project.save() document.save() document_file.save() assert document.belongs_to(user)