darkhelm/notifications-api
1
0
Fork 0
mirror of https://github.com/GSA/notifications-api.git synced 2026-09-07 15:58:25 -04:00
Code Issues Packages Projects Releases Wiki Activity

fix more sqlalchemy

Browse Source
This commit is contained in:
Kenneth Kehl
2024-11-14 11:26:53 -08:00
parent 99a92e705e
commit 6d8fdab5a3
4 changed files with 48 additions and 22 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
tests/app/celery/test_reporting_tasks.py
View File

@@ -4,7 +4,7 @@ from uuid import UUID
import pytest
from freezegun import freeze_time
from sqlalchemy import select
from sqlalchemy import func, select
from app import db
from app.celery.reporting_tasks import (
@@ -363,9 +363,12 @@ def test_create_nightly_billing_for_day_use_BST(
rate_multiplier=1.0,
billable_units=4,
)
assert Notification.query.count() == 3
assert FactBilling.query.count() == 0
stmt = select(func.count()).select_from(Notification)
count = db.session.execute(stmt).scalar() or 0
assert count == 3
stmt = select(func.count()).select_from(FactBilling)
count = db.session.execute(stmt).scalar() or 0
assert count == 0
create_nightly_billing_for_day("2018-03-25")
records = FactBilling.query.order_by(FactBilling.local_date).all()