test_example1

test_example1.py

from models import SaleRecords
from sqlalchemy import func


def sold_product_percent(session) -> dict:
    sold_product = session.query(func.sum(SaleRecords.count), SaleRecords.product_name
                          ).group_by(SaleRecords.product_name).all()
    sum_of_products = sum(count for count, name in sold_product)
    percents = {item[1]: round((item[0] / sum_of_products) * 100, 2)
                for item in sold_product}
    return percents

test_example_2.py

import unittest
from models import Base
from sqlalchemy import create_engine
from sqlalchemy.orm import sessionmaker
from main import sold_product_percent


class TestMain(unittest.TestCase):
    
    def test_sold_product_percent(self):
        # Arrange
        db_url = "postgresql://user_name:password@127.0.0.1:5432/test"
        engine = create_engine(db_url, echo=False)
        Base.metadata.create_all(engine)
        engine.connect()
        session_local = sessionmaker(autocommit=False, autoflush=False, bind=engine)
        session = session_local()
        # We must have used the mocks here
        
        # Act
        output = sold_product_percent(session)

        # Assert
        assert type(output) == dict
        assert 99 < sum(output.values()) < 100