Python get_celery_task_signature Exemples

Exemple #1

Fichier : test_sql_injection_protection.py Projet : ThanKarab/MIP-Engine

import json
import uuid

import pytest

from mipengine.common.node_tasks_DTOs import ColumnInfo, TableSchema
from tests.integration_tests.nodes_communication import get_celery_app
from tests.integration_tests.nodes_communication import get_celery_task_signature

local_node = get_celery_app("localnode1")
local_node_create_table = get_celery_task_signature(local_node, "create_table")
local_node_get_tables = get_celery_task_signature(local_node, "get_tables")
local_node_get_table_schema = get_celery_task_signature(
    local_node, "get_table_schema")
local_node_get_table_data = get_celery_task_signature(local_node,
                                                      "get_table_data")
local_node_create_merge_table = get_celery_task_signature(
    local_node, "create_merge_table")
local_node_get_merge_tables = get_celery_task_signature(
    local_node, "get_merge_tables")
local_node_create_pathology_view = get_celery_task_signature(
    local_node, "create_pathology_view")
local_node_get_views = get_celery_task_signature(local_node, "get_views")
local_node_create_remote_table = get_celery_task_signature(
    local_node, "create_remote_table")
local_node_get_remote_tables = get_celery_task_signature(
    local_node, "get_remote_tables")
local_node_cleanup = get_celery_task_signature(local_node, "clean_up")

context_id = "HISTOGRAMS"

Exemple #2

Fichier : test_tables.py Projet : madgik/MIP-Engine

import pytest
import uuid as uuid

from mipengine.common.node_tasks_DTOs import ColumnInfo
from mipengine.common.node_tasks_DTOs import TableData
from mipengine.common.node_tasks_DTOs import TableSchema
from tests.integration_tests.nodes_communication import get_celery_app
from tests.integration_tests.nodes_communication import get_celery_task_signature

local_node = get_celery_app("localnode1")
local_node_create_table = get_celery_task_signature(local_node, "create_table")
local_node_get_tables = get_celery_task_signature(local_node, "get_tables")
local_node_insert_data_to_table = get_celery_task_signature(
    local_node, "insert_data_to_table")

local_node_get_table_schema = get_celery_task_signature(
    local_node, "get_table_schema")
local_node_get_table_data = get_celery_task_signature(local_node,
                                                      "get_table_data")
local_node_cleanup = get_celery_task_signature(local_node, "clean_up")


@pytest.fixture(autouse=True)
def context_id():
    context_id = "test_tables_" + str(uuid.uuid4()).replace("-", "")

    yield context_id

    local_node_cleanup.delay(context_id=context_id).get()

Exemple #3

Fichier : test_local_global_nodes_flow.py Projet : ThanKarab/MIP-Engine

from mipengine.common.node_catalog import node_catalog
from mipengine.common.node_tasks_DTOs import ColumnInfo, TableData
from mipengine.common.node_tasks_DTOs import TableInfo
from mipengine.common.node_tasks_DTOs import TableSchema
from tests.integration_tests.nodes_communication import get_celery_task_signature
from tests.integration_tests.nodes_communication import get_celery_app

local_node_1_id = "localnode1"
local_node_2_id = "localnode2"
global_node_id = "globalnode"
local_node_1 = get_celery_app(local_node_1_id)
local_node_2 = get_celery_app(local_node_2_id)
global_node = get_celery_app(global_node_id)

local_node_1_create_table = get_celery_task_signature(local_node_1,
                                                      "create_table")
local_node_2_create_table = get_celery_task_signature(local_node_2,
                                                      "create_table")
local_node_1_insert_data_to_table = get_celery_task_signature(
    local_node_1, "insert_data_to_table")
local_node_2_insert_data_to_table = get_celery_task_signature(
    local_node_2, "insert_data_to_table")
global_node_create_remote_table = get_celery_task_signature(
    global_node, "create_remote_table")
global_node_get_remote_tables = get_celery_task_signature(
    global_node, "get_remote_tables")
global_node_create_merge_table = get_celery_task_signature(
    global_node, "create_merge_table")
global_node_get_merge_tables = get_celery_task_signature(
    global_node, "get_merge_tables")
global_node_get_merge_table_data = get_celery_task_signature(

Exemple #4

import uuid

import pytest

from mipengine.common.node_tasks_DTOs import ColumnInfo
from mipengine.common.node_tasks_DTOs import TableData
from mipengine.common.node_tasks_DTOs import TableSchema
from tests.integration_tests.nodes_communication import get_celery_task_signature
from tests.integration_tests.nodes_communication import get_celery_app

local_node_id = "localnode1"
local_node = get_celery_app(local_node_id)
local_node_create_table = get_celery_task_signature(local_node, "create_table")
local_node_insert_data_to_table = get_celery_task_signature(
    local_node, "insert_data_to_table")
local_node_create_pathology_view = get_celery_task_signature(
    local_node, "create_pathology_view")
local_node_create_view = get_celery_task_signature(local_node, "create_view")
local_node_get_views = get_celery_task_signature(local_node, "get_views")
local_node_get_view_data = get_celery_task_signature(local_node,
                                                     "get_table_data")
local_node_get_view_schema = get_celery_task_signature(local_node,
                                                       "get_table_schema")
local_node_cleanup = get_celery_task_signature(local_node, "clean_up")


@pytest.fixture(autouse=True)
def context_id():
    context_id = "test_views_" + str(uuid.uuid4()).replace("-", "")

    yield context_id

Exemple #5

Fichier : test_remote_tables.py Projet : madgik/MIP-Engine

import uuid

import pytest

from mipengine.common.node_tasks_DTOs import ColumnInfo
from mipengine.common.node_tasks_DTOs import TableInfo
from mipengine.common.node_tasks_DTOs import TableSchema
from tests.integration_tests.nodes_communication import get_celery_task_signature
from tests.integration_tests.nodes_communication import get_celery_app
from mipengine.node_registry import NodeRegistryClient

global_node_id = "globalnode"
local_node_id = "localnode1"
global_node = get_celery_app(global_node_id)
local_node = get_celery_app(local_node_id)
local_node_create_table = get_celery_task_signature(local_node, "create_table")
global_node_create_remote_table = get_celery_task_signature(
    global_node, "create_remote_table")
global_node_get_remote_tables = get_celery_task_signature(
    global_node, "get_remote_tables")
local_node_cleanup = get_celery_task_signature(local_node, "clean_up")
global_node_cleanup = get_celery_task_signature(global_node, "clean_up")


@pytest.fixture(autouse=True)
def context_id():
    context_id = "test_remote_tables_" + str(uuid.uuid4()).replace("-", "")

    yield context_id

    local_node_cleanup.delay(context_id=context_id.lower()).get()