IDEA-154389 Remove and restore superclass leads to wrong state of ClassToSubclasses mappings

java/java-tests/testData/compileServer/incremental/common/integrateOnSuperclassRemovedAndRestored.log

@@ -0,0 +1,13 @@
+Cleaning output files:
+out/production/IntegrateOnSuperclassRemovedAndRestored/A.class
+End of files
+Cleaning output files:
+out/production/IntegrateOnSuperclassRemovedAndRestored/B.class
+End of files
+Compiling files:
+src/B.java
+End of files
+Compiling files:
+src/A.java
+src/B.java
+End of files

java/java-tests/testData/compileServer/incremental/common/integrateOnSuperclassRemovedAndRestored/src/A.java

@@ -0,0 +1 @@
+class A {}

java/java-tests/testData/compileServer/incremental/common/integrateOnSuperclassRemovedAndRestored/src/A.java.new1

@@ -0,0 +1 @@
+class A {}

java/java-tests/testData/compileServer/incremental/common/integrateOnSuperclassRemovedAndRestored/src/A.java.remove

@@ -0,0 +1 @@
+class A {}

java/java-tests/testData/compileServer/incremental/common/integrateOnSuperclassRemovedAndRestored/src/B.java

@@ -0,0 +1,3 @@
+class B extends A {
+
+}

jps/jps-builders/src/org/jetbrains/jps/builders/java/dependencyView/Mappings.java

@@ -2164,6 +2164,31 @@ public class Mappings {
             }
           }
 
+          // Now that the list of added classes is complete,
+          // check that super-classes of compiled classes are among newly added ones.
+          // Even if compiled class did not change, we should register 'added' superclass
+          // Consider situation for class B extends A:
+          // 1. file A is removed, make fails with error in file B
+          // 2. A is added back, B and A are compiled together in the second make session
+          // 3. Even if B did not change, A is considered as newly added and should be registered again in ClassToSubclasses dependencies
+          //    Without this code such registration will not happen because list of B's parents did not change
+          final Set<ClassRepr> addedClasses = myDelta.getAddedClasses();
+          if (!addedClasses.isEmpty()) {
+            final TIntHashSet addedNames = new TIntHashSet();
+            for (ClassRepr repr : addedClasses) {
+              addedNames.add(repr.name);
+            }
+            for (FileClasses compiledFile : newClasses) {
+              for (ClassRepr aClass : compiledFile.myFileClasses) {
+                for (int parent : aClass.getSupers()) {
+                  if (addedNames.contains(parent)) {
+                    myDelta.registerAddedSuperClass(aClass.name, parent);
+                  }
+                }
+              }
+            }
+          }
+
           debug("End of Differentiate.");
 
           if (myEasyMode) {

jps/jps-builders/testSrc/org/jetbrains/ether/CommonTest.java

@@ -149,7 +149,12 @@ public class CommonTest extends IncrementalTestCase {
     JpsModuleRootModificationUtil.addDependency(moduleB, moduleA);
     doTestBuild(1).assertSuccessful();
   }
-  
+
+  public void testIntegrateOnSuperclassRemovedAndRestored() throws Exception {
+    setupInitialProject();
+
+    doTestBuild(2);
+  }
   public void testMoveToplevelClassToAnotherFile() throws Exception {
     doTest();
   }