From d3258e2da4ae8e7ce71ff39c07c2531a7612c01a Mon Sep 17 00:00:00 2001
From: Alex Yatskov <alex@foosoft.net>
Date: Tue, 8 May 2018 17:55:45 -0700
Subject: [PATCH] add media tests

---
 tests/test_media.py | 28 ++++++++++++++++++++++++++++
 1 file changed, 28 insertions(+)
 create mode 100755 tests/test_media.py

diff --git a/tests/test_media.py b/tests/test_media.py
new file mode 100755
index 0000000..7870fbc
--- /dev/null
+++ b/tests/test_media.py
@@ -0,0 +1,28 @@
+#!/usr/bin/env python
+
+import unittest
+import util
+
+
+class TestMedia(unittest.TestCase):
+    def runTest(self):
+        filename = '_test.txt'
+        data = 'test'
+
+        # storeMediaFile
+        util.invoke('storeMediaFile', filename='_test.txt', data=data)
+
+        # retrieveMediaFile (part 1)
+        media = util.invoke('retrieveMediaFile', filename=filename)
+        self.assertEqual(media, data)
+
+        # deleteMediaFile
+        util.invoke('deleteMediaFile', filename=filename)
+
+        # retrieveMediaFile (part 2)
+        media = util.invoke('retrieveMediaFile', filename=filename)
+        self.assertFalse(media)
+
+
+if __name__ == '__main__':
+    unittest.main()