mirror of https://github.com/zulip/zulip.git
119 lines
4.5 KiB
Python
119 lines
4.5 KiB
Python
from typing import Dict
|
|
|
|
import orjson
|
|
|
|
from zerver.lib.actions import do_add_realm_playground
|
|
from zerver.lib.test_classes import ZulipTestCase
|
|
from zerver.models import RealmPlayground, get_realm
|
|
|
|
|
|
class RealmPlaygroundTests(ZulipTestCase):
|
|
def json_serialize(self, payload: Dict[str, str]) -> Dict[str, str]:
|
|
payload["url_prefix"] = orjson.dumps(payload["url_prefix"]).decode()
|
|
payload["pygments_language"] = orjson.dumps(payload["pygments_language"]).decode()
|
|
return payload
|
|
|
|
def test_create_one_playground_entry(self) -> None:
|
|
iago = self.example_user("iago")
|
|
|
|
payload = {
|
|
"name": "Python playground",
|
|
"pygments_language": "Python",
|
|
"url_prefix": "https://python.example.com",
|
|
}
|
|
# Now send a POST request to the API endpoint.
|
|
resp = self.api_post(iago, "/json/realm/playgrounds", self.json_serialize(payload))
|
|
self.assert_json_success(resp)
|
|
|
|
# Check if the actual object exists
|
|
realm = get_realm("zulip")
|
|
self.assertTrue(
|
|
RealmPlayground.objects.filter(realm=realm, name="Python playground").exists()
|
|
)
|
|
|
|
def test_create_multiple_playgrounds_for_same_language(self) -> None:
|
|
iago = self.example_user("iago")
|
|
|
|
data = [
|
|
{
|
|
"name": "Python playground 1",
|
|
"pygments_language": "Python",
|
|
"url_prefix": "https://python.example.com",
|
|
},
|
|
{
|
|
"name": "Python playground 2",
|
|
"pygments_language": "Python",
|
|
"url_prefix": "https://python2.example.com",
|
|
},
|
|
]
|
|
for payload in data:
|
|
resp = self.api_post(iago, "/json/realm/playgrounds", self.json_serialize(payload))
|
|
self.assert_json_success(resp)
|
|
|
|
realm = get_realm("zulip")
|
|
self.assertTrue(
|
|
RealmPlayground.objects.filter(realm=realm, name="Python playground 1").exists()
|
|
)
|
|
self.assertTrue(
|
|
RealmPlayground.objects.filter(realm=realm, name="Python playground 2").exists()
|
|
)
|
|
|
|
def test_invalid_params(self) -> None:
|
|
iago = self.example_user("iago")
|
|
|
|
payload = {
|
|
"name": "Invalid URL",
|
|
"pygments_language": "Python",
|
|
"url_prefix": "https://invalid-url",
|
|
}
|
|
resp = self.api_post(iago, "/json/realm/playgrounds", self.json_serialize(payload))
|
|
self.assert_json_error(resp, "url_prefix is not a URL")
|
|
|
|
payload["url_prefix"] = "https://python.example.com"
|
|
payload["pygments_language"] = "a$b$c"
|
|
resp = self.api_post(iago, "/json/realm/playgrounds", self.json_serialize(payload))
|
|
self.assert_json_error(resp, "Invalid characters in pygments language")
|
|
|
|
def test_create_already_existing_playground(self) -> None:
|
|
iago = self.example_user("iago")
|
|
|
|
payload = {
|
|
"name": "Python playground",
|
|
"pygments_language": "Python",
|
|
"url_prefix": "https://python.example.com",
|
|
}
|
|
serialized_payload = self.json_serialize(payload)
|
|
resp = self.api_post(iago, "/json/realm/playgrounds", serialized_payload)
|
|
self.assert_json_success(resp)
|
|
|
|
resp = self.api_post(iago, "/json/realm/playgrounds", serialized_payload)
|
|
self.assert_json_error(resp, "Realm playground with this Realm and Name already exists.")
|
|
|
|
def test_not_realm_admin(self) -> None:
|
|
hamlet = self.example_user("hamlet")
|
|
|
|
resp = self.api_post(hamlet, "/json/realm/playgrounds")
|
|
self.assert_json_error(resp, "Must be an organization administrator")
|
|
|
|
resp = self.api_delete(hamlet, "/json/realm/playgrounds/1")
|
|
self.assert_json_error(resp, "Must be an organization administrator")
|
|
|
|
def test_delete_realm_playground(self) -> None:
|
|
iago = self.example_user("iago")
|
|
realm = get_realm("zulip")
|
|
|
|
playground_info = dict(
|
|
name="Python playground",
|
|
pygments_language="Python",
|
|
url_prefix="https://python.example.com",
|
|
)
|
|
playground_id = do_add_realm_playground(realm, **playground_info)
|
|
self.assertTrue(RealmPlayground.objects.filter(name="Python playground").exists())
|
|
|
|
result = self.api_delete(iago, f"/json/realm/playgrounds/{playground_id + 1}")
|
|
self.assert_json_error(result, "Invalid playground")
|
|
|
|
result = self.api_delete(iago, f"/json/realm/playgrounds/{playground_id}")
|
|
self.assert_json_success(result)
|
|
self.assertFalse(RealmPlayground.objects.filter(name="Python").exists())
|