#298 - Add JPA example for composable repositories.

jpa/example/src/test/java/example/springdata/jpa/compositions/ComposedRepositoryTests.java

@@ -0,0 +1,132 @@
+/*
+ * Copyright 2017 the original author or authors.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *      http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+package example.springdata.jpa.compositions;
+
+import static org.assertj.core.api.Assertions.*;
+
+import java.util.Arrays;
+
+import org.junit.Test;
+import org.junit.runner.RunWith;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.boot.test.context.SpringBootTest;
+import org.springframework.test.context.junit4.SpringRunner;
+import org.springframework.transaction.annotation.Transactional;
+
+/**
+ * Integration test showing the usage of a composite repository via {@link UserRepository}.
+ *
+ * @author Mark Paluch
+ */
+@RunWith(SpringRunner.class)
+@Transactional
+@SpringBootTest
+public class ComposedRepositoryTests {
+
+	@Autowired UserRepository repository;
+
+	/**
+	 * Tests inserting a user and asserts it can be loaded again.
+	 */
+	@Test
+	public void testInsert() {
+
+		User user = new User();
+		user.setUsername("username");
+
+		user = repository.save(user);
+
+		assertThat(repository.findById(user.getId())).hasValue(user);
+	}
+
+	/**
+	 * Testing {@link ContactRepository} fragment.
+	 */
+	@Test
+	public void testContactRepository() {
+
+		User walter = new User();
+		walter.setUsername("heisenberg");
+		walter.setFirstname("Walter");
+		walter.setLastname("White");
+
+		User walterJr = new User();
+		walterJr.setUsername("flynn");
+		walterJr.setFirstname("Walter Jr.");
+		walterJr.setLastname("White");
+
+		repository.saveAll(Arrays.asList(walter, walterJr));
+
+		assertThat(repository.findRelatives(walter)).contains(walterJr);
+	}
+
+	/**
+	 * Testing {@link EmployeeRepository} fragment.
+	 */
+	@Test
+	public void testFindCoworkers() {
+
+		User gustavo = new User();
+		gustavo.setUsername("pollosh");
+		gustavo.setFirstname("Gustavo");
+		gustavo.setLastname("Fring");
+
+		User walter = new User();
+		walter.setUsername("heisenberg");
+		walter.setFirstname("Walter");
+		walter.setLastname("White");
+		walter.setManager(gustavo);
+
+		User jesse = new User();
+		jesse.setUsername("capncook");
+		jesse.setFirstname("Jesse");
+		jesse.setLastname("Pinkman");
+		jesse.setManager(gustavo);
+
+		repository.saveAll(Arrays.asList(gustavo, walter, jesse));
+
+		assertThat(repository.findCoworkers(walter)).contains(jesse);
+	}
+
+	/**
+	 * Testing {@link EmployeeRepository} fragment.
+	 */
+	@Test
+	public void testFindSubordinates() {
+
+		User gustavo = new User();
+		gustavo.setUsername("pollosh");
+		gustavo.setFirstname("Gustavo");
+		gustavo.setLastname("Fring");
+
+		User walter = new User();
+		walter.setUsername("heisenberg");
+		walter.setFirstname("Walter");
+		walter.setLastname("White");
+		walter.setManager(gustavo);
+
+		User jesse = new User();
+		jesse.setUsername("capncook");
+		jesse.setFirstname("Jesse");
+		jesse.setLastname("Pinkman");
+		jesse.setManager(gustavo);
+
+		repository.saveAll(Arrays.asList(gustavo, walter, jesse));
+
+		assertThat(repository.findSubordinates(gustavo)).contains(walter, jesse);
+		assertThat(repository.findSubordinates(walter)).isEmpty();
+	}
+}