def setUp(self):
""" PREREQUISITES """
PlanAdapter.create_table()
ServiceTypeAdapter.create_table()
PlanServiceTypeAdapter.create_table()
# ManifestAdapter.create_table()
self.service = ServiceTypeAdapter.sample_model('instance1')
ServiceTypeAdapter.save(self.service)
Adapter.create_table()
self.test_model = Adapter.sample_model('instance1')
self.id_name = Adapter.get_id(self.test_model)
_, self.result = SQLStore.add_service_instance(self.test_model)
def add_service_instance(instance: ServiceInstance) -> tuple:
id_name = ServiceInstanceAdapter.get_id(instance)
if ServiceInstanceAdapter.exists_in_db(id_name):
LOG.warning(
'An existing instance was attempted to be saved. Updating it...'
)
SQLStore.delete_service_instance(id_name)
# return 'The Instance already exists in the catalog.', 409
''' Attempt to Create Table '''
PlanAdapter.create_table()
ServiceTypeAdapter.create_table()
ManifestAdapter.create_table()
ServiceInstanceAdapter.create_table()
LastOperationAdapter.create_table()
ServiceInstanceAdapter.save(instance)
id_name = ServiceInstanceAdapter.get_id(instance)
if ServiceInstanceAdapter.exists_in_db(id_name):
LOG.warning('Instance added successfully...')
return 'Instance added successfully', 200
else:
LOG.warning('Could not save the Instance in the DB...')
return 'Could not save the Instance in the DB', 500
def delete_last_operation(self, instance_id: str = None) -> tuple:
if ServiceInstanceAdapter.exists_in_db(instance_id):
instance = ServiceInstanceAdapter.find_by_id_name(instance_id)
''' Attempt to Create Table '''
PlanAdapter.create_table()
ServiceTypeAdapter.create_table()
ServiceTypeAdapter.create_table()
ServiceInstanceAdapter.create_table()
LastOperationAdapter.create_table()
instance.state = None
ServiceInstanceAdapter.save(instance)
id_name = ServiceInstanceAdapter.get_id(instance)
if ServiceInstanceAdapter.exists_in_db(id_name):
return 'Instance added successfully', 200
else:
return 'Could not save the Instance in the DB', 500
else:
raise Exception('Service Instance not found')
def tearDown(self):
SQLStore.delete_service_instance(Adapter.get_id(self.test_model))
ServiceTypeAdapter.delete(self.service.id)
