Coverage for manila/tests/cmd/test_scheduler.py: 100%

1# Copyright 2015 Mirantis Inc. 

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 

16import sys 

17 

18from manila.cmd import scheduler as manila_scheduler 

19from manila import test 

20from manila import version 

21 

22CONF = manila_scheduler.CONF 

23 

24 

25class ManilaCmdSchedulerTestCase(test.TestCase): 

26 

27 def test_main(self): 

28 sys.argv = ['manila-scheduler'] 

29 self.mock_object(manila_scheduler.log, 'setup') 

30 self.mock_object(manila_scheduler.log, 'register_options') 

31 self.mock_object(manila_scheduler.utils, 'monkey_patch') 

32 self.mock_object(manila_scheduler.service.Service, 'create') 

33 self.mock_object(manila_scheduler.service, 'serve') 

34 self.mock_object(manila_scheduler.service, 'wait') 

35 

36 manila_scheduler.main() 

37 

38 self.assertEqual('manila', CONF.project) 

39 self.assertEqual(version.version_string(), CONF.version) 

40 manila_scheduler.log.setup.assert_called_once_with(CONF, "manila") 

41 manila_scheduler.log.register_options.assert_called_once_with(CONF) 

42 manila_scheduler.utils.monkey_patch.assert_called_once_with() 

43 manila_scheduler.service.Service.create.assert_called_once_with( 

44 binary='manila-scheduler', coordination=True) 

45 manila_scheduler.service.wait.assert_called_once_with() 

46 manila_scheduler.service.serve.assert_called_once_with( 

47 manila_scheduler.service.Service.create.return_value)