Python Container示例

示例#1

文件： test_s3_get_command.py 项目： iwayamak/jetline

 def setUpClass(cls) -> None:
     queue = CommandQueue()
     queue.add_command(TouchFileCommand(PREPARE_TEST_DATA_PATH))
     queue.add_command(
         S3PutCommand(Container().component('S3_COMPONENT.ID=UT'),
                      PREPARE_TEST_DATA_PATH,
                      os.path.basename(PREPARE_TEST_DATA_PATH)))
     queue.execute()

示例#2

文件： test_s3_put_command.py 项目： iwayamak/jetline

 def __init__(self, *args, **kwargs):
     self._component = \
         Container().component('S3_COMPONENT.ID=UT')
     self._test_data_path = \
         os.path.join(
             os.path.dirname(__file__), 'test_data'
         )
     super().__init__(*args, **kwargs)

示例#3

文件： test_scp_get_command.py 项目： iwayamak/jetline

 def setUpClass(cls) -> None:
     queue = CommandQueue()
     for test_data_path in TEST_DATA_PATH_LIST:
         queue.add_command(TouchFileCommand(test_data_path))
     queue.add_command(
         ScpPutCommand(Container().component('SCP_COMPONENT.ID=UT'),
                       TEST_DATA_PATH_LIST, REMOTE_DIR, False, False))
     queue.execute()

示例#4

文件： test_postgresql_processing_count_command.py 项目： iwayamak/jetline

 def __init__(self, *args, **kwargs):
     self._component = Container().component('POSTGRESQL_COMPONENT.ID=UT')
     self._sql_str = (
         'select count(*) from '
         '(select current_timestamp'
         ' union all'
         ' select current_timestamp union all select current_timestamp) t;')
     super().__init__(*args, **kwargs)

示例#5

 def __init__(self, *args, **kwargs):
     self._component = Container().component('CONFLUENCE_COMPONENT.ID=UT')
     self._page_title = os.path.splitext(os.path.basename(__file__))[0]
     self._space_key = 'SPC'
     self._headers = {'content-type': 'application/json'}
     self._type = 'page'
     self._ancestors_id = '2392085'
     self._html_str = f'<div style="font-size:500%">{self._page_title}</div>'
     self._description = f'created by {os.path.basename(__file__)}'
     super().__init__(*args, **kwargs)

示例#6

 def __init__(self, *args, **kwargs):
     self._component = Container().component('CONFLUENCE_COMPONENT.ID=UT')
     self._page_title = os.path.splitext(os.path.basename(__file__))[0]
     self._space_key = 'SPC'
     self._headers = {'content-type': 'application/json'}
     self._type = 'page'
     self._ancestors_id = '2392085'
     self._description = f'created by {os.path.basename(__file__)}'
     self._test_data_path = \
         os.path.join(
             os.path.dirname(__file__), TEST_DATA_DIR
         )
     super().__init__(*args, **kwargs)

示例#7

 def setUpClass(cls) -> None:
     queue = CommandQueue()
     for test_data_path in TEST_DATA_PATH_LIST:
         queue.add_command(
             TouchFileCommand(test_data_path)
         )
         queue.add_command(
             S3PutCommand(
                 Container().component('S3_COMPONENT.ID=UT'),
                 test_data_path,
                 f'test_s3_list_command/{os.path.basename(test_data_path)}'
             )
         )
     queue.execute()

示例#8

 def __init__(self, kwargs: dict):
     component = Container.component(kwargs['component_key'])
     sql_str = FileUtil.file_to_str(kwargs['sql_file_name'])
     self._output_dir = kwargs['output_dir']
     super().__init__(component, sql_str)

示例#9

文件： test_s3_get_command.py 项目： iwayamak/jetline

 def __init__(self, *args, **kwargs):
     self._component = \
         Container().component('S3_COMPONENT.ID=UT')
     super().__init__(*args, **kwargs)

示例#10

 def __init__(self, *args, **kwargs):
     self._component = Container.component('POSTGRESQL_COMPONENT.ID=UT')
     self._sql_str = 'select current_timestamp'
     super().__init__(*args, **kwargs)

示例#11

文件： test_plugin_component.py 项目： iwayamak/jetline

 def __init__(self, *args, **kwargs):
     self._component = Container.component(
         'PLUGIN_COMPONENT.ID=TEST_COMPONENT')
     super().__init__(*args, **kwargs)

示例#12

文件： test_container.py 项目： iwayamak/jetline

 def test_container(self):
     component = \
         Container().component(
             'POSTGRESQL_COMPONENT.ID=TEST_COMPONENT'
         )
     self.assertEqual(component.database, 'test_database')

示例#13

# -*- coding: utf-8 -*-

import os
from ....abc.base_test_case import BaseTestCase
from jetline.command.command_queue import CommandQueue
from jetline.command.db.postgresql.postgresql_processing_command import PostgreSQLProcessingCommand
from jetline.command.db.postgresql.postgresql_copy_from_command import PostgreSQLCopyFromCommand
from jetline.container.container import Container
from jetline.share_parameter.share_parameter import ShareParameter

COMPONENT = Container().component('POSTGRESQL_COMPONENT.ID=UT')
TABLE_NAME = f'{COMPONENT.schema}.test_postgresql_copy_from_command'
TEST_DATA_DIR = 'test_data'


class TestPostgreSQLCopyFromCommand(BaseTestCase):
    def __init__(self, *args, **kwargs):
        self._test_data_path = \
            os.path.join(
                os.path.dirname(__file__), TEST_DATA_DIR
            )
        super().__init__(*args, **kwargs)

    @classmethod
    def setUpClass(cls):
        ShareParameter.dry_run_mode = False
        queue = CommandQueue()
        sql_str = 'drop table if exists {table_name}'
        queue.add_command(
            PostgreSQLProcessingCommand(
                COMPONENT,