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)