Exemple #1
0
def emulator(request) -> None:
    # Lazy imports here to avoid slow top-level import when running pytest for any tests
    # pylint: disable=import-outside-toplevel
    from recidiviz.utils import pubsub_helper
    from recidiviz.ingest.scrape import sessions

    datastore_emulator, pubsub_emulator = _start_emulators()

    # These environment variables can only be set while these tests are
    # running as they also impact ndb/testbed behavior
    prior_environs = _write_emulator_environs()
    sessions.clear_ds()
    pubsub_helper.clear_publisher()
    pubsub_helper.clear_subscriber()

    def cleanup() -> None:
        datastore_emulator.terminate()
        pubsub_emulator.terminate()

        _restore_environs(prior_environs)
        sessions.clear_ds()
        pubsub_helper.clear_publisher()
        pubsub_helper.clear_subscriber()

    request.addfinalizer(cleanup)
Exemple #2
0
    def cleanup() -> None:
        datastore_emulator.terminate()
        pubsub_emulator.terminate()

        _restore_environs(prior_environs)
        sessions.clear_ds()
        pubsub_helper.clear_publisher()
        pubsub_helper.clear_subscriber()
Exemple #3
0
    def cleanup():
        os.killpg(datastore_emulator.pid, signal.SIGTERM)
        os.killpg(pubsub_emulator.pid, signal.SIGTERM)

        _restore_environs(prior_environs)
        sessions.clear_ds()
        pubsub_helper.clear_publisher()
        pubsub_helper.clear_subscriber()
Exemple #4
0
def emulator(request) -> None:
    datastore_emulator, pubsub_emulator = _start_emulators()

    # These environment variables can only be set while these tests are
    # running as they also impact ndb/testbed behavior
    prior_environs = _write_emulator_environs()
    sessions.clear_ds()
    pubsub_helper.clear_publisher()
    pubsub_helper.clear_subscriber()

    def cleanup() -> None:
        datastore_emulator.terminate()
        pubsub_emulator.terminate()

        _restore_environs(prior_environs)
        sessions.clear_ds()
        pubsub_helper.clear_publisher()
        pubsub_helper.clear_subscriber()

    request.addfinalizer(cleanup)
Exemple #5
0
def emulator(request):
    datastore_emulator, pubsub_emulator = _start_emulators()

    # These environment variables can only be set while these tests are
    # running as they also impact ndb/testbed behavior
    prior_environs = _write_emulator_environs()
    sessions.clear_ds()
    secrets.clear_ds()
    pubsub_helper.clear_publisher()
    pubsub_helper.clear_subscriber()

    def cleanup():
        os.killpg(datastore_emulator.pid, signal.SIGTERM)
        os.killpg(pubsub_emulator.pid, signal.SIGTERM)

        _restore_environs(prior_environs)
        sessions.clear_ds()
        secrets.clear_ds()
        pubsub_helper.clear_publisher()
        pubsub_helper.clear_subscriber()

    request.addfinalizer(cleanup)
 def teardown_method(self, _test_method):
     sessions.clear_ds()
 def setup_method(self, _test_method):
     sessions.clear_ds()
 def teardown_method(self, _test_method: Callable) -> None:
     sessions.clear_ds()
 def setup_method(self, _test_method: Callable) -> None:
     sessions.clear_ds()