Add test setup

Author: Birne94

.gitignore

@@ -87,3 +87,5 @@ ENV/
 
 # Rope project settings
 .ropeproject
+
+.idea

app.py

@@ -0,0 +1,35 @@
+from flask import Flask
+from flask.ext.sqlalchemy import SQLAlchemy
+from models import db, TestTicket, TicketStatus, Ticket
+
+app = Flask(__name__)
+
+db.init_app(app)
+
+with app.app_context():
+    db.create_all()
+    db.session.commit()
+
+
+@app.route('/')
+def index():
+    # This will fail on the first run
+    try:
+        result = db.session.query(Ticket) \
+            .filter(Ticket.status == TicketStatus.open) \
+            .all()
+    except:
+        print('query failed')
+    else:
+        print('query successful')
+
+    # If I query the subclass, everything will work fine
+    db.session.query(TestTicket) \
+        .filter(Ticket.status == TicketStatus.open) \
+        .all()
+
+    return '', 200
+
+
+if __name__ == '__main__':
+    app.run(debug=True)

models.py

@@ -0,0 +1,52 @@
+import enum
+
+from flask.ext.sqlalchemy import SQLAlchemy
+from sqlalchemy import func
+from sqlalchemy.ext.declarative import AbstractConcreteBase
+from sqlalchemy.ext.declarative import declared_attr
+
+
+db = SQLAlchemy()
+
+
+class Base(db.Model):
+    __abstract__ = True
+
+    @declared_attr
+    def __tablename__(cls):
+        return cls.__name__.lower()
+
+    id = db.Column(db.Integer, primary_key=True)
+
+
+class TimestampMixin(object):
+    created_on = db.Column(db.DateTime, server_default=func.now())
+    updated_on = db.Column(db.DateTime, server_default=func.now(),
+                           onupdate=func.now())
+
+
+class TicketStatus(enum.Enum):
+    open = 'open'
+    in_progess = 'in_progress'
+    rejected = 'rejected'
+    closed = 'closed'
+
+    all = (open, in_progess, rejected, closed)
+
+
+class Ticket(AbstractConcreteBase, TimestampMixin, Base):
+    status = db.Column(db.Enum(TicketStatus), nullable=False,
+                       default=TicketStatus.open)
+
+    @declared_attr
+    def __mapper_args__(cls):
+        if cls == Ticket:
+            return {'concrete': False}
+        return {
+            'polymorphic_identity': cls.__name__,
+            'concrete': True
+        }
+
+
+class TestTicket(Ticket):
+    data = db.Column(db.String, nullable=False)