Python FakeAthenaResource示例

示例#1

文件： test_resources.py 项目： zuik/dagster

def test_execute_query(mock_athena_client):
    athena = FakeAthenaResource(client=mock_athena_client)
    assert athena.execute_query("SELECT 1", fetch_results=True) == [("1", )]
    assert athena.execute_query("SELECT * FROM foo",
                                fetch_results=True,
                                expected_results=[(1, None),
                                                  (2, 3)]) == [("1", ),
                                                               ("2", "3")]

示例#2

def test_execute_query_succeeds_on_last_poll(mock_athena_client):
    athena = FakeAthenaResource(client=mock_athena_client, max_polls=1)
    athena.execute_query("SELECT 1", expected_states=["SUCCEEDED"])

示例#3

def test_execute_query_timeout(mock_athena_client):
    athena = FakeAthenaResource(client=mock_athena_client, max_polls=1)
    with pytest.raises(AthenaTimeout):
        athena.execute_query("SELECT 1")

示例#4

def test_execute_query_raises(mock_athena_client, expected_states):
    athena = FakeAthenaResource(client=mock_athena_client)
    with pytest.raises(AthenaError, match="state change reason"):
        athena.execute_query("SELECT 1", expected_states=expected_states)

示例#5

def test_execute_query_state_transitions(mock_athena_client, expected_states):
    athena = FakeAthenaResource(client=mock_athena_client)
    athena.execute_query("SELECT 1", expected_states=expected_states)