|
| 1 | +from datetime import datetime |
| 2 | + |
| 3 | +from ravendb.infrastructure.orders import Order, Company |
| 4 | +from ravendb.tests.test_base import TestBase |
| 5 | + |
| 6 | + |
| 7 | +class TestRavenDB15000(TestBase): |
| 8 | + def setUp(self): |
| 9 | + super(TestRavenDB15000, self).setUp() |
| 10 | + |
| 11 | + def test_can_include_time_series_without_providing_from_and_to_dates_via_load(self): |
| 12 | + with self.store.open_session() as session: |
| 13 | + session.store(Order(company="companies/1-A"), "orders/1-A") |
| 14 | + session.store(Company(name="HR"), "companies/1-A") |
| 15 | + session.time_series_for("orders/1-A", "Heartrate").append_single(datetime(1999, 11, 14), 1) |
| 16 | + session.save_changes() |
| 17 | + |
| 18 | + with self.store.open_session() as session: |
| 19 | + order = session.load( |
| 20 | + "orders/1-A", Order, lambda i: i.include_documents("company").include_time_series("Heartrate") |
| 21 | + ) |
| 22 | + |
| 23 | + # should not go to server |
| 24 | + company = session.load(order.company, Company) |
| 25 | + |
| 26 | + self.assertEqual(1, session.advanced.number_of_requests) |
| 27 | + self.assertEqual("HR", company.name) |
| 28 | + |
| 29 | + # should not go to server |
| 30 | + vals = session.time_series_for_entity(order, "Heartrate").get() |
| 31 | + |
| 32 | + self.assertEqual(1, session.advanced.number_of_requests) |
| 33 | + self.assertEqual(1, len(vals)) |
| 34 | + |
| 35 | + def test_can_include_time_series_without_providing_from_and_to_dates_via_query(self): |
| 36 | + with self.store.open_session() as session: |
| 37 | + session.store(Order(company="companies/1-A"), "orders/1-A") |
| 38 | + session.store(Company(name="HR"), "companies/1-A") |
| 39 | + session.time_series_for("orders/1-A", "Heartrate").append_single(datetime(1999, 11, 14), 1) |
| 40 | + session.save_changes() |
| 41 | + |
| 42 | + with self.store.open_session() as session: |
| 43 | + order = ( |
| 44 | + session.query(object_type=Order) |
| 45 | + .include(lambda i: i.include_documents("company").include_time_series("Heartrate")) |
| 46 | + .first() |
| 47 | + ) |
| 48 | + |
| 49 | + # should not go to server |
| 50 | + company = session.load(order.company, Company) |
| 51 | + |
| 52 | + self.assertEqual(1, session.advanced.number_of_requests) |
| 53 | + self.assertEqual("HR", company.name) |
| 54 | + |
| 55 | + # should not go to server |
| 56 | + vals = session.time_series_for_entity(order, "Heartrate").get() |
| 57 | + |
| 58 | + self.assertEqual(1, session.advanced.number_of_requests) |
| 59 | + self.assertEqual(1, len(vals)) |
0 commit comments