You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

277 lines
13 KiB

  1. # Copyright 2016-2017 Therp BV
  2. # License AGPL-3.0 or later (http://www.gnu.org/licenses/agpl.html).
  3. from openerp.exceptions import ValidationError
  4. from .test_partner_relation_common import TestPartnerRelationCommon
  5. class TestPartnerRelation(TestPartnerRelationCommon):
  6. def setUp(self):
  7. super(TestPartnerRelation, self).setUp()
  8. # Create a new relation type which will not have valid relations:
  9. category_nobody = self.category_model.create({
  10. 'name': 'Nobody'})
  11. (self.type_nobody,
  12. self.selection_nobody,
  13. self.selection_nobody_inverse) = (
  14. self._create_relation_type_selection({
  15. 'name': 'has relation with nobody',
  16. 'name_inverse': 'nobody has relation with',
  17. 'contact_type_left': 'c',
  18. 'contact_type_right': 'p',
  19. 'partner_category_left': category_nobody.id,
  20. 'partner_category_right': category_nobody.id}))
  21. def _get_empty_relation(self):
  22. """Get empty relation record for onchange tests."""
  23. # Need English, because we will compare text
  24. return self.relation_all_model.with_context(lang='en_US').new({})
  25. def test_get_partner_types(self):
  26. """Partner types should contain at least 'c' and 'p'."""
  27. partner_types = self.selection_model.get_partner_types()
  28. type_codes = [ptype[0] for ptype in partner_types]
  29. self.assertTrue('c' in type_codes)
  30. self.assertTrue('p' in type_codes)
  31. def test_create_with_active_id(self):
  32. """Test creation with this_partner_id from active_id."""
  33. # Check wether we can create connection from company to person,
  34. # taking the particular company from the active records:
  35. relation = self.relation_all_model.with_context(
  36. active_id=self.partner_02_company.id,
  37. active_ids=self.partner_02_company.ids).create({
  38. 'other_partner_id': self.partner_01_person.id,
  39. 'type_selection_id': self.selection_company2person.id})
  40. self.assertTrue(relation)
  41. self.assertEqual(relation.this_partner_id, self.partner_02_company)
  42. # Partner should have one relation now:
  43. self.assertEqual(self.partner_01_person.relation_count, 1)
  44. # Test create without type_selection_id:
  45. with self.assertRaises(ValidationError):
  46. self.relation_all_model.create({
  47. 'this_partner_id': self.partner_02_company.id,
  48. 'other_partner_id': self.partner_01_person.id})
  49. def test_display_name(self):
  50. """Test display name"""
  51. relation = self._create_company2person_relation()
  52. self.assertEqual(
  53. relation.display_name, '%s %s %s' % (
  54. relation.this_partner_id.name,
  55. relation.type_selection_id.name,
  56. relation.other_partner_id.name))
  57. def test__regular_write(self):
  58. """Test write with valid data."""
  59. relation = self._create_company2person_relation()
  60. relation.write({'date_start': '2014-09-01'})
  61. relation.invalidate_cache(ids=relation.ids)
  62. self.assertEqual(relation.date_start, '2014-09-01')
  63. def test_write_incompatible_dates(self):
  64. """Test write with date_end before date_start."""
  65. relation = self._create_company2person_relation()
  66. with self.assertRaises(ValidationError):
  67. relation.write({
  68. 'date_start': '2016-09-01',
  69. 'date_end': '2016-08-01'})
  70. def test_validate_overlapping_01(self):
  71. """Test create overlapping with no start / end dates."""
  72. relation = self._create_company2person_relation()
  73. with self.assertRaises(ValidationError):
  74. # New relation with no start / end should give error
  75. self.relation_all_model.create({
  76. 'this_partner_id': relation.this_partner_id.id,
  77. 'type_selection_id': relation.type_selection_id.id,
  78. 'other_partner_id': relation.other_partner_id.id})
  79. def test_validate_overlapping_02(self):
  80. """Test create overlapping with start / end dates."""
  81. relation = self.relation_all_model.create({
  82. 'this_partner_id': self.partner_02_company.id,
  83. 'type_selection_id': self.selection_company2person.id,
  84. 'other_partner_id': self.partner_01_person.id,
  85. 'date_start': '2015-09-01',
  86. 'date_end': '2016-08-31'})
  87. # New relation with overlapping start / end should give error
  88. with self.assertRaises(ValidationError):
  89. self.relation_all_model.create({
  90. 'this_partner_id': relation.this_partner_id.id,
  91. 'type_selection_id': relation.type_selection_id.id,
  92. 'other_partner_id': relation.other_partner_id.id,
  93. 'date_start': '2016-08-01',
  94. 'date_end': '2017-07-30'})
  95. def test_validate_overlapping_03(self):
  96. """Test create not overlapping."""
  97. relation = self.relation_all_model.create({
  98. 'this_partner_id': self.partner_02_company.id,
  99. 'type_selection_id': self.selection_company2person.id,
  100. 'other_partner_id': self.partner_01_person.id,
  101. 'date_start': '2015-09-01',
  102. 'date_end': '2016-08-31'})
  103. relation_another_record = self.relation_all_model.create({
  104. 'this_partner_id': relation.this_partner_id.id,
  105. 'type_selection_id': relation.type_selection_id.id,
  106. 'other_partner_id': relation.other_partner_id.id,
  107. 'date_start': '2016-09-01',
  108. 'date_end': '2017-08-31'})
  109. self.assertTrue(relation_another_record)
  110. def test_inverse_record(self):
  111. """Test creation of inverse record."""
  112. relation = self._create_company2person_relation()
  113. inverse_relation = self.relation_all_model.search([
  114. ('this_partner_id', '=', relation.other_partner_id.id),
  115. ('other_partner_id', '=', relation.this_partner_id.id)])
  116. self.assertEqual(len(inverse_relation), 1)
  117. self.assertEqual(
  118. inverse_relation.type_selection_id.name,
  119. self.selection_person2company.name)
  120. def test_inverse_creation(self):
  121. """Test creation of record through inverse selection."""
  122. relation = self.relation_all_model.create({
  123. 'this_partner_id': self.partner_01_person.id,
  124. 'type_selection_id': self.selection_person2company.id,
  125. 'other_partner_id': self.partner_02_company.id})
  126. # Check wether display name is what we should expect:
  127. self.assertEqual(
  128. relation.display_name, '%s %s %s' % (
  129. self.partner_01_person.name,
  130. self.selection_person2company.name,
  131. self.partner_02_company.name))
  132. def test_inverse_creation_type_id(self):
  133. """Test creation of record through inverse selection with type_id."""
  134. relation = self.relation_all_model.create({
  135. 'this_partner_id': self.partner_01_person.id,
  136. 'type_id': self.selection_person2company.type_id.id,
  137. 'is_inverse': True,
  138. 'other_partner_id': self.partner_02_company.id})
  139. # Check wether display name is what we should expect:
  140. self.assertEqual(
  141. relation.display_name, '%s %s %s' % (
  142. self.partner_01_person.name,
  143. self.selection_person2company.name,
  144. self.partner_02_company.name))
  145. def test_unlink(self):
  146. """Unlinking derived relation should unlink base relation."""
  147. # Check wether underlying record is removed when record is removed:
  148. relation = self._create_company2person_relation()
  149. base_model = self.env[relation.res_model]
  150. base_relation = base_model.browse([relation.res_id])
  151. relation.unlink()
  152. self.assertFalse(base_relation.exists())
  153. def test_on_change_type_selection(self):
  154. """Test on_change_type_selection."""
  155. # 1. Test call with empty relation
  156. relation_empty = self._get_empty_relation()
  157. result = relation_empty.onchange_type_selection_id()
  158. self.assertTrue('domain' in result)
  159. self.assertFalse('warning' in result)
  160. self.assertTrue('this_partner_id' in result['domain'])
  161. self.assertFalse(result['domain']['this_partner_id'])
  162. self.assertTrue('other_partner_id' in result['domain'])
  163. self.assertFalse(result['domain']['other_partner_id'])
  164. # 2. Test call with company 2 person relation
  165. relation = self._create_company2person_relation()
  166. domain = relation.onchange_type_selection_id()['domain']
  167. self.assertTrue(
  168. ('is_company', '=', False) in domain['other_partner_id'])
  169. # 3. Test with relation needing categories,
  170. # take active partner from active_id:
  171. relation_ngo_volunteer = self.relation_all_model.with_context(
  172. active_id=self.partner_03_ngo.id).create({
  173. 'type_selection_id': self.selection_ngo2volunteer.id,
  174. 'other_partner_id': self.partner_04_volunteer.id})
  175. domain = relation_ngo_volunteer.onchange_type_selection_id()['domain']
  176. self.assertTrue(
  177. ('category_id', 'in', [self.category_01_ngo.id]) in
  178. domain['this_partner_id'])
  179. self.assertTrue(
  180. ('category_id', 'in', [self.category_02_volunteer.id]) in
  181. domain['other_partner_id'])
  182. # 4. Test with invalid or impossible combinations
  183. relation_nobody = self._get_empty_relation()
  184. with self.env.do_in_draft():
  185. relation_nobody.type_selection_id = self.selection_nobody
  186. warning = relation_nobody.onchange_type_selection_id()['warning']
  187. self.assertTrue('message' in warning)
  188. self.assertTrue('No this partner available' in warning['message'])
  189. with self.env.do_in_draft():
  190. relation_nobody.this_partner_id = self.partner_02_company
  191. warning = relation_nobody.onchange_type_selection_id()['warning']
  192. self.assertTrue('message' in warning)
  193. self.assertTrue('incompatible' in warning['message'])
  194. # Allow left partner and check message for other partner:
  195. self.type_nobody.write({'partner_category_left': False})
  196. self.selection_nobody.invalidate_cache(ids=self.selection_nobody.ids)
  197. warning = relation_nobody.onchange_type_selection_id()['warning']
  198. self.assertTrue('message' in warning)
  199. self.assertTrue('No other partner available' in warning['message'])
  200. def test_on_change_partner_id(self):
  201. """Test on_change_partner_id."""
  202. # 1. Test call with empty relation
  203. relation_empty = self._get_empty_relation()
  204. result = relation_empty.onchange_partner_id()
  205. self.assertTrue('domain' in result)
  206. self.assertFalse('warning' in result)
  207. self.assertTrue('type_selection_id' in result['domain'])
  208. self.assertFalse(result['domain']['type_selection_id'])
  209. # 2. Test call with company 2 person relation
  210. relation = self._create_company2person_relation()
  211. domain = relation.onchange_partner_id()['domain']
  212. self.assertTrue(
  213. ('contact_type_this', '=', 'c') in domain['type_selection_id'])
  214. # 3. Test with invalid or impossible combinations
  215. relation_nobody = self._get_empty_relation()
  216. with self.env.do_in_draft():
  217. relation_nobody.this_partner_id = self.partner_02_company
  218. relation_nobody.type_selection_id = self.selection_nobody
  219. warning = relation_nobody.onchange_partner_id()['warning']
  220. self.assertTrue('message' in warning)
  221. self.assertTrue('incompatible' in warning['message'])
  222. def test_write(self):
  223. """Test write. Special attention for changing type."""
  224. relation_company2person = self._create_company2person_relation()
  225. company_partner = relation_company2person.this_partner_id
  226. # First get another worker:
  227. partner_extra_person = self.partner_model.create({
  228. 'name': 'A new worker',
  229. 'is_company': False,
  230. 'ref': 'NW01'})
  231. relation_company2person.write({
  232. 'other_partner_id': partner_extra_person.id})
  233. self.assertEqual(
  234. relation_company2person.other_partner_id.name,
  235. partner_extra_person.name)
  236. # We will also change to a type going from person to company:
  237. (type_worker2company,
  238. selection_worker2company,
  239. selection_company2worker) = self._create_relation_type_selection({
  240. 'name': 'works for',
  241. 'name_inverse': 'has worker',
  242. 'contact_type_left': 'p',
  243. 'contact_type_right': 'c'})
  244. relation_company2person.write({
  245. 'this_partner_id': partner_extra_person.id,
  246. 'type_selection_id': selection_worker2company.id,
  247. 'other_partner_id': company_partner.id})
  248. self.assertEqual(
  249. relation_company2person.this_partner_id.id,
  250. partner_extra_person.id)
  251. self.assertEqual(
  252. relation_company2person.type_selection_id.id,
  253. selection_worker2company.id)
  254. self.assertEqual(
  255. relation_company2person.other_partner_id.id,
  256. company_partner.id)