def test_statement_expected_result_type(self): valid_inputs = [ SqlExpectedResultType.ROWS, SqlExpectedResultType.UPDATE_COUNT, "ROWS", "ANY", ] for valid in valid_inputs: statement = _SqlStatement("something", []) statement.expected_result_type = valid self.assertEqual( try_to_get_enum_value(valid, SqlExpectedResultType), statement.expected_result_type) invalid_inputs = [None, 123, "hey"] for invalid in invalid_inputs: with self.assertRaises(TypeError): statement = _SqlStatement("something") statement.expected_result_type = invalid
def expected_result_type(self, expected_result_type): self._expected_result_type = try_to_get_enum_value( expected_result_type, SqlExpectedResultType)
def reconnect_mode(self, value): self._reconnect_mode = try_to_get_enum_value(value, ReconnectMode)
def ssl_protocol(self, value): self._ssl_protocol = try_to_get_enum_value(value, SSLProtocol)
def type(self, value): self._type = try_to_get_enum_value(value, IndexType)
def unique_key_transformation(self, value): self._unique_key_transformation = try_to_get_enum_value(value, UniqueKeyTransformation)
def unique_key(self, value): self._unique_key = try_to_get_enum_value(value, QueryConstants)
def overload_policy(self, value): self._overload_policy = try_to_get_enum_value(value, TopicOverloadPolicy)
def eviction_policy(self, value): self._eviction_policy = try_to_get_enum_value(value, EvictionPolicy)
def in_memory_format(self, value): self._in_memory_format = try_to_get_enum_value(value, InMemoryFormat)
def default_int_type(self, value): self._default_int_type = try_to_get_enum_value(value, IntType)