def __modify_domain_in_database(self):
modified_domain = Database.get_instance().db['domains'].get(
self.domain_name, None)
modified_domain.add_associated_ip(self.domain_ip)
Database.get_instance().db['domains'][
self.domain_name] = modified_domain
return modified_domain
def persist_external_domain_with_two_ips():
domain = ExternalDomain(DEFAULT_DOMAIN_NAME, DEFAULT_TTL)
domain.add_associated_ip(DEFAULT_EXTERNAL_DOMAIN_IP_1)
domain.add_associated_ip(DEFAULT_EXTERNAL_DOMAIN_IP_2)
domains = Database.get_instance().db['domains']
domains[domain.name] = domain
return domain
def validate_time_to_live(self):
valid_domain = self.domain.is_valid()
if not valid_domain:
self.domain = self.dns_query_service.get_response_as_external_domain(
)
Database.get_instance().db['domains'][
self.domain.name] = self.domain
def do_expire_ttl(domain):
Database.get_instance().db['domains'][domain.name].time_to_live = 0
time.sleep(0.1)
def persist_custom_domain():
domain = CustomDomain(DEFAULT_DOMAIN_NAME)
domain.add_associated_ip(DEFAULT_DOMAIN_IP_1)
domains = Database.get_instance().db['domains']
domains[domain.name] = domain
return domain
def tearDown(self):
super().tearDown()
Database.get_instance().db = {'domains': {}}
def get_custom_domains(self):
database_domains = Database.get_instance().db['domains'].values()
for current_domain in database_domains:
if current_domain.is_custom(
) and self.domain_substring in current_domain.name:
self.domains.append(current_domain)
def __custom_domain_already_exists(self):
domain = Database.get_instance().db['domains'].get(
self.domain_name, None)
if domain is not None and domain.is_custom():
return True
return False
def __delete_domain_from_database(self):
deleted_domain = Database.get_instance().db['domains'].pop(
self.domain_name, None)
return deleted_domain
def __add_domain_to_database(self):
new_domain = CustomDomain(self.domain_name)
new_domain.add_associated_ip(self.domain_ip)
Database.get_instance().db['domains'][self.domain_name] = new_domain
return new_domain
def __init__(self, domain_name, dns_query_service):
self.domain_name = domain_name
self.domain = Database.get_instance().db['domains'].get(
self.domain_name)
self.dns_query_service = dns_query_service