def test_context_creation_and_retrieval(self): """ Context should be created and retrieved successfully. """ tracer_id = "d551573a-01dc-41b2-b197-ea8afb7fbac1".replace("-", "") with new_context(tracer_id=tracer_id): context = get_context() nose.tools.eq_(tracer_id, str(context.tracer_id))
def main(argv): if len(argv) < 4: msg = "You must specify the environment, the external_user_id, " \ "the external_product_id, and the impression date" log.error(msg) return {"success": False, "message": msg} try: # command-line arguments env = argv[0] user = argv[1] product = argv[2] impression_date = dateutil.parser.parse(argv[3]) impression = {"external_user_id": user, "external_product_id": product, "created_at": impression_date} session = init_session(env) maintenance.process_impression(session, impression) return {"success": True} except Exception: log.exception('Exception on {0}:'.format(__name__)) return {"success": False, "message": traceback.format_exc()} if __name__ == '__main__': with new_context(): print(main(sys.argv[1:]))
def work(): env = args[0] if len(args) > 0 and is_valid_customer_identifier(args[0]) else None with new_context(tracer_id=self._get_tracer_id(), endpoint=self._get_endpoint_name(), environment=env): return method(*args)