[impv] Do not warnings in dev version (#33)

Author: zhongjiajie

src/pydolphinscheduler/java_gateway.py

@@ -18,6 +18,7 @@
 """Module java gateway, contain gateway behavior."""
 
 import contextlib
+import warnings
 from logging import getLogger
 from typing import Any, Optional
 
@@ -70,12 +71,14 @@ def __init__(
             # 1. Java gateway version is too old: doesn't have method 'getGatewayVersion()'
             # 2. Error connecting to Java gateway
             gateway_version = self.get_gateway_version()
-        if gateway_version != __version__:
-            logger.warning(
+        if not __version__.endswith("dev") and gateway_version != __version__:
+            warnings.warn(
                 f"Using unmatched version of pydolphinscheduler (version {__version__}) "
                 f"and Java gateway (version {gateway_version}) may cause errors. "
                 "We strongly recommend you to find the matched version "
-                "(check: https://pypi.org/project/apache-dolphinscheduler)"
+                "(check: https://pypi.org/project/apache-dolphinscheduler)",
+                UserWarning,
+                stacklevel=2,
             )
 
     @property

tests/integration/test_java_gateway.py

@@ -16,9 +16,14 @@
 # under the License.
 
 """Test pydolphinscheduler java gateway."""
+import importlib
+import warnings
+from unittest.mock import PropertyMock, patch
+
 import pytest
 from py4j.java_gateway import GatewayParameters, JavaGateway, java_import
 
+from pydolphinscheduler import java_gateway
 from tests.testing.constants import TOKEN
 
 gateway_parameters = GatewayParameters(auth_token=TOKEN)
@@ -58,3 +63,30 @@ def test_python_client_java_import_package():
     # test if jvm view have some common utils
     for util in ("FileUtils", "OSUtils", "DateUtils"):
         assert hasattr(gateway.jvm, util)
+
+
+@pytest.mark.parametrize(
+    "version, is_warning",
+    [
+        ("dev", False),
+        ("1.0.0-dev", False),
+        ("1.0.0", True),
+    ],
+)
+@patch("pydolphinscheduler.__version__", new_callable=PropertyMock)
+def test_gateway_entry_point_version_warning(
+    mock_version, version: str, is_warning: bool
+):
+    """Test whether gateway entry point will raise version warning or not."""
+    mock_version.return_value = version
+    mock_version.endswith.return_value = version.endswith("dev")
+
+    importlib.reload(java_gateway)
+    with warnings.catch_warnings(record=True) as w:
+        _ = java_gateway.GatewayEntryPoint()
+        if is_warning:
+            assert len(w) == 1
+            assert issubclass(w[-1].category, UserWarning)
+            assert "Using unmatched version of pydolphinscheduler" in str(w[-1].message)
+        else:
+            assert len(w) == 0