mirror of https://gitlab.com/litecord/litecord.git
92 lines
2.0 KiB
Python
92 lines
2.0 KiB
Python
"""
|
|
|
|
Litecord
|
|
Copyright (C) 2018-2019 Luna Mendes
|
|
|
|
This program is free software: you can redistribute it and/or modify
|
|
it under the terms of the GNU General Public License as published by
|
|
the Free Software Foundation, version 3 of the License.
|
|
|
|
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 General Public License for more details.
|
|
|
|
You should have received a copy of the GNU General Public License
|
|
along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
|
|
"""
|
|
|
|
from litecord.schemas import validate
|
|
from litecord.embed.schemas import EMBED_OBJECT
|
|
from litecord.embed.sanitizer import path_exists
|
|
|
|
|
|
def validate_embed(embed):
|
|
return validate(embed, EMBED_OBJECT)
|
|
|
|
|
|
def valid(embed: dict):
|
|
try:
|
|
validate_embed(embed)
|
|
return True
|
|
except Exception:
|
|
return False
|
|
|
|
|
|
def invalid(embed):
|
|
return not valid(embed)
|
|
|
|
|
|
def test_empty_embed():
|
|
assert valid({})
|
|
|
|
|
|
def test_basic_embed():
|
|
assert valid(
|
|
{
|
|
"title": "test",
|
|
"description": "acab",
|
|
"url": "https://www.w3.org",
|
|
"color": 123,
|
|
}
|
|
)
|
|
|
|
|
|
def test_footer_embed():
|
|
assert invalid({"footer": {}})
|
|
|
|
assert valid({"title": "test", "footer": {"text": "abcdef"}})
|
|
|
|
|
|
def test_image():
|
|
assert invalid({"image": {}})
|
|
|
|
assert valid({"image": {"url": "https://www.w3.org"}})
|
|
|
|
|
|
def test_author():
|
|
assert invalid({"author": {"name": ""}})
|
|
|
|
assert valid({"author": {"name": "abcdef"}})
|
|
|
|
|
|
def test_fields():
|
|
assert valid(
|
|
{
|
|
"fields": [
|
|
{"name": "a", "value": "b"},
|
|
{"name": "c", "value": "d", "inline": False},
|
|
]
|
|
}
|
|
)
|
|
|
|
assert invalid({"fields": [{"name": "a"}]})
|
|
|
|
|
|
def test_path_exists():
|
|
"""Test the path_exists() function for embed sanitization."""
|
|
assert path_exists({"a": {"b": 2}}, "a.b")
|
|
assert path_exists({"a": "b"}, "a")
|
|
assert not path_exists({"a": "b"}, "a.b")
|