def setUp(self):
"""Create User"""
user = User.objects.create_user(email='*****@*****.**', password='******', is_staff=True, is_superuser=True)
"""Clean Storage by Traversal"""
CleanStorage(storage=TEMP_STORAGE)
CleanStorage(storage=CASE_STORAGE)
"""Load fixtures"""
call_command('loaddata', 'region', verbosity=0)
call_command('loaddata', 'type', verbosity=0)
call_command('loaddata', 'case.test.yaml', verbosity=0)
self.case = Case.objects.first()
"""Set Up Default File"""
"""Upload File to Temp Storage"""
file = SimpleUploadedFile('test1.txt', open(ROOT_DIR('apps/files/test.txt'), 'rb').read())
temp = models.TempFile()
temp.user = user
temp.case_uuid = self.case.uuid
temp.file = file
temp.file.storage = TEMP_STORAGE
temp.save()
file2 = SimpleUploadedFile('test2.txt', open(ROOT_DIR('apps/files/test.txt'), 'rb').read())
temp2 = models.TempFile()
temp2.user = user
temp2.case_uuid = self.case.uuid
temp2.file = file2
temp2.file.storage = TEMP_STORAGE
temp2.save()
self.objs = models.TempFile.objects.filter(case_uuid=self.case.uuid)
def setUp(self):
"""Create User"""
user = User.objects.create_user(email='*****@*****.**', password='******', is_staff=True, is_superuser=True)
"""Clean Storage by Traversal"""
CleanStorage(storage=TEMP_STORAGE)
CleanStorage(storage=CASE_STORAGE)
"""Set Up Default File"""
"""Upload File to Temp Storage"""
file = SimpleUploadedFile('test.txt', open(ROOT_DIR('apps/files/test.txt'), 'rb').read())
temp = models.TempFile()
temp.user = user
temp.case_uuid = uuid.uuid4()
temp.file = file
temp.file.storage = TEMP_STORAGE
temp.save()
"""Upload File to Case Storage"""
file2 = SimpleUploadedFile('test.txt', open(ROOT_DIR('apps/files/test.txt'), 'rb').read())
temp2 = models.TempFile()
temp2.user = user
temp2.case_uuid = uuid.uuid4()
temp2.file = file2
temp2.file.storage = CASE_STORAGE
temp2.save()
def test_duplicate(self):
"""
當某個案件上傳重複檔名的檔案時，預期會出現錯誤
"""
file2 = SimpleUploadedFile('test.txt', open(ROOT_DIR('apps/files/test.txt'), 'rb').read())
temp2 = models.TempFile()
temp2.user = self.user
temp2.case_uuid = self.uuid
temp2.file = file2
temp2.file.storage = TEMP_STORAGE
with self.assertRaises(ValidationError):
temp2.save()