From 32c4844aed3e2deecbd50294dcfd34d68eddb967 Mon Sep 17 00:00:00 2001
From: Cayo Puigdefabregas <cayo@ribaguifi.com>
Date: Wed, 17 Feb 2021 23:14:40 +0100
Subject: [PATCH] adding test for verify stamp

---
 tests/test_basic.py     |  1 -
 tests/test_documents.py | 40 +++++++++++++++++++++++++++++++++++++++-
 2 files changed, 39 insertions(+), 2 deletions(-)

diff --git a/tests/test_basic.py b/tests/test_basic.py
index b5e52805..77103f1b 100644
--- a/tests/test_basic.py
+++ b/tests/test_basic.py
@@ -27,7 +27,6 @@ def test_dependencies():
 def test_api_docs(client: Client):
     """Tests /apidocs correct initialization."""
     docs, _ = client.get('/apidocs')
-    # import pdb; pdb.set_trace()
     assert set(docs['paths'].keys()) == {
         '/actions/',
         '/apidocs',
diff --git a/tests/test_documents.py b/tests/test_documents.py
index fc4f8ea2..173be232 100644
--- a/tests/test_documents.py
+++ b/tests/test_documents.py
@@ -1,8 +1,9 @@
 import csv
 import hashlib
 from datetime import datetime
-from io import StringIO
+from io import StringIO, BytesIO
 from pathlib import Path
+from flask import url_for
 
 import pytest
 from werkzeug.exceptions import Unauthorized
@@ -459,3 +460,40 @@ def test_get_document_lots(user: UserClient, user2: UserClient):
     assert export_csv[1][3] == 'comments,lot1,testcomment-lot1,' or 'comments,lot2,testcomment-lot2,'
     assert export2_csv[1][1] == 'Lot3-User2'
     assert export2_csv[1][3] == 'comments,lot3,testcomment-lot3,'
+
+
+@pytest.mark.mvp 
+def test_verify_stamp(user: UserClient, client: Client):
+    """Test verify stamp of one export device information in a csv file."""
+    snapshot, _ = user.post(file('basic.snapshot'), res=Snapshot)
+    csv_str, _ = user.get(res=documents.DocumentDef.t,
+                          item='devices/',
+                          accept='text/csv',
+                          query=[('filter', {'type': ['Computer']})])
+    
+    response, _ = client.post(res=documents.DocumentDef.t,
+            item='stamps/',
+            content_type='multipart/form-data',
+            accept='text/html',
+            data={'docUpload': [(BytesIO(bytes(csv_str, 'utf-8')), 'example.csv')]}, 
+            status=200)
+    assert "alert alert-info" in response
+    assert not "alert alert-danger" in response
+
+    response, _ = client.post(res=documents.DocumentDef.t,
+            item='stamps/',
+            content_type='multipart/form-data',
+            accept='text/html',
+            data={'docUpload': [(BytesIO(b'abc'), 'example.csv')]},
+            status=200)
+
+    assert not "alert alert-info" in response
+    assert "alert alert-danger" in response
+
+    response, _ = client.get(res=documents.DocumentDef.t,
+            item='stamps/',
+            accept='text/html',
+            status=200)
+
+    assert not "alert alert-info" in response
+    assert not "alert alert-danger" in response