create dummy loa document on-the-fly with very-basic pdf content

Author: mahtin

CloudFlare/tests/test_loa_documents.py

@@ -4,6 +4,9 @@
 import sys
 import time
 import random
+import tempfile
+
+from utils import dummy_loa_document
 
 sys.path.insert(0, os.path.abspath('.'))
 sys.path.insert(0, os.path.abspath('..'))
@@ -68,31 +71,38 @@ def test_addressing_loa_documents():
     """ test_addressing_loa_documents """
     loa_documents = cf.accounts.addressing.loa_documents(account_id)
     assert isinstance(loa_documents, list)
-    for loa_document in loa_documents:
+    for loa_document in loa_documents[-4:]:
         assert isinstance(loa_document, dict)
         assert 'id' in loa_document
+        assert 'created' in loa_document
         assert 'filename' in loa_document
         assert 'verified' in loa_document
         assert 'size_bytes' in loa_document
-        print('%s: filename=%s size_bytes=%d verified=%s' % (
+        print('%s: %s filename=%s size_bytes=%d verified=%s' % (
            loa_document['id'],
+           loa_document['created'],
            loa_document['filename'],
            loa_document['size_bytes'],
            loa_document['verified']
         ), file=sys.stderr)
 
 def test_addressing_loa_documents_upload(filename=None):
     """ test_addressing_loa_documents_upload """
-    if not filename:
-        filename = 'CloudFlare/tests/dummy_loa_document.pdf'
-    try:
-        pdf_file = open(filename, 'rb')
-    except (FileNotFoundError, IsADirectoryError, PermissionError) as e:
-        print('%s: %s' % (filename, e), file=sys.stderr)
-        assert False
+    if filename:
+        # use provided file
+        try:
+            pdf_file = open(filename, 'rb')
+        except (FileNotFoundError, IsADirectoryError, PermissionError) as e:
+            print('%s: %s' % (filename, e), file=sys.stderr)
+            assert False
+    else:
+        # create a dummy temporary file
+        pdf_file = tempfile.NamedTemporaryFile(mode='w+b', prefix='dummy-loa-document-', suffix='.pdf', delete=False)
+        pdf_file.write(dummy_loa_document.encode())
+        pdf_file.seek(0)
 
     size_bytes = os.fstat(pdf_file.fileno()).st_size
-    print('filename=%s size_bytes=%d' % (filename, size_bytes), file=sys.stderr)
+    print('filename=%s size_bytes=%d' % (pdf_file.name, size_bytes), file=sys.stderr)
 
     files = {'loa_document': pdf_file}
     try:
@@ -115,6 +125,8 @@ def test_addressing_loa_documents_upload(filename=None):
 
 def ispdf(s):
     """ ispdf """
+    if isinstance(s, str):
+        s = s.encode()
     idx = 0
     while s[idx] in [b'\r', b'\n']:
         idx += 1
@@ -129,21 +141,24 @@ def ispdf(s):
     if b'%PDF-' in s[0:1024]:
         return True
     # give up!
+    print('ispdf: failing with content="%s..."' % (s[0:50]), file=sys.stderr)
     return False
 
 def test_addressing_loa_documents_download():
     """ test_addressing_loa_documents_download """
     loa_documents = cf.accounts.addressing.loa_documents(account_id)
     assert isinstance(loa_documents, list)
-    for loa_document in loa_documents:
+    for loa_document in loa_documents[-4:]:
         assert isinstance(loa_document, dict)
         assert 'id' in loa_document
+        assert 'created' in loa_document
         assert 'filename' in loa_document
         assert 'verified' in loa_document
         assert 'size_bytes' in loa_document
         assert isinstance(loa_document['size_bytes'], int)
-        print('%s: filename=%s size_bytes=%d verified=%s' % (
+        print('%s: %s filename=%s size_bytes=%d verified=%s' % (
             loa_document['id'],
+            loa_document['created'],
             loa_document['filename'],
             loa_document['size_bytes'],
             loa_document['verified']
@@ -152,7 +167,6 @@ def test_addressing_loa_documents_download():
         size_bytes = loa_document['size_bytes']
         pdf_content = cf.accounts.addressing.loa_documents.download(account_id, loa_document_identifier)
         assert size_bytes == len(pdf_content)
-        print('%s: %s' % (loa_document_identifier, pdf_content[0:10]))
         assert ispdf(pdf_content)
 
 if __name__ == '__main__':

CloudFlare/tests/utils.py

@@ -0,0 +1,46 @@
+""" misc utilities for Cloudflare test code """
+
+dummy_loa_document = """%PDF-1.4
+1 0 obj << /Length 559 >> stream
+ 1 0 0 rg
+ BT /F1 24 Tf 72 680 Td (LOA DOCUMENT) Tj ET
+ .75 .75 .75 RG 1 w
+ 72 676 m 540 676 l s
+ 0 0 0 rg
+ BT /F1 12 Tf 72 648 Td (See /tests/ folder under python-cloudflare on GitHub) Tj ET
+ BT /F1 12 Tf 72 624 Td (THIS DOCUMENT IS ONLY USED FOR TESTING) Tj ET
+ BT /F1 12 Tf 72 604 Td (Please ignore and delete upon receipt) Tj ET
+ BT /F1 12 Tf 72 584 Td (See URL:) Tj ET
+ 0 0 1 rg
+ BT /F2 12 Tf 130 584 Td (http://github.com/cloudflare/python-cloudflare) Tj ET
+ .75 .75 .75 RG 1 w
+ 72 560 m 540 560 l s
+ .75 .75 .75 RG 1 w
+ 36 756 m 576 756 l 576 36 l 36 36 l 36 756 l s
+endstream
+endobj
+2 0 obj << /Type /Catalog /Pages 3 0 R >> endobj
+3 0 obj << /Type /Pages /Kids [4 0 R ] /Count 1 >> endobj
+4 0 obj << /Type /Page /Parent 3 0 R /MediaBox [0 0 612 792] /Contents 1 0 R
+ /Resources <<
+  /ProcSet 5 0 R
+  /Font << /F1 6 0 R >>
+  /Font << /F2 7 0 R >>
+ >>
+>>
+endobj
+5 0 obj [/PDF /Text] endobj
+6 0 obj << /Type /Font /Subtype /Type1 /Name /F1 /BaseFont /Arial >> endobj
+7 0 obj << /Type /Font /Subtype /Type1 /Name /F2 /BaseFont /Courier >> endobj
+8 0 obj << /Creator (https://github.com/cloudflare/python-cloudflare)
+ /Producer (Hand coded for python-cloudflare)
+ /Title (dummy_loa_document.pdf)
+ /Author (Martin J Levy)
+ /Subject (Dummy LOA Document - please delete)
+ /Keywords (LOA)
+ /CreationDate (D:20240101120000Z)
+ /ModDate (D:20240101120000Z)
+>> endobj
+trailer << /Size 8 /Root 2 0 R /Info 8 0 R >>
+%%EOF"""
+