mirror of
https://github.com/valitydev/osquery-1.git
synced 2024-11-08 02:18:53 +00:00
56 lines
1.9 KiB
Python
Executable File
56 lines
1.9 KiB
Python
Executable File
#!/usr/bin/env python
|
|
|
|
# Copyright (c) 2014, Facebook, Inc.
|
|
# All rights reserved.
|
|
#
|
|
# This source code is licensed under the BSD-style license found in the
|
|
# LICENSE file in the root directory of this source tree. An additional grant
|
|
# of patent rights can be found in the PATENTS file in the same directory.
|
|
|
|
from __future__ import absolute_import
|
|
from __future__ import division
|
|
from __future__ import print_function
|
|
from __future__ import unicode_literals
|
|
|
|
import unittest
|
|
|
|
# osquery-specific testing utils
|
|
import test_base
|
|
|
|
class WatchdogTests(test_base.ProcessGenerator, unittest.TestCase):
|
|
def test_1_daemon_without_watchdog(self):
|
|
config = test_base.CONFIG.copy()
|
|
config["options"]["disable_watchdog"] = "true"
|
|
config["options"]["disable_extensions"] = "true"
|
|
daemon = self._run_daemon(config)
|
|
self.assertTrue(daemon.isAlive())
|
|
daemon.kill()
|
|
|
|
def test_2_daemon_with_watchdog(self):
|
|
config = test_base.CONFIG.copy()
|
|
config["options"]["disable_watchdog"] = "false"
|
|
daemon = self._run_daemon(config)
|
|
self.assertTrue(daemon.isAlive())
|
|
|
|
# Check that the daemon spawned a child process
|
|
children = daemon.getChildren()
|
|
self.assertTrue(len(children) > 0)
|
|
daemon.kill()
|
|
|
|
# This will take a few moments to make sure the client process
|
|
# dies when the watcher goes away
|
|
self.assertTrue(daemon.isDead(children[0].pid))
|
|
|
|
def test_3_catastrophic_worker_failure(self):
|
|
config = test_base.CONFIG.copy()
|
|
# A bad DB path will cause the worker to fail.
|
|
config["options"]["db_path"] = "/tmp/this/does/not/exists.db"
|
|
config["options"]["disable_watchdog"] = "false"
|
|
daemon = self._run_daemon(config)
|
|
daemon.isAlive(5)
|
|
self.assertTrue(daemon.isDead(daemon.pid))
|
|
daemon.kill()
|
|
|
|
if __name__ == '__main__':
|
|
test_base.Tester().run()
|