service-layer test for add batch [test_add_batch]

hjwp · hjwp · commit b7e15c46e8d4 · 2020-01-08T20:41:36.000Z
diff --git a/test_services.py b/test_services.py
@@ -32,6 +32,13 @@ def commit(self):
         self.committed = True
 
 
+def test_add_batch():
+    repo, session = FakeRepository([]), FakeSession()
+    services.add_batch('b1', 'CRUNCHY-ARMCHAIR', 100, None, repo, session)
+    assert repo.get('b1') is not None
+    assert session.committed
+
+
 def test_returns_allocation():
     repo = FakeRepository.for_batch("batch1", "COMPLICATED-LAMP", 100, eta=None)
     result = services.allocate("o1", "COMPLICATED-LAMP", 10, repo, FakeSession())
