Python assert_called_once_with 예제들

예제 #1

def test_real_main_method(ArgumentParser, _main):
    """
    Test parsing of arguments from main method
    """
    ArgumentParser.return_value.parse_args.return_value = Mock(
        concurrency=3,
        queues=["email1"],
        interval=1,
        batchsize=10,
        logging_level="WARN",
        region='us-east-1',
        prefetch_multiplier=2,
        access_key_id=None,
        secret_access_key=None)
    main()

    _main.assert_called_once_with(queue_prefixes=['email1'],
                                  concurrency=3,
                                  interval=1,
                                  batchsize=10,
                                  logging_level="WARN",
                                  region='us-east-1',
                                  prefetch_multiplier=2,
                                  access_key_id=None,
                                  secret_access_key=None)

예제 #2

파일: test_manager_worker.py 프로젝트: spulec/PyQS

def test_real_main_method(ArgumentParser, _main):
    """
    Test parsing of arguments from main method
    """
    ArgumentParser.return_value.parse_args.return_value = Mock(
        concurrency=3, queues=["email1"], interval=1, batchsize=10,
        logging_level="WARN", region='us-east-1', prefetch_multiplier=2,
        access_key_id=None, secret_access_key=None,
    )
    main()

    _main.assert_called_once_with(
        queue_prefixes=['email1'], concurrency=3, interval=1, batchsize=10,
        logging_level="WARN", region='us-east-1', prefetch_multiplier=2,
        access_key_id=None, secret_access_key=None,
    )