| Total Complexity | 29 |
| Total Lines | 837 |
| Duplicated Lines | 58.9 % |
Duplicate code is one of the most pungent code smells. A rule that is often used is to re-structure code once it is duplicated in three or more places.
Common duplication problems, and corresponding solutions are:
| 1 | # -*- coding: utf-8 -*- |
||
| 126 | @override_settings( |
||
| 127 | ACCOUNT_ACTIVATION_DAYS=7, |
||
| 128 | REGISTRATION_OPEN=True, |
||
| 129 | REGISTRATION_SUPPLEMENT_CLASS=None, |
||
| 130 | REGISTRATION_BACKEND_CLASS=( |
||
| 131 | 'registration.backends.default.DefaultRegistrationBackend'), |
||
| 132 | ) |
||
| 133 | class RegistrationProfileManagerTestCase(TestCase): |
||
| 134 | user_info = { |
||
| 135 | 'username': 'alice', |
||
| 136 | 'email': '[email protected]', |
||
| 137 | } |
||
| 138 | |||
| 139 | def setUp(self): |
||
| 140 | self.mock_site = mock_site() |
||
| 141 | |||
| 142 | def test_register(self): |
||
| 143 | new_user = RegistrationProfile.objects.register(site=self.mock_site, |
||
| 144 | **self.user_info) |
||
| 145 | self.assertEqual(new_user.username, 'alice') |
||
| 146 | self.assertEqual(new_user.email, '[email protected]') |
||
| 147 | self.failIf(new_user.is_active) |
||
| 148 | self.failIf(new_user.has_usable_password()) |
||
| 149 | |||
| 150 | def test_register_email(self): |
||
| 151 | RegistrationProfile.objects.register(site=self.mock_site, |
||
| 152 | **self.user_info) |
||
| 153 | |||
| 154 | self.assertEqual(len(mail.outbox), 1) |
||
| 155 | |||
| 156 | def test_register_no_email(self): |
||
| 157 | RegistrationProfile.objects.register(site=self.mock_site, |
||
| 158 | send_email=False, |
||
| 159 | **self.user_info) |
||
| 160 | |||
| 161 | self.assertEqual(len(mail.outbox), 0) |
||
| 162 | |||
| 163 | def test_acceptance(self): |
||
| 164 | new_user = RegistrationProfile.objects.register(site=self.mock_site, |
||
| 165 | send_email=False, |
||
| 166 | **self.user_info) |
||
| 167 | |||
| 168 | profile = new_user.registration_profile |
||
| 169 | RegistrationProfile.objects.accept_registration(profile, |
||
| 170 | site=self.mock_site) |
||
| 171 | |||
| 172 | self.assertEqual(profile.status, 'accepted') |
||
| 173 | self.assertNotEqual(profile.activation_key, None) |
||
| 174 | |||
| 175 | def test_acceptance_email(self): |
||
| 176 | new_user = RegistrationProfile.objects.register(site=self.mock_site, |
||
| 177 | send_email=False, |
||
| 178 | **self.user_info) |
||
| 179 | |||
| 180 | profile = new_user.registration_profile |
||
| 181 | RegistrationProfile.objects.accept_registration(profile, |
||
| 182 | site=self.mock_site) |
||
| 183 | |||
| 184 | self.assertEqual(len(mail.outbox), 1) |
||
| 185 | |||
| 186 | def test_acceptance_no_email(self): |
||
| 187 | new_user = RegistrationProfile.objects.register(site=self.mock_site, |
||
| 188 | send_email=False, |
||
| 189 | **self.user_info) |
||
| 190 | |||
| 191 | profile = new_user.registration_profile |
||
| 192 | RegistrationProfile.objects.accept_registration(profile, |
||
| 193 | site=self.mock_site, |
||
| 194 | send_email=False) |
||
| 195 | |||
| 196 | self.assertEqual(len(mail.outbox), 0) |
||
| 197 | |||
| 198 | def test_acceptance_force(self): |
||
| 199 | user1 = RegistrationProfile.objects.register( |
||
| 200 | username='user1', email='[email protected]', |
||
| 201 | site=self.mock_site, send_email=False, |
||
| 202 | ) |
||
| 203 | user2 = RegistrationProfile.objects.register( |
||
| 204 | username='user2', email='[email protected]', |
||
| 205 | site=self.mock_site, send_email=False, |
||
| 206 | ) |
||
| 207 | RegistrationProfile.objects.accept_registration( |
||
| 208 | user1.registration_profile, site=self.mock_site |
||
| 209 | ) |
||
| 210 | RegistrationProfile.objects.reject_registration( |
||
| 211 | user2.registration_profile, site=self.mock_site |
||
| 212 | ) |
||
| 213 | |||
| 214 | # from accepted => accepted |
||
| 215 | profile = user1.registration_profile |
||
| 216 | previous_activation_key = profile.activation_key |
||
| 217 | RegistrationProfile.objects.accept_registration(profile, |
||
| 218 | site=self.mock_site, |
||
| 219 | force=True) |
||
| 220 | self.assertEqual(profile.status, 'accepted') |
||
| 221 | self.assertNotEqual(profile.activation_key, previous_activation_key) |
||
| 222 | |||
| 223 | # from rejected => accepted |
||
| 224 | profile = user2.registration_profile |
||
| 225 | RegistrationProfile.objects.accept_registration(profile, |
||
| 226 | site=self.mock_site, |
||
| 227 | force=True) |
||
| 228 | self.assertEqual(profile.status, 'accepted') |
||
| 229 | self.assertNotEqual(profile.activation_key, None) |
||
| 230 | |||
| 231 | def test_rejection(self): |
||
| 232 | new_user = RegistrationProfile.objects.register(site=self.mock_site, |
||
| 233 | send_email=False, |
||
| 234 | **self.user_info) |
||
| 235 | |||
| 236 | profile = new_user.registration_profile |
||
| 237 | RegistrationProfile.objects.reject_registration(profile, |
||
| 238 | site=self.mock_site) |
||
| 239 | |||
| 240 | self.assertEqual(profile.status, 'rejected') |
||
| 241 | self.assertEqual(profile.activation_key, None) |
||
| 242 | |||
| 243 | def test_rejection_email(self): |
||
| 244 | new_user = RegistrationProfile.objects.register(site=self.mock_site, |
||
| 245 | send_email=False, |
||
| 246 | **self.user_info) |
||
| 247 | |||
| 248 | profile = new_user.registration_profile |
||
| 249 | RegistrationProfile.objects.reject_registration(profile, |
||
| 250 | site=self.mock_site) |
||
| 251 | |||
| 252 | self.assertEqual(len(mail.outbox), 1) |
||
| 253 | |||
| 254 | def test_rejection_no_email(self): |
||
| 255 | new_user = RegistrationProfile.objects.register(site=self.mock_site, |
||
| 256 | send_email=False, |
||
| 257 | **self.user_info) |
||
| 258 | |||
| 259 | profile = new_user.registration_profile |
||
| 260 | RegistrationProfile.objects.reject_registration(profile, |
||
| 261 | site=self.mock_site, |
||
| 262 | send_email=False) |
||
| 263 | |||
| 264 | self.assertEqual(len(mail.outbox), 0) |
||
| 265 | |||
| 266 | def test_acceptance_after_rejection_success(self): |
||
| 267 | new_user = RegistrationProfile.objects.register(site=self.mock_site, |
||
| 268 | send_email=False, |
||
| 269 | **self.user_info) |
||
| 270 | |||
| 271 | profile = new_user.registration_profile |
||
| 272 | |||
| 273 | # reject |
||
| 274 | result = RegistrationProfile.objects.reject_registration( |
||
| 275 | profile, site=self.mock_site |
||
| 276 | ) |
||
| 277 | self.failUnless(result) |
||
| 278 | self.assertEqual(profile.status, 'rejected') |
||
| 279 | self.assertEqual(profile.activation_key, None) |
||
| 280 | |||
| 281 | # accept should work even after rejection |
||
| 282 | result = RegistrationProfile.objects.accept_registration( |
||
| 283 | profile, site=self.mock_site |
||
| 284 | ) |
||
| 285 | self.failUnless(result) |
||
| 286 | self.assertEqual(profile.status, 'accepted') |
||
| 287 | self.assertNotEqual(profile.activation_key, None) |
||
| 288 | |||
| 289 | View Code Duplication | def test_acceptance_after_acceptance_fail(self): |
|
|
|
|||
| 290 | new_user = RegistrationProfile.objects.register(site=self.mock_site, |
||
| 291 | send_email=False, |
||
| 292 | **self.user_info) |
||
| 293 | |||
| 294 | profile = new_user.registration_profile |
||
| 295 | |||
| 296 | # accept |
||
| 297 | result = RegistrationProfile.objects.accept_registration( |
||
| 298 | profile, site=self.mock_site |
||
| 299 | ) |
||
| 300 | self.failUnless(result) |
||
| 301 | self.assertEqual(profile.status, 'accepted') |
||
| 302 | self.assertNotEqual(profile.activation_key, None) |
||
| 303 | |||
| 304 | # accept should not work |
||
| 305 | result = RegistrationProfile.objects.accept_registration( |
||
| 306 | profile, site=self.mock_site |
||
| 307 | ) |
||
| 308 | self.failIf(result) |
||
| 309 | self.assertEqual(profile.status, 'accepted') |
||
| 310 | self.assertNotEqual(profile.activation_key, None) |
||
| 311 | |||
| 312 | def test_rejection_after_acceptance_fail(self): |
||
| 313 | new_user = RegistrationProfile.objects.register(site=self.mock_site, |
||
| 314 | send_email=False, |
||
| 315 | **self.user_info) |
||
| 316 | |||
| 317 | profile = new_user.registration_profile |
||
| 318 | |||
| 319 | # accept |
||
| 320 | result = RegistrationProfile.objects.accept_registration( |
||
| 321 | profile, site=self.mock_site |
||
| 322 | ) |
||
| 323 | self.failUnless(result) |
||
| 324 | self.assertEqual(profile.status, 'accepted') |
||
| 325 | self.assertNotEqual(profile.activation_key, None) |
||
| 326 | |||
| 327 | # reject should not work |
||
| 328 | result = RegistrationProfile.objects.reject_registration( |
||
| 329 | profile, site=self.mock_site |
||
| 330 | ) |
||
| 331 | self.failIf(result) |
||
| 332 | self.assertEqual(profile.status, 'accepted') |
||
| 333 | self.assertNotEqual(profile.activation_key, None) |
||
| 334 | |||
| 335 | View Code Duplication | def test_rejection_after_rejection_fail(self): |
|
| 336 | new_user = RegistrationProfile.objects.register(site=self.mock_site, |
||
| 337 | send_email=False, |
||
| 338 | **self.user_info) |
||
| 339 | |||
| 340 | profile = new_user.registration_profile |
||
| 341 | |||
| 342 | # accept |
||
| 343 | result = RegistrationProfile.objects.reject_registration( |
||
| 344 | profile, site=self.mock_site |
||
| 345 | ) |
||
| 346 | self.failUnless(result) |
||
| 347 | self.assertEqual(profile.status, 'rejected') |
||
| 348 | self.assertEqual(profile.activation_key, None) |
||
| 349 | |||
| 350 | # reject should not work |
||
| 351 | result = RegistrationProfile.objects.reject_registration( |
||
| 352 | profile, site=self.mock_site |
||
| 353 | ) |
||
| 354 | self.failIf(result) |
||
| 355 | self.assertEqual(profile.status, 'rejected') |
||
| 356 | self.assertEqual(profile.activation_key, None) |
||
| 357 | |||
| 358 | View Code Duplication | def test_activation_with_password(self): |
|
| 359 | new_user = RegistrationProfile.objects.register(site=self.mock_site, |
||
| 360 | send_email=False, |
||
| 361 | **self.user_info) |
||
| 362 | |||
| 363 | profile = new_user.registration_profile |
||
| 364 | RegistrationProfile.objects.accept_registration( |
||
| 365 | profile, site=self.mock_site, send_email=False |
||
| 366 | ) |
||
| 367 | activated = RegistrationProfile.objects.activate_user( |
||
| 368 | activation_key=profile.activation_key, |
||
| 369 | site=self.mock_site, |
||
| 370 | password='swordfish', |
||
| 371 | send_email=False) |
||
| 372 | |||
| 373 | self.failUnless(activated) |
||
| 374 | |||
| 375 | activated_user, password, is_generated = activated |
||
| 376 | |||
| 377 | self.assertEqual(new_user, activated_user) |
||
| 378 | self.assertEqual(password, 'swordfish') |
||
| 379 | self.assertEqual(is_generated, False) |
||
| 380 | # the user should be activated with the password |
||
| 381 | self.failUnless(activated_user.is_active) |
||
| 382 | self.failUnless(activated_user.has_usable_password()) |
||
| 383 | self.failUnless(activated_user.check_password(password)) |
||
| 384 | # inspection profile should be deleted |
||
| 385 | self.failIf(RegistrationProfile.objects.filter(pk=profile.pk).exists()) |
||
| 386 | |||
| 387 | View Code Duplication | def test_activation_without_password(self): |
|
| 388 | new_user = RegistrationProfile.objects.register(site=self.mock_site, |
||
| 389 | send_email=False, |
||
| 390 | **self.user_info) |
||
| 391 | |||
| 392 | profile = new_user.registration_profile |
||
| 393 | RegistrationProfile.objects.accept_registration( |
||
| 394 | profile, site=self.mock_site, send_email=False |
||
| 395 | ) |
||
| 396 | activated = RegistrationProfile.objects.activate_user( |
||
| 397 | activation_key=profile.activation_key, |
||
| 398 | site=self.mock_site, |
||
| 399 | send_email=False |
||
| 400 | ) |
||
| 401 | |||
| 402 | self.failUnless(activated) |
||
| 403 | |||
| 404 | activated_user, password, is_generated = activated |
||
| 405 | |||
| 406 | self.assertEqual(new_user, activated_user) |
||
| 407 | self.assertEqual(len(password), |
||
| 408 | settings.REGISTRATION_DEFAULT_PASSWORD_LENGTH) |
||
| 409 | self.assertEqual(is_generated, True) |
||
| 410 | # the user should be activated with the password |
||
| 411 | self.failUnless(activated_user.is_active) |
||
| 412 | self.failUnless(activated_user.has_usable_password()) |
||
| 413 | self.failUnless(activated_user.check_password(password)) |
||
| 414 | # inspection profile should be deleted |
||
| 415 | self.failIf(RegistrationProfile.objects.filter(pk=profile.pk).exists()) |
||
| 416 | |||
| 417 | View Code Duplication | def test_activation_email(self): |
|
| 418 | new_user = RegistrationProfile.objects.register(site=self.mock_site, |
||
| 419 | send_email=False, |
||
| 420 | **self.user_info) |
||
| 421 | |||
| 422 | profile = new_user.registration_profile |
||
| 423 | RegistrationProfile.objects.accept_registration( |
||
| 424 | profile, site=self.mock_site, send_email=False |
||
| 425 | ) |
||
| 426 | RegistrationProfile.objects.activate_user( |
||
| 427 | activation_key=profile.activation_key, |
||
| 428 | site=self.mock_site, |
||
| 429 | ) |
||
| 430 | |||
| 431 | self.assertEqual(len(mail.outbox), 1) |
||
| 432 | |||
| 433 | View Code Duplication | def test_activation_no_email(self): |
|
| 434 | new_user = RegistrationProfile.objects.register(site=self.mock_site, |
||
| 435 | send_email=False, |
||
| 436 | **self.user_info) |
||
| 437 | |||
| 438 | profile = new_user.registration_profile |
||
| 439 | RegistrationProfile.objects.accept_registration( |
||
| 440 | profile, site=self.mock_site, send_email=False |
||
| 441 | ) |
||
| 442 | RegistrationProfile.objects.activate_user( |
||
| 443 | activation_key=profile.activation_key, |
||
| 444 | site=self.mock_site, |
||
| 445 | send_email=False |
||
| 446 | ) |
||
| 447 | |||
| 448 | self.assertEqual(len(mail.outbox), 0) |
||
| 449 | |||
| 450 | View Code Duplication | def test_activation_with_untreated_fail(self): |
|
| 451 | new_user = RegistrationProfile.objects.register(site=self.mock_site, |
||
| 452 | send_email=False, |
||
| 453 | **self.user_info) |
||
| 454 | |||
| 455 | profile = new_user.registration_profile |
||
| 456 | |||
| 457 | result = RegistrationProfile.objects.activate_user( |
||
| 458 | activation_key=profile.activation_key, |
||
| 459 | site=self.mock_site, |
||
| 460 | password='swordfish' |
||
| 461 | ) |
||
| 462 | |||
| 463 | self.failIf(result) |
||
| 464 | # the user should not be activated |
||
| 465 | self.failIf(new_user.is_active) |
||
| 466 | self.failIf(new_user.has_usable_password()) |
||
| 467 | self.failIf(new_user.check_password('swordfish')) |
||
| 468 | # inspection profile should not be deleted |
||
| 469 | self.failUnless( |
||
| 470 | RegistrationProfile.objects.filter(pk=profile.pk).exists() |
||
| 471 | ) |
||
| 472 | |||
| 473 | View Code Duplication | def test_activation_with_rejected_fail(self): |
|
| 474 | new_user = RegistrationProfile.objects.register(site=self.mock_site, |
||
| 475 | send_email=False, |
||
| 476 | **self.user_info) |
||
| 477 | |||
| 478 | profile = new_user.registration_profile |
||
| 479 | RegistrationProfile.objects.reject_registration( |
||
| 480 | profile, site=self.mock_site |
||
| 481 | ) |
||
| 482 | |||
| 483 | result = RegistrationProfile.objects.activate_user( |
||
| 484 | activation_key=profile.activation_key, |
||
| 485 | site=self.mock_site, |
||
| 486 | password='swordfish' |
||
| 487 | ) |
||
| 488 | |||
| 489 | self.failIf(result) |
||
| 490 | # the user should not be activated |
||
| 491 | self.failIf(new_user.is_active) |
||
| 492 | self.failIf(new_user.has_usable_password()) |
||
| 493 | self.failIf(new_user.check_password('swordfish')) |
||
| 494 | # inspection profile should not be deleted |
||
| 495 | self.failUnless( |
||
| 496 | RegistrationProfile.objects.filter(pk=profile.pk).exists() |
||
| 497 | ) |
||
| 498 | |||
| 499 | View Code Duplication | def test_activation_with_expired_fail(self): |
|
| 500 | new_user = RegistrationProfile.objects.register(site=self.mock_site, |
||
| 501 | send_email=False, |
||
| 502 | **self.user_info) |
||
| 503 | |||
| 504 | profile = new_user.registration_profile |
||
| 505 | RegistrationProfile.objects.accept_registration( |
||
| 506 | profile, site=self.mock_site |
||
| 507 | ) |
||
| 508 | |||
| 509 | new_user.date_joined -= datetime.timedelta( |
||
| 510 | days=settings.ACCOUNT_ACTIVATION_DAYS + 1 |
||
| 511 | ) |
||
| 512 | new_user.save() |
||
| 513 | |||
| 514 | result = RegistrationProfile.objects.activate_user( |
||
| 515 | activation_key=profile.activation_key, |
||
| 516 | site=self.mock_site, |
||
| 517 | password='swordfish' |
||
| 518 | ) |
||
| 519 | |||
| 520 | self.failIf(result) |
||
| 521 | # the user should not be activated |
||
| 522 | self.failIf(new_user.is_active) |
||
| 523 | self.failIf(new_user.has_usable_password()) |
||
| 524 | self.failIf(new_user.check_password('swordfish')) |
||
| 525 | # inspection profile should not be deleted |
||
| 526 | self.failUnless( |
||
| 527 | RegistrationProfile.objects.filter(pk=profile.pk).exists() |
||
| 528 | ) |
||
| 529 | |||
| 530 | View Code Duplication | def test_activation_with_invalid_key_fail(self): |
|
| 531 | new_user = RegistrationProfile.objects.register(site=self.mock_site, |
||
| 532 | send_email=False, |
||
| 533 | **self.user_info) |
||
| 534 | |||
| 535 | profile = new_user.registration_profile |
||
| 536 | RegistrationProfile.objects.accept_registration( |
||
| 537 | profile, site=self.mock_site |
||
| 538 | ) |
||
| 539 | |||
| 540 | result = RegistrationProfile.objects.activate_user( |
||
| 541 | activation_key='foo', |
||
| 542 | site=self.mock_site, |
||
| 543 | password='swordfish' |
||
| 544 | ) |
||
| 545 | |||
| 546 | self.failIf(result) |
||
| 547 | # the user should not be activated |
||
| 548 | self.failIf(new_user.is_active) |
||
| 549 | self.failIf(new_user.has_usable_password()) |
||
| 550 | self.failIf(new_user.check_password('swordfish')) |
||
| 551 | # inspection profile should not be deleted |
||
| 552 | self.failUnless( |
||
| 553 | RegistrationProfile.objects.filter(pk=profile.pk).exists() |
||
| 554 | ) |
||
| 555 | |||
| 556 | View Code Duplication | def test_expired_user_deletion(self): |
|
| 557 | RegistrationProfile.objects.register( |
||
| 558 | username='new_untreated_user', |
||
| 559 | email='[email protected]', |
||
| 560 | site=self.mock_site, |
||
| 561 | ) |
||
| 562 | new_accepted_user = RegistrationProfile.objects.register( |
||
| 563 | username='new_accepted_user', |
||
| 564 | email='[email protected]', |
||
| 565 | site=self.mock_site, |
||
| 566 | ) |
||
| 567 | new_rejected_user = RegistrationProfile.objects.register( |
||
| 568 | username='new_rejected_user', |
||
| 569 | email='[email protected]', |
||
| 570 | site=self.mock_site, |
||
| 571 | ) |
||
| 572 | expired_untreated_user = RegistrationProfile.objects.register( |
||
| 573 | username='expired untreated user', |
||
| 574 | email='[email protected]', |
||
| 575 | site=self.mock_site, |
||
| 576 | ) |
||
| 577 | expired_accepted_user = RegistrationProfile.objects.register( |
||
| 578 | username='expired_accepted_user', |
||
| 579 | email='[email protected]', |
||
| 580 | site=self.mock_site, |
||
| 581 | ) |
||
| 582 | expired_rejected_user = RegistrationProfile.objects.register( |
||
| 583 | username='expired_rejected_user', |
||
| 584 | email='[email protected]', |
||
| 585 | site=self.mock_site, |
||
| 586 | ) |
||
| 587 | |||
| 588 | RegistrationProfile.objects.accept_registration( |
||
| 589 | new_accepted_user.registration_profile, |
||
| 590 | site=self.mock_site, |
||
| 591 | ) |
||
| 592 | RegistrationProfile.objects.reject_registration( |
||
| 593 | new_rejected_user.registration_profile, |
||
| 594 | site=self.mock_site, |
||
| 595 | ) |
||
| 596 | RegistrationProfile.objects.accept_registration( |
||
| 597 | expired_accepted_user.registration_profile, |
||
| 598 | site=self.mock_site, |
||
| 599 | ) |
||
| 600 | RegistrationProfile.objects.reject_registration( |
||
| 601 | expired_rejected_user.registration_profile, |
||
| 602 | site=self.mock_site, |
||
| 603 | ) |
||
| 604 | |||
| 605 | delta = datetime.timedelta(days=settings.ACCOUNT_ACTIVATION_DAYS+1) |
||
| 606 | |||
| 607 | expired_untreated_user.date_joined -= delta |
||
| 608 | expired_untreated_user.save() |
||
| 609 | expired_accepted_user.date_joined -= delta |
||
| 610 | expired_accepted_user.save() |
||
| 611 | expired_rejected_user.date_joined -= delta |
||
| 612 | expired_rejected_user.save() |
||
| 613 | |||
| 614 | RegistrationProfile.objects.delete_expired_users() |
||
| 615 | # Only expired_accepted_user is deleted |
||
| 616 | User = get_user_model() |
||
| 617 | self.assertEqual(RegistrationProfile.objects.count(), 5) |
||
| 618 | self.assertRaises(User.DoesNotExist, |
||
| 619 | User.objects.get, |
||
| 620 | username='expired_accepted_user') |
||
| 621 | |||
| 622 | View Code Duplication | def test_rejected_user_deletion(self): |
|
| 623 | RegistrationProfile.objects.register( |
||
| 624 | username='new_untreated_user', |
||
| 625 | email='[email protected]', |
||
| 626 | site=self.mock_site, |
||
| 627 | ) |
||
| 628 | new_accepted_user = RegistrationProfile.objects.register( |
||
| 629 | username='new_accepted_user', |
||
| 630 | email='[email protected]', |
||
| 631 | site=self.mock_site, |
||
| 632 | ) |
||
| 633 | new_rejected_user = RegistrationProfile.objects.register( |
||
| 634 | username='new_rejected_user', |
||
| 635 | email='[email protected]', |
||
| 636 | site=self.mock_site, |
||
| 637 | ) |
||
| 638 | expired_untreated_user = RegistrationProfile.objects.register( |
||
| 639 | username='expired untreated user', |
||
| 640 | email='[email protected]', |
||
| 641 | site=self.mock_site, |
||
| 642 | ) |
||
| 643 | expired_accepted_user = RegistrationProfile.objects.register( |
||
| 644 | username='expired_accepted_user', |
||
| 645 | email='[email protected]', |
||
| 646 | site=self.mock_site, |
||
| 647 | ) |
||
| 648 | expired_rejected_user = RegistrationProfile.objects.register( |
||
| 649 | username='expired_rejected_user', |
||
| 650 | email='[email protected]', |
||
| 651 | site=self.mock_site, |
||
| 652 | ) |
||
| 653 | |||
| 654 | RegistrationProfile.objects.accept_registration( |
||
| 655 | new_accepted_user.registration_profile, |
||
| 656 | site=self.mock_site, |
||
| 657 | ) |
||
| 658 | RegistrationProfile.objects.reject_registration( |
||
| 659 | new_rejected_user.registration_profile, |
||
| 660 | site=self.mock_site, |
||
| 661 | ) |
||
| 662 | RegistrationProfile.objects.accept_registration( |
||
| 663 | expired_accepted_user.registration_profile, |
||
| 664 | site=self.mock_site, |
||
| 665 | ) |
||
| 666 | RegistrationProfile.objects.reject_registration( |
||
| 667 | expired_rejected_user.registration_profile, |
||
| 668 | site=self.mock_site, |
||
| 669 | ) |
||
| 670 | |||
| 671 | delta = datetime.timedelta(days=settings.ACCOUNT_ACTIVATION_DAYS+1) |
||
| 672 | |||
| 673 | expired_untreated_user.date_joined -= delta |
||
| 674 | expired_untreated_user.save() |
||
| 675 | expired_accepted_user.date_joined -= delta |
||
| 676 | expired_accepted_user.save() |
||
| 677 | expired_rejected_user.date_joined -= delta |
||
| 678 | expired_rejected_user.save() |
||
| 679 | |||
| 680 | RegistrationProfile.objects.delete_rejected_users() |
||
| 681 | # new_rejected_user and expired_rejected_user are deleted |
||
| 682 | User = get_user_model() |
||
| 683 | self.assertEqual(RegistrationProfile.objects.count(), 4) |
||
| 684 | self.assertRaises(User.DoesNotExist, |
||
| 685 | User.objects.get, |
||
| 686 | username='new_rejected_user') |
||
| 687 | self.assertRaises(User.DoesNotExist, |
||
| 688 | User.objects.get, |
||
| 689 | username='expired_rejected_user') |
||
| 690 | |||
| 691 | View Code Duplication | def test_management_command_cleanup_expired_registrations(self): |
|
| 692 | RegistrationProfile.objects.register( |
||
| 693 | username='new_untreated_user', |
||
| 694 | email='[email protected]', |
||
| 695 | site=self.mock_site, |
||
| 696 | ) |
||
| 697 | new_accepted_user = RegistrationProfile.objects.register( |
||
| 698 | username='new_accepted_user', |
||
| 699 | email='[email protected]', |
||
| 700 | site=self.mock_site, |
||
| 701 | ) |
||
| 702 | new_rejected_user = RegistrationProfile.objects.register( |
||
| 703 | username='new_rejected_user', |
||
| 704 | email='[email protected]', |
||
| 705 | site=self.mock_site, |
||
| 706 | ) |
||
| 707 | expired_untreated_user = RegistrationProfile.objects.register( |
||
| 708 | username='expired untreated user', |
||
| 709 | email='[email protected]', |
||
| 710 | site=self.mock_site, |
||
| 711 | ) |
||
| 712 | expired_accepted_user = RegistrationProfile.objects.register( |
||
| 713 | username='expired_accepted_user', |
||
| 714 | email='[email protected]', |
||
| 715 | site=self.mock_site, |
||
| 716 | ) |
||
| 717 | expired_rejected_user = RegistrationProfile.objects.register( |
||
| 718 | username='expired_rejected_user', |
||
| 719 | email='[email protected]', |
||
| 720 | site=self.mock_site, |
||
| 721 | ) |
||
| 722 | |||
| 723 | RegistrationProfile.objects.accept_registration( |
||
| 724 | new_accepted_user.registration_profile, |
||
| 725 | site=self.mock_site, |
||
| 726 | ) |
||
| 727 | RegistrationProfile.objects.reject_registration( |
||
| 728 | new_rejected_user.registration_profile, |
||
| 729 | site=self.mock_site, |
||
| 730 | ) |
||
| 731 | RegistrationProfile.objects.accept_registration( |
||
| 732 | expired_accepted_user.registration_profile, |
||
| 733 | site=self.mock_site, |
||
| 734 | ) |
||
| 735 | RegistrationProfile.objects.reject_registration( |
||
| 736 | expired_rejected_user.registration_profile, |
||
| 737 | site=self.mock_site, |
||
| 738 | ) |
||
| 739 | |||
| 740 | delta = datetime.timedelta(days=settings.ACCOUNT_ACTIVATION_DAYS+1) |
||
| 741 | |||
| 742 | expired_untreated_user.date_joined -= delta |
||
| 743 | expired_untreated_user.save() |
||
| 744 | expired_accepted_user.date_joined -= delta |
||
| 745 | expired_accepted_user.save() |
||
| 746 | expired_rejected_user.date_joined -= delta |
||
| 747 | expired_rejected_user.save() |
||
| 748 | |||
| 749 | management.call_command('cleanup_expired_registrations') |
||
| 750 | # Only expired_accepted_user is deleted |
||
| 751 | User = get_user_model() |
||
| 752 | self.assertEqual(RegistrationProfile.objects.count(), 5) |
||
| 753 | self.assertRaises(User.DoesNotExist, User.objects.get, |
||
| 754 | username='expired_accepted_user') |
||
| 755 | |||
| 756 | View Code Duplication | def test_management_command_cleanup_rejected_registrations(self): |
|
| 757 | RegistrationProfile.objects.register( |
||
| 758 | username='new_untreated_user', |
||
| 759 | email='[email protected]', |
||
| 760 | site=self.mock_site, |
||
| 761 | ) |
||
| 762 | new_accepted_user = RegistrationProfile.objects.register( |
||
| 763 | username='new_accepted_user', |
||
| 764 | email='[email protected]', |
||
| 765 | site=self.mock_site, |
||
| 766 | ) |
||
| 767 | new_rejected_user = RegistrationProfile.objects.register( |
||
| 768 | username='new_rejected_user', |
||
| 769 | email='[email protected]', |
||
| 770 | site=self.mock_site, |
||
| 771 | ) |
||
| 772 | expired_untreated_user = RegistrationProfile.objects.register( |
||
| 773 | username='expired untreated user', |
||
| 774 | email='[email protected]', |
||
| 775 | site=self.mock_site, |
||
| 776 | ) |
||
| 777 | expired_accepted_user = RegistrationProfile.objects.register( |
||
| 778 | username='expired_accepted_user', |
||
| 779 | email='[email protected]', |
||
| 780 | site=self.mock_site, |
||
| 781 | ) |
||
| 782 | expired_rejected_user = RegistrationProfile.objects.register( |
||
| 783 | username='expired_rejected_user', |
||
| 784 | email='[email protected]', |
||
| 785 | site=self.mock_site, |
||
| 786 | ) |
||
| 787 | |||
| 788 | RegistrationProfile.objects.accept_registration( |
||
| 789 | new_accepted_user.registration_profile, |
||
| 790 | site=self.mock_site, |
||
| 791 | ) |
||
| 792 | RegistrationProfile.objects.reject_registration( |
||
| 793 | new_rejected_user.registration_profile, |
||
| 794 | site=self.mock_site, |
||
| 795 | ) |
||
| 796 | RegistrationProfile.objects.accept_registration( |
||
| 797 | expired_accepted_user.registration_profile, |
||
| 798 | site=self.mock_site, |
||
| 799 | ) |
||
| 800 | RegistrationProfile.objects.reject_registration( |
||
| 801 | expired_rejected_user.registration_profile, |
||
| 802 | site=self.mock_site, |
||
| 803 | ) |
||
| 804 | |||
| 805 | delta = datetime.timedelta(days=settings.ACCOUNT_ACTIVATION_DAYS+1) |
||
| 806 | |||
| 807 | expired_untreated_user.date_joined -= delta |
||
| 808 | expired_untreated_user.save() |
||
| 809 | expired_accepted_user.date_joined -= delta |
||
| 810 | expired_accepted_user.save() |
||
| 811 | expired_rejected_user.date_joined -= delta |
||
| 812 | expired_rejected_user.save() |
||
| 813 | |||
| 814 | management.call_command('cleanup_rejected_registrations') |
||
| 815 | # new_rejected_user and expired_rejected_user are deleted |
||
| 816 | User = get_user_model() |
||
| 817 | self.assertEqual(RegistrationProfile.objects.count(), 4) |
||
| 818 | self.assertRaises(User.DoesNotExist, User.objects.get, |
||
| 819 | username='new_rejected_user') |
||
| 820 | self.assertRaises(User.DoesNotExist, User.objects.get, |
||
| 821 | username='expired_rejected_user') |
||
| 822 | |||
| 823 | def test_management_command_cleanup_registrations(self): |
||
| 824 | RegistrationProfile.objects.register( |
||
| 825 | username='new_untreated_user', |
||
| 826 | email='[email protected]', |
||
| 827 | site=self.mock_site, |
||
| 828 | ) |
||
| 829 | new_accepted_user = RegistrationProfile.objects.register( |
||
| 830 | username='new_accepted_user', |
||
| 831 | email='[email protected]', |
||
| 832 | site=self.mock_site, |
||
| 833 | ) |
||
| 834 | new_rejected_user = RegistrationProfile.objects.register( |
||
| 835 | username='new_rejected_user', |
||
| 836 | email='[email protected]', |
||
| 837 | site=self.mock_site, |
||
| 838 | ) |
||
| 839 | expired_untreated_user = RegistrationProfile.objects.register( |
||
| 840 | username='expired untreated user', |
||
| 841 | email='[email protected]', |
||
| 842 | site=self.mock_site, |
||
| 843 | ) |
||
| 844 | expired_accepted_user = RegistrationProfile.objects.register( |
||
| 845 | username='expired_accepted_user', |
||
| 846 | email='[email protected]', |
||
| 847 | site=self.mock_site, |
||
| 848 | ) |
||
| 849 | expired_rejected_user = RegistrationProfile.objects.register( |
||
| 850 | username='expired_rejected_user', |
||
| 851 | email='[email protected]', |
||
| 852 | site=self.mock_site, |
||
| 853 | ) |
||
| 854 | |||
| 855 | RegistrationProfile.objects.accept_registration( |
||
| 856 | new_accepted_user.registration_profile, |
||
| 857 | site=self.mock_site, |
||
| 858 | ) |
||
| 859 | RegistrationProfile.objects.reject_registration( |
||
| 860 | new_rejected_user.registration_profile, |
||
| 861 | site=self.mock_site, |
||
| 862 | ) |
||
| 863 | RegistrationProfile.objects.accept_registration( |
||
| 864 | expired_accepted_user.registration_profile, |
||
| 865 | site=self.mock_site, |
||
| 866 | ) |
||
| 867 | RegistrationProfile.objects.reject_registration( |
||
| 868 | expired_rejected_user.registration_profile, |
||
| 869 | site=self.mock_site, |
||
| 870 | ) |
||
| 871 | |||
| 872 | delta = datetime.timedelta(days=settings.ACCOUNT_ACTIVATION_DAYS+1) |
||
| 873 | |||
| 874 | expired_untreated_user.date_joined -= delta |
||
| 875 | expired_untreated_user.save() |
||
| 876 | expired_accepted_user.date_joined -= delta |
||
| 877 | expired_accepted_user.save() |
||
| 878 | expired_rejected_user.date_joined -= delta |
||
| 879 | expired_rejected_user.save() |
||
| 880 | |||
| 881 | management.call_command('cleanup_registrations') |
||
| 882 | # new_rejected_user, expired rejected_user and expired_accepted_user |
||
| 883 | # are deleted |
||
| 884 | User = get_user_model() |
||
| 885 | self.assertEqual(RegistrationProfile.objects.count(), 3) |
||
| 886 | self.assertRaises(User.DoesNotExist, User.objects.get, |
||
| 887 | username='new_rejected_user') |
||
| 888 | self.assertRaises(User.DoesNotExist, User.objects.get, |
||
| 889 | username='expired_rejected_user') |
||
| 890 | self.assertRaises(User.DoesNotExist, User.objects.get, |
||
| 891 | username='expired_accepted_user') |
||
| 892 | |||
| 893 | def test_management_command_cleanupregistration(self): |
||
| 894 | RegistrationProfile.objects.register( |
||
| 895 | username='new_untreated_user', |
||
| 896 | email='[email protected]', |
||
| 897 | site=self.mock_site, |
||
| 898 | ) |
||
| 899 | new_accepted_user = RegistrationProfile.objects.register( |
||
| 900 | username='new_accepted_user', |
||
| 901 | email='[email protected]', |
||
| 902 | site=self.mock_site, |
||
| 903 | ) |
||
| 904 | new_rejected_user = RegistrationProfile.objects.register( |
||
| 905 | username='new_rejected_user', |
||
| 906 | email='[email protected]', |
||
| 907 | site=self.mock_site, |
||
| 908 | ) |
||
| 909 | expired_untreated_user = RegistrationProfile.objects.register( |
||
| 910 | username='expired untreated user', |
||
| 911 | email='[email protected]', |
||
| 912 | site=self.mock_site, |
||
| 913 | ) |
||
| 914 | expired_accepted_user = RegistrationProfile.objects.register( |
||
| 915 | username='expired_accepted_user', |
||
| 916 | email='[email protected]', |
||
| 917 | site=self.mock_site, |
||
| 918 | ) |
||
| 919 | expired_rejected_user = RegistrationProfile.objects.register( |
||
| 920 | username='expired_rejected_user', |
||
| 921 | email='[email protected]', |
||
| 922 | site=self.mock_site, |
||
| 923 | ) |
||
| 924 | |||
| 925 | RegistrationProfile.objects.accept_registration( |
||
| 926 | new_accepted_user.registration_profile, |
||
| 927 | site=self.mock_site, |
||
| 928 | ) |
||
| 929 | RegistrationProfile.objects.reject_registration( |
||
| 930 | new_rejected_user.registration_profile, |
||
| 931 | site=self.mock_site, |
||
| 932 | ) |
||
| 933 | RegistrationProfile.objects.accept_registration( |
||
| 934 | expired_accepted_user.registration_profile, |
||
| 935 | site=self.mock_site, |
||
| 936 | ) |
||
| 937 | RegistrationProfile.objects.reject_registration( |
||
| 938 | expired_rejected_user.registration_profile, |
||
| 939 | site=self.mock_site, |
||
| 940 | ) |
||
| 941 | |||
| 942 | delta = datetime.timedelta(days=settings.ACCOUNT_ACTIVATION_DAYS+1) |
||
| 943 | |||
| 944 | expired_untreated_user.date_joined -= delta |
||
| 945 | expired_untreated_user.save() |
||
| 946 | expired_accepted_user.date_joined -= delta |
||
| 947 | expired_accepted_user.save() |
||
| 948 | expired_rejected_user.date_joined -= delta |
||
| 949 | expired_rejected_user.save() |
||
| 950 | |||
| 951 | # django-registration compatibility |
||
| 952 | management.call_command('cleanupregistration') |
||
| 953 | # new_rejected_user, expired rejected_user and expired_accepted_user |
||
| 954 | # are deleted |
||
| 955 | User = get_user_model() |
||
| 956 | self.assertEqual(RegistrationProfile.objects.count(), 3) |
||
| 957 | self.assertRaises(User.DoesNotExist, User.objects.get, |
||
| 958 | username='new_rejected_user') |
||
| 959 | self.assertRaises(User.DoesNotExist, User.objects.get, |
||
| 960 | username='expired_rejected_user') |
||
| 961 | self.assertRaises(User.DoesNotExist, User.objects.get, |
||
| 962 | username='expired_accepted_user') |
||
| 963 |