/**
* Copyright 2011 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 org.springframework.data.neo4j.support;
import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;
import org.junit.Assert;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.neo4j.graphdb.*;
import org.neo4j.helpers.collection.IteratorUtil;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.dao.InvalidDataAccessApiUsageException;
import org.springframework.data.neo4j.Friendship;
import org.springframework.data.neo4j.*;
import org.springframework.data.neo4j.Group;
import static org.springframework.data.neo4j.Person.persistedPerson;
import org.springframework.data.neo4j.Person;
import org.springframework.data.neo4j.repository.DirectGraphRepositoryFactory;
import org.springframework.data.neo4j.support.node.Neo4jHelper;
import org.springframework.test.context.ContextConfiguration;
import org.springframework.test.context.junit4.SpringJUnit4ClassRunner;
import org.springframework.test.context.transaction.BeforeTransaction;
import org.springframework.transaction.annotation.Transactional;
import java.util.*;
import static org.junit.Assert.assertEquals;
@RunWith(SpringJUnit4ClassRunner.class)
@ContextConfiguration(locations = {"classpath:org/springframework/data/neo4j/support/Neo4jGraphPersistenceTest-context.xml"})
public class NodeEntityRelationshipTest {
protected final Log log = LogFactory.getLog(getClass());
@Autowired
private GraphDatabaseContext graphDatabaseContext;
@Autowired
private DirectGraphRepositoryFactory graphRepositoryFactory;
@BeforeTransaction
public void cleanDb() {
Neo4jHelper.cleanDb(graphDatabaseContext);
}
@Test
@Transactional
public void testCreateRelationshipWithoutAnnotationOnSet() {
Person p = persistedPerson("Michael", 35);
Person spouse = persistedPerson("Tina", 36);
p.setSpouse(spouse);
Node spouseNode=p.getPersistentState().getSingleRelationship(DynamicRelationshipType.withName("spouse"), Direction.OUTGOING).getEndNode();
assertEquals(spouse.getPersistentState(), spouseNode);
assertEquals(spouse, p.getSpouse());
}
@Test
@Transactional
public void testCreateRelationshipWithAnnotationOnSet() {
Person p = persistedPerson("Michael", 35);
Person mother = persistedPerson("Gabi", 60);
p.setMother(mother);
Node motherNode = p.getPersistentState().getSingleRelationship(DynamicRelationshipType.withName("mother"), Direction.OUTGOING).getEndNode();
assertEquals(mother.getPersistentState(), motherNode);
assertEquals(mother, p.getMother());
}
@Test
@Transactional
public void testDeleteRelationship() {
Person p = persistedPerson("Michael", 35);
Person spouse = persistedPerson("Tina", 36);
p.setSpouse(spouse);
p.setSpouse(null);
Assert.assertNull(p.getPersistentState().getSingleRelationship(DynamicRelationshipType.withName("spouse"), Direction.OUTGOING));
Assert.assertNull(p.getSpouse());
}
@Test
@Transactional
public void testDeletePreviousRelationshipOnNewRelationship() {
Person p = persistedPerson("Michael", 35);
Person spouse = persistedPerson("Tina", 36);
Person friend = persistedPerson("Helga", 34);
p.setSpouse(spouse);
p.setSpouse(friend);
assertEquals(friend.getPersistentState(), p.getPersistentState().getSingleRelationship(DynamicRelationshipType.withName("spouse"), Direction.OUTGOING).getEndNode());
assertEquals(friend, p.getSpouse());
}
@Test
@Transactional
public void testCreateIncomingRelationshipWithAnnotationOnSet() {
Person p = persistedPerson("David", 25);
Person boss = persistedPerson("Emil", 32);
p.setBoss(boss);
assertEquals(boss.getPersistentState(), p.getPersistentState().getSingleRelationship(DynamicRelationshipType.withName("boss"), Direction.INCOMING).getStartNode());
assertEquals(boss, p.getBoss());
}
@Transactional
public void testAllowsCircularRelationship() {
Person p = persistedPerson("Michael", 35);
p.setBoss(p);
assertEquals("created self-referencing relationship",p,p.getBoss());
}
@Test
@Transactional
public void testSetOneToManyRelationship() {
Person michael = persistedPerson("Michael", 35);
Person david = persistedPerson("David", 25);
Group group = new Group().persist();
Set<Person> persons = new HashSet<Person>(Arrays.asList(michael, david));
group.setPersons(persons);
Relationship michaelRel = michael.getPersistentState().getSingleRelationship(DynamicRelationshipType.withName("persons"), Direction.INCOMING);
Relationship davidRel = david.getPersistentState().getSingleRelationship(DynamicRelationshipType.withName("persons"), Direction.INCOMING);
assertEquals(group.getPersistentState(), michaelRel.getStartNode());
assertEquals(group.getPersistentState(), davidRel.getStartNode());
}
@Test
@Transactional
public void testGetOneToManyRelationship() {
Person michael = persistedPerson("Michael", 35);
Person david = persistedPerson("David", 25);
Group group = new Group().persist();
Set<Person> persons = new HashSet<Person>(Arrays.asList(michael, david));
group.setPersons(persons);
Collection<Person> personsFromGet = group.getPersons();
assertEquals(persons, personsFromGet);
Assert.assertTrue(Set.class.isAssignableFrom(personsFromGet.getClass()));
}
@Test
@Transactional
public void testAddToOneToManyRelationship() {
Person michael = persistedPerson("Michael", 35);
Person david = persistedPerson("David", 25);
Group group = new Group().persist();
group.setPersons(new HashSet<Person>());
group.getPersons().add(michael);
group.getPersons().add(david);
Collection<Person> personsFromGet = group.getPersons();
assertEquals(new HashSet<Person>(Arrays.asList(david,michael)), personsFromGet);
Assert.assertTrue(Set.class.isAssignableFrom(personsFromGet.getClass()));
}
@Test
public void testAddToOneToManyRelationshipOutsideOfTransaction() {
Person michael = persistedPerson("Michael", 35);
Group group = new Group().persist();
group.getPersons().add(michael);
group = group.persist();
Collection<Person> personsFromGet = group.getPersons();
assertEquals(new HashSet<Person>(Arrays.asList(michael)), personsFromGet);
}
@Test
@Transactional
public void testRemoveFromOneToManyRelationship() {
Person michael = persistedPerson("Michael", 35);
Person david = persistedPerson("David", 25);
Group group = new Group().persist();
group.setPersons(new HashSet<Person>(Arrays.asList(michael, david)));
group.getPersons().remove(david);
assertEquals(Collections.singleton(michael), group.getPersons());
}
@Test
@Transactional
public void testRemoveAllFromOneToManyRelationship() {
Person michael = persistedPerson("Michael", 35);
Person david = persistedPerson("David", 25);
Group group = new Group().persist();
group.setPersons(new HashSet<Person>(Arrays.asList(michael, david)));
group.getPersons().removeAll(Collections.singleton(david));
assertEquals(Collections.singleton(michael), group.getPersons());
}
@Test
@Transactional
public void testRetainAllFromOneToManyRelationship() {
Person michael = persistedPerson("Michael", 35);
Person david = persistedPerson("David", 25);
Group group = new Group().persist();
group.setPersons(new HashSet<Person>(Arrays.asList(michael, david)));
group.getPersons().retainAll(Collections.singleton(david));
assertEquals(Collections.singleton(david), group.getPersons());
}
@Test
@Transactional
public void testClearFromOneToManyRelationship() {
Person michael = persistedPerson("Michael", 35);
Person david = persistedPerson("David", 25);
Group group = new Group().persist();
group.setPersons(new HashSet<Person>(Arrays.asList(michael, david)));
group.getPersons().clear();
assertEquals(Collections.<Person>emptySet(), group.getPersons());
}
@Test
@Transactional
public void testRelationshipGetEntities() {
Person p = persistedPerson("Michael", 35);
Person p2 = persistedPerson("David", 25);
Person p3 = persistedPerson("Emil", 32);
Friendship f2 = p.knows(p2);
Friendship f3 = p.knows(p3);
assertEquals(new HashSet<Friendship>(Arrays.asList(f2, f3)), IteratorUtil.addToCollection(p.getFriendships().iterator(), new HashSet<Friendship>()));
}
@Test(expected = InvalidDataAccessApiUsageException.class)
@Transactional
public void testRelationshipSetEntitiesShouldThrowException() {
Person p = persistedPerson("Michael", 35);
p.setFriendships(new HashSet<Friendship>());
}
@Test
@Transactional
public void testOneToManyReadOnly() {
Person michael = persistedPerson("Michael", 35);
Person david = persistedPerson("David", 25);
Group group = new Group().persist();
Set<Person> persons = new HashSet<Person>(Arrays.asList(michael, david));
group.setPersons(persons);
assertEquals(persons, IteratorUtil.addToCollection(group.getReadOnlyPersons().iterator(), new HashSet<Person>()));
}
@Test(expected = InvalidDataAccessApiUsageException.class)
@Transactional
public void testOneToManyReadOnlyShouldThrowExceptionOnSet() {
Group group = new Group().persist();
group.setReadOnlyPersons(new HashSet<Person>());
}
}