Minors comments changes and new test_get_all_lots was added

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')

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,'