get allocations working

hjwp · hjwp · commit 0800c6148888 · 2021-02-24T20:19:55.000Z
diff --git a/src/djangoproject/alloc/models.py b/src/djangoproject/alloc/models.py
@@ -20,14 +20,23 @@ def from_domain(self, batch: domain_model.Batch):
         return b
 
     def to_domain(self):
-        return domain_model.Batch(ref=self.reference, sku=self.sku, qty=self.qty)
+        b = domain_model.Batch(
+            ref=self.reference, sku=self.sku, qty=self.qty, eta=self.eta
+        )
+        b._allocations = set(a.line.to_domain() for a in self.allocation_set.all())
+        return b
 
 
 class OrderLine(models.Model):
     orderid = models.CharField(max_length=255)
     sku = models.CharField(max_length=255)
     qty = models.IntegerField()
 
+    def to_domain(self):
+        return domain_model.OrderLine(
+            orderid=self.orderid, sku=self.sku, qty=self.qty
+        )
+
 
 class Allocation(models.Model):
     batch = models.ForeignKey(Batch, on_delete=models.CASCADE)
diff --git a/tests/integration/test_repository.py b/tests/integration/test_repository.py
@@ -26,7 +26,6 @@ def test_repository_can_save_a_batch(django_models):
     assert saved_batch.eta == batch.eta
 
 
-@pytest.mark.skip("foo")
 @pytest.mark.django_db
 def test_repository_can_retrieve_a_batch_with_allocations(django_models):
     sku = "PONY-STATUE"
