Fix NPE in vertical class merger in presence of missing classes

Bug: 191124967
Change-Id: I100e759b413098aedbf09b9fbf438292e66b5150
diff --git a/src/main/java/com/android/tools/r8/shaking/VerticalClassMerger.java b/src/main/java/com/android/tools/r8/shaking/VerticalClassMerger.java
index 800332f..cae8f0e 100644
--- a/src/main/java/com/android/tools/r8/shaking/VerticalClassMerger.java
+++ b/src/main/java/com/android/tools/r8/shaking/VerticalClassMerger.java
@@ -1203,7 +1203,7 @@
           }
           holder =
               holder.superType != null
-                  ? appInfo.definitionFor(holder.superType).asProgramClass()
+                  ? asProgramClassOrNull(appInfo.definitionFor(holder.superType))
                   : null;
         }
       }
diff --git a/src/test/java/com/android/tools/r8/classmerging/vertical/VerticalClassMergingWithMissingSuperClassTest.java b/src/test/java/com/android/tools/r8/classmerging/vertical/VerticalClassMergingWithMissingSuperClassTest.java
index d7091fb..53fbaa5 100644
--- a/src/test/java/com/android/tools/r8/classmerging/vertical/VerticalClassMergingWithMissingSuperClassTest.java
+++ b/src/test/java/com/android/tools/r8/classmerging/vertical/VerticalClassMergingWithMissingSuperClassTest.java
@@ -4,13 +4,11 @@
 
 package com.android.tools.r8.classmerging.vertical;
 
-import static org.junit.Assert.assertEquals;
 
 import com.android.tools.r8.NoVerticalClassMerging;
 import com.android.tools.r8.TestBase;
 import com.android.tools.r8.TestParameters;
 import com.android.tools.r8.TestParametersCollection;
-import com.android.tools.r8.utils.codeinspector.AssertUtils;
 import org.junit.Test;
 import org.junit.runner.RunWith;
 import org.junit.runners.Parameterized;
@@ -29,21 +27,18 @@
 
   @Test
   public void test() throws Exception {
-    AssertUtils.assertFailsCompilation(
-        () ->
-            testForR8(parameters.getBackend())
-                .addProgramClasses(Main.class, A.class, B.class, C.class)
-                .addKeepMainRule(Main.class)
-                .addDontWarn(MissingClass.class)
-                .addVerticallyMergedClassesInspector(
-                    inspector -> inspector.assertMergedIntoSubtype(B.class))
-                .enableNoVerticalClassMergingAnnotations()
-                .setMinApi(parameters.getApiLevel())
-                .compile()
-                .addRunClasspathFiles(buildOnDexRuntime(parameters, MissingClass.class))
-                .run(parameters.getRuntime(), Main.class)
-                .assertSuccessWithOutputLines("C", "A", "B"),
-        exception -> assertEquals(NullPointerException.class, exception.getCause().getClass()));
+    testForR8(parameters.getBackend())
+        .addProgramClasses(Main.class, A.class, B.class, C.class)
+        .addKeepMainRule(Main.class)
+        .addDontWarn(MissingClass.class)
+        .addVerticallyMergedClassesInspector(
+            inspector -> inspector.assertMergedIntoSubtype(B.class))
+        .enableNoVerticalClassMergingAnnotations()
+        .setMinApi(parameters.getApiLevel())
+        .compile()
+        .addRunClasspathFiles(buildOnDexRuntime(parameters, MissingClass.class))
+        .run(parameters.getRuntime(), Main.class)
+        .assertSuccessWithOutputLines("C", "A", "B");
   }
 
   static class Main {