mirror of
https://github.com/langgenius/dify.git
synced 2026-10-04 13:47:55 +08:00
Byron.wangandautofix-ci[bot]
c1e94f8ee7
Co-authored-by: autofix-ci[bot] <114827586+autofix-ci[bot]@users.noreply.github.com>
32 lines
823 B
Python
32 lines
823 B
Python
import pytest
|
|
|
|
from services.entities.feature_entities import LicenseLimitationModel
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
("enabled", "size", "limit", "required", "expected"),
|
|
[
|
|
(False, 5, 10, 3, True),
|
|
(False, 5, 10, 10, True),
|
|
(True, 5, 0, 3, True),
|
|
(True, 5, 0, 100, True),
|
|
(True, 5, 10, 3, True),
|
|
(True, 5, 10, 5, True),
|
|
(True, 5, 10, 1, True),
|
|
(True, 8, 10, 3, False),
|
|
(True, 8, 10, 2, True),
|
|
(True, 8, 10, 1, True),
|
|
(True, 7, 10, 3, True),
|
|
],
|
|
)
|
|
def test_license_limitation_availability(
|
|
enabled: bool,
|
|
size: int,
|
|
limit: int,
|
|
required: int,
|
|
expected: bool,
|
|
) -> None:
|
|
limitation = LicenseLimitationModel(enabled=enabled, size=size, limit=limit)
|
|
|
|
assert limitation.is_available(required) is expected
|