diff --git a/core/tests.py b/core/tests.py index b217084..266f305 100644 --- a/core/tests.py +++ b/core/tests.py @@ -209,9 +209,13 @@ class DeleteUserTestCase(TestCase): password = 'testing' ) - assign_role(self.user, 'studend') + assign_role(self.user, 'student') + self.url = reverse('core:home') + + def tearDown(test): + User.objects.get(email='testing@amadeus.com').delete() - def tearDown(test): - pass + def test_delete_ok(self): + pass diff --git a/users/urls.py b/users/urls.py index dc7904d..3cb7469 100644 --- a/users/urls.py +++ b/users/urls.py @@ -10,5 +10,6 @@ urlpatterns = [ url(r'^profile/$', views.Profile.as_view(), name='profile'), url(r'^profile/editar/(?P[\w_-]+)/$', views.EditProfile.as_view(), name='edit_profile'), # - url(r'^profile/update/$', views.UpdateUser.as_view(), name='update_profile'), + url(r'^profile/update/$', views.UpdateUser.as_view(), name='update_profile'), + url(r'^profile/delete/$', views.DeleteUser.as_view(), name='delete_profile'), ] diff --git a/users/views.py b/users/views.py index 56dc5ff..9a2fe4f 100644 --- a/users/views.py +++ b/users/views.py @@ -150,3 +150,14 @@ class UpdateUser(LoginRequiredMixin, generic.edit.UpdateView): messages.success(self.request, _('Profile edited successfully!')) return super(UpdateUser, self).form_valid(form) + +class DeleteUser(LoginRequiredMixin, generic.edit.DeleteView): + allowed_roles = ['student'] + login_url = reverse_lazy("core:home") + model = User + success_url = reverse_lazy('core:index') + success_message = "Deleted Successfully" + + def get_queryset(self): + user = get_object_or_404(User, username = self.request.user.username) + return user -- libgit2 0.21.2