Fix WeightedHost
WeightedHosts don't have zone attributes. Let me know if I'm missing something and should be adding a hasattr check instead of removing the zone reference. Change-Id: If76f23fa46c965fc9004e35dd10da33e2bbb48dc
This commit is contained in:
@@ -63,8 +63,6 @@ class WeightedHost(object):
|
|||||||
x = dict(weight=self.weight)
|
x = dict(weight=self.weight)
|
||||||
if self.host_state:
|
if self.host_state:
|
||||||
x['host'] = self.host_state.host
|
x['host'] = self.host_state.host
|
||||||
if self.zone:
|
|
||||||
x['zone'] = self.zone
|
|
||||||
return x
|
return x
|
||||||
|
|
||||||
|
|
||||||
|
|||||||
@@ -16,6 +16,7 @@
|
|||||||
Tests For Least Cost functions.
|
Tests For Least Cost functions.
|
||||||
"""
|
"""
|
||||||
from nova import context
|
from nova import context
|
||||||
|
from nova.scheduler import host_manager
|
||||||
from nova.scheduler import least_cost
|
from nova.scheduler import least_cost
|
||||||
from nova import test
|
from nova import test
|
||||||
from nova.tests.scheduler import fakes
|
from nova.tests.scheduler import fakes
|
||||||
@@ -89,3 +90,24 @@ class LeastCostTestCase(test.TestCase):
|
|||||||
options)
|
options)
|
||||||
self.assertEqual(weighted_host.weight, 10512)
|
self.assertEqual(weighted_host.weight, 10512)
|
||||||
self.assertEqual(weighted_host.host_state.host, 'host1')
|
self.assertEqual(weighted_host.host_state.host, 'host1')
|
||||||
|
|
||||||
|
|
||||||
|
class TestWeightedHost(test.TestCase):
|
||||||
|
|
||||||
|
def setUp(self):
|
||||||
|
pass
|
||||||
|
|
||||||
|
def tearDown(self):
|
||||||
|
pass
|
||||||
|
|
||||||
|
def test_dict_conversion_without_host_state(self):
|
||||||
|
host = least_cost.WeightedHost('someweight')
|
||||||
|
expected = {'weight': 'someweight'}
|
||||||
|
self.assertDictMatch(host.to_dict(), expected)
|
||||||
|
|
||||||
|
def test_dict_conversion_with_host_state(self):
|
||||||
|
host_state = host_manager.HostState('somehost', 'sometopic')
|
||||||
|
host = least_cost.WeightedHost('someweight', host_state)
|
||||||
|
expected = {'weight': 'someweight',
|
||||||
|
'host': 'somehost'}
|
||||||
|
self.assertDictMatch(host.to_dict(), expected)
|
||||||
|
|||||||
Reference in New Issue
Block a user