from unittest import TestCase

from hub_api.config import Config
from hub.persistence.models import City, Application, CityObject, SimulationResults, User


class TestApi(TestCase):
  def test_create_database(self):
    config = Config()
    Application.__table__.create(bind=config.repository.engine, checkfirst=True)
    User.__table__.create(bind=config.repository.engine, checkfirst=True)
    City.__table__.create(bind=config.repository.engine, checkfirst=True)
    CityObject.__table__.create(bind=config.repository.engine, checkfirst=True)
    SimulationResults.__table__.create(bind=config.repository.engine, checkfirst=True)
    config.database.create_user('Administrator', )
    self.assertTrue(True)