async def test_get_tentacle_nodes_on_exchange_name_and_tentacle_type():
matrix = Matrix()
Matrices.instance().add_matrix(matrix)
created_node_1 = matrix.matrix.get_or_create_node(
get_tentacle_path(tentacle_type="NO_TYPE",
tentacle_name="Test-TA",
exchange_name="binance"))
created_node_2 = matrix.matrix.get_or_create_node(
get_tentacle_path(tentacle_type="TEST_TYPE",
tentacle_name="Test-TA-2",
exchange_name="binance"))
assert get_tentacle_nodes(matrix.matrix_id, exchange_name="binance") == [
matrix.get_node_at_path(
get_tentacle_path(exchange_name="binance",
tentacle_type="NO_TYPE")),
matrix.get_node_at_path(
get_tentacle_path(exchange_name="binance",
tentacle_type="TEST_TYPE"))
]
assert get_tentacle_nodes(matrix.matrix_id,
exchange_name="binance",
tentacle_type="NO_TYPE") == [created_node_1]
assert get_tentacle_nodes(matrix.matrix_id,
exchange_name="binance",
tentacle_type="TEST_TYPE") == [created_node_2]
assert get_tentacle_nodes(matrix.matrix_id, exchange_name="bitfinex") == []
Matrices.instance().del_matrix(matrix.matrix_id)
async def test_get_tentacle_nodes_on_tentacle_type():
matrix = Matrix()
Matrices.instance().add_matrix(matrix)
created_node_1 = matrix.matrix.get_or_create_node(
get_tentacle_path(tentacle_type="NO_TYPE", tentacle_name="Test-TA"))
created_node_2 = matrix.matrix.get_or_create_node(
get_tentacle_path(tentacle_type="TEST_TYPE",
tentacle_name="Test-TA-2"))
assert get_tentacle_nodes(matrix.matrix_id, tentacle_type="TA") == []
assert get_tentacle_nodes(matrix.matrix_id,
tentacle_type="NO_TYPE") == [created_node_1]
assert get_tentacle_nodes(matrix.matrix_id,
tentacle_type="TEST_TYPE") == [created_node_2]
Matrices.instance().del_matrix(matrix.matrix_id)
async def test_get_tentacle_nodes_on_multiple_tentacle_type():
matrix = Matrix()
Matrices.instance().add_matrix(matrix)
created_node_1 = matrix.matrix.get_or_create_node(
get_tentacle_path(tentacle_type="TA", tentacle_name="Test-TA"))
created_node_2 = matrix.matrix.get_or_create_node(
get_tentacle_path(tentacle_type="TA", tentacle_name="Test-TA-2"))
created_node_3 = matrix.matrix.get_or_create_node(
get_tentacle_path(tentacle_type="TA", tentacle_name="Test-TA-3"))
assert get_tentacle_nodes(matrix.matrix_id, tentacle_type="TA") == [
created_node_1, created_node_2, created_node_3
]
Matrices.instance().del_matrix(matrix.matrix_id)
async def test_get_tentacle_nodes_mixed():
matrix = Matrix()
Matrices.instance().add_matrix(matrix)
created_node_1 = matrix.matrix.get_or_create_node(
get_tentacle_path(tentacle_type="TA",
tentacle_name="Test-TA",
exchange_name="binance"))
created_node_2 = matrix.matrix.get_or_create_node(
get_tentacle_path(tentacle_type="TA",
tentacle_name="Test-TA-2",
exchange_name="bitfinex"))
created_node_3 = matrix.matrix.get_or_create_node(
get_tentacle_path(tentacle_type="TA",
tentacle_name="Test-TA-3",
exchange_name="binance"))
created_node_4 = matrix.matrix.get_or_create_node(
get_tentacle_path(tentacle_type="TEST-TYPE",
tentacle_name="Test-TA-4",
exchange_name="binance"))
created_node_5 = matrix.matrix.get_or_create_node(
get_tentacle_path(tentacle_type="TEST-TYPE",
tentacle_name="Test-TA-5",
exchange_name="bitfinex"))
created_node_6 = matrix.matrix.get_or_create_node(
get_tentacle_path(tentacle_type="TEST-TYPE",
tentacle_name="Test-TA-6"))
created_node_7 = matrix.matrix.get_or_create_node(
get_tentacle_path(tentacle_name="Test-TA-7", exchange_name="bitfinex"))
created_node_8 = matrix.matrix.get_or_create_node(
get_tentacle_path(tentacle_name="Test-TA-8"))
created_node_9 = matrix.matrix.get_or_create_node(
get_tentacle_path(tentacle_name="Test-TA-9", exchange_name="binance"))
assert get_tentacle_nodes(matrix.matrix_id,
exchange_name="binance",
tentacle_type="TA") == [
created_node_1, created_node_3
]
assert get_tentacle_nodes(matrix.matrix_id,
exchange_name="binance",
tentacle_type="TEST-TYPE") == [created_node_4]
assert get_tentacle_nodes(matrix.matrix_id,
exchange_name="bitfinex",
tentacle_type="TEST-TYPE") == [created_node_5]
assert get_tentacle_nodes(matrix.matrix_id, exchange_name="bitfinex") == [
matrix.get_node_at_path(
get_tentacle_path(exchange_name="bitfinex", tentacle_type="TA")),
matrix.get_node_at_path(
get_tentacle_path(exchange_name="bitfinex",
tentacle_type="TEST-TYPE")), created_node_7
]
assert get_tentacle_nodes(matrix.matrix_id,
tentacle_type="TEST-TYPE") == [created_node_6]
assert get_tentacle_nodes(matrix.matrix_id, exchange_name="binance") == [
matrix.get_node_at_path(
get_tentacle_path(exchange_name="binance", tentacle_type="TA")),
matrix.get_node_at_path(
get_tentacle_path(exchange_name="binance",
tentacle_type="TEST-TYPE")), created_node_9
]
Matrices.instance().del_matrix(matrix.matrix_id)