def test_cannot_set_socket_and_port():
workspace_yaml = """
load_from:
- grpc_server:
socket: myname
port: 5678
"""
with pytest.raises(CheckError,
match="must supply either a socket or a port"):
location_origins_from_config(
yaml.safe_load(workspace_yaml),
# fake out as if it were loaded by a yaml file in this directory
file_relative_path(__file__, "not_a_real.yaml"),
)
def test_ssl_grpc_server_workspace():
server_process = GrpcServerProcess(force_port=True)
try:
with server_process.create_ephemeral_client() as client:
assert client.heartbeat(echo="Hello")
port = server_process.port
ssl_yaml = f"""
load_from:
- grpc_server:
host: localhost
port: {port}
ssl: true
"""
origins = location_origins_from_config(
yaml.safe_load(ssl_yaml),
# fake out as if it were loaded by a yaml file in this directory
file_relative_path(__file__, "not_a_real.yaml"),
)
origin = list(origins.values())[0]
assert origin.use_ssl
# Actually connecting to the server will fail since it's expecting SSL
# and we didn't set up the server with SSL
try:
with origin.create_location():
assert False
except grpc._channel._InactiveRpcError: # pylint: disable=protected-access
pass
finally:
server_process.wait()
def test_multi_location_origins(config_source):
fake_executable = "/var/fake/executable"
origins = location_origins_from_config(
yaml.safe_load(config_source(fake_executable)),
file_relative_path(__file__, "not_a_real.yaml"),
)
assert len(origins) == 6
assert sorted(origins.keys()) == sorted([
"loaded_from_file",
"loaded_from_module",
"named_loaded_from_file",
"named_loaded_from_module",
"named_loaded_from_module_attribute",
"named_loaded_from_file_attribute",
])
assert all([
origin.loadable_target_origin.executable_path == fake_executable
for origin in origins.values()
])
assert origins[
"loaded_from_file"].loadable_target_origin.python_file == file_relative_path(
__file__, "hello_world_repository.py")
assert (origins["loaded_from_module"].loadable_target_origin.module_name ==
"dagster.utils.test.hello_world_repository")
assert (origins["named_loaded_from_file_attribute"].loadable_target_origin.
attribute == "named_hello_world_repository")
assert (origins["named_loaded_from_module_attribute"].
loadable_target_origin.attribute == "named_hello_world_repository")
def test_grpc_socket_workspace():
first_server_process = GrpcServerProcess()
with first_server_process.create_ephemeral_client() as first_server:
second_server_process = GrpcServerProcess()
with second_server_process.create_ephemeral_client() as second_server:
first_socket = first_server.socket
second_socket = second_server.socket
workspace_yaml = """
load_from:
- grpc_server:
host: localhost
socket: {socket_one}
- grpc_server:
socket: {socket_two}
location_name: 'local_port_default_host'
""".format(socket_one=first_socket, socket_two=second_socket)
origins = location_origins_from_config(
yaml.safe_load(workspace_yaml),
# fake out as if it were loaded by a yaml file in this directory
file_relative_path(__file__, "not_a_real.yaml"),
)
with ExitStack() as stack:
repository_location_handles = {
name: stack.enter_context(origin.create_handle())
for name, origin in origins.items()
}
assert len(repository_location_handles) == 2
default_location_name = "grpc:localhost:{socket}".format(
socket=first_socket)
assert repository_location_handles.get(default_location_name)
local_port = repository_location_handles.get(
default_location_name)
assert local_port.socket == first_socket
assert local_port.host == "localhost"
assert local_port.port is None
assert repository_location_handles.get(
"local_port_default_host")
local_port_default_host = repository_location_handles.get(
"local_port_default_host")
assert local_port_default_host.socket == second_socket
assert local_port_default_host.host == "localhost"
assert local_port_default_host.port is None
assert all(
map(lambda x: x.location_name,
repository_location_handles.values()))
second_server_process.wait()
first_server_process.wait()
def test_grpc_multi_location_workspace(config_source):
origins = location_origins_from_config(
yaml.safe_load(config_source(sys.executable)),
# fake out as if it were loaded by a yaml file in this directory
file_relative_path(__file__, "not_a_real.yaml"),
)
with ExitStack() as stack:
repository_locations = {
name: stack.enter_context(origin.create_test_location())
for name, origin in origins.items()
}
assert len(repository_locations) == 6
assert "loaded_from_file" in repository_locations
assert "loaded_from_module" in repository_locations
loaded_from_file_location = repository_locations.get("loaded_from_file")
assert isinstance(loaded_from_file_location, GrpcServerRepositoryLocation)
assert loaded_from_file_location.repository_names == {"hello_world_repository"}
loaded_from_module_location = repository_locations.get("loaded_from_module")
assert isinstance(loaded_from_module_location, GrpcServerRepositoryLocation)
assert loaded_from_module_location.repository_names == {"hello_world_repository"}
named_loaded_from_file_location = repository_locations.get("named_loaded_from_file")
assert named_loaded_from_file_location.repository_names == {"hello_world_repository_name"}
assert isinstance(named_loaded_from_file_location, GrpcServerRepositoryLocation)
named_loaded_from_module_location = repository_locations.get("named_loaded_from_module")
assert named_loaded_from_module_location.repository_names == {"hello_world_repository_name"}
assert isinstance(named_loaded_from_module_location, GrpcServerRepositoryLocation)
named_loaded_from_module_attribute_location = repository_locations.get(
"named_loaded_from_module_attribute"
)
assert named_loaded_from_module_attribute_location.repository_names == {
"hello_world_repository_name"
}
assert isinstance(named_loaded_from_module_attribute_location, GrpcServerRepositoryLocation)
named_loaded_from_file_attribute_location = repository_locations.get(
"named_loaded_from_file_attribute"
)
assert named_loaded_from_file_attribute_location.repository_names == {
"hello_world_repository_name"
}
assert isinstance(named_loaded_from_file_attribute_location, GrpcServerRepositoryLocation)
def test_grpc_server_env_vars():
with environ({
"FOO_PORT": "1234",
"FOO_SOCKET": "barsocket",
"FOO_HOST": "barhost",
}):
valid_yaml = """
load_from:
- grpc_server:
host:
env: FOO_HOST
port:
env: FOO_PORT
location_name: 'my_grpc_server_port'
- grpc_server:
host:
env: FOO_HOST
socket:
env: FOO_SOCKET
location_name: 'my_grpc_server_socket'
"""
origins = location_origins_from_config(
yaml.safe_load(valid_yaml),
file_relative_path(__file__, "not_a_real.yaml"),
)
assert len(origins) == 2
port_origin = origins["my_grpc_server_port"]
assert isinstance(origins["my_grpc_server_port"],
GrpcServerRepositoryLocationOrigin)
assert port_origin.port == 1234
assert port_origin.host == "barhost"
socket_origin = origins["my_grpc_server_socket"]
assert isinstance(origins["my_grpc_server_socket"],
GrpcServerRepositoryLocationOrigin)
assert socket_origin.socket == "barsocket"
assert socket_origin.host == "barhost"