def write(self, vals): """Extend to update related company with company registry. Also formats company_registry value. """ if vals.get(REGISTRY_KEY): vals[REGISTRY_KEY] = strip_space(vals[REGISTRY_KEY]) res = super().write(vals) # Note this is only valid for write, because on create, we know # there won't be any related company (this is different than # `company_id` relation). if REGISTRY_KEY in vals: # Value to set for related company. self._set_company_registry_on_company(vals[REGISTRY_KEY]) return res
def create(self, vals_list): """Extend to format company_registry without any spaces.""" for vals in vals_list: if vals.get(REGISTRY_KEY): vals[REGISTRY_KEY] = strip_space(vals[REGISTRY_KEY]) return super().create(vals_list)
def test_54_strip_space(self): """Strip string with white space only.""" s = formatting.strip_space('\tabc ') self.assertEqual(s, 'abc')
def test_55_strip_space(self): """Strip string with white space and space around chars.""" s = formatting.strip_space('\tab c \nd\r') self.assertEqual(s, 'abcd')
def test_53_strip_space(self): """Strip string without spaces.""" s = formatting.strip_space('abc') self.assertEqual(s, 'abc')