From 674f8ecf706cee90767f2b4f4e577ade7e941c5e Mon Sep 17 00:00:00 2001
From: "Ilya.Kazakevich" <Ilya.Kazakevich@jetbrains.com>
Date: Wed, 28 Jun 2017 19:33:37 +0300
Subject: [PATCH] PY-24784: Use module SDK for newly created configurations

---
 python/src/com/jetbrains/python/testing/PyTestsShared.kt         | 1 +
 .../env/python/testing/CreateConfigurationTestTask.java          | 1 +
 2 files changed, 2 insertions(+)

diff --git a/python/src/com/jetbrains/python/testing/PyTestsShared.kt b/python/src/com/jetbrains/python/testing/PyTestsShared.kt
index 70625456c9a7..8f6e33a4fbc9 100644
--- a/python/src/com/jetbrains/python/testing/PyTestsShared.kt
+++ b/python/src/com/jetbrains/python/testing/PyTestsShared.kt
@@ -626,6 +626,7 @@ object PyTestsConfigurationProducer : com.jetbrains.python.testing.AbstractPytho
 
     val location = context?.location
     configuration.module = context?.module
+    configuration.isUseModuleSdk = true
     if (location is com.jetbrains.python.testing.PyTargetBasedPsiLocation) {
       location.target.copyTo(configuration.target)
     }
diff --git a/python/testSrc/com/jetbrains/env/python/testing/CreateConfigurationTestTask.java b/python/testSrc/com/jetbrains/env/python/testing/CreateConfigurationTestTask.java
index e6f2af7a7fd4..a3ecd14b858b 100644
--- a/python/testSrc/com/jetbrains/env/python/testing/CreateConfigurationTestTask.java
+++ b/python/testSrc/com/jetbrains/env/python/testing/CreateConfigurationTestTask.java
@@ -83,6 +83,7 @@ public abstract class CreateConfigurationTestTask<T extends AbstractPythonTestRu
 
         @SuppressWarnings("unchecked") // Checked one line above
         final T typedConfiguration = createConfigurationByElement(elementToRightClickOn, myExpectedConfigurationType);
+        Assert.assertTrue("Should use module sdk", typedConfiguration.isUseModuleSdk());
         checkConfiguration(typedConfiguration, elementToRightClickOn);
       }
     }), ModalityState.NON_MODAL);