mirror of
https://github.com/valitydev/osquery-1.git
synced 2024-11-07 01:55:20 +00:00
63 lines
1.8 KiB
Python
Executable File
63 lines
1.8 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
|
|
# Copyright (c) 2014-present, Facebook, Inc.
|
|
# All rights reserved.
|
|
#
|
|
# This source code is licensed in accordance with the terms specified in
|
|
# the LICENSE file found in the root directory of this source tree.
|
|
|
|
import json
|
|
import os
|
|
import shutil
|
|
import time
|
|
import unittest
|
|
import sys
|
|
|
|
# osquery-specific testing utils
|
|
import test_base
|
|
import utils
|
|
|
|
|
|
class ExampleQueryTests(test_base.QueryTester):
|
|
@test_base.flaky
|
|
def test_cross_platform_queries(self):
|
|
self._execute_set(PLATFORM_EXAMPLES["specs"])
|
|
|
|
@test_base.flaky
|
|
def test_platform_specific_queries(self):
|
|
posix = ["darwin", "linux"]
|
|
if utils.platform() in posix:
|
|
self._execute_set(PLATFORM_EXAMPLES["posix"])
|
|
self._execute_set(PLATFORM_EXAMPLES[utils.platform()])
|
|
|
|
@test_base.flaky
|
|
def test_utility_queries(self):
|
|
self._execute_set(PLATFORM_EXAMPLES["utility"])
|
|
|
|
if __name__ == '__main__':
|
|
# Import the API generation code for example query introspection.
|
|
SCRIPT_DIR = os.path.dirname(os.path.realpath(__file__))
|
|
SOURCE_DIR = os.path.abspath(SCRIPT_DIR + "/../../")
|
|
sys.path.append(SOURCE_DIR + "/tools/codegen")
|
|
from genapi import gen_api
|
|
API = gen_api(SOURCE_DIR + "/specs")
|
|
|
|
# Organize example queries by platform
|
|
PLATFORM_EXAMPLES = {}
|
|
for category in API:
|
|
PLATFORM_EXAMPLES[category["key"]] = []
|
|
for table in category["tables"]:
|
|
if len(table["examples"]) > 0:
|
|
PLATFORM_EXAMPLES[category["key"]] += table["examples"]
|
|
else:
|
|
PLATFORM_EXAMPLES[category["key"]] += [
|
|
"select * from %s limit 1" % table["name"]
|
|
]
|
|
|
|
module = test_base.Tester()
|
|
|
|
# Find and import the thrift-generated python interface
|
|
test_base.loadThriftFromBuild(test_base.ARGS.build)
|
|
|
|
module.run()
|