How to use

DatastoreQueryForcedResponseTest

in

io.yawp.repository.query

@Test
public void testForcedResponseList() {
  BasicObject xpto = yawp.save(new BasicObject("xpto"));
  List<BasicObject> nonForcedList = yawp(BasicObject.class).list();
  assertEquals(1, nonForcedList.size());
  List<BasicObject> forcedList = yawp(BasicObject.class).forceResult(QueryType.LIST, Arrays.asList(xpto, xpto)).list();
  assertEquals(2, forcedList.size());
}

@Test
public void testForcedResponseIds() {
  BasicObject xpto = yawp.save(new BasicObject("xpto"));
  List<IdRef<BasicObject>> nonForcedList = yawp(BasicObject.class).ids();
  assertEquals(1, nonForcedList.size());
  List<IdRef<BasicObject>> forcedList = yawp(BasicObject.class).forceResult(QueryType.IDS, Arrays.asList(xpto.getId(), xpto.getId())).ids();
  assertEquals(2, forcedList.size());
}

  @Test
  public void testClearForcedResponse() {
    QueryBuilder<BasicObject> q = yawp(BasicObject.class).forceResult(QueryType.IDS, Collections.<IdRef<BasicObject>>emptyList()).clearForcedResults();
    assertNull(q.getForcedResult(QueryType.IDS));
  }
}

@Test
public void testForcedResponseListViaOnly() {
  yawp.save(new BasicObject("xpto"));
  BasicObject fake = new BasicObject("fake");
  BasicObject nonForcedObj = yawp(BasicObject.class).only();
  assertEquals("xpto", nonForcedObj.getStringValue());
  BasicObject forcedObj = yawp(BasicObject.class).forceResult(QueryType.LIST, Arrays.asList(fake)).only();
  assertEquals("fake", forcedObj.getStringValue());
}

@Test
public void testForcedResponseListViaFirst() {
  yawp.save(new BasicObject("xpto"));
  BasicObject fake = new BasicObject("fake");
  BasicObject nonForcedObj = yawp(BasicObject.class).only();
  assertEquals("xpto", nonForcedObj.getStringValue());
  BasicObject forcedObj = yawp(BasicObject.class).forceResult(QueryType.LIST, Arrays.asList(fake)).first();
  assertEquals("fake", forcedObj.getStringValue());
}

@Test
public void testForcedResponseIdsViaOnlyId() {
  BasicObject xpto = yawp.save(new BasicObject("xpto"));
  BasicObject fake = new BasicObject("fake");
  fake.setId(IdRef.create(yawp, BasicObject.class, "oni"));
  IdRef<BasicObject> nonForcedId = yawp(BasicObject.class).onlyId();
  assertEquals(xpto.getId(), nonForcedId);
  IdRef<BasicObject> forcedId = yawp(BasicObject.class).forceResult(QueryType.IDS, Arrays.asList(fake.getId())).onlyId();
  assertEquals("/basic_objects/oni", forcedId.toString());
}

@Test
public void testForcedResponseById() {
  BasicObject xpto = yawp.save(new BasicObject("xpto"));
  BasicObject fake = new BasicObject("fake");
  BasicObject nonForcedObj = yawp.query(BasicObject.class).fetch(xpto.getId());
  assertEquals("xpto", nonForcedObj.getStringValue());
  BasicObject forcedObj = yawp(BasicObject.class).forceResult(QueryType.FETCH, fake).fetch(xpto.getId());
  assertEquals("fake", forcedObj.getStringValue());
}