package org.tests.basic;

import io.ebean.BaseTestCase;
import io.ebean.BeanState;
import io.ebean.CacheMode;
import io.ebean.DB;
import org.junit.jupiter.api.Test;
import org.tests.model.basic.Address;
import org.tests.model.basic.Customer;
import org.tests.model.basic.ResetBasicData;

import java.util.Map;
import java.util.Set;

import static org.junit.jupiter.api.Assertions.*;

public class TestLazyLoadInCache extends BaseTestCase {

  @Test
  public void testLoadInCache() {

    ResetBasicData.reset();

    Map<Integer, Customer> map = DB.find(Customer.class)
      .select("id, name")
      .setBeanCacheMode(CacheMode.PUT)
      .setReadOnly(true)
      .order().asc("id")
      .findMap();

    assertFalse(map.isEmpty());

    Object id = map.keySet().iterator().next();

    Customer cust1 = map.get(id);

    Customer cust1B = DB.find(Customer.class)
      .setReadOnly(true)
      .setId(id)
      .findOne();

    assertNotSame(cust1, cust1B);

    Set<String> loadedProps = DB.beanState(cust1).loadedProps();

    assertTrue(loadedProps.contains("name"));
    assertFalse(loadedProps.contains("status"));

    cust1.getStatus();

    // a readOnly reference
    Address billingAddress = cust1.getBillingAddress();
    BeanState billAddrState = DB.beanState(billingAddress);
    assertTrue(billAddrState.isReference());
    assertTrue(billAddrState.isReadOnly());

    // lazy load .. no longer a reference
    billingAddress.getCity();
    assertFalse(billAddrState.isReference());

  }

}