test: Added test for created_at

app/factories/access_code.py

@@ -11,6 +11,7 @@ class AccessCodeFactory(factory.alchemy.SQLAlchemyModelFactory):
     class Meta:
         model = AccessCode
         sqlalchemy_session = db.session
+        sqlalchemy_session_persistence = 'commit'
 
     tickets = factory.RelatedFactory(TicketFactory)
     user = factory.RelatedFactory(UserFactory)
@@ -21,7 +22,5 @@ class Meta:
     tickets_number = 30
     min_quantity = 10
     max_quantity = 20
-    marketer_id = 1
     valid_from = common.date_
     valid_till = common.dateEnd_
-    event_id = 1

app/factories/discount_code.py

@@ -11,6 +11,7 @@ class DiscountCodeFactory(factory.alchemy.SQLAlchemyModelFactory):
     class Meta:
         model = DiscountCode
         sqlalchemy_session = db.session
+        sqlalchemy_session_persistence = 'commit'
 
     marketer = factory.RelatedFactory(UserFactory)
     code = common.string_
@@ -24,8 +25,6 @@ class Meta:
     valid_from = common.date_
     valid_till = common.dateEnd_
     used_for = "event"
-    marketer_id = 1
-    event_id = None
 
 
 class DiscountCodeTicketFactory(factory.alchemy.SQLAlchemyModelFactory):

app/factories/event.py

@@ -8,6 +8,7 @@ class EventFactoryBasic(factory.alchemy.SQLAlchemyModelFactory):
     class Meta:
         model = Event
         sqlalchemy_session = db.session
+        sqlalchemy_session_persistence = 'commit'
 
     name = common.string_
     external_event_url = common.url_

app/factories/event_invoice.py

@@ -11,6 +11,7 @@ class EventInvoiceFactory(factory.alchemy.SQLAlchemyModelFactory):
     class Meta:
         model = EventInvoice
         sqlalchemy_session = db.session
+        sqlalchemy_session_persistence = 'commit'
 
     event = factory.RelatedFactory(EventFactoryBasic)
     user = factory.RelatedFactory(UserFactory)
@@ -30,6 +31,3 @@ class Meta:
     last4 = "1234"
     stripe_token = common.string_
     paypal_token = common.string_
-    event_id = 1
-    user_id = 2
-    discount_code_id = 1

app/factories/order.py

@@ -9,8 +9,8 @@ class OrderFactory(factory.alchemy.SQLAlchemyModelFactory):
     class Meta:
         model = Order
         sqlalchemy_session = db.session
+        sqlalchemy_session_persistence = 'commit'
 
     event = factory.RelatedFactory(EventFactoryBasic)
-    event_id = 1
     payment_mode = 'free'
     status = 'initializing'

app/factories/role_invite.py

@@ -10,13 +10,11 @@ class RoleInviteFactory(factory.alchemy.SQLAlchemyModelFactory):
     class Meta:
         model = RoleInvite
         sqlalchemy_session = db.session
+        sqlalchemy_session_persistence = 'commit'
 
     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_
-    event_id = 1
-    role_id = 1

app/factories/session.py

@@ -12,6 +12,7 @@ class SessionFactoryBase(factory.alchemy.SQLAlchemyModelFactory):
     class Meta:
         model = Session
         sqlalchemy_session = db.session
+        sqlalchemy_session_persistence = 'commit'
 
     title = common.string_
     subtitle = common.string_
@@ -27,13 +28,8 @@ 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
-    session_type_id = 1
-    track_id = 1
-    microlocation_id = 1
 
 
 class SessionFactory(SessionFactoryBase):

app/factories/ticket_holder.py

@@ -0,0 +1,35 @@
+import factory
+
+import app.factories.common as common
+from app.models.ticket_holder import TicketHolder, db
+
+
+class TicketHolderFactory(factory.alchemy.SQLAlchemyModelFactory):
+    class Meta:
+        model = TicketHolder
+        sqlalchemy_session = db.session
+        sqlalchemy_session_persistence = 'commit'
+
+    firstname = 'John'
+    lastname = 'Doe'
+    email = common.email_
+    address = common.string_
+    city = common.string_
+    state = common.string_
+    country = common.country_
+    job_title = common.string_
+    phone = common.string_
+    billing_address = common.string_
+    home_address = common.string_
+    shipping_address = common.string_
+    company = common.string_
+    work_address = common.string_
+    work_phone = common.string_
+    website = common.url_
+    blog = common.url_
+    twitter = common.socialUrl_('twitter')
+    facebook = common.socialUrl_('facebook')
+    github = common.socialUrl_('github')
+    age_group = common.string_
+    birth_date = common.date_
+    pdf_url = common.url_

app/factories/user.py

@@ -8,6 +8,7 @@ class UserFactory(factory.alchemy.SQLAlchemyModelFactory):
     class Meta:
         model = User
         sqlalchemy_session = db.session
+        sqlalchemy_session_persistence = 'commit'
 
     email = common.email_
     password = 'password'

app/factories/user_token_blacklist.py

@@ -0,0 +1,13 @@
+import factory
+
+from app.models.user_token_blacklist import UserTokenBlackListTime, db
+from app.factories.user import UserFactory
+
+
+class UserTokenBlacklistFactory(factory.alchemy.SQLAlchemyModelFactory):
+    class Meta:
+        model = UserTokenBlackListTime
+        sqlalchemy_session = db.session
+        sqlalchemy_session_persistence = 'commit'
+
+    user = factory.SubFactory(UserFactory)

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/unit/api/validation/test_createdat.py

@@ -0,0 +1,55 @@
+import unittest
+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.ticket_holder import TicketHolderFactory
+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,
+                TicketHolderFactory,
+                SessionFactory,
+                UserTokenBlacklistFactory,
+            ]
+            models_with_utcnow = [
+                TicketHolderFactory,
+                DiscountCodeFactory,
+                EventInvoiceFactory,
+            ]
+            for model_factory in model_factories:
+                with self.subTest(model_factory=model_factory):
+                    test_model = model_factory()
+                    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=500)
+                    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()