remove duplicate class name

MorphiaOrg · Aug 19, 2024 · 62a9c4d · 62a9c4d
1 parent 18c3113
commit 62a9c4d
Showing 1 changed file with 4 additions and 9 deletions.
diff --git a/core/src/test/java/dev/morphia/test/TestDiscriminatorLookup.java b/core/src/test/java/dev/morphia/test/TestDiscriminatorLookup.java
@@ -1,6 +1,8 @@
 package dev.morphia.test;
 
 import dev.morphia.annotations.Entity;
+import dev.morphia.test.models.Shape;
+import dev.morphia.test.models.Square;
 import dev.morphia.test.models.TestEntity;
 
 import org.testng.annotations.Test;
@@ -14,14 +16,14 @@ public class TestDiscriminatorLookup extends TestBase {
     public void testLookup() {
         withConfig(buildConfig(SomeEntity.class), () -> {
             final SomeEntity entity = new SomeEntity();
-            entity.setShape(new Shape.Square());
+            entity.setShape(new Square());
 
             getDs().save(entity);
         });
 
         final SomeEntity entity = getDs().find(SomeEntity.class).first();
         assertNotNull(entity);
-        assertTrue(Shape.Square.class.isInstance(entity.getShape()));
+        assertTrue(Square.class.isInstance(entity.getShape()));
     }
 
     @Entity
@@ -36,11 +38,4 @@ public void setShape(Shape shape) {
             this.shape = shape;
         }
     }
-
-    @Entity
-    public static abstract class Shape {
-        public static class Square extends Shape {
-            public double side;
-        }
-    }
 }