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
@@ -209,9 +209,13 @@ class DeleteUserTestCase(TestCase): @@ -209,9 +209,13 @@ class DeleteUserTestCase(TestCase):
209 password = 'testing' 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
@@ -10,5 +10,6 @@ urlpatterns = [ @@ -10,5 +10,6 @@ urlpatterns = [
10 url(r'^profile/$', views.Profile.as_view(), name='profile'), 10 url(r'^profile/$', views.Profile.as_view(), name='profile'),
11 url(r'^profile/editar/(?P<username>[\w_-]+)/$', views.EditProfile.as_view(), name='edit_profile'), 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,3 +150,14 @@ class UpdateUser(LoginRequiredMixin, generic.edit.UpdateView):
150 messages.success(self.request, _('Profile edited successfully!')) 150 messages.success(self.request, _('Profile edited successfully!'))
151 151
152 return super(UpdateUser, self).form_valid(form) 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