comparison src/test/test_memory_memory.py @ 833:9bac2fc74968

memory: bug fix to not register twice frontends parameters + added some tests for param update
author souliane <souliane@mailoo.org>
date Fri, 07 Feb 2014 17:10:24 +0100
parents 1fe00f0c9a91
children
comparison
equal deleted inserted replaced
832:c4b22aedb7d7 833:9bac2fc74968
29 class MemoryTest(unittest.TestCase): 29 class MemoryTest(unittest.TestCase):
30 30
31 def setUp(self): 31 def setUp(self):
32 self.host = helpers.FakeSAT() 32 self.host = helpers.FakeSAT()
33 33
34 def _getParamXML(self, name=None, category=None, security_level=None): 34 def _getParamXML(self, param="1", security_level=None):
35 if not name: 35 def getParam(name):
36 name = Const.ENABLE_UNIBOX_PARAM 36 return """
37 if not category: 37 <param name="%(param_name)s" label="%(param_label)s" value="true" type="bool" %(security)s/>
38 category = Const.ENABLE_UNIBOX_KEY 38 """ % {'param_name': name,
39 'param_label': _(name),
40 'security': '' if security_level is None else ('security="%d"' % security_level)
41 }
42 if param == "1":
43 params = getParam(Const.ENABLE_UNIBOX_PARAM)
44 elif param == "2":
45 params = getParam(Const.PARAM_IN_QUOTES)
46 else:
47 params = getParam(Const.ENABLE_UNIBOX_PARAM) + getParam(Const.PARAM_IN_QUOTES)
39 return """ 48 return """
40 <params> 49 <params>
41 <individual> 50 <individual>
42 <category name="%(category_name)s" label="%(category_label)s"> 51 <category name="%(category_name)s" label="%(category_label)s">
43 <param name="%(param_name)s" label="%(param_label)s" value="true" type="bool" %(security)s/> 52 %(params)s
44 </category> 53 </category>
45 </individual> 54 </individual>
46 </params> 55 </params>
47 """ % { 56 """ % {
48 'category_name': category, 57 'category_name': Const.COMPOSITION_KEY,
49 'category_label': _(category), 58 'category_label': _(Const.COMPOSITION_KEY),
50 'param_name': name, 59 'params': params
51 'param_label': _(name),
52 'security': '' if security_level is None else ('security="%d"' % security_level)
53 } 60 }
54 61
55 def _paramExists(self, name=None, category=None, src=None): 62 def _paramExists(self, param="1", src=None):
56 """@return: True is the param (category, name) exists""" 63 """@return: True is the param (category, name) exists"""
57 if not name: 64 if param == "1":
58 name = Const.ENABLE_UNIBOX_PARAM 65 name = Const.ENABLE_UNIBOX_PARAM
59 if not category: 66 else:
60 category = Const.ENABLE_UNIBOX_KEY 67 name = Const.PARAM_IN_QUOTES
68 category = Const.COMPOSITION_KEY
61 if src is None: 69 if src is None:
62 src = self.host.memory.params.dom.documentElement 70 src = self.host.memory.params.dom.documentElement
63 for type_node in src.childNodes: 71 for type_node in src.childNodes:
64 # when src comes self.host.memory.params.dom, we have here 72 # when src comes self.host.memory.params.dom, we have here
65 # some "individual" or "general" elements, when it comes 73 # some "individual" or "general" elements, when it comes
72 for param in cat_node.childNodes: 80 for param in cat_node.childNodes:
73 if param.nodeName == "param" and param.getAttribute("name") == name: 81 if param.nodeName == "param" and param.getAttribute("name") == name:
74 return True 82 return True
75 return False 83 return False
76 84
77 def assertParam_generic(self, src=None, exists=True, deferred=False): 85 def assertParam_generic(self, param="1", src=None, exists=True, deferred=False):
78 msg = "Expected parameter not found!\n" if exists else "Unexpected parameter found!\n" 86 msg = "Expected parameter not found!\n" if exists else "Unexpected parameter found!\n"
79 if deferred == False: 87 if deferred == False:
80 # in this stack we can see the line where the error came from, 88 # in this stack we can see the line where the error came from,
81 # if limit=5, 6 is not enough you can increase the value 89 # if limit=5, 6 is not enough you can increase the value
82 msg += "\n".join(traceback.format_stack(limit=5 if exists else 6)) 90 msg += "\n".join(traceback.format_stack(limit=5 if exists else 6))
83 assertion = self._paramExists(src=src) 91 assertion = self._paramExists(param, src)
84 getattr(self, "assert%s" % exists)(assertion, msg) 92 getattr(self, "assert%s" % exists)(assertion, msg)
85 93
86 def assertParamExists(self, src=None): 94 def assertParamExists(self, param="1", src=None):
87 self.assertParam_generic(src, True) 95 self.assertParam_generic(param, src, True)
88 96
89 def assertParamNotExists(self, src=None): 97 def assertParamNotExists(self, param="1", src=None):
90 self.assertParam_generic(src, False) 98 self.assertParam_generic(param, src, False)
91 99
92 def assertParamExists_async(self, src): 100 def assertParamExists_async(self, src, param="1"):
93 """@param src: a deferred result from Memory.getParams""" 101 """@param src: a deferred result from Memory.getParams"""
94 self.assertParam_generic(minidom.parseString(src.encode("utf-8")), True, True) 102 self.assertParam_generic(param, minidom.parseString(src.encode("utf-8")), True, True)
95 103
96 def assertParamNotExists_async(self, src): 104 def assertParamNotExists_async(self, src, param="1"):
97 """@param src: a deferred result from Memory.getParams""" 105 """@param src: a deferred result from Memory.getParams"""
98 self.assertParam_generic(minidom.parseString(src.encode("utf-8")), False, True) 106 self.assertParam_generic(param, minidom.parseString(src.encode("utf-8")), False, True)
99 107
100 def _getParams(self, security_limit, app='', profile_key='@NONE@'): 108 def _getParams(self, security_limit, app='', profile_key='@NONE@'):
101 if profile_key == '@NONE@': 109 if profile_key == '@NONE@':
102 profile_key = '@DEFAULT@' 110 profile_key = '@DEFAULT@'
103 return self.host.memory.getParams(security_limit, app, profile_key) 111 return self.host.memory.getParams(security_limit, app, profile_key)
108 self.host.memory.updateParams(self._getParamXML()) 116 self.host.memory.updateParams(self._getParamXML())
109 self.assertParamExists() 117 self.assertParamExists()
110 previous = self.host.memory.params.dom.cloneNode(True) 118 previous = self.host.memory.params.dom.cloneNode(True)
111 # now check if it is really updated and not duplicated 119 # now check if it is really updated and not duplicated
112 self.host.memory.updateParams(self._getParamXML()) 120 self.host.memory.updateParams(self._getParamXML())
121 self.assertEqual(previous.toxml().encode("utf-8"), self.host.memory.params.dom.toxml().encode("utf-8"))
122
123 self.host.memory.init()
124 # check successive updates (without intersection)
125 self.host.memory.updateParams(self._getParamXML('1'))
126 self.assertParamExists("1")
127 self.assertParamNotExists("2")
128 self.host.memory.updateParams(self._getParamXML('2'))
129 self.assertParamExists("1")
130 self.assertParamExists("2")
131
132 previous = self.host.memory.params.dom.cloneNode(True) # save for later
133
134 self.host.memory.init()
135 # check successive updates (with intersection)
136 self.host.memory.updateParams(self._getParamXML('1'))
137 self.assertParamExists("1")
138 self.assertParamNotExists("2")
139 self.host.memory.updateParams(self._getParamXML('both'))
140 self.assertParamExists("1")
141 self.assertParamExists("2")
142
143 # successive updates with or without intersection should have the same result
113 self.assertEqual(previous.toxml().encode("utf-8"), self.host.memory.params.dom.toxml().encode("utf-8")) 144 self.assertEqual(previous.toxml().encode("utf-8"), self.host.memory.params.dom.toxml().encode("utf-8"))
114 145
115 def test_getParams(self): 146 def test_getParams(self):
116 # tests with no security level on the parameter (most secure) 147 # tests with no security level on the parameter (most secure)
117 params = self._getParamXML() 148 params = self._getParamXML()
153 register(params, 0, Const.APP_NAME) 184 register(params, 0, Const.APP_NAME)
154 self.assertParamNotExists() 185 self.assertParamNotExists()
155 self.host.memory.init() 186 self.host.memory.init()
156 register(params, 1, Const.APP_NAME) 187 register(params, 1, Const.APP_NAME)
157 self.assertParamNotExists() 188 self.assertParamNotExists()
189
158 # tests with security level 0 on the parameter (not secure) 190 # tests with security level 0 on the parameter (not secure)
159 params = self._getParamXML(security_level=0) 191 params = self._getParamXML(security_level=0)
160 self.host.memory.init() 192 self.host.memory.init()
161 register(params, Const.NO_SECURITY_LIMIT, Const.APP_NAME) 193 register(params, Const.NO_SECURITY_LIMIT, Const.APP_NAME)
162 self.assertParamExists() 194 self.assertParamExists()
164 register(params, 0, Const.APP_NAME) 196 register(params, 0, Const.APP_NAME)
165 self.assertParamExists() 197 self.assertParamExists()
166 self.host.memory.init() 198 self.host.memory.init()
167 register(params, 1, Const.APP_NAME) 199 register(params, 1, Const.APP_NAME)
168 self.assertParamExists() 200 self.assertParamExists()
201
169 # tests with security level 1 on the parameter (more secure) 202 # tests with security level 1 on the parameter (more secure)
170 params = self._getParamXML(security_level=1) 203 params = self._getParamXML(security_level=1)
171 self.host.memory.init() 204 self.host.memory.init()
172 register(params, Const.NO_SECURITY_LIMIT, Const.APP_NAME) 205 register(params, Const.NO_SECURITY_LIMIT, Const.APP_NAME)
173 self.assertParamExists() 206 self.assertParamExists()