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)
