test_main.py

import os
import pytest
from orderbooks.db_models import OhlcvdataModel

def add_record(session):
    new_rec = OhlcvdataModel(
        name='bitfinex'
    )
    session.add(new_rec)
    session.commit()
    return new_rec


@pytest.fixture()
def test_db(transacted_postgresql_db):
    transacted_postgresql_db.create_table(OhlcvdataModel)
    transacted_postgresql_db.session.commit()
    return transacted_postgresql_db

def test_add_record(test_db):
    ohlcv_rec = add_record(test_db.session)
    assert type(ohlcv_rec.session_id) == int
    assert ohlcv_rec.session_id == 1