Python execute_query_one Examples

Example #1

File: test_psycopg2_connector.py Project: peopledoc/procrastinate

def test_execute_query_arg(psycopg2_connector):
    psycopg2_connector.execute_query("SELECT %(arg)s", arg=1)
    result = psycopg2_connector.execute_query_one("SELECT %(arg)s", arg=1)
    assert result == {"?column?": 1}

    result = psycopg2_connector.execute_query_all("SELECT %(arg)s", arg=1)
    assert result == [{"?column?": 1}]

Example #2

File: test_psycopg2_connector.py Project: peopledoc/procrastinate

def test_execute_query_percent(psycopg2_connector):
    psycopg2_connector.execute_query("SELECT '%'")
    result = psycopg2_connector.execute_query_one("SELECT '%'")
    assert result == {"?column?": "%"}

    result = psycopg2_connector.execute_query_all("SELECT '%'")
    assert result == [{"?column?": "%"}]

Example #3

File: test_psycopg2_connector.py Project: peopledoc/procrastinate

def test_execute_query(psycopg2_connector):
    psycopg2_connector.execute_query(
        "COMMENT ON TABLE \"procrastinate_jobs\" IS 'foo'")
    result = psycopg2_connector.execute_query_one(
        "SELECT obj_description('public.procrastinate_jobs'::regclass)")
    assert result == {"obj_description": "foo"}

    result = psycopg2_connector.execute_query_all(
        "SELECT obj_description('public.procrastinate_jobs'::regclass)")
    assert result == [{"obj_description": "foo"}]