def test_server_version(stub): # --- given --- c = pymkts.GRPCClient() # --- when --- c.server_version() # --- then --- assert c.stub.ServerVersion.called == 1
def test_list_symbols(stub): # --- given --- c = pymkts.GRPCClient() # --- when --- c.list_symbols() # --- then --- assert c.stub.ListSymbols.called == 1
def test_destroy(stub): # --- given --- c = pymkts.GRPCClient() tbk = 'TEST/1Min/TICK' # --- when --- c.destroy(tbk) # --- then --- assert c.stub.Destroy.called == 1
def test_query(stub): # --- given --- c = pymkts.GRPCClient() p = pymkts.Params('BTC', '1Min', 'OHLCV') # --- when --- c.query(p) # --- then --- assert c.stub.Query.called == 1
def test_sql(stub): # --- given --- c = pymkts.GRPCClient() s = "SELECT * FROM `BTC/1Min/OHLCV`" # --- when --- c.sql(s) # --- then --- assert c.stub.Query.called == 1
def test_write(stub): # --- given --- c = pymkts.GRPCClient() data = np.array([(1, 0)], dtype=[('Epoch', 'i8'), ('Ask', 'f4')]) tbk = 'TEST/1Min/TICK' # --- when --- c.write(data, tbk) # --- then --- assert c.stub.Write.called == 1
def test_create(stub): # --- given --- c = pymkts.GRPCClient() dtype = [('Epoch', 'i8'), ('Bid', 'f4'), ('Ask', 'f4')] tbk = 'TEST/1Min/TICK' # --- when --- c.create(tbk=tbk, dtype=dtype, isvariablelength=False) # --- then --- assert c.stub.Create.called == 1
def test_create(stub): # --- given --- c = pymkts.GRPCClient() dtype = [('Epoch', 'i8'), ('Bid', 'f4'), ('Ask', 'f4')] tbk = 'TEST/1Min/TICK' # --- when --- c.create(tbk=tbk, data_shape=pymkts.DataShape(dtype)) # --- then --- assert c.stub.Create.called == 1
def test_build_query(): # --- given --- c = pymkts.GRPCClient(endpoint="127.0.0.1:5995") p = pymkts.Params('TSLA', '1Min', 'OHLCV', 1500000000, 4294967296) # --- when --- query = c.build_query([p]) # --- then --- assert query == MultiQueryRequest(requests=[ QueryRequest(destination="TSLA/1Min/OHLCV", epoch_start=1500000000, epoch_end=4294967296) ])
def test_grpc_client_init(): c = pymkts.GRPCClient("127.0.0.1:5995") assert c.endpoint == "127.0.0.1:5995" assert isinstance(c.stub, marketstore_pb2_grpc.MarketstoreStub)