sqlc-gen-python/examples/src/tests/test_authors.py at main · sqlc-dev/sqlc-gen-python · GitHub

History

56 lines (40 loc) · 1.8 KB

Raw

import os

import pytest

import sqlalchemy.ext.asyncio

from authors import query

from dbtest.migrations import apply_migrations, apply_migrations_async

def test_authors(db: sqlalchemy.engine.Connection):

apply_migrations(db, [os.path.dirname(__file__) + "/../authors/schema.sql"])

querier = query.Querier(db)

authors = list(querier.list_authors())

assert authors == []

author_name = "Brian Kernighan"

author_bio = "Co-author of The C Programming Language and The Go Programming Language"

new_author = querier.create_author(name=author_name, bio=author_bio)

assert new_author.id > 0

assert new_author.name == author_name

assert new_author.bio == author_bio

db_author = querier.get_author(id=new_author.id)

assert db_author == new_author

author_list = list(querier.list_authors())

assert len(author_list) == 1

assert author_list[0] == new_author

@pytest.mark.asyncio

async def test_authors_async(async_db: sqlalchemy.ext.asyncio.AsyncConnection):

await apply_migrations_async(async_db, [os.path.dirname(__file__) + "/../authors/schema.sql"])

querier = query.AsyncQuerier(async_db)

async for _ in querier.list_authors():

assert False, "No authors should exist"

author_name = "Brian Kernighan"

author_bio = "Co-author of The C Programming Language and The Go Programming Language"

new_author = await querier.create_author(name=author_name, bio=author_bio)

assert new_author.id > 0

assert new_author.name == author_name

assert new_author.bio == author_bio

db_author = await querier.get_author(id=new_author.id)

assert db_author == new_author

author_list = []

async for author in querier.list_authors():

author_list.append(author)

assert len(author_list) == 1

assert author_list[0] == new_author

sqlc-gen-python/examples/src/tests/test_authors.py at main · sqlc-dev/sqlc-gen-python · GitHub

Uh oh!

Expand file tree

Breadcrumbs

test_authors.py

Latest commit

History

Breadcrumbs

test_authors.py

File metadata and controls