def test_get_workbooks(self):
created0 = db_api.create_workbook(WORKBOOKS[0])
created1 = db_api.create_workbook(WORKBOOKS[1])
fetched = db_api.get_workbooks()
self.assertEqual(2, len(fetched))
self.assertEqual(created0, fetched[0])
self.assertEqual(created1, fetched[1])
def test_get_workbooks(self):
created0 = db_api.create_workbook(WORKBOOKS[0])
created1 = db_api.create_workbook(WORKBOOKS[1])
fetched = db_api.get_workbooks()
self.assertEqual(2, len(fetched))
self.assertEqual(created0, fetched[0])
self.assertEqual(created1, fetched[1])
def test_create_workbook_duplicate_without_auth(self):
cfg.CONF.set_default('auth_enable', False, group='pecan')
db_api.create_workbook(WORKBOOKS[0])
self.assertRaises(
exc.DBDuplicateEntryException,
db_api.create_workbook,
WORKBOOKS[0]
)
def test_workbooks_in_two_projects(self):
created = db_api.create_workbook(WORKBOOKS[1])
fetched = db_api.get_workbooks()
self.assertEqual(1, len(fetched))
self.assertEqual(created, fetched[0])
# Create a new user.
auth_context.set_ctx(test_base.get_context(default=False))
created = db_api.create_workbook(WORKBOOKS[1])
fetched = db_api.get_workbooks()
self.assertEqual(1, len(fetched))
self.assertEqual(created, fetched[0])
def test_workbooks_in_two_projects(self):
created = db_api.create_workbook(WORKBOOKS[1])
fetched = db_api.get_workbooks()
self.assertEqual(1, len(fetched))
self.assertEqual(created, fetched[0])
# Create a new user.
auth_context.set_ctx(test_base.get_context(default=False))
created = db_api.create_workbook(WORKBOOKS[1])
fetched = db_api.get_workbooks()
self.assertEqual(1, len(fetched))
self.assertEqual(created, fetched[0])
def test_commit_multiple_objects(self):
db_api.start_tx()
try:
created = db_api.create_workflow_execution(WF_EXECS[0])
fetched = db_api.get_workflow_execution(created.id)
self.assertEqual(created, fetched)
created_wb = db_api.create_workbook(WORKBOOKS[0])
fetched_wb = db_api.get_workbook(created_wb.name)
self.assertEqual(created_wb, fetched_wb)
self.assertTrue(self.is_db_session_open())
db_api.commit_tx()
finally:
db_api.end_tx()
self.assertFalse(self.is_db_session_open())
fetched = db_api.get_workflow_execution(created.id)
self.assertEqual(created, fetched)
fetched_wb = db_api.get_workbook(created_wb.name)
self.assertEqual(created_wb, fetched_wb)
self.assertFalse(self.is_db_session_open())
def test_rollback_multiple_objects(self):
db_api.start_tx()
try:
created = db_api.create_workflow_execution(WF_EXECS[0])
fetched = db_api.get_workflow_execution(created['id'])
self.assertEqual(created, fetched)
created_wb = db_api.create_workbook(WORKBOOKS[0])
fetched_wb = db_api.get_workbook(created_wb.name)
self.assertEqual(created_wb, fetched_wb)
self.assertTrue(self.is_db_session_open())
db_api.rollback_tx()
finally:
db_api.end_tx()
self.assertFalse(self.is_db_session_open())
self.assertRaises(exc.NotFoundException, db_api.get_workflow_execution,
created.id)
self.assertRaises(exc.NotFoundException, db_api.get_workbook,
created_wb.name)
self.assertFalse(self.is_db_session_open())
def test_commit_multiple_objects(self):
db_api.start_tx()
try:
created = db_api.create_workflow_execution(WF_EXECS[0])
fetched = db_api.get_workflow_execution(created.id)
self.assertEqual(created, fetched)
created_wb = db_api.create_workbook(WORKBOOKS[0])
fetched_wb = db_api.get_workbook(created_wb.name)
self.assertEqual(created_wb, fetched_wb)
self.assertTrue(self.is_db_session_open())
db_api.commit_tx()
finally:
db_api.end_tx()
self.assertFalse(self.is_db_session_open())
fetched = db_api.get_workflow_execution(created.id)
self.assertEqual(created, fetched)
fetched_wb = db_api.get_workbook(created_wb.name)
self.assertEqual(created_wb, fetched_wb)
self.assertFalse(self.is_db_session_open())
def test_rollback_multiple_objects(self):
db_api.start_tx()
try:
created = db_api.create_workflow_execution(WF_EXECS[0])
fetched = db_api.get_workflow_execution(created['id'])
self.assertEqual(created, fetched)
created_wb = db_api.create_workbook(WORKBOOKS[0])
fetched_wb = db_api.get_workbook(created_wb.name)
self.assertEqual(created_wb, fetched_wb)
self.assertTrue(self.is_db_session_open())
db_api.rollback_tx()
finally:
db_api.end_tx()
self.assertFalse(self.is_db_session_open())
self.assertRaises(
exc.NotFoundException,
db_api.get_workflow_execution,
created.id
)
self.assertRaises(
exc.NotFoundException,
db_api.get_workbook,
created_wb.name
)
self.assertFalse(self.is_db_session_open())
def test_workbook_public(self):
# Create a workbook(scope=public) as under one project
# then make sure it's visible for other projects.
created0 = db_api.create_workbook(WORKBOOKS[0])
fetched = db_api.get_workbooks()
self.assertEqual(1, len(fetched))
self.assertEqual(created0, fetched[0])
# Assert that the project_id stored is actually the context's
# project_id not the one given.
self.assertEqual(created0.project_id, auth_context.ctx().project_id)
self.assertNotEqual(WORKBOOKS[0]['project_id'],
auth_context.ctx().project_id)
# Create a new user.
ctx = auth_context.MistralContext(
user_id='9-0-44-5',
project_id='99-88-33',
user_name='test-user',
project_name='test-another',
is_admin=False
)
auth_context.set_ctx(ctx)
fetched = db_api.get_workbooks()
self.assertEqual(1, len(fetched))
self.assertEqual(created0, fetched[0])
self.assertEqual('public', created0.scope)
def test_rollback_transaction(self):
try:
with db_api.transaction():
created = db_api.create_workbook(WORKBOOKS[0])
fetched = db_api.get_workbook(created.name)
self.assertEqual(created, fetched)
self.assertTrue(self.is_db_session_open())
db_api.create_workbook(WORKBOOKS[0])
except exc.DBDuplicateEntry:
pass
self.assertFalse(self.is_db_session_open())
self.assertRaises(exc.NotFoundException, db_api.get_workbook,
created.name)
def test_delete_workbook(self):
created = db_api.create_workbook(WORKBOOKS[0])
fetched = db_api.get_workbook(created.name)
self.assertEqual(created, fetched)
db_api.delete_workbook(created.name)
self.assertRaises(exc.NotFoundException, db_api.get_workbook,
created.name)
def test_rollback_transaction(self):
try:
with db_api.transaction():
created = db_api.create_workbook(WORKBOOKS[0])
fetched = db_api.get_workbook(created.name)
self.assertEqual(created, fetched)
self.assertTrue(self.is_db_session_open())
db_api.create_workbook(WORKBOOKS[0])
except exc.DBDuplicateEntryException:
pass
self.assertFalse(self.is_db_session_open())
self.assertRaises(
exc.NotFoundException,
db_api.get_workbook,
created.name
)
def test_create_and_get_and_load_workbook(self):
created = db_api.create_workbook(WORKBOOKS[0])
fetched = db_api.get_workbook(created['name'])
self.assertEqual(created, fetched)
fetched = db_api.load_workbook(created.name)
self.assertEqual(created, fetched)
self.assertIsNone(db_api.load_workbook("not-existing-wb"))
def test_create_and_get_and_load_workbook(self):
created = db_api.create_workbook(WORKBOOKS[0])
fetched = db_api.get_workbook(created['name'])
self.assertEqual(created, fetched)
fetched = db_api.load_workbook(created.name)
self.assertEqual(created, fetched)
self.assertIsNone(db_api.load_workbook("not-existing-wb"))
def test_commit_transaction(self):
with db_api.transaction():
created = db_api.create_workbook(WORKBOOKS[0])
fetched = db_api.get_workbook(created.name)
self.assertEqual(created, fetched)
self.assertTrue(self.is_db_session_open())
self.assertFalse(self.is_db_session_open())
fetched = db_api.get_workbook(created.name)
self.assertEqual(created, fetched)
self.assertFalse(self.is_db_session_open())
def test_delete_workbook(self):
created = db_api.create_workbook(WORKBOOKS[0])
fetched = db_api.get_workbook(created.name)
self.assertEqual(created, fetched)
db_api.delete_workbook(created.name)
self.assertRaises(
exc.NotFoundException,
db_api.get_workbook,
created.name
)
def test_workbooks_in_two_projects(self):
created = db_api.create_workbook(WORKBOOKS[1])
fetched = db_api.get_workbooks()
self.assertEqual(1, len(fetched))
self.assertEqual(created, fetched[0])
# Create a new user.
ctx = auth_context.MistralContext(
user_id='9-0-44-5',
project_id='99-88-33',
user_name='test-user',
project_name='test-another',
is_admin=False
)
auth_context.set_ctx(ctx)
created = db_api.create_workbook(WORKBOOKS[1])
fetched = db_api.get_workbooks()
self.assertEqual(1, len(fetched))
self.assertEqual(created, fetched[0])
def test_update_workbook(self):
created = db_api.create_workbook(WORKBOOKS[0])
self.assertIsNone(created.updated_at)
updated = db_api.update_workbook(created.name,
{'definition': 'my new definition'})
self.assertEqual('my new definition', updated.definition)
fetched = db_api.get_workbook(created['name'])
self.assertEqual(updated, fetched)
self.assertIsNotNone(fetched.updated_at)
def test_commit_transaction(self):
with db_api.transaction():
created = db_api.create_workbook(WORKBOOKS[0])
fetched = db_api.get_workbook(created.name)
self.assertEqual(created, fetched)
self.assertTrue(self.is_db_session_open())
self.assertFalse(self.is_db_session_open())
fetched = db_api.get_workbook(created.name)
self.assertEqual(created, fetched)
self.assertFalse(self.is_db_session_open())
def test_workbook_private(self):
# Create a workbook(scope=private) as under one project
# then make sure it's NOT visible for other projects.
created1 = db_api.create_workbook(WORKBOOKS[1])
fetched = db_api.get_workbooks()
self.assertEqual(1, len(fetched))
self.assertEqual(created1, fetched[0])
# Create a new user.
auth_context.set_ctx(test_base.get_context(default=False))
fetched = db_api.get_workbooks()
self.assertEqual(0, len(fetched))
def test_workbook_private(self):
# Create a workbook(scope=private) as under one project
# then make sure it's NOT visible for other projects.
created1 = db_api.create_workbook(WORKBOOKS[1])
fetched = db_api.get_workbooks()
self.assertEqual(1, len(fetched))
self.assertEqual(created1, fetched[0])
# Create a new user.
auth_context.set_ctx(test_base.get_context(default=False))
fetched = db_api.get_workbooks()
self.assertEqual(0, len(fetched))
def test_update_workbook(self):
created = db_api.create_workbook(WORKBOOKS[0])
self.assertIsNone(created.updated_at)
updated = db_api.update_workbook(
created.name,
{'definition': 'my new definition'}
)
self.assertEqual('my new definition', updated.definition)
fetched = db_api.get_workbook(created['name'])
self.assertEqual(updated, fetched)
self.assertIsNotNone(fetched.updated_at)
def test_rollback(self):
db_api.start_tx()
try:
created = db_api.create_workbook(WORKBOOKS[0])
fetched = db_api.get_workbook(created.name)
self.assertEqual(created, fetched)
self.assertTrue(self.is_db_session_open())
db_api.rollback_tx()
finally:
db_api.end_tx()
self.assertFalse(self.is_db_session_open())
self.assertRaises(exc.NotFoundException, db_api.get_workbook,
created['id'])
self.assertFalse(self.is_db_session_open())
def test_commit(self):
db_api.start_tx()
try:
created = db_api.create_workbook(WORKBOOKS[0])
fetched = db_api.get_workbook(created.name)
self.assertEqual(created, fetched)
self.assertTrue(self.is_db_session_open())
db_api.commit_tx()
finally:
db_api.end_tx()
self.assertFalse(self.is_db_session_open())
fetched = db_api.get_workbook(created.name)
self.assertEqual(created, fetched)
self.assertFalse(self.is_db_session_open())
def test_commit(self):
db_api.start_tx()
try:
created = db_api.create_workbook(WORKBOOKS[0])
fetched = db_api.get_workbook(created.name)
self.assertEqual(created, fetched)
self.assertTrue(self.is_db_session_open())
db_api.commit_tx()
finally:
db_api.end_tx()
self.assertFalse(self.is_db_session_open())
fetched = db_api.get_workbook(created.name)
self.assertEqual(created, fetched)
self.assertFalse(self.is_db_session_open())
def test_rollback(self):
db_api.start_tx()
try:
created = db_api.create_workbook(WORKBOOKS[0])
fetched = db_api.get_workbook(created.name)
self.assertEqual(created, fetched)
self.assertTrue(self.is_db_session_open())
db_api.rollback_tx()
finally:
db_api.end_tx()
self.assertFalse(self.is_db_session_open())
self.assertRaises(
exc.NotFoundException,
db_api.get_workbook,
created['id']
)
self.assertFalse(self.is_db_session_open())
def test_workbook_private(self):
# Create a workbook(scope=private) as under one project
# then make sure it's NOT visible for other projects.
created1 = db_api.create_workbook(WORKBOOKS[1])
fetched = db_api.get_workbooks()
self.assertEqual(1, len(fetched))
self.assertEqual(created1, fetched[0])
# Create a new user.
ctx = auth_context.MistralContext(
user_id='9-0-44-5',
project_id='99-88-33',
user_name='test-user',
project_name='test-another',
is_admin=False
)
auth_context.set_ctx(ctx)
fetched = db_api.get_workbooks()
self.assertEqual(0, len(fetched))
def test_workbook_public(self):
# Create a workbook(scope=public) as under one project
# then make sure it's visible for other projects.
created0 = db_api.create_workbook(WORKBOOKS[0])
fetched = db_api.get_workbooks()
self.assertEqual(1, len(fetched))
self.assertEqual(created0, fetched[0])
# Assert that the project_id stored is actually the context's
# project_id not the one given.
self.assertEqual(created0.project_id, auth_context.ctx().project_id)
self.assertNotEqual(WORKBOOKS[0]['project_id'],
auth_context.ctx().project_id)
# Create a new user.
auth_context.set_ctx(test_base.get_context(default=False))
fetched = db_api.get_workbooks()
self.assertEqual(1, len(fetched))
self.assertEqual(created0, fetched[0])
self.assertEqual('public', created0.scope)
def test_workbook_public(self):
# Create a workbook(scope=public) as under one project
# then make sure it's visible for other projects.
created0 = db_api.create_workbook(WORKBOOKS[0])
fetched = db_api.get_workbooks()
self.assertEqual(1, len(fetched))
self.assertEqual(created0, fetched[0])
# Assert that the project_id stored is actually the context's
# project_id not the one given.
self.assertEqual(created0.project_id, auth_context.ctx().project_id)
self.assertNotEqual(WORKBOOKS[0]['project_id'],
auth_context.ctx().project_id)
# Create a new user.
auth_context.set_ctx(test_base.get_context(default=False))
fetched = db_api.get_workbooks()
self.assertEqual(1, len(fetched))
self.assertEqual(created0, fetched[0])
self.assertEqual('public', created0.scope)
def test_workbook_repr(self):
s = db_api.create_workbook(WORKBOOKS[0]).__repr__()
self.assertIn('Workbook ', s)
self.assertIn("'name': 'my_workbook1'", s)
def test_workbook_repr(self):
s = db_api.create_workbook(WORKBOOKS[0]).__repr__()
self.assertIn('Workbook ', s)
self.assertIn("'name': 'my_workbook1'", s)
def test_create_workbook_duplicate_without_auth(self):
cfg.CONF.set_default('auth_enable', False, group='pecan')
db_api.create_workbook(WORKBOOKS[0])
self.assertRaises(exc.DBDuplicateEntry, db_api.create_workbook,
WORKBOOKS[0])
