mirror of
https://github.com/vee1e/speakeasy.git
synced 2026-09-02 19:37:28 +00:00
63 lines
2 KiB
Python
63 lines
2 KiB
Python
# Copyright (C) 2020 FireEye, Inc. All Rights Reserved.
|
|
|
|
import os
|
|
import json
|
|
import os.path
|
|
|
|
import unittest
|
|
import jsonschema
|
|
|
|
import speakeasy
|
|
import speakeasy.speakeasy
|
|
|
|
|
|
def get_default_config():
|
|
fpath = os.path.join(os.path.dirname(speakeasy.__file__), 'configs', 'default.json')
|
|
with open(fpath, 'r') as ff:
|
|
return json.load(ff)
|
|
|
|
|
|
class TestConfig(unittest.TestCase):
|
|
def test_speakeasy_configs(self):
|
|
'''
|
|
Make sure all of the bundled configs shipped with speakeasy conform
|
|
'''
|
|
config_dir = os.path.join(os.path.dirname(speakeasy.__file__), 'configs')
|
|
self.assertTrue(os.path.isdir(config_dir))
|
|
for fname in os.listdir(config_dir):
|
|
if not fname.endswith('.json'):
|
|
continue
|
|
fpath = os.path.join(config_dir, fname)
|
|
self.assertTrue(os.path.isfile(fpath))
|
|
with open(fpath, 'r') as ff:
|
|
config = json.load(ff)
|
|
speakeasy.speakeasy.validate_config(config)
|
|
|
|
def test_validation(self):
|
|
'''
|
|
Tweak the default config, validate that errors are caught
|
|
'''
|
|
conf0 = get_default_config()
|
|
self.assertTrue('emu_engine' in conf0)
|
|
speakeasy.speakeasy.validate_config(conf0)
|
|
|
|
# validating a non enum
|
|
conf0['emu_engine'] = 'alternate_engine'
|
|
with self.assertRaises(jsonschema.exceptions.ValidationError):
|
|
speakeasy.speakeasy.validate_config(conf0)
|
|
|
|
# validate a missing required field
|
|
conf1 = get_default_config()
|
|
conf1.pop('emu_engine', None)
|
|
with self.assertRaises(jsonschema.exceptions.ValidationError):
|
|
speakeasy.speakeasy.validate_config(conf1)
|
|
|
|
# validate an incorrect type
|
|
conf2 = get_default_config()
|
|
conf2['emu_engine'] = 1.0
|
|
with self.assertRaises(jsonschema.exceptions.ValidationError):
|
|
speakeasy.speakeasy.validate_config(conf2)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main()
|