You've already forked mariadb-columnstore-engine
mirror of
https://github.com/mariadb-corporation/mariadb-columnstore-engine.git
synced 2025-07-29 08:21:15 +03:00
MCOL-5496: Merge CMAPI code to engine repo.
[add] cmapi code to engine
This commit is contained in:
committed by
Alan Mologorsky
parent
77eedd1756
commit
a079a2c944
29
cmapi/mcs_node_control/test/test_dbrm_socket.py
Normal file
29
cmapi/mcs_node_control/test/test_dbrm_socket.py
Normal file
@ -0,0 +1,29 @@
|
||||
import io
|
||||
import logging
|
||||
import unittest
|
||||
|
||||
from mcs_node_control.models.dbrm_socket import MAGIC_BYTES, DBRMSocketHandler
|
||||
|
||||
|
||||
logging.basicConfig(level='DEBUG')
|
||||
|
||||
|
||||
class TestDBRMSocketHandler(unittest.TestCase):
|
||||
|
||||
def test_myreceive_to_magic(self):
|
||||
response_data = b'\x01\x00\x00\x00\x00'
|
||||
valid_magic = b'%s%s' % (MAGIC_BYTES, response_data)
|
||||
first_unknow = b'A%s%s' % (MAGIC_BYTES, response_data)
|
||||
partial_first_magic = b'%s%s%s' % (
|
||||
MAGIC_BYTES[:3], MAGIC_BYTES, response_data
|
||||
)
|
||||
sock_responses = [valid_magic, first_unknow, partial_first_magic]
|
||||
for sock_response in sock_responses:
|
||||
with self.subTest(sock_response=sock_response):
|
||||
data_stream = io.BytesIO(sock_response)
|
||||
data_stream.recv = data_stream.read
|
||||
dbrm_socket = DBRMSocketHandler()
|
||||
# pylint: disable=protected-access
|
||||
dbrm_socket._socket = data_stream
|
||||
dbrm_socket._receive_magic()
|
||||
self.assertEqual(data_stream.read(), response_data)
|
Reference in New Issue
Block a user