Esempi in Python per BotoClientProxy

Esempio n. 1

File: test_boto_proxy.py Progetto: alterrebe/senza

def test_proxy(mock_boto_client: MagicMock):
    proxy = BotoClientProxy('test')
    mock_boto_client.assert_called_once_with('test')
    proxy.random_test(42)
    mock_boto_client.random_test.assert_called_once_with(42)
    assert proxy.random_test is not mock_boto_client.random_test
    assert proxy.p is mock_boto_client.p

Esempio n. 2

File: test_boto_proxy.py Progetto: zeus911/senza

def test_proxy(mock_boto_client: MagicMock):
    proxy = BotoClientProxy('test')
    mock_boto_client.assert_called_once_with('test')
    proxy.random_test(42)
    mock_boto_client.random_test.assert_called_once_with(42)
    assert proxy.random_test is not mock_boto_client.random_test
    assert proxy.p is mock_boto_client.p

Esempio n. 3

File: test_boto_proxy.py Progetto: alterrebe/senza

def test_random_error(mock_boto_client: MagicMock, monkeypatch):
    monkeypatch.setattr('senza.manaus.boto_proxy.sleep', MagicMock())

    def throttled(arg):
        raise botocore.exceptions.ClientError(
            {'Error': {'Code': "everyday i'm shuffling"}},
            'testing'
        )

    mock_boto_client.throttled.side_effect = throttled
    proxy = BotoClientProxy('test')

    with pytest.raises(botocore.exceptions.ClientError):
        proxy.throttled(42)
    mock_boto_client.throttled.assert_called_with(42)
    assert mock_boto_client.throttled.call_count == 1

Esempio n. 4

File: test_boto_proxy.py Progetto: zeus911/senza

def test_random_error(mock_boto_client: MagicMock, monkeypatch):
    monkeypatch.setattr('senza.manaus.boto_proxy.sleep', MagicMock())

    def throttled(arg):
        raise botocore.exceptions.ClientError(
            {'Error': {
                'Code': "everyday i'm shuffling"
            }}, 'testing')

    mock_boto_client.throttled.side_effect = throttled
    proxy = BotoClientProxy('test')

    with pytest.raises(botocore.exceptions.ClientError):
        proxy.throttled(42)
    mock_boto_client.throttled.assert_called_with(42)
    assert mock_boto_client.throttled.call_count == 1

Esempio n. 5

File: test_boto_proxy.py Progetto: alterrebe/senza

def test_throttling(mock_boto_client: MagicMock, monkeypatch):
    monkeypatch.setattr('senza.manaus.boto_proxy.sleep', MagicMock())
    i = 0

    def throttled(arg):
        nonlocal i
        if i < 3:
            i += 1
            raise botocore.exceptions.ClientError(
                {'Error': {'Code': 'Throttling'}},
                'testing'
            )
        else:
            return arg

    mock_boto_client.throttled.side_effect = throttled
    proxy = BotoClientProxy('test')
    mock_boto_client.assert_called_once_with('test')
    assert proxy.throttled(42) == 42
    mock_boto_client.throttled.assert_called_with(42)
    assert mock_boto_client.throttled.call_count == 4

Esempio n. 6

File: test_boto_proxy.py Progetto: zeus911/senza

def test_throttling(mock_boto_client: MagicMock, monkeypatch):
    monkeypatch.setattr('senza.manaus.boto_proxy.sleep', MagicMock())
    i = 0

    def throttled(arg):
        nonlocal i
        if i < 3:
            i += 1
            raise botocore.exceptions.ClientError(
                {'Error': {
                    'Code': 'Throttling'
                }}, 'testing')
        else:
            return arg

    mock_boto_client.throttled.side_effect = throttled
    proxy = BotoClientProxy('test')
    mock_boto_client.assert_called_once_with('test')
    assert proxy.throttled(42) == 42
    mock_boto_client.throttled.assert_called_with(42)
    assert mock_boto_client.throttled.call_count == 4