Python Lizzy.get_trafficの例

コード例 #1

def test_get_traffic(monkeypatch):
    mock_request = MagicMock()
    mock_request.return_value = FakeResponse(200, '{"weight": 100.0}')
    monkeypatch.setattr('requests.get', mock_request)

    lizzy = Lizzy('https://lizzy.example', '7E5770K3N')
    lizzy.get_traffic('lizzy-test')

    header = make_header('7E5770K3N')
    mock_request.assert_called_once_with(
        'https://lizzy.example/api/stacks/lizzy-test/traffic',
        None,
        headers=header,
        verify=False)

    mock_request.reset_mock()
    mock_request.return_value = FakeResponse(200, '["stack1","stack2"]')

    lizzy = Lizzy('https://lizzy.example', '7E5770K3N')
    lizzy.get_traffic('574CC', region='ab-foo-7')

    header = make_header('7E5770K3N')
    url = 'https://lizzy.example/api/stacks/574CC/traffic?region=ab-foo-7'
    mock_request.assert_called_once_with(url,
                                         None,
                                         headers=header,
                                         verify=False)

コード例 #2

ファイル: test_lizzy.py プロジェクト: mfelsche/lizzy-client

def test_get_traffic(monkeypatch):
    mock_request = MagicMock()
    mock_request.return_value = FakeResponse(200, '{"weight": 100.0}')
    monkeypatch.setattr('requests.get', mock_request)

    lizzy = Lizzy('https://lizzy.example', '7E5770K3N')
    lizzy.get_traffic('lizzy-test')

    header = make_header('7E5770K3N')
    mock_request.assert_called_once_with(
        'https://lizzy.example/api/stacks/lizzy-test/traffic', None,
        headers=header, verify=False)

    mock_request.reset_mock()
    mock_request.return_value = FakeResponse(200, '["stack1","stack2"]')

    lizzy = Lizzy('https://lizzy.example', '7E5770K3N')
    lizzy.get_traffic('574CC', region='ab-foo-7')

    header = make_header('7E5770K3N')
    url = 'https://lizzy.example/api/stacks/574CC/traffic?region=ab-foo-7'
    mock_request.assert_called_once_with(url, None, headers=header,
                                         verify=False)