Matthew Treinish | 87f772c | 2013-12-16 20:07:42 +0000 | [diff] [blame] | 1 | # Copyright 2013 IBM Corp. |
| 2 | # |
| 3 | # Licensed under the Apache License, Version 2.0 (the "License"); you may |
| 4 | # not use this file except in compliance with the License. You may obtain |
| 5 | # a copy of the License at |
| 6 | # |
| 7 | # http://www.apache.org/licenses/LICENSE-2.0 |
| 8 | # |
| 9 | # Unless required by applicable law or agreed to in writing, software |
| 10 | # distributed under the License is distributed on an "AS IS" BASIS, WITHOUT |
| 11 | # WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the |
| 12 | # License for the specific language governing permissions and limitations |
| 13 | # under the License. |
| 14 | |
| 15 | import os |
| 16 | import unittest |
| 17 | |
| 18 | |
| 19 | def load_tests(loader, tests, pattern): |
| 20 | suite = unittest.TestSuite() |
| 21 | base_path = os.path.split(os.path.dirname(os.path.abspath(__file__)))[0] |
| 22 | base_path = os.path.split(base_path)[0] |
| 23 | for test_dir in ['./tempest/api', './tempest/cli', './tempest/scenario', |
| 24 | './tempest/thirdparty']: |
| 25 | if not pattern: |
| 26 | suite.addTests(loader.discover(test_dir, top_level_dir=base_path)) |
| 27 | else: |
| 28 | suite.addTests(loader.discover(test_dir, pattern=pattern, |
| 29 | top_level_dir=base_path)) |
| 30 | return suite |