| /******************************************************************************* |
| * Licensed to the Apache Software Foundation (ASF) under one |
| * or more contributor license agreements. See the NOTICE file |
| * distributed with this work for additional information |
| * regarding copyright ownership. The ASF licenses this file |
| * to you 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.apache.olingo.odata2.jpa.processor.core.jpql; |
| |
| import static org.junit.Assert.assertEquals; |
| import static org.junit.Assert.assertNotNull; |
| import static org.junit.Assert.assertTrue; |
| import static org.junit.Assert.fail; |
| |
| import java.util.ArrayList; |
| import java.util.List; |
| |
| import org.apache.olingo.odata2.api.edm.EdmAssociation; |
| import org.apache.olingo.odata2.api.edm.EdmAssociationEnd; |
| import org.apache.olingo.odata2.api.edm.EdmEntitySet; |
| import org.apache.olingo.odata2.api.edm.EdmEntityType; |
| import org.apache.olingo.odata2.api.edm.EdmException; |
| import org.apache.olingo.odata2.api.edm.EdmMapping; |
| import org.apache.olingo.odata2.api.edm.EdmMultiplicity; |
| import org.apache.olingo.odata2.api.edm.EdmNavigationProperty; |
| import org.apache.olingo.odata2.api.edm.EdmProperty; |
| import org.apache.olingo.odata2.api.edm.EdmSimpleType; |
| import org.apache.olingo.odata2.api.edm.EdmSimpleTypeKind; |
| import org.apache.olingo.odata2.api.uri.KeyPredicate; |
| import org.apache.olingo.odata2.api.uri.NavigationSegment; |
| import org.apache.olingo.odata2.api.uri.info.GetEntityUriInfo; |
| import org.apache.olingo.odata2.jpa.processor.api.access.JPAJoinClause; |
| import org.apache.olingo.odata2.jpa.processor.api.exception.ODataJPAModelException; |
| import org.apache.olingo.odata2.jpa.processor.api.exception.ODataJPARuntimeException; |
| import org.apache.olingo.odata2.jpa.processor.core.jpql.JPQLJoinSelectSingleContext.JPQLJoinSelectSingleContextBuilder; |
| import org.apache.olingo.odata2.jpa.processor.core.model.JPAEdmMappingImpl; |
| import org.easymock.EasyMock; |
| import org.junit.After; |
| import org.junit.AfterClass; |
| import org.junit.BeforeClass; |
| import org.junit.Test; |
| |
| public class JPQLJoinSelectSingleContextTest { |
| |
| GetEntityUriInfo entityUriInfo; |
| |
| @BeforeClass |
| public static void setUpBeforeClass() throws Exception {} |
| |
| @AfterClass |
| public static void tearDownAfterClass() throws Exception {} |
| |
| public void setUp(final boolean toThrowException) throws Exception { |
| entityUriInfo = EasyMock.createMock(GetEntityUriInfo.class); |
| EdmEntitySet edmEntitySet = EasyMock.createMock(EdmEntitySet.class); |
| EdmEntityType edmEntityType = EasyMock.createMock(EdmEntityType.class); |
| List<NavigationSegment> navigationSegments = new ArrayList<NavigationSegment>(); |
| final EdmNavigationProperty navigationProperty = createNavigationProperty("a"); |
| final EdmNavigationProperty navigationProperty1 = createNavigationProperty("b"); |
| final List<KeyPredicate> keyPredicates = createKeyPredicates(toThrowException); |
| NavigationSegment navigationSegment = new NavigationSegment() { |
| |
| @Override |
| public EdmNavigationProperty getNavigationProperty() { |
| return navigationProperty; |
| } |
| |
| @Override |
| public List<KeyPredicate> getKeyPredicates() { |
| return keyPredicates; |
| } |
| |
| @Override |
| public EdmEntitySet getEntitySet() { |
| // TODO Auto-generated method stub |
| return null; |
| } |
| }; |
| NavigationSegment navigationSegment1 = new NavigationSegment() { |
| |
| @Override |
| public EdmNavigationProperty getNavigationProperty() { |
| return navigationProperty1; |
| } |
| |
| @Override |
| public List<KeyPredicate> getKeyPredicates() { |
| return keyPredicates; |
| } |
| |
| @Override |
| public EdmEntitySet getEntitySet() { |
| // TODO Auto-generated method stub |
| return null; |
| } |
| }; |
| navigationSegments.add(navigationSegment); |
| navigationSegments.add(navigationSegment1); |
| EasyMock.expect(entityUriInfo.getNavigationSegments()).andStubReturn(navigationSegments); |
| EasyMock.expect(entityUriInfo.getSelect()).andStubReturn(null); |
| EasyMock.expect(entityUriInfo.getFilter()).andStubReturn(null); |
| EasyMock.expect(entityUriInfo.getKeyPredicates()).andStubReturn(createKeyPredicates(toThrowException)); |
| EasyMock.expect(entityUriInfo.getTargetEntitySet()).andStubReturn(edmEntitySet); |
| EdmEntitySet startEdmEntitySet = EasyMock.createMock(EdmEntitySet.class); |
| EdmEntityType startEdmEntityType = EasyMock.createMock(EdmEntityType.class); |
| EasyMock.expect(startEdmEntityType.getMapping()).andStubReturn(null); |
| EasyMock.expect(startEdmEntityType.getName()).andStubReturn("SOHeader"); |
| EasyMock.expect(startEdmEntitySet.getEntityType()).andStubReturn(startEdmEntityType); |
| EasyMock.expect(entityUriInfo.getStartEntitySet()).andStubReturn(startEdmEntitySet); |
| EasyMock.replay(startEdmEntityType, startEdmEntitySet); |
| EasyMock.expect(edmEntitySet.getEntityType()).andStubReturn(edmEntityType); |
| EasyMock.expect(edmEntityType.getMapping()).andStubReturn(null); |
| EasyMock.expect(edmEntityType.getName()).andStubReturn("SOHeader"); |
| EasyMock.replay(edmEntityType, edmEntitySet, entityUriInfo); |
| |
| } |
| |
| @After |
| public void tearDown() throws Exception {} |
| |
| @Test |
| public void testGetJPAOuterJoinClauses() throws Exception { |
| setUp(false); |
| |
| JPQLJoinSelectSingleContext joinContext = new JPQLJoinSelectSingleContext(); |
| JPQLJoinSelectSingleContextBuilder joinContextBuilder = joinContext.new JPQLJoinSelectSingleContextBuilder(); |
| try { |
| joinContextBuilder.entityView = entityUriInfo; |
| joinContextBuilder.build(); |
| } catch (ODataJPAModelException e) { |
| fail("Should not come here"); |
| } catch (ODataJPARuntimeException e) { |
| fail("Should not come here"); |
| } |
| List<JPAJoinClause> joinClauses = joinContext.getJPAJoinClauses(); |
| assertNotNull(joinClauses); |
| assertTrue(joinClauses.size() > 0); |
| assertEquals("E1", joinClauses.get(0).getEntityAlias()); |
| assertEquals("SOHeader", joinClauses.get(0).getEntityName()); |
| assertEquals("s_Itema", joinClauses.get(1).getEntityRelationShip()); |
| assertEquals("R1", joinClauses.get(1).getEntityRelationShipAlias()); |
| } |
| |
| @Test |
| public void testExceptionThrown() throws Exception { |
| setUp(true); |
| JPQLJoinSelectSingleContext joinContext = new JPQLJoinSelectSingleContext(); |
| JPQLJoinSelectSingleContextBuilder joinContextBuilder = joinContext.new JPQLJoinSelectSingleContextBuilder(); |
| try { |
| joinContextBuilder.entityView = entityUriInfo; |
| joinContextBuilder.build(); |
| fail("Should not come here"); |
| } catch (ODataJPAModelException e) { |
| fail("Should not come here"); |
| } catch (ODataJPARuntimeException e) { |
| assertTrue(true); |
| } |
| } |
| |
| private EdmNavigationProperty createNavigationProperty(final String z) throws EdmException { |
| EdmNavigationProperty navigationProperty = EasyMock.createMock(EdmNavigationProperty.class); |
| EdmAssociation association = EasyMock.createMock(EdmAssociation.class); |
| EdmAssociationEnd associationEnd = EasyMock.createMock(EdmAssociationEnd.class); |
| EasyMock.expect(navigationProperty.getFromRole()).andStubReturn("roleA" + z); |
| EasyMock.expect(navigationProperty.getToRole()).andStubReturn("roleB" + z); |
| EasyMock.expect(navigationProperty.getName()).andStubReturn("navP" + z); |
| EasyMock.expect(navigationProperty.getName()).andStubReturn("navP" + z); |
| EasyMock.expect(navigationProperty.getMultiplicity()).andStubReturn(EdmMultiplicity.ONE); |
| EdmEntityType edmEntityType = EasyMock.createMock(EdmEntityType.class); |
| EdmMapping edmMapping = EasyMock.createMock(EdmMapping.class); |
| EasyMock.expect(edmMapping.getInternalName()).andStubReturn("sItem" + z); |
| EasyMock.expect(edmEntityType.getMapping()).andStubReturn(edmMapping); |
| EasyMock.expect(edmEntityType.getName()).andStubReturn("soItem" + z); |
| EasyMock.expect(associationEnd.getEntityType()).andStubReturn(edmEntityType); |
| EasyMock.expect(association.getEnd("roleA" + z)).andStubReturn(associationEnd); |
| EasyMock.expect(navigationProperty.getRelationship()).andStubReturn(association); |
| EdmMapping edmMapping1 = EasyMock.createMock(EdmMapping.class); |
| EasyMock.expect(edmMapping1.getInternalName()).andStubReturn("s_Item" + z); |
| EasyMock.expect(navigationProperty.getMapping()).andStubReturn(edmMapping1); |
| EasyMock.replay(edmMapping, edmMapping1, edmEntityType, associationEnd, association, navigationProperty); |
| return navigationProperty; |
| } |
| |
| private List<KeyPredicate> createKeyPredicates(final boolean toThrowException) throws EdmException { |
| KeyPredicate keyPredicate = EasyMock.createMock(KeyPredicate.class); |
| EasyMock.expect(keyPredicate.getLiteral()).andStubReturn("1"); |
| EdmProperty edmProperty = EasyMock.createMock(EdmProperty.class); |
| JPAEdmMappingImpl edmMapping = EasyMock.createMock(JPAEdmMappingImpl.class); |
| EasyMock.expect(edmMapping.getJPAType()) |
| .andStubReturn(null); |
| EasyMock.expect(edmMapping.getInternalName()).andStubReturn("soid"); |
| EasyMock.expect(edmProperty.getMapping()).andStubReturn(edmMapping); |
| EasyMock.expect(edmProperty.getName()).andStubReturn("soid"); |
| EdmSimpleType edmType = EdmSimpleTypeKind.Int32.getEdmSimpleTypeInstance(); |
| if (toThrowException) { |
| EasyMock.expect(edmProperty.getType()).andStubThrow(new EdmException(null)); |
| } else { |
| EasyMock.expect(edmProperty.getType()).andStubReturn(edmType); |
| } |
| EasyMock.expect(keyPredicate.getProperty()).andStubReturn(edmProperty); |
| |
| EasyMock.replay(edmMapping, edmProperty, keyPredicate); |
| List<KeyPredicate> keyPredicates = new ArrayList<KeyPredicate>(); |
| keyPredicates.add(keyPredicate); |
| return keyPredicates; |
| } |
| |
| } |