3719 adding some more tests

testapp/requirements.txt

@@ -1,4 +1,7 @@
-Django>=1.11
+Django>=3.2.23,<=4.2.9
+coverage
+pyclamd
+six
 coverage==3.7.1
 python-magic>=0.4.15
 python-magic-bin>=0.4.14

testapp/settings.py

@@ -13,6 +13,7 @@
     'django.contrib.messages',
     'django.contrib.staticfiles',
     'attachments',
+    'bootstrap',
     'testapp',
 ]
 
@@ -26,6 +27,24 @@
     'django.middleware.clickjacking.XFrameOptionsMiddleware',
 ]
 
+TEMPLATES = [
+    {
+        'BACKEND': 'django.template.backends.django.DjangoTemplates',
+        'OPTIONS': {
+            'context_processors': [
+                'django.contrib.auth.context_processors.auth',
+                'django.template.context_processors.static',
+                'django.template.context_processors.request',
+                'django.contrib.messages.context_processors.messages',
+            ],
+            'loaders': [
+                'django.template.loaders.filesystem.Loader',
+                'django.template.loaders.app_directories.Loader',
+            ],
+        },
+    },
+]
+
 ROOT_URLCONF = 'testapp.urls'
 
 WSGI_APPLICATION = 'testapp.wsgi.application'

testapp/tests.py

@@ -5,6 +5,7 @@
 from attachments.utils import session, url_filename
 
 from .models import Document
+from .utils import get_storage
 
 import io
 
@@ -41,3 +42,37 @@ def test_upload(self):
         upload = sess.uploads.get()
         self.assertEqual(upload.file_name, att.name)
         self.assertEqual(upload.file_size, len(att_data))
+
+    def test_delete(self):
+        att_data = b'some data'
+        request = RequestFactory().get('/test/page/')
+        sess = session(request)
+        att = io.BytesIO(att_data)
+        att.name = 'testfile'
+        response = self.client.post('/attachments/%s/' % sess.uuid, {'attachment': att})
+        upload = sess.uploads.get()
+        response = self.client.post(f'/attachments/delete/upload/{sess.uuid}/{upload.upload_id}', {'attachment': att})
+        self.assertEqual(response.json(), {'ok': True})
+
+    def test_download(self):
+        storage = get_storage()
+        att_data = b'some data'
+        request = RequestFactory().get('/test/page/')
+        sess = session(request)
+        att = io.BytesIO(att_data)
+        att.name = 'testfile'
+        response = self.client.post(f'/attachments/download/{att.id}/{att.name}', {'attachment': att})
+        self.assertEqual(
+            response.json(), 
+            {'X-Sendfile': att.file_path,
+             'Content-Length': storage.size(att.file_path),
+             'Content-Disposition': 'attachment; filename="%s"' % att.name})
+
+    def test_update_attachment(self):
+        att_data = b'some data'
+        request = RequestFactory().get('/test/page/')
+        sess = session(request)
+        att = io.BytesIO(att_data)
+        att.name = 'testfile'
+        response = self.client.post(f'/attachments/update/{att.id}', {'attachment': att})
+        self.assertEqual(response.json(), {'ok': True})