adding file is PDF and ZIP functions

dmutils/__init__.py

@@ -3,4 +3,4 @@
 
 import flask_featureflags
 
-__version__ = '15.9.1'
+__version__ = '15.10.0'

dmutils/documents.py

@@ -140,6 +140,20 @@ def file_is_open_document_format(file_object):
     ]
 
 
+def file_is_pdf(file_object):
+    """Checks file extension as being PDF."""
+    return get_extension(file_object.filename) in [
+        ".pdf", ".pda"
+    ]
+
+
+def file_is_zip(file_object):
+    """Checks file extension as being ZIP."""
+    return get_extension(file_object.filename) in [
+        ".zip"
+    ]
+
+
 def generate_file_name(framework_slug, bucket_short_name, supplier_id, service_id, field, filename, suffix=None):
     if suffix is None:
         suffix = default_file_suffix()
@@ -191,6 +205,7 @@ def get_agreement_document_path(framework_slug, supplier_id, supplier_name, docu
 
 
 def sanitise_supplier_name(supplier_name):
+    """Replace ampersands with 'and' and spaces with a single underscore."""
     sanitised_supplier_name = supplier_name.strip().replace(' ', '_').replace('&', 'and')
     for bad_char in BAD_SUPPLIER_NAME_CHARACTERS:
         sanitised_supplier_name = sanitised_supplier_name.replace(bad_char, '')

tests/test_documents.py

@@ -16,7 +16,7 @@
     validate_documents,
     upload_document, upload_service_documents,
     get_signed_url, get_agreement_document_path,
-    sanitise_supplier_name)
+    sanitise_supplier_name, file_is_pdf, file_is_zip)
 
 
 class TestGenerateFilename(unittest.TestCase):
@@ -71,6 +71,14 @@ def test_file_is_open_document_format(self):
     def test_file_is_not_open_document_format(self):
         self.assertFalse(file_is_open_document_format(mock_file('file1.doc', 1)))
 
+    def test_file_is_pdf(self):
+        self.assertTrue(file_is_pdf(mock_file('file.pdf', 1)))
+        self.assertFalse(file_is_pdf(mock_file('file.doc', 1)))
+
+    def test_file_is_zip(self):
+        self.assertTrue(file_is_zip(mock_file('file.zip', 1)))
+        self.assertFalse(file_is_zip(mock_file('file.sit', 1)))
+
     def test_validate_documents(self):
         self.assertEqual(
             validate_documents({'file1': mock_file('file1.pdf', 1)}),