def tearDown(self):
""" Reset all environment variables of consequence """
if "AWS_EXECUTION_ENV" in os.environ:
os.environ.pop("AWS_EXECUTION_ENV")
if "INSTANA_EXTRA_HTTP_HEADERS" in os.environ:
os.environ.pop("INSTANA_EXTRA_HTTP_HEADERS")
if "INSTANA_ENDPOINT_URL" in os.environ:
os.environ.pop("INSTANA_ENDPOINT_URL")
if "INSTANA_AGENT_KEY" in os.environ:
os.environ.pop("INSTANA_AGENT_KEY")
if "INSTANA_ZONE" in os.environ:
os.environ.pop("INSTANA_ZONE")
if "INSTANA_TAGS" in os.environ:
os.environ.pop("INSTANA_TAGS")
set_agent(self.original_agent)
set_tracer(self.original_tracer)
def create_agent_and_setup_tracer(self):
self.agent = AWSFargateAgent()
self.span_recorder = StanRecorder(self.agent)
self.tracer = InstanaTracer(recorder=self.span_recorder)
set_agent(self.agent)
set_tracer(self.tracer)
# Manually set the ECS Metadata API results on the collector
with open(self.pwd + '/../data/fargate/1.3.0/root_metadata.json',
'r') as json_file:
self.agent.collector.root_metadata = json.load(json_file)
with open(self.pwd + '/../data/fargate/1.3.0/task_metadata.json',
'r') as json_file:
self.agent.collector.task_metadata = json.load(json_file)
with open(self.pwd + '/../data/fargate/1.3.0/stats_metadata.json',
'r') as json_file:
self.agent.collector.stats_metadata = json.load(json_file)
with open(self.pwd + '/../data/fargate/1.3.0/task_stats_metadata.json',
'r') as json_file:
self.agent.collector.task_stats_metadata = json.load(json_file)
def tearDown(self):
""" Reset all environment variables of consequence """
if "AWS_EXECUTION_ENV" in os.environ:
os.environ.pop("AWS_EXECUTION_ENV")
if "LAMBDA_HANDLER" in os.environ:
os.environ.pop("LAMBDA_HANDLER")
if "INSTANA_EXTRA_HTTP_HEADERS" in os.environ:
os.environ.pop("INSTANA_EXTRA_HTTP_HEADERS")
if "INSTANA_ENDPOINT_URL" in os.environ:
os.environ.pop("INSTANA_ENDPOINT_URL")
if "INSTANA_ENDPOINT_PROXY" in os.environ:
os.environ.pop("INSTANA_ENDPOINT_PROXY")
if "INSTANA_AGENT_KEY" in os.environ:
os.environ.pop("INSTANA_AGENT_KEY")
if "INSTANA_SERVICE_NAME" in os.environ:
os.environ.pop("INSTANA_SERVICE_NAME")
if "INSTANA_DEBUG" in os.environ:
os.environ.pop("INSTANA_DEBUG")
if "INSTANA_LOG_LEVEL" in os.environ:
os.environ.pop("INSTANA_LOG_LEVEL")
set_agent(self.original_agent)
set_tracer(self.original_tracer)
def create_agent_and_setup_tracer(self):
self.agent = AWSFargateAgent()
self.span_recorder = StanRecorder(self.agent)
self.tracer = InstanaTracer(recorder=self.span_recorder)
set_agent(self.agent)
set_tracer(self.tracer)
