/* * Copyright 2016 SteelBridge Laboratories, LLC. * * 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. * * For more information: http://steelbridgelabs.com */ package com.steelbridgelabs.oss.neo4j.structure; import org.junit.Assert; import org.junit.Test; import org.junit.runner.RunWith; import org.mockito.Mock; import org.mockito.Mockito; import org.mockito.runners.MockitoJUnitRunner; import org.neo4j.driver.Driver; import org.neo4j.driver.Result; import org.neo4j.driver.Session; import org.neo4j.driver.Transaction; import java.util.Collections; /** * @author Rogelio J. Baucells */ @RunWith(MockitoJUnitRunner.class) public class Neo4JGraphWhileExecuteTest { @Mock private Driver driver; @Mock private Session session; @Mock private Transaction transaction; @Mock private Neo4JElementIdProvider provider; @Mock private Result statementResult; @Test @SuppressWarnings("unchecked") public void givenTextCypherStatementShouldExecuteIt() { // arrange Mockito.when(driver.session(Mockito.any())).thenReturn(session); Mockito.when(session.beginTransaction()).thenAnswer(invocation -> transaction); Mockito.when(transaction.run(Mockito.any(String.class), Mockito.anyMap())).thenAnswer(invocation -> statementResult); Mockito.when(provider.fieldName()).thenAnswer(invocation -> "id"); try (Neo4JGraph graph = new Neo4JGraph(driver, provider, provider)) { // act Result result = graph.execute("MATCH (n) RETURN n"); // assert Assert.assertNotNull("Failed to execute CYPHER statement", result); Mockito.verify(transaction, Mockito.times(1)).run(Mockito.any(String.class), Mockito.anyMap()); } } @Test @SuppressWarnings("unchecked") public void givenCypherStatementShouldExecuteIt() { // arrange Mockito.when(driver.session(Mockito.any())).thenReturn(session); Mockito.when(session.beginTransaction()).thenAnswer(invocation -> transaction); Mockito.when(transaction.run(Mockito.any(String.class), Mockito.anyMap())).thenAnswer(invocation -> statementResult); Mockito.when(provider.fieldName()).thenAnswer(invocation -> "id"); try (Neo4JGraph graph = new Neo4JGraph(driver, provider, provider)) { // act Result result = graph.execute("MATCH (n) RETURN n"); // assert Assert.assertNotNull("Failed to execute CYPHER statement", result); Mockito.verify(transaction, Mockito.times(1)).run(Mockito.any(String.class), Mockito.anyMap()); } } @Test @SuppressWarnings("unchecked") public void givenCypherStatementWithParametersShouldExecuteIt() { // arrange Mockito.when(driver.session(Mockito.any())).thenReturn(session); Mockito.when(session.beginTransaction()).thenAnswer(invocation -> transaction); Mockito.when(transaction.run(Mockito.any(String.class), Mockito.anyMap())).thenAnswer(invocation -> statementResult); Mockito.when(provider.fieldName()).thenAnswer(invocation -> "id"); try (Neo4JGraph graph = new Neo4JGraph(driver, provider, provider)) { // act Result result = graph.execute("MATCH (n{id: $id}) RETURN n", Collections.singletonMap("id", 10)); // assert Assert.assertNotNull("Failed to execute CYPHER statement", result); Mockito.verify(transaction, Mockito.times(1)).run(Mockito.any(String.class), Mockito.anyMap()); } } }