dify

dify/api/tests/unit_tests/libs/test_custom_inputs.py

@@ -0,0 +1,68 @@
+"""Unit tests for custom input types."""
+
+import pytest
+
+from libs.custom_inputs import time_duration
+
+
+class TestTimeDuration:
+    """Test time_duration input validator."""
+
+    def test_valid_days(self):
+        """Test valid days format."""
+        result = time_duration("7d")
+        assert result == "7d"
+
+    def test_valid_hours(self):
+        """Test valid hours format."""
+        result = time_duration("4h")
+        assert result == "4h"
+
+    def test_valid_minutes(self):
+        """Test valid minutes format."""
+        result = time_duration("30m")
+        assert result == "30m"
+
+    def test_valid_seconds(self):
+        """Test valid seconds format."""
+        result = time_duration("30s")
+        assert result == "30s"
+
+    def test_uppercase_conversion(self):
+        """Test uppercase units are converted to lowercase."""
+        result = time_duration("7D")
+        assert result == "7d"
+
+        result = time_duration("4H")
+        assert result == "4h"
+
+    def test_invalid_format_no_unit(self):
+        """Test invalid format without unit."""
+        with pytest.raises(ValueError, match="Invalid time duration format"):
+            time_duration("7")
+
+    def test_invalid_format_wrong_unit(self):
+        """Test invalid format with wrong unit."""
+        with pytest.raises(ValueError, match="Invalid time duration format"):
+            time_duration("7days")
+
+        with pytest.raises(ValueError, match="Invalid time duration format"):
+            time_duration("7x")
+
+    def test_invalid_format_no_number(self):
+        """Test invalid format without number."""
+        with pytest.raises(ValueError, match="Invalid time duration format"):
+            time_duration("d")
+
+        with pytest.raises(ValueError, match="Invalid time duration format"):
+            time_duration("abc")
+
+    def test_empty_string(self):
+        """Test empty string."""
+        with pytest.raises(ValueError, match="Time duration cannot be empty"):
+            time_duration("")
+
+    def test_none(self):
+        """Test None value."""
+        with pytest.raises(ValueError, match="Time duration cannot be empty"):
+            time_duration(None)