123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868 |
- # Copyright 2020 Dirk Klimpel
- # Copyright 2021 The Matrix.org Foundation C.I.C.
- #
- # Licensed under the Apache License, Version 2.0 (the "License");
- # you may not use this file except in compliance with the License.
- # You may obtain a copy of the License at
- #
- # http://www.apache.org/licenses/LICENSE-2.0
- #
- # Unless required by applicable law or agreed to in writing, software
- # distributed under the License is distributed on an "AS IS" BASIS,
- # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
- # See the License for the specific language governing permissions and
- # limitations under the License.
- import os
- from typing import Dict
- from parameterized import parameterized
- from twisted.test.proto_helpers import MemoryReactor
- from twisted.web.resource import Resource
- import synapse.rest.admin
- from synapse.api.errors import Codes
- from synapse.media.filepath import MediaFilePaths
- from synapse.rest.client import login, profile, room
- from synapse.server import HomeServer
- from synapse.util import Clock
- from tests import unittest
- from tests.test_utils import SMALL_PNG
- VALID_TIMESTAMP = 1609459200000 # 2021-01-01 in milliseconds
- INVALID_TIMESTAMP_IN_S = 1893456000 # 2030-01-01 in seconds
- class _AdminMediaTests(unittest.HomeserverTestCase):
- servlets = [
- synapse.rest.admin.register_servlets,
- synapse.rest.admin.register_servlets_for_media_repo,
- login.register_servlets,
- ]
- def create_resource_dict(self) -> Dict[str, Resource]:
- resources = super().create_resource_dict()
- resources["/_matrix/media"] = self.hs.get_media_repository_resource()
- return resources
- class DeleteMediaByIDTestCase(_AdminMediaTests):
- def prepare(self, reactor: MemoryReactor, clock: Clock, hs: HomeServer) -> None:
- self.server_name = hs.hostname
- self.admin_user = self.register_user("admin", "pass", admin=True)
- self.admin_user_tok = self.login("admin", "pass")
- self.filepaths = MediaFilePaths(hs.config.media.media_store_path)
- def test_no_auth(self) -> None:
- """
- Try to delete media without authentication.
- """
- url = "/_synapse/admin/v1/media/%s/%s" % (self.server_name, "12345")
- channel = self.make_request("DELETE", url, b"{}")
- self.assertEqual(
- 401,
- channel.code,
- msg=channel.json_body,
- )
- self.assertEqual(Codes.MISSING_TOKEN, channel.json_body["errcode"])
- def test_requester_is_no_admin(self) -> None:
- """
- If the user is not a server admin, an error is returned.
- """
- self.other_user = self.register_user("user", "pass")
- self.other_user_token = self.login("user", "pass")
- url = "/_synapse/admin/v1/media/%s/%s" % (self.server_name, "12345")
- channel = self.make_request(
- "DELETE",
- url,
- access_token=self.other_user_token,
- )
- self.assertEqual(403, channel.code, msg=channel.json_body)
- self.assertEqual(Codes.FORBIDDEN, channel.json_body["errcode"])
- def test_media_does_not_exist(self) -> None:
- """
- Tests that a lookup for a media that does not exist returns a 404
- """
- url = "/_synapse/admin/v1/media/%s/%s" % (self.server_name, "12345")
- channel = self.make_request(
- "DELETE",
- url,
- access_token=self.admin_user_tok,
- )
- self.assertEqual(404, channel.code, msg=channel.json_body)
- self.assertEqual(Codes.NOT_FOUND, channel.json_body["errcode"])
- def test_media_is_not_local(self) -> None:
- """
- Tests that a lookup for a media that is not a local returns a 400
- """
- url = "/_synapse/admin/v1/media/%s/%s" % ("unknown_domain", "12345")
- channel = self.make_request(
- "DELETE",
- url,
- access_token=self.admin_user_tok,
- )
- self.assertEqual(400, channel.code, msg=channel.json_body)
- self.assertEqual("Can only delete local media", channel.json_body["error"])
- def test_delete_media(self) -> None:
- """
- Tests that delete a media is successfully
- """
- # Upload some media into the room
- response = self.helper.upload_media(
- SMALL_PNG,
- tok=self.admin_user_tok,
- expect_code=200,
- )
- # Extract media ID from the response
- server_and_media_id = response["content_uri"][6:] # Cut off 'mxc://'
- server_name, media_id = server_and_media_id.split("/")
- self.assertEqual(server_name, self.server_name)
- # Attempt to access media
- channel = self.make_request(
- "GET",
- f"/_matrix/media/v3/download/{server_and_media_id}",
- shorthand=False,
- access_token=self.admin_user_tok,
- )
- # Should be successful
- self.assertEqual(
- 200,
- channel.code,
- msg=(
- "Expected to receive a 200 on accessing media: %s" % server_and_media_id
- ),
- )
- # Test if the file exists
- local_path = self.filepaths.local_media_filepath(media_id)
- self.assertTrue(os.path.exists(local_path))
- url = "/_synapse/admin/v1/media/%s/%s" % (self.server_name, media_id)
- # Delete media
- channel = self.make_request(
- "DELETE",
- url,
- access_token=self.admin_user_tok,
- )
- self.assertEqual(200, channel.code, msg=channel.json_body)
- self.assertEqual(1, channel.json_body["total"])
- self.assertEqual(
- media_id,
- channel.json_body["deleted_media"][0],
- )
- # Attempt to access media
- channel = self.make_request(
- "GET",
- f"/_matrix/media/v3/download/{server_and_media_id}",
- shorthand=False,
- access_token=self.admin_user_tok,
- )
- self.assertEqual(
- 404,
- channel.code,
- msg=(
- "Expected to receive a 404 on accessing deleted media: %s"
- % server_and_media_id
- ),
- )
- # Test if the file is deleted
- self.assertFalse(os.path.exists(local_path))
- class DeleteMediaByDateSizeTestCase(_AdminMediaTests):
- servlets = [
- synapse.rest.admin.register_servlets,
- synapse.rest.admin.register_servlets_for_media_repo,
- login.register_servlets,
- profile.register_servlets,
- room.register_servlets,
- ]
- def prepare(self, reactor: MemoryReactor, clock: Clock, hs: HomeServer) -> None:
- self.media_repo = hs.get_media_repository_resource()
- self.server_name = hs.hostname
- self.admin_user = self.register_user("admin", "pass", admin=True)
- self.admin_user_tok = self.login("admin", "pass")
- self.filepaths = MediaFilePaths(hs.config.media.media_store_path)
- self.url = "/_synapse/admin/v1/media/delete"
- self.legacy_url = "/_synapse/admin/v1/media/%s/delete" % self.server_name
- # Move clock up to somewhat realistic time
- self.reactor.advance(1000000000)
- def test_no_auth(self) -> None:
- """
- Try to delete media without authentication.
- """
- channel = self.make_request("POST", self.url, b"{}")
- self.assertEqual(401, channel.code, msg=channel.json_body)
- self.assertEqual(Codes.MISSING_TOKEN, channel.json_body["errcode"])
- def test_requester_is_no_admin(self) -> None:
- """
- If the user is not a server admin, an error is returned.
- """
- self.other_user = self.register_user("user", "pass")
- self.other_user_token = self.login("user", "pass")
- channel = self.make_request(
- "POST",
- self.url,
- access_token=self.other_user_token,
- )
- self.assertEqual(403, channel.code, msg=channel.json_body)
- self.assertEqual(Codes.FORBIDDEN, channel.json_body["errcode"])
- def test_media_is_not_local(self) -> None:
- """
- Tests that a lookup for media that is not local returns a 400
- """
- url = "/_synapse/admin/v1/media/%s/delete" % "unknown_domain"
- channel = self.make_request(
- "POST",
- url + f"?before_ts={VALID_TIMESTAMP}",
- access_token=self.admin_user_tok,
- )
- self.assertEqual(400, channel.code, msg=channel.json_body)
- self.assertEqual("Can only delete local media", channel.json_body["error"])
- def test_missing_parameter(self) -> None:
- """
- If the parameter `before_ts` is missing, an error is returned.
- """
- channel = self.make_request(
- "POST",
- self.url,
- access_token=self.admin_user_tok,
- )
- self.assertEqual(400, channel.code, msg=channel.json_body)
- self.assertEqual(Codes.MISSING_PARAM, channel.json_body["errcode"])
- self.assertEqual(
- "Missing integer query parameter 'before_ts'", channel.json_body["error"]
- )
- def test_invalid_parameter(self) -> None:
- """
- If parameters are invalid, an error is returned.
- """
- channel = self.make_request(
- "POST",
- self.url + "?before_ts=-1234",
- access_token=self.admin_user_tok,
- )
- self.assertEqual(400, channel.code, msg=channel.json_body)
- self.assertEqual(Codes.INVALID_PARAM, channel.json_body["errcode"])
- self.assertEqual(
- "Query parameter before_ts must be a positive integer.",
- channel.json_body["error"],
- )
- channel = self.make_request(
- "POST",
- self.url + f"?before_ts={INVALID_TIMESTAMP_IN_S}",
- access_token=self.admin_user_tok,
- )
- self.assertEqual(400, channel.code, msg=channel.json_body)
- self.assertEqual(Codes.INVALID_PARAM, channel.json_body["errcode"])
- self.assertEqual(
- "Query parameter before_ts you provided is from the year 1970. "
- + "Double check that you are providing a timestamp in milliseconds.",
- channel.json_body["error"],
- )
- channel = self.make_request(
- "POST",
- self.url + f"?before_ts={VALID_TIMESTAMP}&size_gt=-1234",
- access_token=self.admin_user_tok,
- )
- self.assertEqual(400, channel.code, msg=channel.json_body)
- self.assertEqual(Codes.INVALID_PARAM, channel.json_body["errcode"])
- self.assertEqual(
- "Query parameter size_gt must be a string representing a positive integer.",
- channel.json_body["error"],
- )
- channel = self.make_request(
- "POST",
- self.url + f"?before_ts={VALID_TIMESTAMP}&keep_profiles=not_bool",
- access_token=self.admin_user_tok,
- )
- self.assertEqual(400, channel.code, msg=channel.json_body)
- self.assertEqual(Codes.INVALID_PARAM, channel.json_body["errcode"])
- self.assertEqual(
- "Boolean query parameter 'keep_profiles' must be one of ['true', 'false']",
- channel.json_body["error"],
- )
- @parameterized.expand([(True,), (False,)])
- def test_delete_media_never_accessed(self, use_legacy_url: bool) -> None:
- """
- Tests that media deleted if it is older than `before_ts` and never accessed
- `last_access_ts` is `NULL` and `created_ts` < `before_ts`
- """
- url = self.legacy_url if use_legacy_url else self.url
- # upload and do not access
- server_and_media_id = self._create_media()
- self.pump(1.0)
- # test that the file exists
- media_id = server_and_media_id.split("/")[1]
- local_path = self.filepaths.local_media_filepath(media_id)
- self.assertTrue(os.path.exists(local_path))
- # timestamp after upload/create
- now_ms = self.clock.time_msec()
- channel = self.make_request(
- "POST",
- url + "?before_ts=" + str(now_ms),
- access_token=self.admin_user_tok,
- )
- self.assertEqual(200, channel.code, msg=channel.json_body)
- self.assertEqual(1, channel.json_body["total"])
- self.assertEqual(
- media_id,
- channel.json_body["deleted_media"][0],
- )
- self._access_media(server_and_media_id, False)
- def test_keep_media_by_date(self) -> None:
- """
- Tests that media is not deleted if it is newer than `before_ts`
- """
- # timestamp before upload
- now_ms = self.clock.time_msec()
- server_and_media_id = self._create_media()
- self._access_media(server_and_media_id)
- channel = self.make_request(
- "POST",
- self.url + "?before_ts=" + str(now_ms),
- access_token=self.admin_user_tok,
- )
- self.assertEqual(200, channel.code, msg=channel.json_body)
- self.assertEqual(0, channel.json_body["total"])
- self._access_media(server_and_media_id)
- # timestamp after upload
- now_ms = self.clock.time_msec()
- channel = self.make_request(
- "POST",
- self.url + "?before_ts=" + str(now_ms),
- access_token=self.admin_user_tok,
- )
- self.assertEqual(200, channel.code, msg=channel.json_body)
- self.assertEqual(1, channel.json_body["total"])
- self.assertEqual(
- server_and_media_id.split("/")[1],
- channel.json_body["deleted_media"][0],
- )
- self._access_media(server_and_media_id, False)
- def test_keep_media_by_size(self) -> None:
- """
- Tests that media is not deleted if its size is smaller than or equal
- to `size_gt`
- """
- server_and_media_id = self._create_media()
- self._access_media(server_and_media_id)
- now_ms = self.clock.time_msec()
- channel = self.make_request(
- "POST",
- self.url + "?before_ts=" + str(now_ms) + "&size_gt=67",
- access_token=self.admin_user_tok,
- )
- self.assertEqual(200, channel.code, msg=channel.json_body)
- self.assertEqual(0, channel.json_body["total"])
- self._access_media(server_and_media_id)
- now_ms = self.clock.time_msec()
- channel = self.make_request(
- "POST",
- self.url + "?before_ts=" + str(now_ms) + "&size_gt=66",
- access_token=self.admin_user_tok,
- )
- self.assertEqual(200, channel.code, msg=channel.json_body)
- self.assertEqual(1, channel.json_body["total"])
- self.assertEqual(
- server_and_media_id.split("/")[1],
- channel.json_body["deleted_media"][0],
- )
- self._access_media(server_and_media_id, False)
- def test_keep_media_by_user_avatar(self) -> None:
- """
- Tests that we do not delete media if is used as a user avatar
- Tests parameter `keep_profiles`
- """
- server_and_media_id = self._create_media()
- self._access_media(server_and_media_id)
- # set media as avatar
- channel = self.make_request(
- "PUT",
- "/profile/%s/avatar_url" % (self.admin_user,),
- content={"avatar_url": "mxc://%s" % (server_and_media_id,)},
- access_token=self.admin_user_tok,
- )
- self.assertEqual(200, channel.code, msg=channel.json_body)
- now_ms = self.clock.time_msec()
- channel = self.make_request(
- "POST",
- self.url + "?before_ts=" + str(now_ms) + "&keep_profiles=true",
- access_token=self.admin_user_tok,
- )
- self.assertEqual(200, channel.code, msg=channel.json_body)
- self.assertEqual(0, channel.json_body["total"])
- self._access_media(server_and_media_id)
- now_ms = self.clock.time_msec()
- channel = self.make_request(
- "POST",
- self.url + "?before_ts=" + str(now_ms) + "&keep_profiles=false",
- access_token=self.admin_user_tok,
- )
- self.assertEqual(200, channel.code, msg=channel.json_body)
- self.assertEqual(1, channel.json_body["total"])
- self.assertEqual(
- server_and_media_id.split("/")[1],
- channel.json_body["deleted_media"][0],
- )
- self._access_media(server_and_media_id, False)
- def test_keep_media_by_room_avatar(self) -> None:
- """
- Tests that we do not delete media if it is used as a room avatar
- Tests parameter `keep_profiles`
- """
- server_and_media_id = self._create_media()
- self._access_media(server_and_media_id)
- # set media as room avatar
- room_id = self.helper.create_room_as(self.admin_user, tok=self.admin_user_tok)
- channel = self.make_request(
- "PUT",
- "/rooms/%s/state/m.room.avatar" % (room_id,),
- content={"url": "mxc://%s" % (server_and_media_id,)},
- access_token=self.admin_user_tok,
- )
- self.assertEqual(200, channel.code, msg=channel.json_body)
- now_ms = self.clock.time_msec()
- channel = self.make_request(
- "POST",
- self.url + "?before_ts=" + str(now_ms) + "&keep_profiles=true",
- access_token=self.admin_user_tok,
- )
- self.assertEqual(200, channel.code, msg=channel.json_body)
- self.assertEqual(0, channel.json_body["total"])
- self._access_media(server_and_media_id)
- now_ms = self.clock.time_msec()
- channel = self.make_request(
- "POST",
- self.url + "?before_ts=" + str(now_ms) + "&keep_profiles=false",
- access_token=self.admin_user_tok,
- )
- self.assertEqual(200, channel.code, msg=channel.json_body)
- self.assertEqual(1, channel.json_body["total"])
- self.assertEqual(
- server_and_media_id.split("/")[1],
- channel.json_body["deleted_media"][0],
- )
- self._access_media(server_and_media_id, False)
- def _create_media(self) -> str:
- """
- Create a media and return media_id and server_and_media_id
- """
- # Upload some media into the room
- response = self.helper.upload_media(
- SMALL_PNG,
- tok=self.admin_user_tok,
- expect_code=200,
- )
- # Extract media ID from the response
- server_and_media_id = response["content_uri"][6:] # Cut off 'mxc://'
- server_name = server_and_media_id.split("/")[0]
- # Check that new media is a local and not remote
- self.assertEqual(server_name, self.server_name)
- return server_and_media_id
- def _access_media(
- self, server_and_media_id: str, expect_success: bool = True
- ) -> None:
- """
- Try to access a media and check the result
- """
- media_id = server_and_media_id.split("/")[1]
- local_path = self.filepaths.local_media_filepath(media_id)
- channel = self.make_request(
- "GET",
- f"/_matrix/media/v3/download/{server_and_media_id}",
- shorthand=False,
- access_token=self.admin_user_tok,
- )
- if expect_success:
- self.assertEqual(
- 200,
- channel.code,
- msg=(
- "Expected to receive a 200 on accessing media: %s"
- % server_and_media_id
- ),
- )
- # Test that the file exists
- self.assertTrue(os.path.exists(local_path))
- else:
- self.assertEqual(
- 404,
- channel.code,
- msg=(
- "Expected to receive a 404 on accessing deleted media: %s"
- % (server_and_media_id)
- ),
- )
- # Test that the file is deleted
- self.assertFalse(os.path.exists(local_path))
- class QuarantineMediaByIDTestCase(_AdminMediaTests):
- def prepare(self, reactor: MemoryReactor, clock: Clock, hs: HomeServer) -> None:
- self.store = hs.get_datastores().main
- self.server_name = hs.hostname
- self.admin_user = self.register_user("admin", "pass", admin=True)
- self.admin_user_tok = self.login("admin", "pass")
- # Upload some media into the room
- response = self.helper.upload_media(
- SMALL_PNG,
- tok=self.admin_user_tok,
- expect_code=200,
- )
- # Extract media ID from the response
- server_and_media_id = response["content_uri"][6:] # Cut off 'mxc://'
- self.media_id = server_and_media_id.split("/")[1]
- self.url = "/_synapse/admin/v1/media/%s/%s/%s"
- @parameterized.expand(["quarantine", "unquarantine"])
- def test_no_auth(self, action: str) -> None:
- """
- Try to protect media without authentication.
- """
- channel = self.make_request(
- "POST",
- self.url % (action, self.server_name, self.media_id),
- b"{}",
- )
- self.assertEqual(401, channel.code, msg=channel.json_body)
- self.assertEqual(Codes.MISSING_TOKEN, channel.json_body["errcode"])
- @parameterized.expand(["quarantine", "unquarantine"])
- def test_requester_is_no_admin(self, action: str) -> None:
- """
- If the user is not a server admin, an error is returned.
- """
- self.other_user = self.register_user("user", "pass")
- self.other_user_token = self.login("user", "pass")
- channel = self.make_request(
- "POST",
- self.url % (action, self.server_name, self.media_id),
- access_token=self.other_user_token,
- )
- self.assertEqual(403, channel.code, msg=channel.json_body)
- self.assertEqual(Codes.FORBIDDEN, channel.json_body["errcode"])
- def test_quarantine_media(self) -> None:
- """
- Tests that quarantining and remove from quarantine a media is successfully
- """
- media_info = self.get_success(self.store.get_local_media(self.media_id))
- assert media_info is not None
- self.assertFalse(media_info["quarantined_by"])
- # quarantining
- channel = self.make_request(
- "POST",
- self.url % ("quarantine", self.server_name, self.media_id),
- access_token=self.admin_user_tok,
- )
- self.assertEqual(200, channel.code, msg=channel.json_body)
- self.assertFalse(channel.json_body)
- media_info = self.get_success(self.store.get_local_media(self.media_id))
- assert media_info is not None
- self.assertTrue(media_info["quarantined_by"])
- # remove from quarantine
- channel = self.make_request(
- "POST",
- self.url % ("unquarantine", self.server_name, self.media_id),
- access_token=self.admin_user_tok,
- )
- self.assertEqual(200, channel.code, msg=channel.json_body)
- self.assertFalse(channel.json_body)
- media_info = self.get_success(self.store.get_local_media(self.media_id))
- assert media_info is not None
- self.assertFalse(media_info["quarantined_by"])
- def test_quarantine_protected_media(self) -> None:
- """
- Tests that quarantining from protected media fails
- """
- # protect
- self.get_success(self.store.mark_local_media_as_safe(self.media_id, safe=True))
- # verify protection
- media_info = self.get_success(self.store.get_local_media(self.media_id))
- assert media_info is not None
- self.assertTrue(media_info["safe_from_quarantine"])
- # quarantining
- channel = self.make_request(
- "POST",
- self.url % ("quarantine", self.server_name, self.media_id),
- access_token=self.admin_user_tok,
- )
- self.assertEqual(200, channel.code, msg=channel.json_body)
- self.assertFalse(channel.json_body)
- # verify that is not in quarantine
- media_info = self.get_success(self.store.get_local_media(self.media_id))
- assert media_info is not None
- self.assertFalse(media_info["quarantined_by"])
- class ProtectMediaByIDTestCase(_AdminMediaTests):
- def prepare(self, reactor: MemoryReactor, clock: Clock, hs: HomeServer) -> None:
- hs.get_media_repository_resource()
- self.store = hs.get_datastores().main
- self.admin_user = self.register_user("admin", "pass", admin=True)
- self.admin_user_tok = self.login("admin", "pass")
- # Upload some media into the room
- response = self.helper.upload_media(
- SMALL_PNG,
- tok=self.admin_user_tok,
- expect_code=200,
- )
- # Extract media ID from the response
- server_and_media_id = response["content_uri"][6:] # Cut off 'mxc://'
- self.media_id = server_and_media_id.split("/")[1]
- self.url = "/_synapse/admin/v1/media/%s/%s"
- @parameterized.expand(["protect", "unprotect"])
- def test_no_auth(self, action: str) -> None:
- """
- Try to protect media without authentication.
- """
- channel = self.make_request("POST", self.url % (action, self.media_id), b"{}")
- self.assertEqual(401, channel.code, msg=channel.json_body)
- self.assertEqual(Codes.MISSING_TOKEN, channel.json_body["errcode"])
- @parameterized.expand(["protect", "unprotect"])
- def test_requester_is_no_admin(self, action: str) -> None:
- """
- If the user is not a server admin, an error is returned.
- """
- self.other_user = self.register_user("user", "pass")
- self.other_user_token = self.login("user", "pass")
- channel = self.make_request(
- "POST",
- self.url % (action, self.media_id),
- access_token=self.other_user_token,
- )
- self.assertEqual(403, channel.code, msg=channel.json_body)
- self.assertEqual(Codes.FORBIDDEN, channel.json_body["errcode"])
- def test_protect_media(self) -> None:
- """
- Tests that protect and unprotect a media is successfully
- """
- media_info = self.get_success(self.store.get_local_media(self.media_id))
- assert media_info is not None
- self.assertFalse(media_info["safe_from_quarantine"])
- # protect
- channel = self.make_request(
- "POST",
- self.url % ("protect", self.media_id),
- access_token=self.admin_user_tok,
- )
- self.assertEqual(200, channel.code, msg=channel.json_body)
- self.assertFalse(channel.json_body)
- media_info = self.get_success(self.store.get_local_media(self.media_id))
- assert media_info is not None
- self.assertTrue(media_info["safe_from_quarantine"])
- # unprotect
- channel = self.make_request(
- "POST",
- self.url % ("unprotect", self.media_id),
- access_token=self.admin_user_tok,
- )
- self.assertEqual(200, channel.code, msg=channel.json_body)
- self.assertFalse(channel.json_body)
- media_info = self.get_success(self.store.get_local_media(self.media_id))
- assert media_info is not None
- self.assertFalse(media_info["safe_from_quarantine"])
- class PurgeMediaCacheTestCase(_AdminMediaTests):
- servlets = [
- synapse.rest.admin.register_servlets,
- synapse.rest.admin.register_servlets_for_media_repo,
- login.register_servlets,
- profile.register_servlets,
- room.register_servlets,
- ]
- def prepare(self, reactor: MemoryReactor, clock: Clock, hs: HomeServer) -> None:
- self.media_repo = hs.get_media_repository_resource()
- self.server_name = hs.hostname
- self.admin_user = self.register_user("admin", "pass", admin=True)
- self.admin_user_tok = self.login("admin", "pass")
- self.filepaths = MediaFilePaths(hs.config.media.media_store_path)
- self.url = "/_synapse/admin/v1/purge_media_cache"
- def test_no_auth(self) -> None:
- """
- Try to delete media without authentication.
- """
- channel = self.make_request("POST", self.url, b"{}")
- self.assertEqual(
- 401,
- channel.code,
- msg=channel.json_body,
- )
- self.assertEqual(Codes.MISSING_TOKEN, channel.json_body["errcode"])
- def test_requester_is_not_admin(self) -> None:
- """
- If the user is not a server admin, an error is returned.
- """
- self.other_user = self.register_user("user", "pass")
- self.other_user_token = self.login("user", "pass")
- channel = self.make_request(
- "POST",
- self.url,
- access_token=self.other_user_token,
- )
- self.assertEqual(403, channel.code, msg=channel.json_body)
- self.assertEqual(Codes.FORBIDDEN, channel.json_body["errcode"])
- def test_invalid_parameter(self) -> None:
- """
- If parameters are invalid, an error is returned.
- """
- channel = self.make_request(
- "POST",
- self.url + "?before_ts=-1234",
- access_token=self.admin_user_tok,
- )
- self.assertEqual(400, channel.code, msg=channel.json_body)
- self.assertEqual(Codes.INVALID_PARAM, channel.json_body["errcode"])
- self.assertEqual(
- "Query parameter before_ts must be a positive integer.",
- channel.json_body["error"],
- )
- channel = self.make_request(
- "POST",
- self.url + f"?before_ts={INVALID_TIMESTAMP_IN_S}",
- access_token=self.admin_user_tok,
- )
- self.assertEqual(400, channel.code, msg=channel.json_body)
- self.assertEqual(Codes.INVALID_PARAM, channel.json_body["errcode"])
- self.assertEqual(
- "Query parameter before_ts you provided is from the year 1970. "
- + "Double check that you are providing a timestamp in milliseconds.",
- channel.json_body["error"],
- )
|