コード例 #1
0
 def test_list_cart(self):
     assert (
         models.Cart.query_by_host("1.1.1.1") is None
     )  # this doesn't match the menu query... this is why we do TDD!
     db_seed.seed()
     test_item = models.MenuItem.query_by_id(1)
     assert test_item.cart_id is None
     models.Cart.add_item("1.1.1.1", test_item)
     assert len(models.Cart.query_by_host("1.1.1.1").items) == 1
     assert test_item.cart_id == "1.1.1.1"
コード例 #2
0
 def test_get_image(self):
     assert models.Image.get_image(0) is None
     db_seed.seed()
     m = models.Image.get_image(1)
     assert all([m.id, m.data])
コード例 #3
0
 def test_add_cart_success(self):
     db_seed.seed()
     models.Cart.add_item("1.1.1.1", models.MenuItem.query_by_id(1))
コード例 #4
0
 def test_menu_query_by_id(self):
     assert models.MenuItem.query_by_id(1) is None
     db_seed.seed()
     assert models.MenuItem.query_by_id(1) is not None
コード例 #5
0
 def test_menu_query(self):
     assert len([_.serialize() for _ in models.MenuItem.query_all()
                 if _]) == 0
     db_seed.seed()
     assert len([_.serialize() for _ in models.MenuItem.query_all() if _
                 ]) >= len(db_seed.menu_start_raw)
コード例 #6
0
from openapi_server import encoder
from flask_cors import CORS
from openapi_server.database import models
from openapi_server.database import db_seed

app = connexion.App(__name__, specification_dir="./openapi/")
app.app.json_encoder = encoder.JSONEncoder
# app.app.config["SQLALCHEMY_DATABASE_URI"] = os.getenv("DATABASE_URL", "postgresql:///test-cfd")
app.app.config["SQLALCHEMY_DATABASE_URI"] = os.getenv(
    "DATABASE_URI",
    "sqlite:///:memory:")  # can this work if there's no db to connect to?
app.app.config["SQLALCHEMY_TRACK_MODIFICATIONS"] = False
app.add_api("openapi.yaml", arguments={"title": "CFD"}, pythonic_params=True)
CORS(app.app)
models.db.init_app(app.app)

# Seed the database if we're running with an in memory-db
if "memory" in app.app.config.get("SQLALCHEMY_DATABASE_URI"):
    app.app.app_context().push()
    models.db.create_all()
    db_seed.seed()


def main():
    app.run(port=8080, debug=True)


if __name__ == "__main__":
    main()