@@ -39,6 +39,7 @@ class TestAmphoraInfo(base.TestCase):
39
39
FAKE_LISTENER_ID_3 = uuidutils .generate_uuid ()
40
40
FAKE_LISTENER_ID_4 = uuidutils .generate_uuid ()
41
41
LB_ID_1 = uuidutils .generate_uuid ()
42
+ LB_ID_2 = uuidutils .generate_uuid ()
42
43
43
44
def setUp (self ):
44
45
super ().setUp ()
@@ -113,6 +114,8 @@ def test_compile_amphora_info_for_udp(self, mock_gethostname,
113
114
@mock .patch ('octavia.amphorae.backends.agent.api_server.util.'
114
115
'get_listeners' , return_value = [FAKE_LISTENER_ID_1 ,
115
116
FAKE_LISTENER_ID_2 ])
117
+ @mock .patch ('octavia.amphorae.backends.agent.api_server.util.'
118
+ 'get_loadbalancers' , return_value = [LB_ID_1 , LB_ID_2 ])
116
119
@mock .patch ('octavia.amphorae.backends.agent.api_server.'
117
120
'amphora_info.AmphoraInfo._get_meminfo' )
118
121
@mock .patch ('octavia.amphorae.backends.agent.api_server.'
@@ -129,7 +132,8 @@ def test_compile_amphora_info_for_udp(self, mock_gethostname,
129
132
@mock .patch ('socket.gethostname' , return_value = 'FAKE_HOST' )
130
133
def test_compile_amphora_details (self , mhostname , m_count , m_pkg_version ,
131
134
m_load , m_get_nets , m_os , m_cpu ,
132
- mget_mem , mget_listener ):
135
+ mget_mem , mget_loadbalancers ,
136
+ mget_listeners ):
133
137
mget_mem .return_value = {'SwapCached' : 0 , 'Buffers' : 344792 ,
134
138
'MemTotal' : 21692784 , 'Cached' : 4271856 ,
135
139
'Slab' : 534384 , 'MemFree' : 12685624 ,
@@ -183,6 +187,7 @@ def test_compile_amphora_details(self, mhostname, m_count, m_pkg_version,
183
187
u'topology_status' : u'OK' }
184
188
actual = self .amp_info .compile_amphora_details ()
185
189
self .assertEqual (expected_dict , actual .json )
190
+ m_count .assert_called_once_with (sorted (mget_loadbalancers ()))
186
191
api_server .VERSION = original_version
187
192
188
193
@mock .patch ('octavia.amphorae.backends.agent.api_server.util.'
0 commit comments