diff options
Diffstat (limited to 'docker/storperf-master/tests/storperf_master_test.py')
| -rw-r--r-- | docker/storperf-master/tests/storperf_master_test.py | 37 |
1 files changed, 12 insertions, 25 deletions
diff --git a/docker/storperf-master/tests/storperf_master_test.py b/docker/storperf-master/tests/storperf_master_test.py index f328982..1edac6d 100644 --- a/docker/storperf-master/tests/storperf_master_test.py +++ b/docker/storperf-master/tests/storperf_master_test.py @@ -7,45 +7,23 @@ # http://www.apache.org/licenses/LICENSE-2.0 ############################################################################## -import os import unittest -import mock +from unittest.mock import patch -from storperf.db.configuration_db import ConfigurationDB from storperf.storperf_master import StorPerfMaster -class MockStack(object): - - def __init__(self): - pass - - def get_stack(self): - return None - - class StorPerfMasterTest(unittest.TestCase): def setUp(self): - ConfigurationDB.db_name = __name__ + '.db' - try: - os.remove(ConfigurationDB.db_name) - except OSError: - pass - with mock.patch("storperf.storperf_master.OSCreds"), \ - mock.patch( + with patch("storperf.storperf_master.OSCreds"), \ + patch( "storperf.storperf_master.OpenStackHeatStack") as oshs: oshs.return_value.get_stack.return_value = None self.storperf = StorPerfMaster() - def tearDown(self): - try: - os.remove(ConfigurationDB.db_name) - except OSError: - pass - def test_agent_count(self): expected = 10 @@ -82,6 +60,15 @@ class StorPerfMasterTest(unittest.TestCase): self.assertEqual( expected, actual, "Did not expect: " + str(actual)) + def test_volume_type(self): + expected = 'tripleo-ceph' + + self.storperf.volume_type = expected + actual = self.storperf.volume_type + + self.assertEqual( + expected, actual, "Did not expect: " + str(actual)) + def test_agent_network(self): expected = "ABCDEF" |