def test_read_socket_from_environment_variables(): fake_socket = 'unix:///path/to/endpoint.sock' os.environ['SPIFFE_ENDPOINT_SOCKET'] = fake_socket setter = ConfigSetter() assert setter.get_config().spiffe_endpoint_socket == fake_socket
def test_socket_parameter_preponderance_over_environment_variable(): fake_socket = 'unix:///path/to/endpoint.sock' os.environ['SPIFFE_ENDPOINT_SOCKET'] = 'env_var_socket' setter = ConfigSetter(spiffe_endpoint_socket=fake_socket) assert setter.get_config().spiffe_endpoint_socket == fake_socket
def __init__(self, spiffe_socket: str = None) -> None: """Creates a new Workload API Client. Args: spiffe_socket: Path to Workload API UDS. If not specified, the SPIFFE_ENDPOINT_SOCKET environment variable must be set. Returns: DefaultWorkloadApiClient: New Workload API Client object. Raises: ArgumentError: If spiffe_socket_path is invalid or not provided and SPIFFE_ENDPOINT_SOCKET environment variable doesn't exist. """ try: self._config = ConfigSetter( spiffe_endpoint_socket=spiffe_socket).get_config() except ArgumentError as e: raise ArgumentError( 'Invalid DefaultWorkloadApiClient configuration: {}'.format( str(e))) self._channel = self._get_spiffe_grpc_channel() self._spiffe_workload_api_stub = workload_pb2_grpc.SpiffeWorkloadAPIStub( self._channel)
def __init__(self, spiffe_socket: str = None) -> None: """Creates a new Workload API Client. Args: spiffe_socket: Path to Workload API UDS. If not specified, the SPIFFE_ENDPOINT_SOCKET environment variable must be set. Returns: DefaultWorkloadApiClient: New Workload API Client object. Raises: ValueError: If spiffe_socket_path is invalid or not provided and SPIFFE_ENDPOINT_SOCKET environment variable doesn't exist. """ try: self._config = ConfigSetter( spiffe_endpoint_socket=spiffe_socket).get_config() except ValueError: raise ValueError( 'SPIFFE socket argument or environment variable invalid in DefaultWorkloadApiClient.' )
def test_path_scheme_is_valid_tcp(): fake_socket = 'tcp://127.0.0.1:8000' setter = ConfigSetter(spiffe_endpoint_socket=fake_socket) assert setter.get_config().spiffe_endpoint_socket == fake_socket
def test_path_scheme_is_valid_unix(): fake_socket = 'unix:///path/to/endpoint.sock' setter = ConfigSetter(spiffe_endpoint_socket=fake_socket) assert setter.get_config().spiffe_endpoint_socket == fake_socket
def test_pass_socket_as_parameter(): fake_socket = 'unix:///path/to/endpoint.sock' setter = ConfigSetter(spiffe_endpoint_socket=fake_socket) assert setter.get_config().spiffe_endpoint_socket == fake_socket
def test_socket_must_be_set(): with pytest.raises(ValueError) as exception: ConfigSetter() assert str(exception.value) == 'SPIFFE endpoint socket: socket must be set.'
def test_invalid_endpoint_socket(test_input, expected): with pytest.raises(ValueError) as exception: ConfigSetter(spiffe_endpoint_socket=test_input) assert str(exception.value) == expected