Coverage for manila/tests/scheduler/filters/test_ignore_attempted_hosts.py: 100%

1# Copyright 2011 OpenStack Foundation. 

2# All Rights Reserved. 

3# 

4# Licensed under the Apache License, Version 2.0 (the "License"); you may 

5# not use this file except in compliance with the License. You may obtain 

6# a copy of the License at 

7# 

8# http://www.apache.org/licenses/LICENSE-2.0 

9# 

10# Unless required by applicable law or agreed to in writing, software 

11# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT 

12# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the 

13# License for the specific language governing permissions and limitations 

14# under the License. 

15 

16""" 

17Tests For IgnoreAttemptedHost filter. 

18""" 

19 

20 

21from manila.scheduler.filters import ignore_attempted_hosts 

22from manila import test 

23from manila.tests.scheduler import fakes 

24 

25 

26class HostFiltersTestCase(test.TestCase): 

27 """Test case for IgnoreAttemptedHost filter.""" 

28 

29 def setUp(self): 

30 super(HostFiltersTestCase, self).setUp() 

31 self.filter = ignore_attempted_hosts.IgnoreAttemptedHostsFilter() 

32 

33 def test_ignore_attempted_hosts_filter_disabled(self): 

34 # Test case where re-scheduling is disabled. 

35 host = fakes.FakeHostState('host1', {}) 

36 filter_properties = {} 

37 self.assertTrue(self.filter.host_passes(host, filter_properties)) 

38 

39 def test_ignore_attempted_hosts_filter_pass(self): 

40 # Node not previously tried. 

41 host = fakes.FakeHostState('host1', {}) 

42 attempted = dict(num_attempts=2, hosts=['host2']) 

43 filter_properties = dict(retry=attempted) 

44 self.assertTrue(self.filter.host_passes(host, filter_properties)) 

45 

46 def test_ignore_attempted_hosts_filter_fail(self): 

47 # Node was already tried. 

48 host = fakes.FakeHostState('host1', {}) 

49 attempted = dict(num_attempts=2, hosts=['host1']) 

50 filter_properties = dict(retry=attempted) 

51 self.assertFalse(self.filter.host_passes(host, filter_properties))