mirror of
https://github.com/cisagov/manage.get.gov.git
synced 2025-05-18 18:39:21 +02:00
283 lines
11 KiB
Python
283 lines
11 KiB
Python
from io import StringIO
|
|
|
|
from django.test import TestCase
|
|
|
|
from registrar.models import (
|
|
User,
|
|
Domain,
|
|
DomainInvitation,
|
|
TransitionDomain,
|
|
DomainInformation,
|
|
UserDomainRole,
|
|
)
|
|
|
|
from django.core.management import call_command
|
|
|
|
from .common import less_console_noise
|
|
|
|
|
|
class TestLogins(TestCase):
|
|
|
|
""" """
|
|
|
|
def setUp(self):
|
|
""" """
|
|
# self.load_transition_domain_script = "load_transition_domain",
|
|
# self.transfer_script = "transfer_transition_domains_to_domains",
|
|
# self.master_script = "load_transition_domain",
|
|
|
|
self.test_data_file_location = "/app/registrar/tests/data"
|
|
self.test_domain_contact_filename = "test_domain_contacts.txt"
|
|
self.test_contact_filename = "test_contacts.txt"
|
|
self.test_domain_status_filename = "test_domain_statuses.txt"
|
|
|
|
def tearDown(self):
|
|
super().tearDown()
|
|
TransitionDomain.objects.all().delete()
|
|
Domain.objects.all().delete()
|
|
DomainInvitation.objects.all().delete()
|
|
DomainInformation.objects.all().delete()
|
|
User.objects.all().delete()
|
|
UserDomainRole.objects.all().delete()
|
|
|
|
def run_load_domains(self):
|
|
call_command(
|
|
"load_transition_domain",
|
|
f"{self.test_data_file_location}/{self.test_domain_contact_filename}",
|
|
f"{self.test_data_file_location}/{self.test_contact_filename}",
|
|
f"{self.test_data_file_location}/{self.test_domain_status_filename}",
|
|
)
|
|
|
|
def run_transfer_domains(self):
|
|
call_command("transfer_transition_domains_to_domains")
|
|
|
|
def run_master_script(self):
|
|
call_command(
|
|
"master_domain_migrations",
|
|
runMigrations=True,
|
|
migrationDirectory=f"{self.test_data_file_location}",
|
|
migrationFilenames=(
|
|
f"{self.test_domain_contact_filename},"
|
|
f"{self.test_contact_filename},"
|
|
f"{self.test_domain_status_filename}"
|
|
),
|
|
)
|
|
|
|
def compare_tables(
|
|
self,
|
|
expected_total_transition_domains,
|
|
expected_total_domains,
|
|
expected_total_domain_informations,
|
|
expected_total_domain_invitations,
|
|
expected_missing_domains,
|
|
expected_duplicate_domains,
|
|
expected_missing_domain_informations,
|
|
expected_missing_domain_invitations,
|
|
):
|
|
"""Does a diff between the transition_domain and the following tables:
|
|
domain, domain_information and the domain_invitation.
|
|
Verifies that the data loaded correctly."""
|
|
|
|
missing_domains = []
|
|
duplicate_domains = []
|
|
missing_domain_informations = []
|
|
missing_domain_invites = []
|
|
for transition_domain in TransitionDomain.objects.all(): # DEBUG:
|
|
transition_domain_name = transition_domain.domain_name
|
|
transition_domain_email = transition_domain.username
|
|
|
|
# Check Domain table
|
|
matching_domains = Domain.objects.filter(name=transition_domain_name)
|
|
# Check Domain Information table
|
|
matching_domain_informations = DomainInformation.objects.filter(
|
|
domain__name=transition_domain_name
|
|
)
|
|
# Check Domain Invitation table
|
|
matching_domain_invitations = DomainInvitation.objects.filter(
|
|
email=transition_domain_email.lower(),
|
|
domain__name=transition_domain_name,
|
|
)
|
|
|
|
if len(matching_domains) == 0:
|
|
missing_domains.append(transition_domain_name)
|
|
elif len(matching_domains) > 1:
|
|
duplicate_domains.append(transition_domain_name)
|
|
if len(matching_domain_informations) == 0:
|
|
missing_domain_informations.append(transition_domain_name)
|
|
if len(matching_domain_invitations) == 0:
|
|
missing_domain_invites.append(transition_domain_name)
|
|
|
|
total_missing_domains = len(missing_domains)
|
|
total_duplicate_domains = len(duplicate_domains)
|
|
total_missing_domain_informations = len(missing_domain_informations)
|
|
total_missing_domain_invitations = len(missing_domain_invites)
|
|
|
|
total_transition_domains = len(TransitionDomain.objects.all())
|
|
total_domains = len(Domain.objects.all())
|
|
total_domain_informations = len(DomainInformation.objects.all())
|
|
total_domain_invitations = len(DomainInvitation.objects.all())
|
|
|
|
print(
|
|
f"""
|
|
total_missing_domains = {len(missing_domains)}
|
|
total_duplicate_domains = {len(duplicate_domains)}
|
|
total_missing_domain_informations = {len(missing_domain_informations)}
|
|
total_missing_domain_invitations = {len(missing_domain_invites)}
|
|
|
|
total_transition_domains = {len(TransitionDomain.objects.all())}
|
|
total_domains = {len(Domain.objects.all())}
|
|
total_domain_informations = {len(DomainInformation.objects.all())}
|
|
total_domain_invitations = {len(DomainInvitation.objects.all())}
|
|
"""
|
|
)
|
|
|
|
self.assertTrue(total_missing_domains == expected_missing_domains)
|
|
self.assertTrue(total_duplicate_domains == expected_duplicate_domains)
|
|
self.assertTrue(
|
|
total_missing_domain_informations == expected_missing_domain_informations
|
|
)
|
|
self.assertTrue(
|
|
total_missing_domain_invitations == expected_missing_domain_invitations
|
|
)
|
|
|
|
self.assertTrue(total_transition_domains == expected_total_transition_domains)
|
|
self.assertTrue(total_domains == expected_total_domains)
|
|
self.assertTrue(total_domain_informations == expected_total_domain_informations)
|
|
self.assertTrue(total_domain_invitations == expected_total_domain_invitations)
|
|
|
|
def test_master_migration_functions(self):
|
|
"""Run the full master migration script using local test data.
|
|
NOTE: This is more of an integration test and so far does not
|
|
follow best practice of limiting the number of assertions per test.
|
|
But for now, this will double-check that the script
|
|
works as intended."""
|
|
|
|
self.run_master_script()
|
|
|
|
# STEP 2: (analyze the tables just like the
|
|
# migration script does, but add assert statements)
|
|
expected_total_transition_domains = 8
|
|
expected_total_domains = 4
|
|
expected_total_domain_informations = 0
|
|
expected_total_domain_invitations = 7
|
|
|
|
expected_missing_domains = 0
|
|
expected_duplicate_domains = 0
|
|
# we expect 8 missing domain invites since the
|
|
# migration does not auto-login new users
|
|
expected_missing_domain_informations = 8
|
|
# we expect 1 missing invite from anomaly.gov (an injected error)
|
|
expected_missing_domain_invitations = 1
|
|
self.compare_tables(
|
|
expected_total_transition_domains,
|
|
expected_total_domains,
|
|
expected_total_domain_informations,
|
|
expected_total_domain_invitations,
|
|
expected_missing_domains,
|
|
expected_duplicate_domains,
|
|
expected_missing_domain_informations,
|
|
expected_missing_domain_invitations,
|
|
)
|
|
|
|
def test_load_transition_domain(self):
|
|
self.run_load_domains()
|
|
|
|
# STEP 2: (analyze the tables just like the migration
|
|
# script does, but add assert statements)
|
|
expected_total_transition_domains = 8
|
|
expected_total_domains = 0
|
|
expected_total_domain_informations = 0
|
|
expected_total_domain_invitations = 0
|
|
|
|
expected_missing_domains = 8
|
|
expected_duplicate_domains = 0
|
|
expected_missing_domain_informations = 8
|
|
expected_missing_domain_invitations = 8
|
|
self.compare_tables(
|
|
expected_total_transition_domains,
|
|
expected_total_domains,
|
|
expected_total_domain_informations,
|
|
expected_total_domain_invitations,
|
|
expected_missing_domains,
|
|
expected_duplicate_domains,
|
|
expected_missing_domain_informations,
|
|
expected_missing_domain_invitations,
|
|
)
|
|
|
|
def test_transfer_transition_domains_to_domains(self):
|
|
# TODO: setup manually instead of calling other script
|
|
self.run_load_domains()
|
|
self.run_transfer_domains()
|
|
|
|
# Analyze the tables
|
|
expected_total_transition_domains = 8
|
|
expected_total_domains = 4
|
|
expected_total_domain_informations = 0
|
|
expected_total_domain_invitations = 7
|
|
|
|
expected_missing_domains = 0
|
|
expected_duplicate_domains = 0
|
|
expected_missing_domain_informations = 8
|
|
expected_missing_domain_invitations = 1
|
|
self.compare_tables(
|
|
expected_total_transition_domains,
|
|
expected_total_domains,
|
|
expected_total_domain_informations,
|
|
expected_total_domain_invitations,
|
|
expected_missing_domains,
|
|
expected_duplicate_domains,
|
|
expected_missing_domain_informations,
|
|
expected_missing_domain_invitations,
|
|
)
|
|
|
|
def test_logins(self):
|
|
# TODO: setup manually instead of calling other scripts
|
|
self.run_load_domains()
|
|
self.run_transfer_domains()
|
|
|
|
# Simluate Logins
|
|
for invite in DomainInvitation.objects.all():
|
|
# get a user with this email address
|
|
user, user_created = User.objects.get_or_create(
|
|
email=invite.email, username=invite.email
|
|
)
|
|
user.first_login()
|
|
|
|
# Analyze the tables
|
|
expected_total_transition_domains = 8
|
|
expected_total_domains = 4
|
|
expected_total_domain_informations = 3
|
|
expected_total_domain_invitations = 7
|
|
|
|
expected_missing_domains = 0
|
|
expected_duplicate_domains = 0
|
|
expected_missing_domain_informations = 1
|
|
expected_missing_domain_invitations = 1
|
|
self.compare_tables(
|
|
expected_total_transition_domains,
|
|
expected_total_domains,
|
|
expected_total_domain_informations,
|
|
expected_total_domain_invitations,
|
|
expected_missing_domains,
|
|
expected_duplicate_domains,
|
|
expected_missing_domain_informations,
|
|
expected_missing_domain_invitations,
|
|
)
|
|
|
|
def test_send_domain_invitations_email(self):
|
|
"""Can send only a single domain invitation email."""
|
|
with less_console_noise():
|
|
self.run_load_domains()
|
|
self.run_transfer_domains()
|
|
|
|
# this is one of the email addresses in data/test_contacts.txt
|
|
output_stream = StringIO()
|
|
# also have to re-point the logging handlers to output_stream
|
|
with less_console_noise(output_stream):
|
|
call_command("send_domain_invitations", "testuser@gmail.com", stdout=output_stream)
|
|
|
|
# Check that we had the right numbers in our output
|
|
output = output_stream.getvalue()
|
|
# should only be one domain we send email for
|
|
self.assertIn("Found 1 transition domains", output)
|
|
self.assertTrue("would send email to testuser@gmail.com", output)
|