← Back to branch summary

~opnfv-team/charms/trusty/odl-controller/Be

« back to all changes in this revision

Viewing changes to unit_tests/test_utils.py

  • Committer: James Page
  • Date: 2015-11-12 11:25:54 UTC
  • mfrom: (10.1.2 odl-controller)
  • Revision ID: james.page@ubuntu.com-20151112112554-xbh2zetd8zzldl0z
Add unit tests, functional tests, Makefiles and tox configuration

Show diffs side-by-side

added added

removed removed

Lines of Context:
unit_tests/test_utils.py
 
1
import logging
 
2
import unittest
 
3
import os
 
4
import yaml
 
5
 
 
6
from contextlib import contextmanager
 
7
from mock import patch, MagicMock
 
8
 
 
9
patch('charmhelpers.contrib.openstack.utils.set_os_workload_status').start()
 
10
patch('charmhelpers.core.hookenv.status_set').start()
 
11
 
 
12
 
 
13
def load_config():
 
14
    '''
 
15
    Walk backwords from __file__ looking for config.yaml, load and return the
 
16
    'options' section'
 
17
    '''
 
18
    config = None
 
19
    f = __file__
 
20
    while config is None:
 
21
        d = os.path.dirname(f)
 
22
        if os.path.isfile(os.path.join(d, 'config.yaml')):
 
23
            config = os.path.join(d, 'config.yaml')
 
24
            break
 
25
        f = d
 
26
 
 
27
    if not config:
 
28
        logging.error('Could not find config.yaml in any parent directory '
 
29
                      'of %s. ' % file)
 
30
        raise Exception
 
31
 
 
32
    return yaml.safe_load(open(config).read())['options']
 
33
 
 
34
 
 
35
def get_default_config():
 
36
    '''
 
37
    Load default charm config from config.yaml return as a dict.
 
38
    If no default is set in config.yaml, its value is None.
 
39
    '''
 
40
    default_config = {}
 
41
    config = load_config()
 
42
    for k, v in config.iteritems():
 
43
        if 'default' in v:
 
44
            default_config[k] = v['default']
 
45
        else:
 
46
            default_config[k] = None
 
47
    return default_config
 
48
 
 
49
 
 
50
class CharmTestCase(unittest.TestCase):
 
51
 
 
52
    def setUp(self, obj, patches):
 
53
        super(CharmTestCase, self).setUp()
 
54
        self.patches = patches
 
55
        self.obj = obj
 
56
        self.test_config = TestConfig()
 
57
        self.test_relation = TestRelation()
 
58
        self.patch_all()
 
59
 
 
60
    def patch(self, method):
 
61
        _m = patch.object(self.obj, method)
 
62
        mock = _m.start()
 
63
        self.addCleanup(_m.stop)
 
64
        return mock
 
65
 
 
66
    def patch_all(self):
 
67
        for method in self.patches:
 
68
            setattr(self, method, self.patch(method))
 
69
 
 
70
 
 
71
class TestConfig(object):
 
72
 
 
73
    def __init__(self):
 
74
        self.config = get_default_config()
 
75
 
 
76
    def get(self, attr=None):
 
77
        if not attr:
 
78
            return self.get_all()
 
79
        try:
 
80
            return self.config[attr]
 
81
        except KeyError:
 
82
            return None
 
83
 
 
84
    def get_all(self):
 
85
        return self.config
 
86
 
 
87
    def set(self, attr, value):
 
88
            if attr not in self.config:
 
89
                raise KeyError
 
90
            self.config[attr] = value
 
91
 
 
92
 
 
93
class TestRelation(object):
 
94
 
 
95
    def __init__(self, relation_data={}):
 
96
        self.relation_data = relation_data
 
97
 
 
98
    def set(self, relation_data):
 
99
        self.relation_data = relation_data
 
100
 
 
101
    def get(self, attribute=None, unit=None, rid=None):
 
102
        if attribute is None:
 
103
            return self.relation_data
 
104
        elif attribute in self.relation_data:
 
105
            return self.relation_data[attribute]
 
106
        return None
 
107
 
 
108
 
 
109
@contextmanager
 
110
def patch_open():
 
111
    '''Patch open() to allow mocking both open() itself and the file that is
 
112
    yielded.
 
113
 
 
114
    Yields the mock for "open" and "file", respectively.'''
 
115
    mock_open = MagicMock(spec=open)
 
116
    mock_file = MagicMock(spec=file)
 
117
 
 
118
    @contextmanager
 
119
    def stub_open(*args, **kwargs):
 
120
        mock_open(*args, **kwargs)
 
121
        yield mock_file
 
122
 
 
123
    with patch('__builtin__.open', stub_open):
 
124
        yield mock_open, mock_file