예제 #1
0
def test_get_tracker_store_from_endpoint_config_error_exit(tmp_path: Path):
    # write config without event broker to file
    endpoints_path = write_endpoint_config_to_yaml(tmp_path, {})

    available_endpoints = rasa_core_utils.read_endpoints_from_path(endpoints_path)

    with pytest.raises(SystemExit):
        # noinspection PyProtectedMember
        assert export._get_tracker_store(available_endpoints)
예제 #2
0
def test_get_event_broker_and_tracker_store_from_endpoint_config(
        tmp_path: Path):
    # write valid config to file
    endpoints_path = write_endpoint_config_to_yaml(tmp_path, {
        "event_broker": {
            "type": "sql"
        },
        "tracker_store": {
            "type": "sql"
        }
    })

    available_endpoints = rasa_core_utils.read_endpoints_from_path(
        endpoints_path)

    # fetching the event broker is successful
    assert export._get_event_broker(available_endpoints)
    assert export._get_tracker_store(available_endpoints)