32 lines
882 B
Python
32 lines
882 B
Python
import pytest
|
|
|
|
from reflex.components.datadisplay.code import CodeBlock
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
"theme, expected", [("light", "one-light"), ("dark", "one-dark")]
|
|
)
|
|
def test_code_light_dark_theme(theme, expected):
|
|
code_block = CodeBlock.create(theme=theme)
|
|
|
|
assert code_block.theme._var_name == expected # type: ignore
|
|
|
|
|
|
def generate_custom_code(language, expected_case):
|
|
return f"SyntaxHighlighter.registerLanguage('{language}', {expected_case})"
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
"language, expected_case",
|
|
[
|
|
("python", "python"),
|
|
("firestore-security-rules", "firestoreSecurityRules"),
|
|
("typescript", "typescript"),
|
|
],
|
|
)
|
|
def test_get_custom_code(language, expected_case):
|
|
code_block = CodeBlock.create(language=language)
|
|
assert code_block._get_custom_code() == generate_custom_code(
|
|
language, expected_case
|
|
)
|