From 570ffa2c48b44b7dd2e3d0a87a09df44ac8943b5 Mon Sep 17 00:00:00 2001
From: Jake Howard <git@theorangeone.net>
Date: Tue, 13 Sep 2016 21:30:41 +0100
Subject: [PATCH] Add basic tests

---
 project/files/tests.py | 22 +++++++++++++++++++++-
 requirements.txt       |  1 +
 2 files changed, 22 insertions(+), 1 deletion(-)

diff --git a/project/files/tests.py b/project/files/tests.py
index 7ce503c..0fc5fcf 100644
--- a/project/files/tests.py
+++ b/project/files/tests.py
@@ -1,3 +1,23 @@
 from django.test import TestCase
+from model_mommy import mommy
+from .models import SharedFile
+from django.core.files.uploadedfile import SimpleUploadedFile
 
-# Create your tests here.
+
+class SharedFileTestCase(TestCase):
+    def setUp(self):
+        asset = SimpleUploadedFile(
+            name='902598_test_file.txt',
+            content='',
+            content_type='text/plain'
+        )
+        self.file = mommy.make(SharedFile, file=asset)
+
+    def test_str(self):
+        self.assertEqual(str(self.file), self.file.get_original_filename())
+
+    def test_get_original_filename(self):
+        self.assertIn('test_file', self.file.get_original_filename())
+
+    def test_absolute_url(self):
+        self.assertEqual(self.file.get_absolute_url(), '/files/{0}/'.format(self.file.short_id))
diff --git a/requirements.txt b/requirements.txt
index def1277..d2914d7 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -6,6 +6,7 @@ django-material==0.8.0
 django-shortuuidfield==0.1.3
 django==1.8.13
 flake8==2.5.4
+model-mommy==1.3.0
 opbeat==3.3.4
 psycopg2==2.6.1
 waitress==0.9.0