def run_test(self):
"""
Common test path for all configs to test against
:return:
"""
print('Basic string operations using get/put/delete')
resp = archive.put(userId=self.test_user_id, bucket=self.test_bucket_id, archiveid='document_1',
data=self.document_1)
print(('Document 1 PUT: {}'.format(resp)))
resp = archive.get(userId=self.test_user_id, bucket=self.test_bucket_id, archiveid='document_1')
self.assertEqual(self.document_1, resp)
self.assertTrue(archive.exists(self.test_user_id, self.test_bucket_id, 'document_1'))
self.assertFalse(archive.exists(self.test_user_id, self.test_bucket_id, 'document_10'))
print('Document operations')
resp = archive.put_document(userId=self.test_user_id, bucket=self.test_bucket_id, archiveId='document_json',
data=self.document_json)
print(('Document JSON PUT Doc: {}'.format(resp)))
resp = archive.get_document(userId=self.test_user_id, bucket=self.test_bucket_id, archiveId='document_json')
print(('Document JSON GET Dock: {}'.format(resp)))
self.assertEqual(self.document_json, resp)
print('Document operations')
resp = archive.put_document(userId=self.test_user_id, bucket=self.test_bucket_id, archiveId='document_json',
data=self.document_1)
print(('Document string PUT Doc: {}'.format(resp)))
resp = archive.get_document(userId=self.test_user_id, bucket=self.test_bucket_id, archiveId='document_json')
print(('Document string GET Dock: {}'.format(resp)))
self.assertEqual(self.document_1, resp)
def add_data(self):
print('Adding data')
for i in range(0, 100):
archiveId = 'doc-{}'.format(i)
print(('Adding document: {}'.format(archiveId)))
archive.put_document(userId='test1',
bucket='testing',
archiveId=archiveId,
data='TESTINGBUCKETDATASMALL'.join(
[str(x) for x in range(100)]))
def save_policy(user_id, policyId, active, policy_bundle, dbsession):
"""
Do the save, outside the context of an api call itself.
:param user_id: str - requesting usera
:param policyId: str - the id for policy
:param active: boolean - is active or not
:param policy_bundle: dict - bundle content
:return:
"""
try:
if archive.put_document(user_id, 'policy_bundles', policyId, policy_bundle):
rc = db_policybundle.update(policyId, user_id, active, policy_bundle, session=dbsession)
else:
rc = False
except Exception as err:
raise anchore_engine.common.helpers.make_anchore_exception(err,
input_message="cannot add policy, failed to update archive/DB",
input_httpcode=500)
if not rc:
raise Exception("DB update failed")
else:
if active:
try:
rc = db_policybundle.set_active_policy(policyId, user_id, session=dbsession)
except Exception as err:
raise Exception("could not set policy as active - exception: " + str(err))
record = db_policybundle.get(user_id, policyId, session=dbsession)
record['policybundle'] = policy_bundle
return record