def test_read_daterange(self):
test_file_offset = 0
test_row_id = 1
test_columns_info = []
col = DbColumn()
col.data_type = datatypes.DATATYPE_DATERANGE
test_columns_info.append(col)
res = self._daterange_reader.read_row(test_file_offset, test_row_id, test_columns_info)
self.assertEqual(str(self._daterange_test_value), str(res[0].raw_object))
def test_read_float2(self):
test_file_offset = 0
test_row_id = 1
test_columns_info = []
col = DbColumn()
col.data_type = datatypes.DATATYPE_REAL
test_columns_info.append(col)
res = self._float_reader2.read_row(test_file_offset, test_row_id, test_columns_info)
self.assertEqual(self._float_test_value2, res[0].raw_object)
def test_read_bytea(self):
test_file_offset = 0
test_row_id = 1
test_columns_info = []
col = DbColumn()
col.data_type = datatypes.DATATYPE_BYTEA
test_columns_info.append(col)
res = self._bytea_reader.read_row(test_file_offset, test_row_id, test_columns_info)
expected = self._bytea_test_value.tobytes()
actual = str(res[0].raw_object)
self.assertEqual(str(expected), actual)
def test_read_json(self):
"""Test json/jsonb string is returned as is"""
test_file_offset = 0
test_row_id = 1
for datatype in [datatypes.DATATYPE_JSON, datatypes.DATATYPE_JSONB]:
col = DbColumn()
col.data_type = datatype
test_columns_info = [col]
reader = ServiceBufferFileStreamReader(self._dict_file_stream)
res = reader.read_row(test_file_offset, test_row_id, test_columns_info)
self.assertEqual(1, len(res))
self.assertEqual(json.dumps(self._dict_test_value), res[0].raw_object)
def test_read_dict(self):
test_file_offset = 0
test_row_id = 1
test_columns_info = []
col = DbColumn()
col.data_type = datatypes.DATATYPE_JSON
test_columns_info.append(col)
res = self._dict_reader.read_row(test_file_offset, test_row_id,
test_columns_info)
actual_raw_object = res[0].raw_object
expected1 = self._dict_test_value["Ser,ver"]
actual1 = actual_raw_object["Ser,ver"]
expected2 = self._dict_test_value["Sche'ma"]
actual2 = actual_raw_object["Sche'ma"]
self.assertEqual(expected1, actual1)
self.assertEqual(expected2, actual2)
def test_read_multiple_cols(self):
test_file_offset = 0
test_row_id = 1
test_columns_info = []
real_column1 = DbColumn()
real_column1.data_type = datatypes.DATATYPE_REAL
integer_column = DbColumn()
integer_column.data_type = datatypes.DATATYPE_INTEGER
text_column = DbColumn()
text_column.data_type = datatypes.DATATYPE_TEXT
real_column2 = DbColumn()
real_column2.data_type = datatypes.DATATYPE_REAL
test_columns_info.append(real_column1)
test_columns_info.append(integer_column)
test_columns_info.append(text_column)
test_columns_info.append(real_column2)
res = self._multiple_cols_reader.read_row(test_file_offset,
test_row_id,
test_columns_info)
self.assertEqual(self._float_test_value1, res[0].raw_object)
self.assertEqual(self._int_test_value, res[1].raw_object)
self.assertEqual(self._str_test_value, res[2].raw_object)
self.assertEqual(self._float_test_value2, res[3].raw_object)