zydronium/manage.get.gov
1
0
Fork 0
mirror of https://github.com/cisagov/manage.get.gov.git synced 2025-07-07 11:43:30 +02:00
Code Issues Projects Releases Packages Wiki Activity

Test changes

Browse source
This commit is contained in:
zandercymatics 2023-10-17 12:40:15 -06:00
parent b6921b3f4c
commit 0dde277c86
No known key found for this signature in database
GPG key ID: FF4636ABEC9682B7
2 changed files with 28 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

17
src/epplibwrapper/tests/test_pool.py
View file

@ -3,9 +3,11 @@ from unittest.mock import MagicMock, patch
from django.conf import settings
from django.test import TestCase
from epplibwrapper.client import EPPLibWrapper
from epplibwrapper.utility.pool import EPPConnectionPool
from registrar.models.domain import Domain
from registrar.models.domain import registry
from contextlib import ExitStack
import logging
@ -120,16 +122,23 @@ class TestConnectionPool(TestCase):
# Fake data for the _pool object
domain, _ = Domain.objects.get_or_create(name="freeman.gov")
def start_fake_connection(self):
registry.pool_status.pool_running = True
registry.pool_status.connection_success = True
registry._pool = registry.get_pool()
# The connection pool will fail to start, start it manually
# so that our mocks can take over
registry.start_connection_pool(try_start_if_invalid=True)
with ExitStack() as stack:
stack.enter_context(patch.object(EPPLibWrapper, "get_pool", self.fake_pool))
stack.enter_context(patch.object(EPPLibWrapper, "start_connection_pool", start_fake_connection))
expected_contact = domain.security_contact
# Pretend that we've connected
registry.pool_status.pool_running = True
registry.pool_status.connection_success = True
# Trigger the getter - should succeed
expected_contact = domain.security_contact
self.assertEqual(registry.pool_status.pool_running, True)
self.assertEqual(registry.pool_status.connection_success, True)
self.assertEqual(len(registry._pool.conn), self.pool_options["size"])
self.assertEqual(len(registry.get_pool().conn), self.pool_options["size"])