Python get_celery_task_signature Examples

Example #1

File: test_sql_injection_protection.py Project: 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"

Example #2

File: test_tables.py Project: 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()

Example #3

File: test_local_global_nodes_flow.py Project: 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(

Example #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

Example #5

File: test_remote_tables.py Project: 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()