Add more langgraph tests.

These exercise the various checkpointers and stores, so we adapt them to
run for the MySQL versions as well.

Makefile

@@ -5,7 +5,7 @@
 ######################
 
 start-mysql:
-	MYSQL_VERSION=${MYSQL_VERSION:-16} docker compose -f tests/compose-mysql.yml up -V --force-recreate --wait
+	MYSQL_VERSION=${MYSQL_VERSION:-8} docker compose -f tests/compose-mysql.yml up -V --force-recreate --wait
 
 stop-mysql:
 	docker compose -f tests/compose-mysql.yml down
@@ -14,7 +14,9 @@ MYSQL_VERSIONS ?= 8
 test_mysql_version:
 	@echo "Testing MySQL $(MYSQL_VERSION)"
 	@MYSQL_VERSION=$(MYSQL_VERSION) make start-mysql
-	@poetry run pytest $(TEST)
+	@poetry run pytest --ignore=tests/langgraph $(TEST)
+	@poetry run pytest -n auto --dist worksteal tests/langgraph
+	@LANGGRAPH_FF_SEND_V2=true poetry run pytest -n auto --dist worksteal tests/langgraph
 	@EXIT_CODE=$$?; \
 	make stop-mysql; \
 	echo "Finished testing MySQL $(MYSQL_VERSION); Exit code: $$EXIT_CODE"; \
@@ -31,7 +33,7 @@ test:
 
 TEST ?= .
 test_watch:
-	POSTGRES_VERSION=${MYSQL_VERSION:-8} make start-mysql; \
+	MYSQL_VERSION=${MYSQL_VERSION:-8} make start-mysql; \
 	poetry run ptw $(TEST); \
 	EXIT_CODE=$$?; \
 	make stop-mysql; \

pyproject.toml

@@ -32,6 +32,10 @@ mypy = "^1.10.0"
 pymysql = "^1.1.1"
 aiomysql = "^0.2.0"
 types-PyMySQL = "^1.1.0"
+langgraph = "0.2.49"
+syrupy = "^4.0.2"
+pytest-repeat = "^0.9.3"
+pytest-xdist = {extras = ["psutil"], version = "^3.6.1"}
 
 [tool.pytest.ini_options]
 # --strict-markers will raise errors on unknown marks.
@@ -57,6 +61,9 @@ lint.select = [
   "I",  # isort
 ]
 lint.ignore = ["E501", "B008", "UP007", "UP006"]
+exclude = [
+  "tests/langgraph"
+]
 
 [tool.mypy]
 # https://mypy.readthedocs.io/en/stable/config_file.html
@@ -67,3 +74,6 @@ warn_unused_ignores = "True"
 warn_redundant_casts = "True"
 allow_redefinition = "True"
 disable_error_code = "typeddict-item, return-value"
+exclude = [
+  "tests/langgraph"
+]

tests/compose-mysql.yml

@@ -15,3 +15,10 @@ services:
       retries: 5
       interval: 60s
       start_interval: 1s
+    configs:
+      - source: init.sql
+        target: /docker-entrypoint-initdb.d/init.sql
+configs:
+  init.sql:
+    content: |
+      GRANT ALL PRIVILEGES ON *.* TO 'mysql'@'%';