def weblink_add_typed(request, folder_id=None, response_format='html'):
"Web link add to preselected folder"
folder = None
if folder_id:
folder = get_object_or_404(Folder, pk=folder_id)
if not request.user.profile.has_permission(folder, mode='x'):
folder = None
if request.POST:
if 'cancel' not in request.POST:
link = WebLink()
form = WebLinkForm(
request.user.profile, folder_id, request.POST, instance=link)
if form.is_valid():
link = form.save()
link.set_user_from_request(request)
return HttpResponseRedirect(reverse('documents_weblink_view', args=[link.id]))
else:
return HttpResponseRedirect(reverse('document_index'))
else:
form = WebLinkForm(request.user.profile, folder_id)
context = _get_default_context(request)
context.update({'form': form,
'folder': folder})
return render_to_response('documents/weblink_add_typed', context,
context_instance=RequestContext(request),
response_format=response_format)
def setUp(self):
self.group, created = Group.objects.get_or_create(name='test')
self.user, created = DjangoUser.objects.get_or_create(username=self.username)
self.user.set_password(self.password)
self.user.save()
perspective, created = Perspective.objects.get_or_create(
name='default')
perspective.set_default_user()
perspective.save()
ModuleSetting.set('default_perspective', perspective.id)
self.folder = Folder(name='test')
self.folder.set_default_user()
self.folder.save()
self.document = Document(title='test_document', folder=self.folder)
self.document.set_default_user()
self.document.save()
self.file = File(name='test_file', folder=self.folder)
self.file.set_default_user()
self.file.save()
self.link = WebLink(title='test', folder=self.folder, url='test')
self.link.set_default_user()
self.link.save()
def weblink_add(request, response_format='html'):
"New web link form"
if request.POST:
if 'cancel' not in request.POST:
link = WebLink()
form = WebLinkForm(
request.user.profile, None, request.POST, instance=link)
if form.is_valid():
link = form.save()
link.set_user_from_request(request)
return HttpResponseRedirect(reverse('documents_weblink_view', args=[link.id]))
else:
return HttpResponseRedirect(reverse('document_index'))
else:
form = WebLinkForm(request.user.profile, None)
context = _get_default_context(request)
context.update({'form': form,
'file': file})
return render_to_response('documents/weblink_add', context,
context_instance=RequestContext(request),
response_format=response_format)
def test_model_weblink(self):
"""Test WebLink Model"""
folder = Folder(name='test')
folder.save()
obj = WebLink(title='test', folder=folder, url='test')
obj.save()
self.assertEquals(folder, obj.folder)
self.assertNotEquals(obj.id, None)
obj.delete()
