Test default consistency being used with CouchbaseRepository findAll.

Related tickets #1243

Author: JesusTheHun

src/test/java/org/springframework/data/couchbase/repository/CouchbaseRepositoryQueryIntegrationTests.java

@@ -57,20 +57,13 @@
 import org.springframework.data.couchbase.core.query.N1QLExpression;
 import org.springframework.data.couchbase.core.query.Query;
 import org.springframework.data.couchbase.core.query.QueryCriteria;
-import org.springframework.data.couchbase.domain.Address;
-import org.springframework.data.couchbase.domain.Airport;
-import org.springframework.data.couchbase.domain.AirportRepository;
-import org.springframework.data.couchbase.domain.Iata;
-import org.springframework.data.couchbase.domain.NaiveAuditorAware;
-import org.springframework.data.couchbase.domain.Person;
-import org.springframework.data.couchbase.domain.PersonRepository;
-import org.springframework.data.couchbase.domain.User;
-import org.springframework.data.couchbase.domain.UserRepository;
+import org.springframework.data.couchbase.domain.*;
 import org.springframework.data.couchbase.domain.time.AuditingDateTimeProvider;
 import org.springframework.data.couchbase.repository.auditing.EnableCouchbaseAuditing;
 import org.springframework.data.couchbase.repository.config.EnableCouchbaseRepositories;
 import org.springframework.data.couchbase.repository.query.CouchbaseQueryMethod;
 import org.springframework.data.couchbase.repository.query.CouchbaseRepositoryQuery;
+import org.springframework.data.couchbase.repository.support.SimpleCouchbaseRepository;
 import org.springframework.data.couchbase.util.Capabilities;
 import org.springframework.data.couchbase.util.ClusterAwareIntegrationTests;
 import org.springframework.data.couchbase.util.ClusterType;
@@ -92,6 +85,7 @@
  * @author Michael Nitschinger
  * @author Michael Reiche
  * @author Jens Schauder
+ * @author Jonathan Massuchetti
  */
 @SpringJUnitConfig(CouchbaseRepositoryQueryIntegrationTests.Config.class)
 @IgnoreWhen(missesCapabilities = Capabilities.QUERY, clusterTypes = ClusterType.MOCKED)
@@ -101,6 +95,9 @@ public class CouchbaseRepositoryQueryIntegrationTests extends ClusterAwareIntegr
 
 	@Autowired AirportRepository airportRepository;
 
+	@Autowired
+	AirportDefaultConsistencyRepository airportDefaultConsistencyRepository;
+
 	@Autowired UserRepository userRepository;
 
 	@Autowired CouchbaseTemplate couchbaseTemplate;
@@ -212,9 +209,9 @@ public void saveNotBounded() {
 		Airport airport2 = null;
 		for (int i = 1; i <= 100; i++) {
 			// set version == 0 so save() will be an upsert, not a replace
-			Airport saved = airportRepository.save(vie.clearVersion());
+			Airport saved = airportDefaultConsistencyRepository.save(vie.clearVersion());
 			try {
-				airport2 = airportRepository.iata(saved.getIata());
+				airport2 = airportDefaultConsistencyRepository.iata(saved.getIata());
 				if (airport2 == null) {
 					break;
 				}
@@ -227,14 +224,14 @@ public void saveNotBounded() {
 				assertEquals(vie.getId(), removeResult.getId());
 				assertTrue(removeResult.getCas() != 0);
 				assertTrue(removeResult.getMutationToken().isPresent());
-				Airport airport3 = airportRepository.iata(vie.getIata());
+				Airport airport3 = airportDefaultConsistencyRepository.iata(vie.getIata());
 				assertNull(airport3, "should have been removed");
 			}
 		}
 		assertNull(airport2, "airport2 should have likely been null at least once");
-		Airport saved = airportRepository.save(vie.clearVersion());
+		Airport saved = airportDefaultConsistencyRepository.save(vie.clearVersion());
 		couchbaseTemplate.findByQuery(Airport.class).withConsistency(REQUEST_PLUS).all();
-		airport2 = airportRepository.iata(vie.getIata());
+		airport2 = airportDefaultConsistencyRepository.iata(vie.getIata());
 		RemoveResult removeResult = couchbaseTemplate.removeById().one(saved.getId());
 		assertNotNull(airport2, "airport2 should have been found");
 	}
@@ -244,7 +241,7 @@ public void saveNotBoundedRequestPlus() {
 		ApplicationContext ac = new AnnotationConfigApplicationContext(ConfigRequestPlus.class);
 		// the Config class has been modified, these need to be loaded again
 		CouchbaseTemplate couchbaseTemplateRP = (CouchbaseTemplate) ac.getBean(COUCHBASE_TEMPLATE);
-		AirportRepository airportRepositoryRP = (AirportRepository) ac.getBean("airportRepository");
+		AirportDefaultConsistencyRepository airportRepositoryRP = (AirportDefaultConsistencyRepository) ac.getBean("airportDefaultConsistencyRepository");
 
 		// save() followed by query with NOT_BOUNDED will result in not finding the document
 		Airport vie = new Airport("airports::vie", "vie", "low9");
@@ -277,6 +274,36 @@ public void saveNotBoundedRequestPlus() {
 		assertFalse(!airports.isEmpty(), "airports should have been empty");
 	}
 
+
+	@Test
+	public void saveNotBoundedRequestPlusWithDefaultRepository() {
+		ApplicationContext ac = new AnnotationConfigApplicationContext(ConfigRequestPlus.class);
+		// the Config class has been modified, these need to be loaded again
+		CouchbaseTemplate couchbaseTemplateRP = (CouchbaseTemplate) ac.getBean(COUCHBASE_TEMPLATE);
+		AirportDefaultConsistencyRepository airportRepositoryRP = (AirportDefaultConsistencyRepository) ac.getBean("airportDefaultConsistencyRepository");
+
+		List<Airport> sizeBeforeTest = airportRepositoryRP.findAll();
+		assertEquals(0, sizeBeforeTest.size());
+
+		List<String> idsToRemove = new ArrayList<>(100);
+		for (int i = 1; i <= 100; i++) {
+			Airport vie = new Airport("airports::vie" + i, "vie" + i, "low9");
+			Airport saved = airportRepositoryRP.save(vie);
+			idsToRemove.add(saved.getId());
+		}
+
+		List<Airport> allSaved = airportRepositoryRP.findAll();
+
+		boolean success = allSaved.size() == 100;
+
+		for (String idToRemove : idsToRemove) {
+			couchbaseTemplateRP.removeById().one(idToRemove);
+		}
+
+		assertTrue(success);
+	}
+
+
 	@Test
 	void findByTypeAlias() {
 		Airport vie = null;