def tearDownClass(cls): cls.set_environment(cls, cls.SANDBOX) cleanup = {'categories': 10000} client = TestClient(env=os.environ['TAP_SQUARE_ENVIRONMENT']) for stream, limit in cleanup.items(): print("Checking if cleanup is required.") all_records = client.get_all(stream, start_date=cls.STATIC_START_DATE) all_ids = [rec.get('id') for rec in all_records if not rec.get('is_deleted')] if len(all_ids) > limit / 2: chunk = int(len(all_ids) - (limit / 2)) print("Cleaning up {} excess records".format(chunk)) client.delete_catalog(all_ids[:chunk])
if test_updates: # Need to reference specific attr from an obj to update print("Testing UPDATE: {}".format(obj)) # import pdb; pdb.set_trace() # UNCOMMENT TO RUN 'INTERACTIVELY' obj_id = created_obj[0].get('id') version = created_obj[0].get('version') updated_obj = client.update(stream=obj, obj_id=obj_id, version=version, obj=created_obj[0], start_date=START_DATE) if updated_obj: print("SUCCESS") if print_objects: print("{}\n".format(updated_obj)) continue print("FAILED") if test_deletes: # Need to reference id from an obj to delete print("Testing UPDATE: {}".format(obj)) # import pdb; pdb.set_trace() # UNCOMMENT TO RUN 'INTERACTIVELY' obj_ids = [created_obj[0].get('id')] version = created_obj[0].get('version') deleted_obj = client.delete_catalog(ids_to_delete=obj_ids) if deleted_obj: print("SUCCESS") if print_objects: print("{}\n".format(deleted_obj)) continue print("FAILED")