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_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)
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)
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_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_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]
