actions/opensearch-pyd
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

Implement teardown functionality for yaml tests

Browse Source
This commit is contained in:
Honza Král
2016-07-11 11:01:56 +02:00
parent a53efe1705
commit 06e9a9851c
1 changed files with 4 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
test_elasticsearch/test_server/test_common.py
+4 -2
View File
@@ -50,6 +50,8 @@ class YamlTestCase(ElasticsearchTestCase):
self._state = {} self._state = {}
def tearDown(self): def tearDown(self):
if hasattr(self, '_teardown_code'):
self.run_code(self._teardown_code)
super(YamlTestCase, self).tearDown() super(YamlTestCase, self).tearDown()
for repo, definition in self.client.snapshot.get_repository(repository='_all').items(): for repo, definition in self.client.snapshot.get_repository(repository='_all').items():
self.client.snapshot.delete_repository(repository=repo) self.client.snapshot.delete_repository(repository=repo)
@@ -255,8 +257,8 @@ def construct_case(filename, name):
i = 0 i = 0
for test in tests: for test in tests:
for test_name, definition in test.items(): for test_name, definition in test.items():
if test_name == 'setup': if test_name in ('setup', 'teardown'):
attrs['_setup_code'] = definition attrs['_%s_code' % test_name] = definition
continue continue
attrs['test_from_yaml_%d' % i] = make_test(test_name, definition, i) attrs['test_from_yaml_%d' % i] = make_test(test_name, definition, i)