Ticket #465: 465-unit-tests-v1.patch

new file mediagoblin/tests/test_attachments.py

@@ +1 @@
+# GNU MediaGoblin -- federated, autonomous media hosting
+# Copyright (C) 2011, 2012 MediaGoblin contributors.  See AUTHORS.
+#
+# This program is free software: you can redistribute it and/or modify
+# it under the terms of the GNU Affero General Public License as published by
+# the Free Software Foundation, either version 3 of the License, or
+# (at your option) any later version.
+#
+# This program is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU Affero General Public License for more details.
+#
+# You should have received a copy of the GNU Affero General Public License
+# along with this program.  If not, see <http://www.gnu.org/licenses/>.
+
+import pytest
+from mediagoblin import mg_globals, auth, storage
+from mediagoblin.db.base import Session
+from mediagoblin.db.models import User, MediaEntry, MediaAttachmentFile
+from mediagoblin.tests.tools import fixture_add_user, fixture_media_entry
+from datetime import datetime
+
+class TestAttachments:
+    @pytest.fixture(autouse=True)
+    def setup(self, test_app):
+        self.test_app = test_app
+        self.user_owner = fixture_add_user(u"owner",
+            privileges=[u"active"])
+        fixture_add_user(u"otheruser",
+            privileges=[u"active"])
+        fixture_add_user(u"admin",
+            privileges=[u"admin",u"active"])
+    
+    def login(self, username):
+        self.test_app.post(
+            "/auth/login/", {
+                "username": username,
+                "password": "toast"})
+            
+    def logout(self):
+        self.test_app.get("/auth/logout/")
+        
+    def test_attachment_file_delete(self, test_app):
+        media_entry = fixture_media_entry(uploader=self.user_owner.id,
+            state=u"processed")
+        Session.add(media_entry)
+        
+        this_storage = storage.storage_system_from_config(
+        {"base_url": "http://example.org/moodia/",
+         "base_dir": "/tmp/"})
+        mg_globals.public_store = this_storage
+        
+        attachment_path = ["attachment_to_delete.txt"]
+        attachment_file = this_storage.get_file(
+                attachment_path, "wb")
+        attachment_file.write("Testing MediaAttachmentFile.delete()");
+        media_entry.attachment_files.append(dict(
+                    name="Attachment to delete",
+                    filepath=attachment_path,
+                    created=datetime.utcnow(),
+                    ))
+               
+        assert MediaAttachmentFile.query.filter_by(name="Attachment to delete").count() == 1
+        assert this_storage.file_exists(attachment_path) == True
+                
+        attachment = MediaAttachmentFile.query.filter_by(name="Attachment to delete").first()
+        attachment.delete();
+        
+        assert MediaAttachmentFile.query.filter_by(name="Attachment to delete").count() == 0
+        assert this_storage.file_exists(attachment_path) == False
+        
+    def test_confirm_delete_attachment(self, test_app):
+        self.login(u"owner")
+        media_entry = fixture_media_entry(uploader=self.user_owner.id,
+            state=u"processed")
+        Session.add(media_entry)
+        
+        media_entry.attachment_files.append(dict(
+            name=u"some name",
+            filepath=[u"does", u"not", u"exist"],
+            ))
+        cda_slug1 = "/u/{username}/m/{media_id}/confirm-delete-attachment/1/".format(
+            username = str(self.user_owner.username),
+            media_id = str(media_entry.id))
+        cda_slug2 = "/u/{username}/m/{media_id}/confirm-delete-attachment/2/".format(
+            username = str(self.user_owner.username),
+            media_id = str(media_entry.id))
+
+        response = test_app.get(cda_slug1)
+        assert response.status == "200 OK"
+                      
+        response = test_app.get(cda_slug2, expect_errors=True)
+        assert response.status == "404 NOT FOUND"
+
+        self.logout()
+        
+        response = test_app.get(cda_slug1)
+        assert response.status == "302 FOUND"
+        
+        self.login(u"otheruser")
+        
+        response = test_app.get(cda_slug1, expect_errors=True)
+        assert response.status == "403 FORBIDDEN"
+        
+        self.logout()
+        self.login(u"admin")
+        
+        Session.add(media_entry)
+        media_entry.attachment_files.append(dict(
+            name=u"some name",
+            filepath=[u"does", u"not", u"exist"],
+            ))
+        cda_slug1 = "/u/{username}/m/{media_id}/confirm-delete-attachment/1/".format(
+            username = str(self.user_owner.username),
+            media_id = str(media_entry.id))
+        
+        response = test_app.get(cda_slug1)
+        assert response.status == "200 OK"