57 lines
1.9 KiB
Python
57 lines
1.9 KiB
Python
from mors.persistence import DbPersistence
|
|
import pytest
|
|
from migrate.versioning.api import upgrade,create,version_control
|
|
from datetime import datetime
|
|
import os
|
|
|
|
db_persistence = None
|
|
TEST_DB="test/test_db11"
|
|
|
|
def setup_module(mod):
|
|
global db_persistence
|
|
DB_URL = "sqlite:///"+TEST_DB
|
|
create(DB_URL, "./mors_repo")
|
|
version_control(DB_URL, "./mors_repo")
|
|
upgrade(DB_URL,"./mors_repo")
|
|
db_persistence = DbPersistence(DB_URL)
|
|
return db_persistence
|
|
|
|
def teardown_module(mod):
|
|
os.unlink(TEST_DB)
|
|
|
|
def test_apis():
|
|
tenant_id = "aasdsadfsadf"
|
|
tenant_user1 = "a@xyz.com"
|
|
expiry_day1 = 3
|
|
tenant_created_date = datetime.utcnow()
|
|
|
|
db_persistence.add_tenant_lease(tenant_id, expiry_day1, tenant_user1, tenant_created_date)
|
|
t_lease = db_persistence.get_tenant_lease(tenant_id)
|
|
assert (t_lease.tenant_uuid == tenant_id)
|
|
assert (t_lease.created_by == tenant_user1)
|
|
assert (t_lease.created_at == tenant_created_date)
|
|
|
|
# Now try update
|
|
tenant_user2 = "b@xyz.com"
|
|
tenant_updated_date = datetime.utcnow()
|
|
db_persistence.update_tenant_lease(tenant_id, expiry_day1, tenant_user2, tenant_updated_date)
|
|
t_lease = db_persistence.get_tenant_lease(tenant_id)
|
|
assert (t_lease.tenant_uuid == tenant_id)
|
|
assert (t_lease.created_by == tenant_user1)
|
|
assert (t_lease.updated_by == tenant_user2)
|
|
assert (t_lease.created_at == tenant_created_date)
|
|
assert (t_lease.updated_at == tenant_updated_date)
|
|
|
|
|
|
# Instance lease now
|
|
instance_uuid = "asdf2-2342-23423"
|
|
now = datetime.utcnow()
|
|
db_persistence.add_instance_lease(instance_uuid, tenant_id, now, tenant_user1, now)
|
|
i_lease = db_persistence.get_instance_lease(instance_uuid, tenant_id)
|
|
assert (i_lease.instance_uuid == instance_uuid)
|
|
assert (i_lease.tenant_uuid == tenant_id)
|
|
assert (i_lease.expiry == now)
|
|
assert (i_lease.created_by == tenant_user1)
|
|
|
|
|