Commit dae43d29f12eece642754480b622f7809dcf6559

Authored by Ailson da Cruz
1 parent c098b737

Remove user #6

Showing 3 changed files with 20 additions and 4 deletions   Show diff stats
core/tests.py
... ... @@ -209,9 +209,13 @@ class DeleteUserTestCase(TestCase):
209 209 password = 'testing'
210 210 )
211 211  
212   - assign_role(self.user, 'studend')
  212 + assign_role(self.user, 'student')
  213 + self.url = reverse('core:home')
  214 +
  215 + def tearDown(test):
  216 + User.objects.get(email='testing@amadeus.com').delete()
213 217  
214   - def tearDown(test):
215   - pass
  218 + def test_delete_ok(self):
  219 + pass
216 220  
217 221  
... ...
users/urls.py
... ... @@ -10,5 +10,6 @@ urlpatterns = [
10 10 url(r'^profile/$', views.Profile.as_view(), name='profile'),
11 11 url(r'^profile/editar/(?P<username>[\w_-]+)/$', views.EditProfile.as_view(), name='edit_profile'),
12 12 #
13   - url(r'^profile/update/$', views.UpdateUser.as_view(), name='update_profile'),
  13 + url(r'^profile/update/$', views.UpdateUser.as_view(), name='update_profile'),
  14 + url(r'^profile/delete/$', views.DeleteUser.as_view(), name='delete_profile'),
14 15 ]
... ...
users/views.py
... ... @@ -150,3 +150,14 @@ class UpdateUser(LoginRequiredMixin, generic.edit.UpdateView):
150 150 messages.success(self.request, _('Profile edited successfully!'))
151 151  
152 152 return super(UpdateUser, self).form_valid(form)
  153 +
  154 +class DeleteUser(LoginRequiredMixin, generic.edit.DeleteView):
  155 + allowed_roles = ['student']
  156 + login_url = reverse_lazy("core:home")
  157 + model = User
  158 + success_url = reverse_lazy('core:index')
  159 + success_message = "Deleted Successfully"
  160 +
  161 + def get_queryset(self):
  162 + user = get_object_or_404(User, username = self.request.user.username)
  163 + return user
... ...