|
12 | 12 | from minos.aggregate import ( |
13 | 13 | AlreadyDeletedException, |
14 | 14 | Condition, |
| 15 | + Delta, |
15 | 16 | DeltaRepositoryException, |
16 | 17 | EntityRepository, |
17 | 18 | NotFoundException, |
@@ -145,18 +146,22 @@ async def test_update(self): |
145 | 146 | self.assertEqual(expected, car) |
146 | 147 | self.assertEqual(car, await self.repository.get(Car, car.uuid)) |
147 | 148 |
|
148 | | - await self.repository.update(car, doors=5) |
| 149 | + observed, delta = await self.repository.update(car, doors=5) |
| 150 | + self.assertIsInstance(delta, Delta) |
| 151 | + |
149 | 152 | expected = Car(5, "red", uuid=car.uuid, version=3, created_at=car.created_at, updated_at=car.updated_at) |
150 | | - self.assertEqual(expected, car) |
151 | | - self.assertEqual(car, await self.repository.get(Car, car.uuid)) |
| 153 | + self.assertEqual(expected, observed) |
| 154 | + self.assertEqual(observed, await self.repository.get(Car, expected.uuid)) |
152 | 155 |
|
153 | 156 | async def test_update_no_changes(self): |
154 | 157 | car, _ = await self.repository.create(Car, doors=3, color="blue") |
155 | 158 |
|
156 | | - await self.repository.update(car, color="blue") |
| 159 | + observed, delta = await self.repository.update(car, color="blue") |
| 160 | + self.assertIsNone(delta) |
| 161 | + |
157 | 162 | expected = Car(3, "blue", uuid=car.uuid, version=1, created_at=car.created_at, updated_at=car.updated_at) |
158 | | - self.assertEqual(expected, car) |
159 | | - self.assertEqual(car, await self.repository.get(Car, car.uuid)) |
| 163 | + self.assertEqual(expected, observed) |
| 164 | + self.assertEqual(observed, await self.repository.get(Car, expected.uuid)) |
160 | 165 |
|
161 | 166 | async def test_update_raises(self): |
162 | 167 | with self.assertRaises(DeltaRepositoryException): |
|
0 commit comments