2
0

tests.py 32 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814
  1. from __future__ import unicode_literals
  2. from copy import deepcopy
  3. import datetime
  4. import uuid
  5. from django.core.exceptions import FieldError
  6. from django.db import connection, transaction, DatabaseError
  7. from django.db.models import F, Value, TimeField, UUIDField
  8. from django.test import TestCase, skipIfDBFeature, skipUnlessDBFeature
  9. from django.test.utils import Approximate
  10. from django.utils import six
  11. from .models import Company, Employee, Number, Experiment, Time, UUID
  12. class BasicExpressionsTests(TestCase):
  13. @classmethod
  14. def setUpTestData(cls):
  15. Company.objects.create(
  16. name="Example Inc.", num_employees=2300, num_chairs=5,
  17. ceo=Employee.objects.create(firstname="Joe", lastname="Smith")
  18. )
  19. Company.objects.create(
  20. name="Foobar Ltd.", num_employees=3, num_chairs=4,
  21. ceo=Employee.objects.create(firstname="Frank", lastname="Meyer")
  22. )
  23. Company.objects.create(
  24. name="Test GmbH", num_employees=32, num_chairs=1,
  25. ceo=Employee.objects.create(firstname="Max", lastname="Mustermann")
  26. )
  27. def setUp(self):
  28. self.company_query = Company.objects.values(
  29. "name", "num_employees", "num_chairs"
  30. ).order_by(
  31. "name", "num_employees", "num_chairs"
  32. )
  33. def test_filter_inter_attribute(self):
  34. # We can filter on attribute relationships on same model obj, e.g.
  35. # find companies where the number of employees is greater
  36. # than the number of chairs.
  37. self.assertQuerysetEqual(
  38. self.company_query.filter(num_employees__gt=F("num_chairs")), [
  39. {
  40. "num_chairs": 5,
  41. "name": "Example Inc.",
  42. "num_employees": 2300,
  43. },
  44. {
  45. "num_chairs": 1,
  46. "name": "Test GmbH",
  47. "num_employees": 32
  48. },
  49. ],
  50. lambda o: o
  51. )
  52. def test_update(self):
  53. # We can set one field to have the value of another field
  54. # Make sure we have enough chairs
  55. self.company_query.update(num_chairs=F("num_employees"))
  56. self.assertQuerysetEqual(
  57. self.company_query, [
  58. {
  59. "num_chairs": 2300,
  60. "name": "Example Inc.",
  61. "num_employees": 2300
  62. },
  63. {
  64. "num_chairs": 3,
  65. "name": "Foobar Ltd.",
  66. "num_employees": 3
  67. },
  68. {
  69. "num_chairs": 32,
  70. "name": "Test GmbH",
  71. "num_employees": 32
  72. }
  73. ],
  74. lambda o: o
  75. )
  76. def test_arithmetic(self):
  77. # We can perform arithmetic operations in expressions
  78. # Make sure we have 2 spare chairs
  79. self.company_query.update(num_chairs=F("num_employees") + 2)
  80. self.assertQuerysetEqual(
  81. self.company_query, [
  82. {
  83. 'num_chairs': 2302,
  84. 'name': 'Example Inc.',
  85. 'num_employees': 2300
  86. },
  87. {
  88. 'num_chairs': 5,
  89. 'name': 'Foobar Ltd.',
  90. 'num_employees': 3
  91. },
  92. {
  93. 'num_chairs': 34,
  94. 'name': 'Test GmbH',
  95. 'num_employees': 32
  96. }
  97. ],
  98. lambda o: o,
  99. )
  100. def test_order_of_operations(self):
  101. # Law of order of operations is followed
  102. self. company_query.update(
  103. num_chairs=F('num_employees') + 2 * F('num_employees')
  104. )
  105. self.assertQuerysetEqual(
  106. self.company_query, [
  107. {
  108. 'num_chairs': 6900,
  109. 'name': 'Example Inc.',
  110. 'num_employees': 2300
  111. },
  112. {
  113. 'num_chairs': 9,
  114. 'name': 'Foobar Ltd.',
  115. 'num_employees': 3
  116. },
  117. {
  118. 'num_chairs': 96,
  119. 'name': 'Test GmbH',
  120. 'num_employees': 32
  121. }
  122. ],
  123. lambda o: o,
  124. )
  125. def test_parenthesis_priority(self):
  126. # Law of order of operations can be overridden by parentheses
  127. self.company_query.update(
  128. num_chairs=((F('num_employees') + 2) * F('num_employees'))
  129. )
  130. self.assertQuerysetEqual(
  131. self.company_query, [
  132. {
  133. 'num_chairs': 5294600,
  134. 'name': 'Example Inc.',
  135. 'num_employees': 2300
  136. },
  137. {
  138. 'num_chairs': 15,
  139. 'name': 'Foobar Ltd.',
  140. 'num_employees': 3
  141. },
  142. {
  143. 'num_chairs': 1088,
  144. 'name': 'Test GmbH',
  145. 'num_employees': 32
  146. }
  147. ],
  148. lambda o: o,
  149. )
  150. def test_update_with_fk(self):
  151. # ForeignKey can become updated with the value of another ForeignKey.
  152. self.assertEqual(
  153. Company.objects.update(point_of_contact=F('ceo')),
  154. 3
  155. )
  156. self.assertQuerysetEqual(
  157. Company.objects.all(), [
  158. "Joe Smith",
  159. "Frank Meyer",
  160. "Max Mustermann",
  161. ],
  162. lambda c: six.text_type(c.point_of_contact),
  163. ordered=False
  164. )
  165. def test_update_with_none(self):
  166. Number.objects.create(integer=1, float=1.0)
  167. Number.objects.create(integer=2)
  168. Number.objects.filter(float__isnull=False).update(float=Value(None))
  169. self.assertQuerysetEqual(
  170. Number.objects.all(), [
  171. None,
  172. None,
  173. ],
  174. lambda n: n.float,
  175. ordered=False
  176. )
  177. def test_filter_with_join(self):
  178. # F Expressions can also span joins
  179. Company.objects.update(point_of_contact=F('ceo'))
  180. c = Company.objects.all()[0]
  181. c.point_of_contact = Employee.objects.create(firstname="Guido", lastname="van Rossum")
  182. c.save()
  183. self.assertQuerysetEqual(
  184. Company.objects.filter(ceo__firstname=F("point_of_contact__firstname")), [
  185. "Foobar Ltd.",
  186. "Test GmbH",
  187. ],
  188. lambda c: c.name,
  189. ordered=False
  190. )
  191. Company.objects.exclude(
  192. ceo__firstname=F("point_of_contact__firstname")
  193. ).update(name="foo")
  194. self.assertEqual(
  195. Company.objects.exclude(
  196. ceo__firstname=F('point_of_contact__firstname')
  197. ).get().name,
  198. "foo",
  199. )
  200. with transaction.atomic():
  201. with self.assertRaises(FieldError):
  202. Company.objects.exclude(
  203. ceo__firstname=F('point_of_contact__firstname')
  204. ).update(name=F('point_of_contact__lastname'))
  205. def test_object_update(self):
  206. # F expressions can be used to update attributes on single objects
  207. test_gmbh = Company.objects.get(name="Test GmbH")
  208. self.assertEqual(test_gmbh.num_employees, 32)
  209. test_gmbh.num_employees = F("num_employees") + 4
  210. test_gmbh.save()
  211. test_gmbh = Company.objects.get(pk=test_gmbh.pk)
  212. self.assertEqual(test_gmbh.num_employees, 36)
  213. def test_object_update_fk(self):
  214. # F expressions cannot be used to update attributes which are foreign
  215. # keys, or attributes which involve joins.
  216. test_gmbh = Company.objects.get(name="Test GmbH")
  217. def test():
  218. test_gmbh.point_of_contact = F("ceo")
  219. self.assertRaises(ValueError, test)
  220. test_gmbh.point_of_contact = test_gmbh.ceo
  221. test_gmbh.save()
  222. test_gmbh.name = F("ceo__last_name")
  223. self.assertRaises(FieldError, test_gmbh.save)
  224. def test_object_update_unsaved_objects(self):
  225. # F expressions cannot be used to update attributes on objects which do
  226. # not yet exist in the database
  227. test_gmbh = Company.objects.get(name="Test GmbH")
  228. acme = Company(
  229. name="The Acme Widget Co.", num_employees=12, num_chairs=5,
  230. ceo=test_gmbh.ceo
  231. )
  232. acme.num_employees = F("num_employees") + 16
  233. self.assertRaises(TypeError, acme.save)
  234. def test_ticket_11722_iexact_lookup(self):
  235. Employee.objects.create(firstname="John", lastname="Doe")
  236. Employee.objects.create(firstname="Test", lastname="test")
  237. queryset = Employee.objects.filter(firstname__iexact=F('lastname'))
  238. self.assertQuerysetEqual(queryset, ["<Employee: Test test>"])
  239. @skipIfDBFeature('has_case_insensitive_like')
  240. def test_ticket_16731_startswith_lookup(self):
  241. Employee.objects.create(firstname="John", lastname="Doe")
  242. e2 = Employee.objects.create(firstname="Jack", lastname="Jackson")
  243. e3 = Employee.objects.create(firstname="Jack", lastname="jackson")
  244. self.assertQuerysetEqual(
  245. Employee.objects.filter(lastname__startswith=F('firstname')),
  246. [e2], lambda x: x)
  247. self.assertQuerysetEqual(
  248. Employee.objects.filter(lastname__istartswith=F('firstname')).order_by('pk'),
  249. [e2, e3], lambda x: x)
  250. def test_ticket_18375_join_reuse(self):
  251. # Test that reverse multijoin F() references and the lookup target
  252. # the same join. Pre #18375 the F() join was generated first, and the
  253. # lookup couldn't reuse that join.
  254. qs = Employee.objects.filter(
  255. company_ceo_set__num_chairs=F('company_ceo_set__num_employees'))
  256. self.assertEqual(str(qs.query).count('JOIN'), 1)
  257. def test_ticket_18375_kwarg_ordering(self):
  258. # The next query was dict-randomization dependent - if the "gte=1"
  259. # was seen first, then the F() will reuse the join generated by the
  260. # gte lookup, if F() was seen first, then it generated a join the
  261. # other lookups could not reuse.
  262. qs = Employee.objects.filter(
  263. company_ceo_set__num_chairs=F('company_ceo_set__num_employees'),
  264. company_ceo_set__num_chairs__gte=1)
  265. self.assertEqual(str(qs.query).count('JOIN'), 1)
  266. def test_ticket_18375_kwarg_ordering_2(self):
  267. # Another similar case for F() than above. Now we have the same join
  268. # in two filter kwargs, one in the lhs lookup, one in F. Here pre
  269. # #18375 the amount of joins generated was random if dict
  270. # randomization was enabled, that is the generated query dependent
  271. # on which clause was seen first.
  272. qs = Employee.objects.filter(
  273. company_ceo_set__num_employees=F('pk'),
  274. pk=F('company_ceo_set__num_employees')
  275. )
  276. self.assertEqual(str(qs.query).count('JOIN'), 1)
  277. def test_ticket_18375_chained_filters(self):
  278. # Test that F() expressions do not reuse joins from previous filter.
  279. qs = Employee.objects.filter(
  280. company_ceo_set__num_employees=F('pk')
  281. ).filter(
  282. company_ceo_set__num_employees=F('company_ceo_set__num_employees')
  283. )
  284. self.assertEqual(str(qs.query).count('JOIN'), 2)
  285. class ExpressionsTests(TestCase):
  286. def test_F_object_deepcopy(self):
  287. """
  288. Make sure F objects can be deepcopied (#23492)
  289. """
  290. f = F("foo")
  291. g = deepcopy(f)
  292. self.assertEqual(f.name, g.name)
  293. def test_f_reuse(self):
  294. f = F('id')
  295. n = Number.objects.create(integer=-1)
  296. c = Company.objects.create(
  297. name="Example Inc.", num_employees=2300, num_chairs=5,
  298. ceo=Employee.objects.create(firstname="Joe", lastname="Smith")
  299. )
  300. c_qs = Company.objects.filter(id=f)
  301. self.assertEqual(c_qs.get(), c)
  302. # Reuse the same F-object for another queryset
  303. n_qs = Number.objects.filter(id=f)
  304. self.assertEqual(n_qs.get(), n)
  305. # The original query still works correctly
  306. self.assertEqual(c_qs.get(), c)
  307. def test_patterns_escape(self):
  308. """
  309. Test that special characters (e.g. %, _ and \) stored in database are
  310. properly escaped when using a pattern lookup with an expression
  311. refs #16731
  312. """
  313. Employee.objects.bulk_create([
  314. Employee(firstname="%Joh\\nny", lastname="%Joh\\n"),
  315. Employee(firstname="Johnny", lastname="%John"),
  316. Employee(firstname="Jean-Claude", lastname="Claud_"),
  317. Employee(firstname="Jean-Claude", lastname="Claude"),
  318. Employee(firstname="Jean-Claude", lastname="Claude%"),
  319. Employee(firstname="Johnny", lastname="Joh\\n"),
  320. Employee(firstname="Johnny", lastname="John"),
  321. Employee(firstname="Johnny", lastname="_ohn"),
  322. ])
  323. self.assertQuerysetEqual(
  324. Employee.objects.filter(firstname__contains=F('lastname')),
  325. ["<Employee: %Joh\\nny %Joh\\n>", "<Employee: Jean-Claude Claude>", "<Employee: Johnny John>"],
  326. ordered=False)
  327. self.assertQuerysetEqual(
  328. Employee.objects.filter(firstname__startswith=F('lastname')),
  329. ["<Employee: %Joh\\nny %Joh\\n>", "<Employee: Johnny John>"],
  330. ordered=False)
  331. self.assertQuerysetEqual(
  332. Employee.objects.filter(firstname__endswith=F('lastname')),
  333. ["<Employee: Jean-Claude Claude>"],
  334. ordered=False)
  335. def test_insensitive_patterns_escape(self):
  336. """
  337. Test that special characters (e.g. %, _ and \) stored in database are
  338. properly escaped when using a case insensitive pattern lookup with an
  339. expression -- refs #16731
  340. """
  341. Employee.objects.bulk_create([
  342. Employee(firstname="%Joh\\nny", lastname="%joh\\n"),
  343. Employee(firstname="Johnny", lastname="%john"),
  344. Employee(firstname="Jean-Claude", lastname="claud_"),
  345. Employee(firstname="Jean-Claude", lastname="claude"),
  346. Employee(firstname="Jean-Claude", lastname="claude%"),
  347. Employee(firstname="Johnny", lastname="joh\\n"),
  348. Employee(firstname="Johnny", lastname="john"),
  349. Employee(firstname="Johnny", lastname="_ohn"),
  350. ])
  351. self.assertQuerysetEqual(
  352. Employee.objects.filter(firstname__icontains=F('lastname')),
  353. ["<Employee: %Joh\\nny %joh\\n>", "<Employee: Jean-Claude claude>", "<Employee: Johnny john>"],
  354. ordered=False)
  355. self.assertQuerysetEqual(
  356. Employee.objects.filter(firstname__istartswith=F('lastname')),
  357. ["<Employee: %Joh\\nny %joh\\n>", "<Employee: Johnny john>"],
  358. ordered=False)
  359. self.assertQuerysetEqual(
  360. Employee.objects.filter(firstname__iendswith=F('lastname')),
  361. ["<Employee: Jean-Claude claude>"],
  362. ordered=False)
  363. class ExpressionsNumericTests(TestCase):
  364. def setUp(self):
  365. Number(integer=-1).save()
  366. Number(integer=42).save()
  367. Number(integer=1337).save()
  368. self.assertEqual(Number.objects.update(float=F('integer')), 3)
  369. def test_fill_with_value_from_same_object(self):
  370. """
  371. We can fill a value in all objects with an other value of the
  372. same object.
  373. """
  374. self.assertQuerysetEqual(
  375. Number.objects.all(),
  376. [
  377. '<Number: -1, -1.000>',
  378. '<Number: 42, 42.000>',
  379. '<Number: 1337, 1337.000>'
  380. ],
  381. ordered=False
  382. )
  383. def test_increment_value(self):
  384. """
  385. We can increment a value of all objects in a query set.
  386. """
  387. self.assertEqual(
  388. Number.objects.filter(integer__gt=0)
  389. .update(integer=F('integer') + 1),
  390. 2)
  391. self.assertQuerysetEqual(
  392. Number.objects.all(),
  393. [
  394. '<Number: -1, -1.000>',
  395. '<Number: 43, 42.000>',
  396. '<Number: 1338, 1337.000>'
  397. ],
  398. ordered=False
  399. )
  400. def test_filter_not_equals_other_field(self):
  401. """
  402. We can filter for objects, where a value is not equals the value
  403. of an other field.
  404. """
  405. self.assertEqual(
  406. Number.objects.filter(integer__gt=0)
  407. .update(integer=F('integer') + 1),
  408. 2)
  409. self.assertQuerysetEqual(
  410. Number.objects.exclude(float=F('integer')),
  411. [
  412. '<Number: 43, 42.000>',
  413. '<Number: 1338, 1337.000>'
  414. ],
  415. ordered=False
  416. )
  417. def test_complex_expressions(self):
  418. """
  419. Complex expressions of different connection types are possible.
  420. """
  421. n = Number.objects.create(integer=10, float=123.45)
  422. self.assertEqual(Number.objects.filter(pk=n.pk).update(
  423. float=F('integer') + F('float') * 2), 1)
  424. self.assertEqual(Number.objects.get(pk=n.pk).integer, 10)
  425. self.assertEqual(Number.objects.get(pk=n.pk).float, Approximate(256.900, places=3))
  426. def test_incorrect_field_expression(self):
  427. with six.assertRaisesRegex(self, FieldError, "Cannot resolve keyword u?'nope' into field.*"):
  428. list(Employee.objects.filter(firstname=F('nope')))
  429. class ExpressionOperatorTests(TestCase):
  430. def setUp(self):
  431. self.n = Number.objects.create(integer=42, float=15.5)
  432. def test_lefthand_addition(self):
  433. # LH Addition of floats and integers
  434. Number.objects.filter(pk=self.n.pk).update(
  435. integer=F('integer') + 15,
  436. float=F('float') + 42.7
  437. )
  438. self.assertEqual(Number.objects.get(pk=self.n.pk).integer, 57)
  439. self.assertEqual(Number.objects.get(pk=self.n.pk).float, Approximate(58.200, places=3))
  440. def test_lefthand_subtraction(self):
  441. # LH Subtraction of floats and integers
  442. Number.objects.filter(pk=self.n.pk).update(integer=F('integer') - 15,
  443. float=F('float') - 42.7)
  444. self.assertEqual(Number.objects.get(pk=self.n.pk).integer, 27)
  445. self.assertEqual(Number.objects.get(pk=self.n.pk).float, Approximate(-27.200, places=3))
  446. def test_lefthand_multiplication(self):
  447. # Multiplication of floats and integers
  448. Number.objects.filter(pk=self.n.pk).update(integer=F('integer') * 15,
  449. float=F('float') * 42.7)
  450. self.assertEqual(Number.objects.get(pk=self.n.pk).integer, 630)
  451. self.assertEqual(Number.objects.get(pk=self.n.pk).float, Approximate(661.850, places=3))
  452. def test_lefthand_division(self):
  453. # LH Division of floats and integers
  454. Number.objects.filter(pk=self.n.pk).update(integer=F('integer') / 2,
  455. float=F('float') / 42.7)
  456. self.assertEqual(Number.objects.get(pk=self.n.pk).integer, 21)
  457. self.assertEqual(Number.objects.get(pk=self.n.pk).float, Approximate(0.363, places=3))
  458. def test_lefthand_modulo(self):
  459. # LH Modulo arithmetic on integers
  460. Number.objects.filter(pk=self.n.pk).update(integer=F('integer') % 20)
  461. self.assertEqual(Number.objects.get(pk=self.n.pk).integer, 2)
  462. self.assertEqual(Number.objects.get(pk=self.n.pk).float, Approximate(15.500, places=3))
  463. def test_lefthand_bitwise_and(self):
  464. # LH Bitwise ands on integers
  465. Number.objects.filter(pk=self.n.pk).update(integer=F('integer').bitand(56))
  466. self.assertEqual(Number.objects.get(pk=self.n.pk).integer, 40)
  467. self.assertEqual(Number.objects.get(pk=self.n.pk).float, Approximate(15.500, places=3))
  468. @skipUnlessDBFeature('supports_bitwise_or')
  469. def test_lefthand_bitwise_or(self):
  470. # LH Bitwise or on integers
  471. Number.objects.filter(pk=self.n.pk).update(integer=F('integer').bitor(48))
  472. self.assertEqual(Number.objects.get(pk=self.n.pk).integer, 58)
  473. self.assertEqual(Number.objects.get(pk=self.n.pk).float, Approximate(15.500, places=3))
  474. def test_lefthand_power(self):
  475. # LH Powert arithmetic operation on floats and integers
  476. Number.objects.filter(pk=self.n.pk).update(integer=F('integer') ** 2,
  477. float=F('float') ** 1.5)
  478. self.assertEqual(Number.objects.get(pk=self.n.pk).integer, 1764)
  479. self.assertEqual(Number.objects.get(pk=self.n.pk).float, Approximate(61.02, places=2))
  480. def test_right_hand_addition(self):
  481. # Right hand operators
  482. Number.objects.filter(pk=self.n.pk).update(integer=15 + F('integer'),
  483. float=42.7 + F('float'))
  484. # RH Addition of floats and integers
  485. self.assertEqual(Number.objects.get(pk=self.n.pk).integer, 57)
  486. self.assertEqual(Number.objects.get(pk=self.n.pk).float, Approximate(58.200, places=3))
  487. def test_right_hand_subtraction(self):
  488. Number.objects.filter(pk=self.n.pk).update(integer=15 - F('integer'),
  489. float=42.7 - F('float'))
  490. # RH Subtraction of floats and integers
  491. self.assertEqual(Number.objects.get(pk=self.n.pk).integer, -27)
  492. self.assertEqual(Number.objects.get(pk=self.n.pk).float, Approximate(27.200, places=3))
  493. def test_right_hand_multiplication(self):
  494. # RH Multiplication of floats and integers
  495. Number.objects.filter(pk=self.n.pk).update(integer=15 * F('integer'),
  496. float=42.7 * F('float'))
  497. self.assertEqual(Number.objects.get(pk=self.n.pk).integer, 630)
  498. self.assertEqual(Number.objects.get(pk=self.n.pk).float, Approximate(661.850, places=3))
  499. def test_right_hand_division(self):
  500. # RH Division of floats and integers
  501. Number.objects.filter(pk=self.n.pk).update(integer=640 / F('integer'),
  502. float=42.7 / F('float'))
  503. self.assertEqual(Number.objects.get(pk=self.n.pk).integer, 15)
  504. self.assertEqual(Number.objects.get(pk=self.n.pk).float, Approximate(2.755, places=3))
  505. def test_right_hand_modulo(self):
  506. # RH Modulo arithmetic on integers
  507. Number.objects.filter(pk=self.n.pk).update(integer=69 % F('integer'))
  508. self.assertEqual(Number.objects.get(pk=self.n.pk).integer, 27)
  509. self.assertEqual(Number.objects.get(pk=self.n.pk).float, Approximate(15.500, places=3))
  510. def test_righthand_power(self):
  511. # RH Powert arithmetic operation on floats and integers
  512. Number.objects.filter(pk=self.n.pk).update(integer=2 ** F('integer'),
  513. float=1.5 ** F('float'))
  514. self.assertEqual(Number.objects.get(pk=self.n.pk).integer, 4398046511104)
  515. self.assertEqual(Number.objects.get(pk=self.n.pk).float, Approximate(536.308, places=3))
  516. class FTimeDeltaTests(TestCase):
  517. def setUp(self):
  518. sday = datetime.date(2010, 6, 25)
  519. stime = datetime.datetime(2010, 6, 25, 12, 15, 30, 747000)
  520. midnight = datetime.time(0)
  521. delta0 = datetime.timedelta(0)
  522. delta1 = datetime.timedelta(microseconds=253000)
  523. delta2 = datetime.timedelta(seconds=44)
  524. delta3 = datetime.timedelta(hours=21, minutes=8)
  525. delta4 = datetime.timedelta(days=10)
  526. # Test data is set so that deltas and delays will be
  527. # strictly increasing.
  528. self.deltas = []
  529. self.delays = []
  530. self.days_long = []
  531. # e0: started same day as assigned, zero duration
  532. end = stime + delta0
  533. e0 = Experiment.objects.create(name='e0', assigned=sday, start=stime,
  534. end=end, completed=end.date(), estimated_time=delta0)
  535. self.deltas.append(delta0)
  536. self.delays.append(e0.start -
  537. datetime.datetime.combine(e0.assigned, midnight))
  538. self.days_long.append(e0.completed - e0.assigned)
  539. # e1: started one day after assigned, tiny duration, data
  540. # set so that end time has no fractional seconds, which
  541. # tests an edge case on sqlite. This Experiment is only
  542. # included in the test data when the DB supports microsecond
  543. # precision.
  544. if connection.features.supports_microsecond_precision:
  545. delay = datetime.timedelta(1)
  546. end = stime + delay + delta1
  547. e1 = Experiment.objects.create(name='e1', assigned=sday,
  548. start=stime + delay, end=end, completed=end.date(), estimated_time=delta1)
  549. self.deltas.append(delta1)
  550. self.delays.append(e1.start -
  551. datetime.datetime.combine(e1.assigned, midnight))
  552. self.days_long.append(e1.completed - e1.assigned)
  553. # e2: started three days after assigned, small duration
  554. end = stime + delta2
  555. e2 = Experiment.objects.create(name='e2',
  556. assigned=sday - datetime.timedelta(3), start=stime, end=end,
  557. completed=end.date(), estimated_time=datetime.timedelta(hours=1))
  558. self.deltas.append(delta2)
  559. self.delays.append(e2.start -
  560. datetime.datetime.combine(e2.assigned, midnight))
  561. self.days_long.append(e2.completed - e2.assigned)
  562. # e3: started four days after assigned, medium duration
  563. delay = datetime.timedelta(4)
  564. end = stime + delay + delta3
  565. e3 = Experiment.objects.create(name='e3',
  566. assigned=sday, start=stime + delay, end=end, completed=end.date(), estimated_time=delta3)
  567. self.deltas.append(delta3)
  568. self.delays.append(e3.start -
  569. datetime.datetime.combine(e3.assigned, midnight))
  570. self.days_long.append(e3.completed - e3.assigned)
  571. # e4: started 10 days after assignment, long duration
  572. end = stime + delta4
  573. e4 = Experiment.objects.create(name='e4',
  574. assigned=sday - datetime.timedelta(10), start=stime, end=end,
  575. completed=end.date(), estimated_time=delta4 - datetime.timedelta(1))
  576. self.deltas.append(delta4)
  577. self.delays.append(e4.start -
  578. datetime.datetime.combine(e4.assigned, midnight))
  579. self.days_long.append(e4.completed - e4.assigned)
  580. self.expnames = [e.name for e in Experiment.objects.all()]
  581. def test_multiple_query_compilation(self):
  582. # Ticket #21643
  583. queryset = Experiment.objects.filter(end__lt=F('start') + datetime.timedelta(hours=1))
  584. q1 = str(queryset.query)
  585. q2 = str(queryset.query)
  586. self.assertEqual(q1, q2)
  587. def test_query_clone(self):
  588. # Ticket #21643
  589. qs = Experiment.objects.filter(end__lt=F('start') + datetime.timedelta(hours=1))
  590. qs2 = qs.all()
  591. list(qs)
  592. list(qs2)
  593. def test_delta_add(self):
  594. for i in range(len(self.deltas)):
  595. delta = self.deltas[i]
  596. test_set = [e.name for e in
  597. Experiment.objects.filter(end__lt=F('start') + delta)]
  598. self.assertEqual(test_set, self.expnames[:i])
  599. test_set = [e.name for e in
  600. Experiment.objects.filter(end__lt=delta + F('start'))]
  601. self.assertEqual(test_set, self.expnames[:i])
  602. test_set = [e.name for e in
  603. Experiment.objects.filter(end__lte=F('start') + delta)]
  604. self.assertEqual(test_set, self.expnames[:i + 1])
  605. def test_delta_subtract(self):
  606. for i in range(len(self.deltas)):
  607. delta = self.deltas[i]
  608. test_set = [e.name for e in
  609. Experiment.objects.filter(start__gt=F('end') - delta)]
  610. self.assertEqual(test_set, self.expnames[:i])
  611. test_set = [e.name for e in
  612. Experiment.objects.filter(start__gte=F('end') - delta)]
  613. self.assertEqual(test_set, self.expnames[:i + 1])
  614. def test_exclude(self):
  615. for i in range(len(self.deltas)):
  616. delta = self.deltas[i]
  617. test_set = [e.name for e in
  618. Experiment.objects.exclude(end__lt=F('start') + delta)]
  619. self.assertEqual(test_set, self.expnames[i:])
  620. test_set = [e.name for e in
  621. Experiment.objects.exclude(end__lte=F('start') + delta)]
  622. self.assertEqual(test_set, self.expnames[i + 1:])
  623. def test_date_comparison(self):
  624. for i in range(len(self.days_long)):
  625. days = self.days_long[i]
  626. test_set = [e.name for e in
  627. Experiment.objects.filter(completed__lt=F('assigned') + days)]
  628. self.assertEqual(test_set, self.expnames[:i])
  629. test_set = [e.name for e in
  630. Experiment.objects.filter(completed__lte=F('assigned') + days)]
  631. self.assertEqual(test_set, self.expnames[:i + 1])
  632. @skipUnlessDBFeature("supports_mixed_date_datetime_comparisons")
  633. def test_mixed_comparisons1(self):
  634. for i in range(len(self.delays)):
  635. delay = self.delays[i]
  636. if not connection.features.supports_microsecond_precision:
  637. delay = datetime.timedelta(delay.days, delay.seconds)
  638. test_set = [e.name for e in
  639. Experiment.objects.filter(assigned__gt=F('start') - delay)]
  640. self.assertEqual(test_set, self.expnames[:i])
  641. test_set = [e.name for e in
  642. Experiment.objects.filter(assigned__gte=F('start') - delay)]
  643. self.assertEqual(test_set, self.expnames[:i + 1])
  644. def test_mixed_comparisons2(self):
  645. delays = [datetime.timedelta(delay.days) for delay in self.delays]
  646. for i in range(len(delays)):
  647. delay = delays[i]
  648. test_set = [e.name for e in
  649. Experiment.objects.filter(start__lt=F('assigned') + delay)]
  650. self.assertEqual(test_set, self.expnames[:i])
  651. test_set = [e.name for e in
  652. Experiment.objects.filter(start__lte=F('assigned') + delay +
  653. datetime.timedelta(1))]
  654. self.assertEqual(test_set, self.expnames[:i + 1])
  655. def test_delta_update(self):
  656. for i in range(len(self.deltas)):
  657. delta = self.deltas[i]
  658. exps = Experiment.objects.all()
  659. expected_durations = [e.duration() for e in exps]
  660. expected_starts = [e.start + delta for e in exps]
  661. expected_ends = [e.end + delta for e in exps]
  662. Experiment.objects.update(start=F('start') + delta, end=F('end') + delta)
  663. exps = Experiment.objects.all()
  664. new_starts = [e.start for e in exps]
  665. new_ends = [e.end for e in exps]
  666. new_durations = [e.duration() for e in exps]
  667. self.assertEqual(expected_starts, new_starts)
  668. self.assertEqual(expected_ends, new_ends)
  669. self.assertEqual(expected_durations, new_durations)
  670. def test_invalid_operator(self):
  671. with self.assertRaises(DatabaseError):
  672. list(Experiment.objects.filter(start=F('start') * datetime.timedelta(0)))
  673. def test_durationfield_add(self):
  674. zeros = [e.name for e in
  675. Experiment.objects.filter(start=F('start') + F('estimated_time'))]
  676. self.assertEqual(zeros, ['e0'])
  677. end_less = [e.name for e in
  678. Experiment.objects.filter(end__lt=F('start') + F('estimated_time'))]
  679. self.assertEqual(end_less, ['e2'])
  680. delta_math = [e.name for e in
  681. Experiment.objects.filter(end__gte=F('start') + F('estimated_time') + datetime.timedelta(hours=1))]
  682. self.assertEqual(delta_math, ['e4'])
  683. @skipUnlessDBFeature("has_native_duration_field")
  684. def test_date_subtraction(self):
  685. under_estimate = [e.name for e in
  686. Experiment.objects.filter(estimated_time__gt=F('end') - F('start'))]
  687. self.assertEqual(under_estimate, ['e2'])
  688. over_estimate = [e.name for e in
  689. Experiment.objects.filter(estimated_time__lt=F('end') - F('start'))]
  690. self.assertEqual(over_estimate, ['e4'])
  691. class ValueTests(TestCase):
  692. def test_update_TimeField_using_Value(self):
  693. Time.objects.create()
  694. Time.objects.update(time=Value(datetime.time(1), output_field=TimeField()))
  695. self.assertEqual(Time.objects.get().time, datetime.time(1))
  696. def test_update_UUIDField_using_Value(self):
  697. UUID.objects.create()
  698. UUID.objects.update(uuid=Value(uuid.UUID('12345678901234567890123456789012'), output_field=UUIDField()))
  699. self.assertEqual(UUID.objects.get().uuid, uuid.UUID('12345678901234567890123456789012'))