From 06bdbb58a79a2cc94d4cdaf05737ee78e9b9f7e4 Mon Sep 17 00:00:00 2001
From: Tomasz Pasternak <tomasz.pasternak@jetbrains.com>
Date: Mon, 4 Nov 2024 11:48:26 +0100
Subject: [PATCH] fix: Python tests (#6954)

* fix: Python tests

* fix: Python tests for CE
---
 intellij_platform_sdk/BUILD.idea241 | 5 +++++
 intellij_platform_sdk/BUILD.idea242 | 5 +++++
 intellij_platform_sdk/BUILD.idea243 | 5 +++++
 python/BUILD                        | 2 ++
 third_party/toml/BUILD              | 3 +++
 5 files changed, 20 insertions(+)

diff --git a/intellij_platform_sdk/BUILD.idea241 b/intellij_platform_sdk/BUILD.idea241
index bf5e1a29dca..17c06bf7826 100644
--- a/intellij_platform_sdk/BUILD.idea241
+++ b/intellij_platform_sdk/BUILD.idea241
@@ -114,6 +114,11 @@ java_import(
     jars = glob(["plugins/cwm-plugin/lib/*.jar"]),
 )
 
+java_import(
+    name = "json",
+    jars = glob(["plugins/json/lib/*.jar"])
+)
+
 # The plugins required by IJwB. We need to include them
 # when running integration tests.
 java_import(
diff --git a/intellij_platform_sdk/BUILD.idea242 b/intellij_platform_sdk/BUILD.idea242
index 157dc3aef82..092dddd4ea4 100644
--- a/intellij_platform_sdk/BUILD.idea242
+++ b/intellij_platform_sdk/BUILD.idea242
@@ -116,6 +116,11 @@ java_import(
     jars = glob(["plugins/cwm-plugin/lib/*.jar"]),
 )
 
+java_import(
+    name = "json",
+    jars = glob(["plugins/json/lib/*.jar"])
+)
+
 # The plugins required by IJwB. We need to include them
 # when running integration tests.
 java_import(
diff --git a/intellij_platform_sdk/BUILD.idea243 b/intellij_platform_sdk/BUILD.idea243
index 157dc3aef82..092dddd4ea4 100644
--- a/intellij_platform_sdk/BUILD.idea243
+++ b/intellij_platform_sdk/BUILD.idea243
@@ -116,6 +116,11 @@ java_import(
     jars = glob(["plugins/cwm-plugin/lib/*.jar"]),
 )
 
+java_import(
+    name = "json",
+    jars = glob(["plugins/json/lib/*.jar"])
+)
+
 # The plugins required by IJwB. We need to include them
 # when running integration tests.
 java_import(
diff --git a/python/BUILD b/python/BUILD
index 19ddccea9e5..fa9dbc545b3 100644
--- a/python/BUILD
+++ b/python/BUILD
@@ -84,6 +84,7 @@ intellij_integration_test_suite(
         "//base",
         "//base:integration_test_utils",
         "//base:unit_test_utils",
+        "//intellij_platform_sdk:json_for_tests",
         "//intellij_platform_sdk:jsr305",
         "//intellij_platform_sdk:plugin_api_for_tests",
         "//intellij_platform_sdk:test_libs",
@@ -92,6 +93,7 @@ intellij_integration_test_suite(
         "//third_party/toml:toml_for_tests",
         "@com_google_guava_guava//jar",
         "@junit//jar",
+        "@org_opentest4j_opentest4j//jar",
     ],
 )
 
diff --git a/third_party/toml/BUILD b/third_party/toml/BUILD
index 28cb9feaf5a..e4878438374 100644
--- a/third_party/toml/BUILD
+++ b/third_party/toml/BUILD
@@ -5,8 +5,11 @@ java_library(
     visibility = ["//visibility:private"],
     exports = select_for_plugin_api({
         "intellij-2024.1": ["@toml_2024_1//:toml"],
+        "intellij-ue-2024.1": ["@toml_2024_1//:toml"],
         "intellij-2024.2": ["@toml_2024_2//:toml"],
+        "intellij-ue-2024.2": ["@toml_2024_2//:toml"],
         "intellij-2024.3": ["@toml_2024_3//:toml"],
+        "intellij-ue-2024.3": ["@toml_2024_3//:toml"],
         "default": [],
     }),
 )