summaryrefslogtreecommitdiffstats
path: root/testsuite/Testsrc/Testlib/TestServer/TestPlugins/TestCfg/TestCfgInfoXML.py
blob: 88dd1f18fc5272e15ece844619c0e2b5e579dc6f (plain)
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
import os
import sys
import lxml.etree
from mock import Mock, MagicMock, patch
from Bcfg2.Server.Plugins.Cfg.CfgInfoXML import *
from Bcfg2.Server.Plugin import InfoXML, PluginExecutionError

# add all parent testsuite directories to sys.path to allow (most)
# relative imports in python 2.4
path = os.path.dirname(__file__)
while path != "/":
    if os.path.basename(path).lower().startswith("test"):
        sys.path.append(path)
    if os.path.basename(path) == "testsuite":
        break
    path = os.path.dirname(path)
from common import XI_NAMESPACE, XI, inPy3k, call, builtins, u, can_skip, \
    skip, skipIf, skipUnless, Bcfg2TestCase, DBModelTestCase, syncdb, \
    patchIf, datastore, re_type
from TestServer.TestPlugins.TestCfg.Test_init import TestCfgInfo


class TestCfgInfoXML(TestCfgInfo):
    test_obj = CfgInfoXML

    def test__init(self):
        TestCfgInfo.test__init(self)
        ci = self.get_obj()
        self.assertIsInstance(ci.infoxml, InfoXML)

    def test_bind_info_to_entry(self):
        entry = lxml.etree.Element("Path", name="/test.txt")
        metadata = Mock()
        ci = self.get_obj()
        ci.infoxml = Mock()
        ci._set_info = Mock()

        self.assertRaises(PluginExecutionError,
                          ci.bind_info_to_entry, entry, metadata)
        ci.infoxml.pnode.Match.assert_called_with(metadata, dict(),
                                                  entry=entry)
        self.assertFalse(ci._set_info.called)

        ci.infoxml.reset_mock()
        ci._set_info.reset_mock()
        mdata_value = Mock()
        def set_mdata(metadata, mdata, entry=None):
            mdata['Info'] = {None: mdata_value}

        ci.infoxml.pnode.Match.side_effect = set_mdata
        ci.bind_info_to_entry(entry, metadata)
        ci.infoxml.pnode.Match.assert_called_with(metadata,
                                                  dict(Info={None: mdata_value}),
                                                  entry=entry)
        ci._set_info.assert_called_with(entry, mdata_value)

    def test_handle_event(self):
        ci = self.get_obj()
        ci.infoxml = Mock()
        ci.handle_event(Mock)
        ci.infoxml.HandleEvent.assert_called_with()

    def test__set_info(self):
        @patch("Bcfg2.Server.Plugins.Cfg.CfgInfo._set_info")
        def inner(mock_set_info):
            ci = self.get_obj()
            entry = Mock()
            info = {"foo": "foo",
                    "__children__": ["one", "two"]}
            ci._set_info(entry, info)
            self.assertItemsEqual(entry.append.call_args_list,
                                  [call(c) for c in info['__children__']])

        inner()
        TestCfgInfo.test__set_info(self)