From a2303498cd9d29768df7be8b5e42a464c56a3c02 Mon Sep 17 00:00:00 2001
From: nad <jnadeu@gmail.com>
Date: Tue, 28 Jul 2020 16:22:17 +0200
Subject: [PATCH] Minors comments changes and new test_get_all_lots was added

---
 ereuse_devicehub/resources/documents/documents.py |  4 ++--
 tests/test_lot.py                                 | 14 ++++++++++++++
 2 files changed, 16 insertions(+), 2 deletions(-)

diff --git a/ereuse_devicehub/resources/documents/documents.py b/ereuse_devicehub/resources/documents/documents.py
index aa61dbb0..57b8844c 100644
--- a/ereuse_devicehub/resources/documents/documents.py
+++ b/ereuse_devicehub/resources/documents/documents.py
@@ -134,7 +134,7 @@ class LotsDocumentView(LotView):
         return self.generate_lots_csv(Lot.query)
 
     def generate_lots_csv(self, query):
-        """Get device query and put information in csv format."""
+        """Get lot query and put information in csv format."""
         data = StringIO()
         cw = csv.writer(data)
         first = True
@@ -154,7 +154,7 @@ class LotRow(OrderedDict):
     def __init__(self, lot: Lot) -> None:
         super().__init__()
         self.lot = lot
-        # General information about device
+        # General information about lot
         self['Id'] = lot.id.hex
         self['Name'] = lot.name
         self['Registered in'] = format(lot.created, '%c')
diff --git a/tests/test_lot.py b/tests/test_lot.py
index c1473496..50f01641 100644
--- a/tests/test_lot.py
+++ b/tests/test_lot.py
@@ -381,3 +381,17 @@ def test_lot_post_add_remove_device_view(app: Devicehub, user: UserClient):
                          query=[('id', device_id)],
                          status=200)
     assert not len(lot['devices'])
+
+
+@pytest.mark.mvp
+def test_get_all_lots(user: UserClient):
+    """Tests submitting and retreiving all lots."""
+    l, _ = user.post({'name': 'Lot1', 'description': 'comments,lot1,testcomment,'}, res=Lot)
+    l, _ = user.post({'name': 'Lot2', 'description': 'comments,lot2,testcomment,'}, res=Lot)
+    l, _ = user.post({'name': 'Lot3', 'description': 'comments3,lot3,testcomment,'}, res=Lot)
+
+    l, _ = user.get(res=Lot)
+    assert l['items'][0]['name'] == 'Lot1'
+    assert l['items'][0]['description'] == 'comments,lot1,testcomment,'
+    assert l['items'][2]['name'] == 'Lot3'
+    assert l['items'][2]['description'] == 'comments,lot3,testcomment,'