FilesExpand file tree

 master

/

utils.py

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

98 lines (79 loc) · 3.4 KB

 master

/

utils.py

Copy path

Top

File metadata and controls

• Code
• Blame

98 lines (79 loc) · 3.4 KB

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

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

# Copyright 2012-2013 OpenStack Foundation

# Copyright 2013 Nebula Inc.

#

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

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

# a copy of the License at

#

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

#

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

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

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

# License for the specific language governing permissions and limitations

# under the License.

#

import os

import fixtures

import sys

import testtools

from openstackclient.tests import fakes

class ParserException(Exception):

pass

class TestCase(testtools.TestCase):

def setUp(self):

testtools.TestCase.setUp(self)

if (os.environ.get("OS_STDOUT_CAPTURE") == "True" or

os.environ.get("OS_STDOUT_CAPTURE") == "1"):

stdout = self.useFixture(fixtures.StringStream("stdout")).stream

self.useFixture(fixtures.MonkeyPatch("sys.stdout", stdout))

if (os.environ.get("OS_STDERR_CAPTURE") == "True" or

os.environ.get("OS_STDERR_CAPTURE") == "1"):

stderr = self.useFixture(fixtures.StringStream("stderr")).stream

self.useFixture(fixtures.MonkeyPatch("sys.stderr", stderr))

def assertNotCalled(self, m, msg=None):

"""Assert a function was not called"""

if m.called:

if not msg:

msg = 'method %s should not have been called' % m

self.fail(msg)

# 2.6 doesn't have the assert dict equals so make sure that it exists

if tuple(sys.version_info)[0:2] < (2, 7):

def assertIsInstance(self, obj, cls, msg=None):

"""self.assertTrue(isinstance(obj, cls)), with a nicer message"""

if not isinstance(obj, cls):

standardMsg = '%s is not an instance of %r' % (obj, cls)

self.fail(self._formatMessage(msg, standardMsg))

def assertDictEqual(self, d1, d2, msg=None):

# Simple version taken from 2.7

self.assertIsInstance(d1, dict,

'First argument is not a dictionary')

self.assertIsInstance(d2, dict,

'Second argument is not a dictionary')

if d1 != d2:

if msg:

self.fail(msg)

else:

standardMsg = '%r != %r' % (d1, d2)

self.fail(standardMsg)

class TestCommand(TestCase):

"""Test cliff command classes"""

def setUp(self):

super(TestCommand, self).setUp()

# Build up a fake app

self.fake_stdout = fakes.FakeStdout()

self.fake_log = fakes.FakeLog()

self.app = fakes.FakeApp(self.fake_stdout, self.fake_log)

self.app.client_manager = fakes.FakeClientManager()

def check_parser(self, cmd, args, verify_args):

cmd_parser = cmd.get_parser('check_parser')

try:

parsed_args = cmd_parser.parse_args(args)

except SystemExit:

raise ParserException("Argument parse failed")

for av in verify_args:

attr, value = av

if attr:

self.assertIn(attr, parsed_args)

self.assertEqual(value, getattr(parsed_args, attr))

return parsed_args