class CollectorTest(BaseTest):
def setUp(self):
super(CollectorTest, self).setUp()
self._server = None
def tearDown(self):
super(CollectorTest, self).tearDown()
self._stop_server()
def _start_server(self, sinks=None):
if self._server is None or not self._server.is_alive():
self._server = CollectorRunner(sinks)
self._server.start()
def _stop_server(self):
if self._server is not None and self._server.is_alive():
self._server.stop()
self._server.join()
self._server = None
def test_session_setup(self):
update({'collector': {'session': 'collector_test'}})
config = get_config()
session_collection_name = SessionCollection.get_collection_name()
col = SessionCollection(self.db[session_collection_name])
self._start_server()
time.sleep(.2)
self.assertEqual(len(col.find({'name': config.collector.session,
'start_time': {'$exists': True},
'end_time': {'$exists': False}})),
1)
self._stop_server()
time.sleep(.2)
self.assertEqual(len(col.find({'name': config.collector.session,
'start_time': {'$exists': True},
'end_time': {'$exists': True}})),
1)
def test_handle(self):
config = get_config()
sock = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
addr = (config.collector.addr, config.collector.port)
sink = _BufferSink()
for i in xrange(5):
sock.sendto('blah', addr)
self._start_server([sink])
# FIXME: adding an event signaling that a socket is bound and ready to
# receive data at the end of BaseServer.start does not seem to
# work, falling back to sleep for the time being
time.sleep(1)
for i in xrange(5):
sock.sendto('blah', addr)
time.sleep(.1)
self._stop_server()
for i in xrange(5):
sock.sendto('blah', addr)
time.sleep(.1)
self.assertEqual([x[0] for x in sink.msgs], ['blah'] * 5)
def run(self):
logging.info('running collector...')
update({'collector': {
'session': self.args.session,
'mongo_uri': self.args.uri,
'addr': self.args.addr,
'port': self.args.port
},
'index_profile_sink': {
'mongo_uri': self.args.uri
},
'query_profile_uri': {
'mongo_uri': self.args.uri
}})
collector = CollectorRunner([IndexProfileSink(), QueryProfileSink()])
collector.start()
while not should_exit:
time.sleep(.1)
collector.stop()
collector.join()
logging.info('collector stopped...')
def _start_server(self, sinks=None):
if self._server is None or not self._server.is_alive():
self._server = CollectorRunner(sinks)
self._server.start()
def _start_server(self, sinks=None):
if self._server is None or not self._server.is_alive():
self._server = CollectorRunner(sinks)
self._server.start()
class CollectorTest(BaseTest):
def setUp(self):
super(CollectorTest, self).setUp()
self._server = None
def tearDown(self):
super(CollectorTest, self).tearDown()
self._stop_server()
def _start_server(self, sinks=None):
if self._server is None or not self._server.is_alive():
self._server = CollectorRunner(sinks)
self._server.start()
def _stop_server(self):
if self._server is not None and self._server.is_alive():
self._server.stop()
self._server.join()
self._server = None
def test_session_setup(self):
update({'collector': {'session': 'collector_test'}})
config = get_config()
session_collection_name = SessionCollection.get_collection_name()
col = SessionCollection(self.db[session_collection_name])
self._start_server()
time.sleep(.2)
self.assertEqual(
len(
col.find({
'name': config.collector.session,
'start_time': {
'$exists': True
},
'end_time': {
'$exists': False
}
})), 1)
self._stop_server()
time.sleep(.2)
self.assertEqual(
len(
col.find({
'name': config.collector.session,
'start_time': {
'$exists': True
},
'end_time': {
'$exists': True
}
})), 1)
def test_handle(self):
config = get_config()
sock = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
addr = (config.collector.addr, config.collector.port)
sink = _BufferSink()
for i in xrange(5):
sock.sendto('blah', addr)
self._start_server([sink])
# FIXME: adding an event signaling that a socket is bound and ready to
# receive data at the end of BaseServer.start does not seem to
# work, falling back to sleep for the time being
time.sleep(1)
for i in xrange(5):
sock.sendto('blah', addr)
time.sleep(.1)
self._stop_server()
for i in xrange(5):
sock.sendto('blah', addr)
time.sleep(.1)
self.assertEqual([x[0] for x in sink.msgs], ['blah'] * 5)
