Weak reference based PersistenceContext for streaming queries

ebean-test/src/test/java/io/ebeaninternal/server/transaction/DefaultPersistenceContextTest.java

@@ -6,16 +6,13 @@ import org.junit.jupiter.api.Test;
 import org.tests.model.basic.*;
 
 import static org.assertj.core.api.Assertions.assertThat;
-import static org.junit.jupiter.api.Assertions.assertFalse;
-import static org.junit.jupiter.api.Assertions.assertTrue;
 
-public class DefaultPersistenceContextTest {
+class DefaultPersistenceContextTest {
 
   private final Customer customer42;
-
   private final Car car1;
 
-  public DefaultPersistenceContextTest() {
+  DefaultPersistenceContextTest() {
     customer42 = new Customer();
     customer42.setId(42);
     car1 = new Car();
@@ -37,8 +34,7 @@ public class DefaultPersistenceContextTest {
   }
 
   @Test
-  public void put_get_withInheritance() {
-
+  void put_get_withInheritance() {
     PersistenceContext pc = pc();
     pc.put(root(Vehicle.class), 1, car1);
 
@@ -47,8 +43,7 @@ public class DefaultPersistenceContextTest {
   }
 
   @Test
-  public void put_get() {
-
+  void put_get() {
     PersistenceContext pc = pc();
     pc.put(Customer.class, customer42.getId(), customer42);
 
@@ -57,8 +52,7 @@ public class DefaultPersistenceContextTest {
   }
 
   @Test
-  public void putIfAbsent_when_absent() {
-
+  void putIfAbsent_when_absent() {
     PersistenceContext pc = pc();
     Object existing = pc.putIfAbsent(Customer.class, customer42.getId(), customer42);
 
@@ -66,8 +60,7 @@ public class DefaultPersistenceContextTest {
   }
 
   @Test
-  public void putIfAbsent_when_notAbsent() {
-
+  void putIfAbsent_when_notAbsent() {
     PersistenceContext pc = pcWith42();
     Object existing = pc.putIfAbsent(Customer.class, customer42.getId(), new Customer());
 
@@ -75,30 +68,28 @@ public class DefaultPersistenceContextTest {
   }
 
   @Test
-  public void get_when_empty() {
+  void get_when_empty() {
     PersistenceContext pc = pc();
     Object found = pc.get(Customer.class, 42);
     assertThat(found).isNull();
   }
 
   @Test
-  public void get_when_there() {
+  void get_when_there() {
     PersistenceContext pc = pcWith42();
     Object found = pc.get(Customer.class, 42);
     assertThat(found).isSameAs(customer42);
   }
 
   @Test
-  public void getWithOption_when_empty() {
-
+  void getWithOption_when_empty() {
     PersistenceContext pc = pc();
     PersistenceContext.WithOption withOption = pc.getWithOption(Customer.class, 42);
     assertThat(withOption).isNull();
   }
 
   @Test
-  public void getWithOption_when_there() {
-
+  void getWithOption_when_there() {
     PersistenceContext pc = pcWith42();
 
     PersistenceContext.WithOption withOption = pc.getWithOption(Customer.class, 42);
@@ -106,8 +97,7 @@ public class DefaultPersistenceContextTest {
   }
 
   @Test
-  public void getWithOption_when_deleted() {
-
+  void getWithOption_when_deleted() {
     PersistenceContext pc = pcWith42();
     pc.deleted(Customer.class, 42);
 
@@ -117,38 +107,33 @@ public class DefaultPersistenceContextTest {
   }
 
   @Test
-  public void size_when_empty() {
-
+  void size_when_empty() {
     PersistenceContext pc = pc();
     assertThat(pc.size(Customer.class)).isEqualTo(0);
   }
 
   @Test
-  public void size_when_some() {
-
+  void size_when_some() {
     PersistenceContext pc = pcWith42();
     assertThat(pc.size(Customer.class)).isEqualTo(1);
   }
 
   @Test
-  public void clear() {
-
+  void clear() {
     PersistenceContext pc = pcWith42();
     pc.clear();
     assertThat(pc.size(Customer.class)).isEqualTo(0);
   }
 
   @Test
-  public void clearClass() {
-
+  void clearClass() {
     PersistenceContext pc = pcWith42();
     pc.clear(Customer.class);
     assertThat(pc.size(Customer.class)).isEqualTo(0);
   }
 
   @Test
-  public void clearClassAndId() {
-
+  void clearClassAndId() {
     PersistenceContext pc = pcWith42();
     pc.put(Customer.class, 43, new Customer());
 
@@ -160,7 +145,7 @@ public class DefaultPersistenceContextTest {
   }
 
   @Test
-  public void forIterate() {
+  void forIterate() {
     final DefaultPersistenceContext pc = pcWith42();
     final Object origCustomer42 = pc.get(Customer.class, 42);
 
@@ -177,7 +162,7 @@ public class DefaultPersistenceContextTest {
   }
 
   @Test
-  public void forIterate_many() {
+  void forIterate_many() {
     DefaultPersistenceContext pc = new DefaultPersistenceContext();
     addCustomers(pc, 1, 100);
     addContacts(pc, 1, 1010);
@@ -191,44 +176,7 @@ public class DefaultPersistenceContextTest {
   }
 
   @Test
-  public void forIterate_resetLimit_forIterateReset() {
-    DefaultPersistenceContext initialPc = new DefaultPersistenceContext();
-    addCustomers(initialPc, 1, 100);
-    addContacts(initialPc, 1, 1010);
-
-    final PersistenceContext pcIterate = initialPc.forIterate();
-    assertFalse(pcIterate.resetLimit());
-
-    // added 900 NEW contact beans
-    addContacts(pcIterate, 2000, 900);
-    assertThat(pcIterate.size(Contact.class)).isEqualTo(1910);
-    assertFalse(pcIterate.resetLimit());
-
-    // boundary, added 1000 NEW contact beans (still false)
-    addContacts(pcIterate, 3000, 100);
-    assertFalse(pcIterate.resetLimit());
-
-    addContacts(pcIterate, 4000, 1);
-    addProducts(pcIterate, 1, 100);
-    // ACT - over 1000 added beans boundary for contacts so returns true
-    assertTrue(pcIterate.resetLimit());
-
-    assertThat(pcIterate.size(Contact.class)).isEqualTo(2011);
-    assertThat(pcIterate.size(Customer.class)).isEqualTo(100);
-    assertThat(pcIterate.size(Product.class)).isEqualTo(100);
-
-    // ACT - obtain new PC forIterateReset
-    PersistenceContext pcReset = pcIterate.forIterateReset();
-
-    // keeps original customer beans as no new added beans there
-    assertThat(pcReset.size(Customer.class)).isEqualTo(100); // customers didn't change
-    // added beans to contacts and products so those where reset
-    assertThat(pcReset.size(Contact.class)).isEqualTo(0);
-    assertThat(pcReset.size(Product.class)).isEqualTo(0);
-  }
-
-  @Test
-  public void toString_sillyTest() {
+  void toString_sillyTest() {
     DefaultPersistenceContext pc = pcWith42();
     assertThat(pc.toString()).contains("org.tests.model.basic.Customer");
   }
@@ -248,12 +196,4 @@ public class DefaultPersistenceContextTest {
       pc.put(Contact.class, i, bean);
     }
   }
-
-  private void addProducts(PersistenceContext pc, int start, int loop) {
-    for (int i = start; i < start + loop; i++) {
-      Product bean = new Product();
-      bean.setId(i);
-      pc.put(Product.class, i, bean);
-    }
-  }
 }