def test_can_upload_file_and_retrieve(self):
# Save file into media
User.objects.create_user('john', '*****@*****.**', 'johnpassword')
user = User.objects.get(username='******')
team = Team.objects.create(team_name='test')
Issue.objects.create(user=user, issue_name='test', issue_content='test contents', team=team)
file_name = 'Test_1'
file = 'Test_File_1'
issue = Issue.objects.get(issue_name='test')
created_file = AttachedFile(file_name=file_name, file=file, user=user, issue=issue)
created_file.save()
# Check if created file is exists in /media/issue_files directory.
created_path = created_file.file.path
expected_path = settings.MEDIA_ROOT + '/Test_File_1'
self.assertEqual(created_path, expected_path)
def add_file(request):
if request.method == 'POST':
user = request.user
issue_name = request.session.get('issue_name')
file_name = request.POST.get('file_name')
issue = Issue.objects.get(issue_name=issue_name)
created_file = AttachedFile(file_name=file_name, file=request.FILES['file'], user=user, issue=issue)
created_file.save()
Message.objects.create(
user=user,
content=file_name,
issue=issue,
file=created_file
)
return redirect(reverse('issue_detail', kwargs={'issue_name': issue_name}))
# return HttpResponse('File upload is success')
# return redirect('/message/get', kwargs={'issue_name': issue_name})
return HttpResponse("File upload is failed")
