def setup_method(self, method):
imp.reload(configreader)
imp.reload(test.config.app)
context.setup(
app='test',
app_config=test.config.app.TestingConfig()
)
context.setup_thread_id()
context.default_log.info(
'%s %s::%s %s' % ('=' * 20, self.__class__.__name__, self._testMethodName, '=' * 20)
)
# modify http client to store http requests
from amplify.agent.util.http import HTTPClient
self.http_requests = []
original_get = HTTPClient.get
original_post = HTTPClient.post
def fake_get(obj, url, *args, **kwargs):
self.http_requests.append(url)
return original_get(obj, url, *args, **kwargs)
def fake_post(obj, url, *args, **kwargs):
self.http_requests.append(url)
return original_post(obj, url, *args, **kwargs)
HTTPClient.get = fake_get
HTTPClient.post = fake_post
def run(self):
# TODO: stop after 3 fatal errors
current_thread().name = 'bridge'
context.setup_thread_id()
try:
while True:
m_size_b, m_rss_b = memusage.report()
self.wait()
context.inc_action_id()
self.flush_meta()
if not self.first_run:
self.flush_metrics()
self.flush_events()
self.flush_configs()
else:
self.first_run = False
gc.collect()
m_size_a, m_rss_a = memusage.report()
context.log.debug('mem before: (%s %s), after (%s, %s)' % (m_size_b, m_rss_b, m_size_a, m_rss_a))
if m_rss_a >= m_rss_b:
context.log.debug('RSS MEMORY same or increased, diff %s' % (m_rss_a-m_rss_b))
elif m_size_a >= m_size_b:
context.log.debug('VSIZE MEMORY same or increased, diff %s' % (m_size_a-m_size_b))
except:
context.default_log.error('failed', exc_info=True)
raise
def setup_method(self, method):
imp.reload(configreader)
imp.reload(test.config.app)
context.setup(
app='test',
app_config=test.config.app.TestingConfig()
)
context.setup_thread_id()
context.default_log.info(
'%s %s::%s %s' % ('=' * 20, self.__class__.__name__, self._testMethodName, '=' * 20)
)
def run(self):
current_thread().name = 'bridge'
context.setup_thread_id()
try:
while True:
self.wait()
context.inc_action_id()
self.flush_all()
gc.collect()
except:
context.default_log.error('failed', exc_info=True)
raise
def run(self):
"""
Common collector cycle
1. Collect data
2. Sleep
3. Stop if object stopped
"""
current_thread().name = self.short_name
context.setup_thread_id()
try:
while True:
context.inc_action_id()
if self.object.running:
self._collect()
self._sleep()
else:
break
except:
context.log.error('%s failed' % self.object.id, exc_info=True)
raise
