test: Added test for created_at (#6728)

Author: me-diru

app/factories/attendee.py

@@ -24,7 +24,6 @@ class Meta:
     event_id = 1
     ticket_id = None
     order_id = None
-    created_at = common.date_
     modified_at = common.date_
 
 

app/factories/role_invite.py

@@ -14,7 +14,6 @@ class Meta:
     event = factory.RelatedFactory(EventFactoryBasic)
     role = factory.RelatedFactory(RoleFactory)
     email = common.email_
-    created_at = common.date_
     hash = common.string_
     status = common.string_
     role_name = common.string_

app/factories/session.py

@@ -27,7 +27,6 @@ class Meta:
     audio_url = common.url_
     signup_url = common.url_
     state = "accepted"
-    created_at = common.date_
     submitted_at = common.date_
     is_mail_sent = True
     event_id = 1

app/factories/user_token_blacklist.py

@@ -0,0 +1,11 @@
+import factory
+
+from app.models.user_token_blacklist import UserTokenBlackListTime, db
+
+
+class UserTokenBlacklistFactory(factory.alchemy.SQLAlchemyModelFactory):
+    class Meta:
+        model = UserTokenBlackListTime
+        sqlalchemy_session = db.session
+
+    user_id = 1

app/models/access_code.py

@@ -1,6 +1,6 @@
 from dataclasses import dataclass
 from datetime import datetime
-
+from sqlalchemy.sql import func
 from app.models import db
 from app.models.base import SoftDeletionModel
 
@@ -29,7 +29,7 @@ class AccessCode(SoftDeletionModel):
         db.Integer, db.ForeignKey('tickets.id', ondelete='CASCADE')
     )
     event_id: int = db.Column(db.Integer, db.ForeignKey('events.id', ondelete='CASCADE'))
-    created_at: datetime = db.Column(db.DateTime(timezone=True))
+    created_at: datetime = db.Column(db.DateTime(timezone=True), default=func.now())
     marketer_id: int = db.Column(
         db.Integer, db.ForeignKey('users.id', ondelete='CASCADE')
     )

app/models/discount_code.py

@@ -1,5 +1,5 @@
 from datetime import datetime
-
+from sqlalchemy.sql import func
 from app.models import db
 from app.models.base import SoftDeletionModel
 
@@ -23,7 +23,7 @@ class DiscountCode(SoftDeletionModel):
     valid_till = db.Column(db.DateTime(timezone=True), nullable=True)
     event_id = db.Column(db.Integer, db.ForeignKey('events.id', ondelete='CASCADE'))
     event = db.relationship('Event', backref='discount_codes', foreign_keys=[event_id])
-    created_at = db.Column(db.DateTime(timezone=True))
+    created_at = db.Column(db.DateTime(timezone=True), default=func.now())
     marketer_id = db.Column(db.Integer, db.ForeignKey('users.id', ondelete='CASCADE'))
     marketer = db.relationship('User', backref='discount_codes_')
 

app/models/role_invite.py

@@ -1,6 +1,6 @@
 import random
 from datetime import datetime, timedelta
-
+from sqlalchemy.sql import func
 import pytz
 from sqlalchemy.schema import UniqueConstraint
 
@@ -31,7 +31,7 @@ class RoleInvite(SoftDeletionModel):
     role = db.relationship("Role")
 
     hash = db.Column(db.String)
-    created_at = db.Column(db.DateTime(timezone=True))
+    created_at = db.Column(db.DateTime(timezone=True), default=func.now())
     status = db.Column(db.String, default="pending")
 
     def __init__(

tests/all/integration/api/helpers/test_csv_jobs_util.py

@@ -65,6 +65,7 @@ def test_export_sessions_with_details_csv(self):
                 long_abstract='long_abstract',
                 comments='comment',
                 level='level',
+                created_at=common.date_,
             )
             db.session.commit()
             field_data = export_sessions_csv([test_session])

tests/all/integration/api/helpers/test_scheduled_jobs.py

@@ -4,6 +4,7 @@
     delete_ticket_holders_no_order_id,
     event_invoices_mark_due,
 )
+import app.factories.common as common
 from app.factories.attendee import AttendeeFactory
 from app.factories.event_invoice import EventInvoiceFactory
 from app.models import db
@@ -39,7 +40,7 @@ def test_delete_ticket_holders_with_no_order_id(self):
         """Method to test deleting ticket holders with no order id after expiry time"""
 
         with self.app.test_request_context():
-            attendee = AttendeeFactory()
+            attendee = AttendeeFactory(created_at=common.date_)
             db.session.commit()
             attendee_id = attendee.id
             delete_ticket_holders_no_order_id()

tests/all/unit/api/validation/test_createdat.py

@@ -0,0 +1,57 @@
+import unittest
+from app.models import db
+from datetime import datetime, timedelta, timezone
+from app.factories.session import SessionFactory
+from app.factories.user import UserFactory
+from app.factories.event import EventFactoryBasic
+from app.factories.access_code import AccessCodeFactory
+from app.factories.event_invoice import EventInvoiceFactory
+from app.factories.attendee import AttendeeFactory
+from app.factories.discount_code import DiscountCodeFactory
+from app.factories.order import OrderFactory
+from app.factories.role_invite import RoleInviteFactory
+from app.factories.user_token_blacklist import UserTokenBlacklistFactory
+from tests.all.integration.utils import OpenEventTestCase
+
+
+class TestCreatedatValidation(OpenEventTestCase):
+    def test_createdat(self):
+        """ Validate time : Tests if created_at is set to current time in all models
+        """
+        with self.app.test_request_context():
+            model_factories = [
+                UserFactory,
+                EventInvoiceFactory,
+                AccessCodeFactory,
+                EventFactoryBasic,
+                OrderFactory,
+                RoleInviteFactory,
+                DiscountCodeFactory,
+                AttendeeFactory,
+                SessionFactory,
+                UserTokenBlacklistFactory,
+            ]
+            models_with_utcnow = [
+                AttendeeFactory,
+                DiscountCodeFactory,
+                EventInvoiceFactory,
+            ]
+            for model_factory in model_factories:
+                with self.subTest(model_factory=model_factory):
+                    test_model = model_factory()
+                    db.session.commit()
+                    if model_factory in models_with_utcnow:
+                        current_time = datetime.utcnow().astimezone()
+                    else:
+                        current_time = datetime.now(timezone.utc).astimezone()
+                    createdat_db = test_model.created_at
+                    time_diff = current_time - createdat_db
+                    allowed_time_lag = timedelta(milliseconds=250)
+                    message = "created_at not set" " to current time in {} \n".format(
+                        model_factory
+                    )
+                    self.assertLessEqual(time_diff, allowed_time_lag, message)
+
+
+if __name__ == "__main__":
+    unittest.main()