multiple: disable socket for unit tests (#29080)

langchain-ai · Jan 7, 2025 · fcc9cdd · fcc9cdd
1 parent 539ebd5
commit fcc9cdd
Show file tree

Hide file tree

Showing 24 changed files with 123 additions and 19 deletions.
diff --git a/libs/partners/anthropic/Makefile b/libs/partners/anthropic/Makefile
@@ -7,7 +7,10 @@ all: help
 TEST_FILE ?= tests/unit_tests/
 integration_test integration_tests: TEST_FILE=tests/integration_tests/
 
-test tests integration_test integration_tests:
+test tests:
+	poetry run pytest -vvv --timeout 30 --disable-socket --allow-unix-socket $(TEST_FILE)
+
+integration_test integration_tests:
 	poetry run pytest -vvv --timeout 30 $(TEST_FILE)
 
 test_watch:

diff --git a/libs/partners/anthropic/poetry.lock b/libs/partners/anthropic/poetry.lock
diff --git a/libs/partners/anthropic/pyproject.toml b/libs/partners/anthropic/pyproject.toml
@@ -66,6 +66,7 @@ pytest-watcher = "^0.3.4"
 pytest-asyncio = "^0.21.1"
 defusedxml = "^0.7.1"
 pytest-timeout = "^2.3.1"
+pytest-socket = "^0.7.0"
 
 [tool.poetry.group.codespell.dependencies]
 codespell = "^2.2.0"

diff --git a/libs/partners/chroma/Makefile b/libs/partners/chroma/Makefile
@@ -7,7 +7,10 @@ all: help
 TEST_FILE ?= tests/unit_tests/
 integration_test integration_tests: TEST_FILE = tests/integration_tests/
 
-test tests integration_test integration_tests:
+test tests:
+	poetry run pytest --disable-socket --allow-unix-socket $(TEST_FILE)
+
+integration_test integration_tests:
 	poetry run pytest $(TEST_FILE)
 
 test_watch:

diff --git a/libs/partners/chroma/poetry.lock b/libs/partners/chroma/poetry.lock
diff --git a/libs/partners/chroma/pyproject.toml b/libs/partners/chroma/pyproject.toml
@@ -80,7 +80,11 @@ syrupy = "^4.0.2"
 pytest-watcher = "^0.3.4"
 pytest-asyncio = "^0.21.1"
 # hack to make sure py3.9 compatible versionof onnxruntime is installed for testing
-onnxruntime = [{version = "<1.20", python = "<3.10"}, {version = "*", python = ">=3.10"}]
+onnxruntime = [
+    { version = "<1.20", python = "<3.10" },
+    { version = "*", python = ">=3.10" },
+]
+pytest-socket = "^0.7.0"
 [[tool.poetry.group.test.dependencies.langchain-core]]
 path = "../../core"
 develop = true
@@ -94,15 +98,22 @@ python = "<3.9"
 path = "../../standard-tests"
 develop = true
 
+
 [tool.poetry.group.codespell.dependencies]
 codespell = "^2.2.0"
 
+
 [tool.poetry.group.test_integration.dependencies]
 
+
 [tool.poetry.group.lint.dependencies]
 ruff = "^0.5"
 # hack to make sure py3.9 compatible versionof onnxruntime is installed for testing
-onnxruntime = [{version = "<1.20", python = "<3.10"}, {version = "*", python = ">=3.10"}]
+onnxruntime = [
+    { version = "<1.20", python = "<3.10" },
+    { version = "*", python = ">=3.10" },
+]
+
 
 [tool.poetry.group.dev.dependencies]
 [[tool.poetry.group.dev.dependencies.langchain-core]]
@@ -114,6 +125,7 @@ python = ">=3.9"
 version = ">=0.1.40,<0.3"
 python = "<3.9"
 
+
 [tool.poetry.group.typing.dependencies]
 mypy = "^1.10"
 types-requests = "^2.31.0.20240406"

diff --git a/libs/partners/fireworks/Makefile b/libs/partners/fireworks/Makefile
@@ -7,7 +7,10 @@ all: help
 TEST_FILE ?= tests/unit_tests/
 integration_test integration_tests: TEST_FILE = tests/integration_tests/
 
-test tests integration_test integration_tests:
+test tests:
+	poetry run pytest --disable-socket --allow-unix-socket $(TEST_FILE)
+
+integration_test integration_tests:
 	poetry run pytest $(TEST_FILE)
 
 test_watch:

diff --git a/libs/partners/fireworks/poetry.lock b/libs/partners/fireworks/poetry.lock
diff --git a/libs/partners/fireworks/pyproject.toml b/libs/partners/fireworks/pyproject.toml
@@ -62,6 +62,7 @@ pytest-mock = "^3.10.0"
 syrupy = "^4.0.2"
 pytest-watcher = "^0.3.4"
 pytest-asyncio = "^0.21.1"
+pytest-socket = "^0.7.0"
 
 [tool.poetry.group.codespell.dependencies]
 codespell = "^2.2.0"

diff --git a/libs/partners/huggingface/Makefile b/libs/partners/huggingface/Makefile
@@ -8,7 +8,10 @@ TEST_FILE ?= tests/unit_tests/
 
 integration_test integration_tests: TEST_FILE=tests/integration_tests/
 
-test tests integration_test integration_tests:
+test tests:
+	poetry run pytest --disable-socket --allow-unix-socket $(TEST_FILE)
+
+integration_test integration_tests:
 	poetry run pytest $(TEST_FILE)
 
 test_watch:

diff --git a/libs/partners/huggingface/poetry.lock b/libs/partners/huggingface/poetry.lock
diff --git a/libs/partners/huggingface/pyproject.toml b/libs/partners/huggingface/pyproject.toml
@@ -59,6 +59,7 @@ optional = true
 pytest = "^7.3.0"
 pytest-asyncio = "^0.21.1"
 pytest-watcher = "^0.3.4"
+pytest-socket = "^0.7.0"
 
 [[tool.poetry.group.test.dependencies.scipy]]
 version = "^1"

diff --git a/libs/partners/pinecone/Makefile b/libs/partners/pinecone/Makefile
@@ -7,7 +7,10 @@ all: help
 TEST_FILE ?= tests/unit_tests/
 integration_test integration_tests: TEST_FILE = tests/integration_tests/
 
-test tests integration_test integration_tests:
+test tests:
+	poetry run pytest --disable-socket --allow-unix-socket $(TEST_FILE)
+
+integration_test integration_tests:
 	poetry run pytest $(TEST_FILE)
 
 test_watch:

diff --git a/libs/partners/pinecone/poetry.lock b/libs/partners/pinecone/poetry.lock
diff --git a/libs/partners/pinecone/pyproject.toml b/libs/partners/pinecone/pyproject.toml
@@ -63,6 +63,7 @@ pytest-mock = "^3.10.0"
 syrupy = "^4.0.2"
 pytest-watcher = "^0.3.4"
 pytest-asyncio = ">=0.25.0,<1"
+pytest-socket = "^0.7.0"
 
 [tool.poetry.group.codespell.dependencies]
 codespell = "^2.2.0"

diff --git a/libs/partners/qdrant/Makefile b/libs/partners/qdrant/Makefile
@@ -8,7 +8,10 @@ TEST_FILE ?= tests/unit_tests/
 
 integration_test integration_tests: TEST_FILE = tests/integration_tests/
 
-test tests integration_test integration_tests:
+test tests:
+	poetry run pytest --disable-socket --allow-unix-socket $(TEST_FILE)
+
+integration_test integration_tests:
 	poetry run pytest $(TEST_FILE)
 
 test_watch:

diff --git a/libs/partners/qdrant/poetry.lock b/libs/partners/qdrant/poetry.lock
diff --git a/libs/partners/qdrant/pyproject.toml b/libs/partners/qdrant/pyproject.toml
@@ -66,6 +66,7 @@ syrupy = "^4.0.2"
 pytest-watcher = "^0.3.4"
 pytest-asyncio = "^0.21.1"
 requests = "^2.31.0"
+pytest-socket = "^0.7.0"
 [[tool.poetry.group.test.dependencies.langchain-core]]
 path = "../../core"
 develop = true

diff --git a/libs/partners/voyageai/Makefile b/libs/partners/voyageai/Makefile
@@ -7,7 +7,10 @@ all: help
 TEST_FILE ?= tests/unit_tests/
 integration_test integration_tests: TEST_FILE=tests/integration_tests/
 
-test tests integration_test integration_tests:
+test tests:
+	poetry run pytest --disable-socket --allow-unix-socket $(TEST_FILE)
+
+integration_test integration_tests:
 	poetry run pytest $(TEST_FILE)
 
 test_watch:

diff --git a/libs/partners/voyageai/poetry.lock b/libs/partners/voyageai/poetry.lock
diff --git a/libs/partners/voyageai/pyproject.toml b/libs/partners/voyageai/pyproject.toml
@@ -57,6 +57,7 @@ pytest-mock = "^3.10.0"
 syrupy = "^4.0.2"
 pytest-watcher = "^0.3.4"
 pytest-asyncio = "^0.21.1"
+pytest-socket = "^0.7.0"
 [[tool.poetry.group.test.dependencies.numpy]]
 version = "^1.24.0"
 python = "<3.12"

diff --git a/libs/partners/xai/Makefile b/libs/partners/xai/Makefile
@@ -8,7 +8,10 @@ TEST_FILE ?= tests/unit_tests/
 
 integration_test integration_tests: TEST_FILE=tests/integration_tests/
 
-test tests integration_test integration_tests:
+test tests:
+	poetry run pytest --disable-socket --allow-unix-socket $(TEST_FILE)
+
+integration_test integration_tests:
 	poetry run pytest $(TEST_FILE)
 
 ######################

diff --git a/libs/partners/xai/poetry.lock b/libs/partners/xai/poetry.lock
diff --git a/libs/partners/xai/pyproject.toml b/libs/partners/xai/pyproject.toml
@@ -75,12 +75,15 @@ docarray = "^0.32.1"
 langchain-openai = { path = "../openai", develop = true }
 langchain-core = { path = "../../core", develop = true }
 langchain-tests = { path = "../../standard-tests", develop = true }
+pytest-socket = "^0.7.0"
+
 
 
 [tool.poetry.group.codespell.dependencies]
 codespell = "^2.2.0"
 
 
+
 [tool.poetry.group.test_integration.dependencies]
 [[tool.poetry.group.test_integration.dependencies.numpy]]
 version = "^1"
@@ -91,15 +94,18 @@ version = "^1.26.0"
 python = ">=3.12"
 
 
+
 [tool.poetry.group.lint.dependencies]
 ruff = "^0.5"
 
 
+
 [tool.poetry.group.typing.dependencies]
 mypy = "^1.10"
 types-requests = "^2"
 langchain-core = { path = "../../core", develop = true }
 
 
+
 [tool.poetry.group.dev.dependencies]
 langchain-core = { path = "../../core", develop = true }