import pytest

import dbconnection.db as dbconnection


@pytest.fixture
def db():
    database = dbconnection.DB()
    database.connect()
    return database

def test_db_connection(db: dbconnection.DB):
    assert db.connection is not None

def test_db_query(db: dbconnection.DB):
    result = db.query('SELECT 1')
    assert result[0] == (1,)