mirror of
https://github.com/valitydev/redash.git
synced 2024-11-07 01:25:16 +00:00
84 lines
2.7 KiB
Python
84 lines
2.7 KiB
Python
from unittest import TestCase
|
|
|
|
from jsonschema import ValidationError
|
|
|
|
from redash.utils.configuration import ConfigurationContainer
|
|
|
|
|
|
configuration_schema = {
|
|
"type": "object",
|
|
"properties": {
|
|
"a": {
|
|
"type": "integer"
|
|
},
|
|
"e": {
|
|
"type": "integer"
|
|
},
|
|
"b": {
|
|
"type": "string"
|
|
}
|
|
},
|
|
"required": ["a"],
|
|
"secret": ["b"]
|
|
}
|
|
|
|
|
|
class TestConfigurationToJson(TestCase):
|
|
def setUp(self):
|
|
self.config = {'a': 1, 'b': 'test'}
|
|
self.container = ConfigurationContainer(self.config, configuration_schema)
|
|
|
|
def test_returns_plain_dict(self):
|
|
self.assertDictEqual(self.config, self.container.to_dict())
|
|
|
|
def test_raises_exception_when_no_schema_set(self):
|
|
self.container.set_schema(None)
|
|
self.assertRaises(RuntimeError, lambda: self.container.to_dict(mask_secrets=True))
|
|
|
|
def test_returns_dict_with_masked_secrets(self):
|
|
d = self.container.to_dict(mask_secrets=True)
|
|
|
|
self.assertEqual(d['a'], self.config['a'])
|
|
self.assertNotEqual(d['b'], self.config['b'])
|
|
|
|
self.assertEqual(self.config['b'], self.container['b'])
|
|
|
|
|
|
class TestConfigurationUpdate(TestCase):
|
|
def setUp(self):
|
|
self.config = {'a': 1, 'b': 'test'}
|
|
self.container = ConfigurationContainer(self.config, configuration_schema)
|
|
|
|
def test_rejects_invalid_new_config(self):
|
|
self.assertRaises(ValidationError, lambda: self.container.update({'c': 3}))
|
|
|
|
def test_fails_if_no_schema_set(self):
|
|
self.container.set_schema(None)
|
|
self.assertRaises(RuntimeError, lambda: self.container.update({'c': 3}))
|
|
|
|
def test_ignores_secret_placehodler(self):
|
|
self.container.update(self.container.to_dict(mask_secrets=True))
|
|
self.assertEqual(self.container['b'], self.config['b'])
|
|
|
|
def test_updates_secret(self):
|
|
new_config = {'a': 2, 'b': 'new'}
|
|
self.container.update(new_config)
|
|
self.assertDictEqual(self.container._config, new_config)
|
|
|
|
def test_doesnt_leave_leftovers(self):
|
|
container = ConfigurationContainer({'a': 1, 'b': 'test', 'e': 3}, configuration_schema)
|
|
new_config = container.to_dict(mask_secrets=True)
|
|
new_config.pop('e')
|
|
container.update(new_config)
|
|
|
|
self.assertEqual(container['a'], 1)
|
|
self.assertEqual('test', container['b'])
|
|
self.assertNotIn('e', container)
|
|
|
|
def test_works_for_schema_without_secret(self):
|
|
secretless = configuration_schema.copy()
|
|
secretless.pop('secret')
|
|
container = ConfigurationContainer({'a': 1, 'b': 'test', 'e': 3}, secretless)
|
|
container.update({'a': 2})
|
|
self.assertEqual(container['a'], 2)
|