def test_list(self):
self.folder = Folder(**{'name': 'Images'})
self.folder.save()
files = ['2210571.jpg', 'images.docx', 'rgba1px.png']
for file in files:
path = 'tests/files/%s' % file
f = open(path, 'rb')
document = Document()
document.folder = self.folder
document.name = file
document.file.save(name=document.name, content=File(f))
document.save()
document.save_thumbnail()
f.close()
url = reverse('model_view_list',
args=['documents.Document', 'folder', self.folder.pk])
response = self.client.get(url)
self.assertEqual(response.status_code, status.HTTP_200_OK)
self.assertEqual(len(response.data), len(files))
url = reverse(
'model_view_detail',
args=['documents.Document', 'folder', self.folder.pk, files[0]])
response = self.client.get(url)
self.assertEqual(response.status_code, status.HTTP_200_OK)
self.assertEqual(response.data['name'], files[0])
def create_document(self):
"""Creates new random documents"""
self.stdout.write("Creating a document")
doc = Document()
doc.name = _faker.text(max_nb_chars=30)
doc.category = random.choice(
[c[0] for c in Document.DOCUMENT_CATEGORIES])
doc.members_only = random.random() < 0.75
doc.file.save("{}.txt".format(doc.name),
ContentFile(_faker.text(max_nb_chars=120)))
doc.save()
def test_delete(self):
name = '2210571.jpg'
url = reverse(
'model_view_detail',
args=['documents.Document', 'folder', self.folder.pk, name])
path = 'tests/files/%s' % name
f = open(path, 'rb')
document = Document()
document.folder = self.folder
document.name = '2210571.jpg'
document.file.save(name=document.name, content=File(f))
document.save()
f.close()
response = self.client.delete(url)
self.assertEqual(response.status_code, status.HTTP_204_NO_CONTENT)
def setUp(self):
self.project = Project(**{'name': 'Customer A'})
self.project.save()
self.folder = Folder(**{'name': 'Images', 'project': self.project})
self.folder.save()
files = ['2210571.jpg', 'images.docx']
for file in files:
path = 'tests/files/%s' % file
f = open(path, 'rb')
document = Document()
document.folder = self.folder
document.name = file
document.file.save(name=document.name, content=File(f))
document.save()
document.save_thumbnail()
f.close()
