1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
|
##############################################################################
# Copyright (c) 2015 EMC and others.
#
# All rights reserved. This program and the accompanying materials
# are made available under the terms of the Apache License, Version 2.0
# which accompanies this distribution, and is available at
# http://www.apache.org/licenses/LICENSE-2.0
##############################################################################
import os
import unittest
import mock
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(
"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
self.storperf.agent_count = expected
actual = self.storperf.agent_count
self.assertEqual(
expected, actual, "Did not expect: " + str(actual))
def test_queue_depths(self):
expected = "1,2,3"
self.storperf.queue_depths = expected
actual = self.storperf.queue_depths
self.assertEqual(
expected, actual, "Did not expect: " + str(actual))
def test_block_sizes(self):
expected = "8,2,1,0"
self.storperf.block_sizes = expected
actual = self.storperf.block_sizes
self.assertEqual(
expected, actual, "Did not expect: " + str(actual))
def test_volume_size(self):
expected = 20
self.storperf.volume_size = expected
actual = self.storperf.volume_size
self.assertEqual(
expected, actual, "Did not expect: " + str(actual))
def test_agent_network(self):
expected = "ABCDEF"
self.storperf.public_network = expected
actual = self.storperf.public_network
self.assertEqual(
expected, actual, "Did not expect: " + str(actual))
def test_agent_flavor(self):
expected = "m1.small"
self.storperf.agent_flavor = expected
actual = self.storperf.agent_flavor
self.assertEqual(
expected, actual, "Did not expect: " + str(actual))
|
