/*******************************************************************************
* 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.core.ep.producer;
import static org.custommonkey.xmlunit.XMLAssert.assertXpathExists;
import static org.custommonkey.xmlunit.XMLAssert.assertXpathNotExists;
import static org.junit.Assert.assertNotNull;
import static org.junit.Assert.assertNull;
import static org.mockito.Mockito.mock;
import static org.mockito.Mockito.when;
import java.io.IOException;
import java.io.InputStream;
import java.net.URI;
import java.net.URISyntaxException;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import org.apache.olingo.odata2.api.ODataCallback;
import org.apache.olingo.odata2.api.edm.*;
import org.apache.olingo.odata2.api.ep.EntityProviderException;
import org.apache.olingo.odata2.api.ep.EntityProviderWriteProperties;
import org.apache.olingo.odata2.api.ep.callback.OnWriteEntryContent;
import org.apache.olingo.odata2.api.ep.callback.OnWriteFeedContent;
import org.apache.olingo.odata2.api.ep.callback.WriteEntryCallbackContext;
import org.apache.olingo.odata2.api.ep.callback.WriteEntryCallbackResult;
import org.apache.olingo.odata2.api.ep.callback.WriteFeedCallbackContext;
import org.apache.olingo.odata2.api.ep.callback.WriteFeedCallbackResult;
import org.apache.olingo.odata2.api.exception.ODataApplicationException;
import org.apache.olingo.odata2.api.exception.ODataException;
import org.apache.olingo.odata2.api.processor.ODataResponse;
import org.apache.olingo.odata2.api.rt.RuntimeDelegate;
import org.apache.olingo.odata2.api.uri.ExpandSelectTreeNode;
import org.apache.olingo.odata2.api.uri.PathSegment;
import org.apache.olingo.odata2.api.uri.UriInfo;
import org.apache.olingo.odata2.core.ODataPathSegmentImpl;
import org.apache.olingo.odata2.core.ep.AbstractProviderTest;
import org.apache.olingo.odata2.core.ep.AtomEntityProvider;
import org.apache.olingo.odata2.core.exception.ODataRuntimeException;
import org.apache.olingo.odata2.core.uri.ExpandSelectTreeCreator;
import org.apache.olingo.odata2.core.uri.UriParserImpl;
import org.apache.olingo.odata2.testutil.helper.StringHelper;
import org.apache.olingo.odata2.testutil.mock.EdmTestProvider;
import org.apache.olingo.odata2.testutil.mock.MockFacade;
import org.custommonkey.xmlunit.exceptions.XpathException;
import org.junit.Test;
import org.xml.sax.SAXException;
public class XmlExpandProducerTest extends AbstractProviderTest {
private static final boolean T = true;
private static final boolean F = false;
private final URI inlineBaseUri;
private String employeeXPathString = "/a:entry/a:link[@href=\"Rooms('1')/nr_Employees\" and @title='nr_Employees']";
private String roomXPathString = "/a:entry/a:link[@href=\"Employees('1')/ne_Room\" and @title='ne_Room']";
private String teamXPathString = "/a:entry/a:link[@href=\"Employees('1')/ne_Team\" and @title='ne_Team']";
private String buildingXPathString = "/a:entry/a:link[@href=\"Buildings('1')/nb_Rooms\" and @title='nb_Rooms']";
public XmlExpandProducerTest(final StreamWriterImplType type) {
super(type);
try {
inlineBaseUri = new URI("http://hubbeldubbel.com/");
} catch (URISyntaxException e) {
throw new ODataRuntimeException(e);
}
}
@Test
public void expandSelectedTeamNullOmitInline() throws Exception {
ExpandSelectTreeNode selectTree = getSelectExpandTree("Employees('1')", "ne_Team", "ne_Team");
HashMap<String, ODataCallback> callbacksEmployee = createCallbacks("Employees");
EntityProviderWriteProperties properties =
EntityProviderWriteProperties.serviceRoot(BASE_URI).expandSelectTree(selectTree).callbacks(callbacksEmployee)
.omitInlineForNullData(true).build();
AtomEntityProvider provider = createAtomEntityProvider();
ODataResponse response =
provider.writeEntry(MockFacade.getMockEdm().getDefaultEntityContainer().getEntitySet("Employees"),
employeeData, properties);
String xmlString = verifyResponse(response);
verifyNavigationProperties(xmlString, F, F, T);
assertXpathExists(teamXPathString, xmlString);
assertXpathNotExists("/a:entry/m:properties", xmlString);
assertXpathNotExists(teamXPathString + "/m:inline", xmlString);
assertXpathNotExists(teamXPathString + "/m:inline/a:entry", xmlString);
}
@Test
public void expandSelectedRoomsNullOmitInline() throws Exception {
ExpandSelectTreeNode selectTree = getSelectExpandTree("Buildings('1')", "nb_Rooms", "nb_Rooms");
HashMap<String, ODataCallback> callbacksEmployee = createCallbacks("Buildings");
EntityProviderWriteProperties properties =
EntityProviderWriteProperties.serviceRoot(BASE_URI).expandSelectTree(selectTree).callbacks(callbacksEmployee)
.omitInlineForNullData(true).build();
AtomEntityProvider provider = createAtomEntityProvider();
ODataResponse response =
provider.writeEntry(MockFacade.getMockEdm().getDefaultEntityContainer().getEntitySet("Buildings"),
buildingData, properties);
String xmlString = verifyResponse(response);
assertXpathNotExists("/a:entry/m:properties", xmlString);
assertXpathExists(buildingXPathString, xmlString);
assertXpathNotExists(buildingXPathString + "/m:inline", xmlString);
assertXpathNotExists(buildingXPathString + "/m:inline/a:feed", xmlString);
assertXpathNotExists(buildingXPathString + "/m:inline/a:feed/a:id", xmlString);
assertXpathNotExists(buildingXPathString + "/m:inline/a:feed/a:title", xmlString);
assertXpathNotExists(buildingXPathString + "/m:inline/a:feed/a:updated", xmlString);
assertXpathNotExists(buildingXPathString + "/m:inline/a:feed/a:author", xmlString);
assertXpathNotExists(buildingXPathString + "/m:inline/a:feed/a:author/a:name", xmlString);
assertXpathNotExists(buildingXPathString + "/m:inline/a:feed/a:link[@href=\"Buildings('1')/nb_Rooms\"]", xmlString);
assertXpathNotExists(buildingXPathString + "/m:inline/a:feed/a:entry", xmlString);
}
@Test
public void expandSelectedEmployees() throws Exception {
ExpandSelectTreeNode selectTree = getSelectExpandTree("Rooms('1')", "nr_Employees", "nr_Employees");
HashMap<String, ODataCallback> callbacksRoom = createCallbacks("Rooms");
EntityProviderWriteProperties properties =
EntityProviderWriteProperties.serviceRoot(BASE_URI).expandSelectTree(selectTree).callbacks(callbacksRoom)
.build();
AtomEntityProvider provider = createAtomEntityProvider();
ODataResponse response =
provider.writeEntry(MockFacade.getMockEdm().getDefaultEntityContainer().getEntitySet("Rooms"), roomData,
properties);
String xmlString = verifyResponse(response);
assertXpathNotExists("/a:entry/m:properties", xmlString);
assertXpathExists("/a:entry/a:link", xmlString);
verifyEmployees(employeeXPathString, xmlString);
}
@Test(expected = EntityProviderException.class)
public void expandSelectedEmployeesWithFacets() throws Exception {
Edm edm = MockFacade.getMockEdm();
EdmTyped imageUrlProperty = edm.getEntityType("RefScenario", "Employee").getProperty("ImageUrl");
EdmFacets facets = mock(EdmFacets.class);
when(facets.getMaxLength()).thenReturn(1);
when(((EdmProperty) imageUrlProperty).getFacets()).thenReturn(facets);
ExpandSelectTreeNode selectTree = getSelectExpandTree("Rooms('1')", "nr_Employees", "nr_Employees");
HashMap<String, ODataCallback> callbacksRoom = createCallbacks("Rooms");
EntityProviderWriteProperties properties =
EntityProviderWriteProperties.serviceRoot(BASE_URI).expandSelectTree(selectTree)
.callbacks(callbacksRoom)
.build();
AtomEntityProvider provider = createAtomEntityProvider();
ODataResponse response =
provider.writeEntry(edm.getDefaultEntityContainer().getEntitySet("Rooms"), roomData,
properties);
String xmlString = verifyResponse(response);
assertXpathNotExists("/a:entry/m:properties", xmlString);
assertXpathExists("/a:entry/a:link", xmlString);
verifyEmployees(employeeXPathString, xmlString);
}
@Test
public void expandSelectedEmployeesIgnoreFacets() throws Exception {
Edm edm = MockFacade.getMockEdm();
EdmTyped imageUrlProperty = edm.getEntityType("RefScenario", "Employee").getProperty("ImageUrl");
EdmFacets facets = mock(EdmFacets.class);
when(facets.getMaxLength()).thenReturn(1);
when(((EdmProperty) imageUrlProperty).getFacets()).thenReturn(facets);
ExpandSelectTreeNode selectTree = getSelectExpandTree("Rooms('1')", "nr_Employees", "nr_Employees");
HashMap<String, ODataCallback> callbacksRoom = createCallbacks("Rooms");
EntityProviderWriteProperties properties =
EntityProviderWriteProperties.serviceRoot(BASE_URI).expandSelectTree(selectTree).callbacks(callbacksRoom)
.callbacks(callbacksRoom).validatingFacets(false)
.build();
AtomEntityProvider provider = createAtomEntityProvider();
ODataResponse response =
provider.writeEntry(edm.getDefaultEntityContainer().getEntitySet("Rooms"), roomData,
properties);
String xmlString = verifyResponse(response);
assertXpathNotExists("/a:entry/m:properties", xmlString);
assertXpathExists("/a:entry/a:link", xmlString);
verifyEmployees(employeeXPathString, xmlString);
}
@Test
public void expandSelectedEmployeesWithBuilder() throws Exception {
EdmEntitySet entitySet = MockFacade.getMockEdm().getDefaultEntityContainer().getEntitySet("Rooms");
List<String> navigationPropertyNames = new ArrayList<String>();
navigationPropertyNames.add("nr_Employees");
ExpandSelectTreeNode selectTree =
ExpandSelectTreeNode.entitySet(entitySet).expandedLinks(navigationPropertyNames).build();
HashMap<String, ODataCallback> callbacksRoom = createCallbacks("Rooms");
EntityProviderWriteProperties properties =
EntityProviderWriteProperties.serviceRoot(BASE_URI).expandSelectTree(selectTree).callbacks(callbacksRoom)
.build();
AtomEntityProvider provider = createAtomEntityProvider();
ODataResponse response =
provider.writeEntry(entitySet, roomData,
properties);
String xmlString = verifyResponse(response);
assertXpathNotExists("/a:entry/m:properties", xmlString);
assertXpathExists("/a:entry/a:link", xmlString);
verifyEmployees(employeeXPathString, xmlString);
}
@Test
public void expandSelectedEmployeesNull() throws Exception {
ExpandSelectTreeNode selectTree = getSelectExpandTree("Rooms('1')", "nr_Employees", "nr_Employees");
HashMap<String, ODataCallback> callbacksRoom = new HashMap<String, ODataCallback>();
ODataCallback employeeCallback = new OnWriteFeedContent() {
@Override
public WriteFeedCallbackResult retrieveFeedResult(final WriteFeedCallbackContext context)
throws ODataApplicationException {
WriteFeedCallbackResult writeFeedCallbackResult = new WriteFeedCallbackResult();
writeFeedCallbackResult.setInlineProperties(DEFAULT_PROPERTIES);
writeFeedCallbackResult.setFeedData(null);
return writeFeedCallbackResult;
}
};
callbacksRoom.put("nr_Employees", employeeCallback);
EntityProviderWriteProperties properties =
EntityProviderWriteProperties.serviceRoot(BASE_URI).expandSelectTree(selectTree).callbacks(callbacksRoom)
.build();
AtomEntityProvider provider = createAtomEntityProvider();
ODataResponse response =
provider.writeEntry(MockFacade.getMockEdm().getDefaultEntityContainer().getEntitySet("Rooms"), roomData,
properties);
String xmlString = verifyResponse(response);
assertXpathNotExists("/a:entry/m:properties", xmlString);
assertXpathExists("/a:entry/a:link", xmlString);
assertXpathExists(employeeXPathString + "/m:inline", xmlString);
assertXpathExists(employeeXPathString + "/m:inline/a:feed", xmlString);
assertXpathNotExists(employeeXPathString + "/m:inline/a:feed/a:entry", xmlString);
}
@Test
public void expandSelectedEmployeesEmpty() throws Exception {
ExpandSelectTreeNode selectTree = getSelectExpandTree("Rooms('1')", "nr_Employees", "nr_Employees");
HashMap<String, ODataCallback> callbacksRoom = new HashMap<String, ODataCallback>();
ODataCallback employeeCallback = new OnWriteFeedContent() {
@Override
public WriteFeedCallbackResult retrieveFeedResult(final WriteFeedCallbackContext context)
throws ODataApplicationException {
WriteFeedCallbackResult writeFeedCallbackResult = new WriteFeedCallbackResult();
writeFeedCallbackResult.setInlineProperties(DEFAULT_PROPERTIES);
writeFeedCallbackResult.setFeedData(new ArrayList<Map<String, Object>>());
return writeFeedCallbackResult;
}
};
callbacksRoom.put("nr_Employees", employeeCallback);
EntityProviderWriteProperties properties =
EntityProviderWriteProperties.serviceRoot(BASE_URI).expandSelectTree(selectTree).callbacks(callbacksRoom)
.build();
AtomEntityProvider provider = createAtomEntityProvider();
ODataResponse response =
provider.writeEntry(MockFacade.getMockEdm().getDefaultEntityContainer().getEntitySet("Rooms"), roomData,
properties);
String xmlString = verifyResponse(response);
assertXpathNotExists("/a:entry/m:properties", xmlString);
assertXpathExists("/a:entry/a:link", xmlString);
assertXpathExists(employeeXPathString + "/m:inline", xmlString);
assertXpathExists(employeeXPathString + "/m:inline/a:feed", xmlString);
}
@Test
public void expandSelectedEmployeesWithSelfLink() throws Exception {
ExpandSelectTreeNode selectTree = getSelectExpandTree("Rooms('1')", "nr_Employees", "nr_Employees");
HashMap<String, ODataCallback> callbacksRoom = createCallbacks("Rooms");
EntityProviderWriteProperties properties =
EntityProviderWriteProperties.serviceRoot(BASE_URI).expandSelectTree(selectTree).callbacks(callbacksRoom)
.build();
AtomEntityProvider provider = createAtomEntityProvider();
ODataResponse response =
provider.writeEntry(MockFacade.getMockEdm().getDefaultEntityContainer().getEntitySet("Rooms"), roomData,
properties);
String xmlString = verifyResponse(response);
assertXpathNotExists("/a:entry/m:properties", xmlString);
assertXpathExists("/a:entry/a:link", xmlString);
verifyEmployees(employeeXPathString, xmlString);
assertXpathExists(employeeXPathString + "/m:inline/a:feed/a:link[@href=\"Rooms('1')/nr_Employees\"]", xmlString);
}
@Test
public void deepExpandSelectedEmployees() throws Exception {
ExpandSelectTreeNode selectTree = getSelectExpandTree("Rooms('1')", "nr_Employees/ne_Room", "nr_Employees/ne_Room");
HashMap<String, ODataCallback> callbacksRoom = createCallbacks("Rooms");
EntityProviderWriteProperties properties =
EntityProviderWriteProperties.serviceRoot(BASE_URI).expandSelectTree(selectTree).callbacks(callbacksRoom)
.build();
AtomEntityProvider provider = createAtomEntityProvider();
ODataResponse response =
provider.writeEntry(MockFacade.getMockEdm().getDefaultEntityContainer().getEntitySet("Rooms"), roomData,
properties);
String xmlString = verifyResponse(response);
assertXpathNotExists("/a:entry/m:properties", xmlString);
assertXpathExists(employeeXPathString, xmlString);
assertXpathExists(employeeXPathString + "/m:inline/a:feed" + roomXPathString, xmlString);
assertXpathExists(employeeXPathString + "/m:inline/a:feed" + roomXPathString
+ "/m:inline/a:entry/a:content/m:properties", xmlString);
}
@Test
public void deepExpandSelectedEmployeesWithRoomId() throws Exception {
ExpandSelectTreeNode selectTree =
getSelectExpandTree("Rooms('1')", "nr_Employees/ne_Room/Id", "nr_Employees/ne_Room");
HashMap<String, ODataCallback> callbacksRoom = createCallbacks("Rooms");
EntityProviderWriteProperties properties =
EntityProviderWriteProperties.serviceRoot(BASE_URI).expandSelectTree(selectTree).callbacks(callbacksRoom)
.build();
AtomEntityProvider provider = createAtomEntityProvider();
ODataResponse response =
provider.writeEntry(MockFacade.getMockEdm().getDefaultEntityContainer().getEntitySet("Rooms"), roomData,
properties);
String xmlString = verifyResponse(response);
assertXpathNotExists("/a:entry/m:properties", xmlString);
assertXpathExists(employeeXPathString, xmlString);
assertXpathExists(employeeXPathString + "/m:inline/a:feed" + roomXPathString, xmlString);
assertXpathExists(employeeXPathString + "/m:inline/a:feed" + roomXPathString + "/m:inline", xmlString);
assertXpathExists(employeeXPathString + "/m:inline/a:feed" + roomXPathString + "/m:inline/a:entry", xmlString);
assertXpathExists(employeeXPathString + "/m:inline/a:feed" + roomXPathString
+ "/m:inline/a:entry/a:content/m:properties/d:Id", xmlString);
}
@Test
public void expandSelectedRoom() throws Exception {
ExpandSelectTreeNode selectTree = getSelectExpandTree("Employees('1')", "ne_Room", "ne_Room");
HashMap<String, ODataCallback> callbacksEmployee = createCallbacks("Employees");
EntityProviderWriteProperties properties =
EntityProviderWriteProperties.serviceRoot(BASE_URI).expandSelectTree(selectTree).callbacks(callbacksEmployee)
.build();
AtomEntityProvider provider = createAtomEntityProvider();
ODataResponse response =
provider.writeEntry(MockFacade.getMockEdm().getDefaultEntityContainer().getEntitySet("Employees"),
employeeData, properties);
String xmlString = verifyResponse(response);
verifyNavigationProperties(xmlString, F, T, F);
assertXpathNotExists("/a:entry/m:properties", xmlString);
verifyRoom(roomXPathString, xmlString);
}
@Test
public void expandSelectedTeamNull() throws Exception {
ExpandSelectTreeNode selectTree = getSelectExpandTree("Employees('1')", "ne_Team", "ne_Team");
HashMap<String, ODataCallback> callbacksEmployee = createCallbacks("Employees");
EntityProviderWriteProperties properties =
EntityProviderWriteProperties.serviceRoot(BASE_URI).expandSelectTree(selectTree).callbacks(callbacksEmployee)
.build();
AtomEntityProvider provider = createAtomEntityProvider();
ODataResponse response =
provider.writeEntry(MockFacade.getMockEdm().getDefaultEntityContainer().getEntitySet("Employees"),
employeeData, properties);
String xmlString = verifyResponse(response);
verifyNavigationProperties(xmlString, F, F, T);
assertXpathNotExists("/a:entry/m:properties", xmlString);
assertXpathExists(teamXPathString + "/m:inline", xmlString);
assertXpathNotExists(teamXPathString + "/m:inline/a:entry", xmlString);
}
@Test
public void expandSelectedTeamEmptyDataMap() throws Exception {
ExpandSelectTreeNode selectTree = getSelectExpandTree("Employees('1')", "ne_Team", "ne_Team");
HashMap<String, ODataCallback> callbacksEmployee = new HashMap<String, ODataCallback>();
OnWriteEntryContent callback = new OnWriteEntryContent() {
@Override
public WriteEntryCallbackResult retrieveEntryResult(final WriteEntryCallbackContext context)
throws ODataApplicationException {
WriteEntryCallbackResult result = new WriteEntryCallbackResult();
result.setInlineProperties(DEFAULT_PROPERTIES);
result.setEntryData(new HashMap<String, Object>());
return result;
}
};
callbacksEmployee.put("ne_Team", callback);
EntityProviderWriteProperties properties =
EntityProviderWriteProperties.serviceRoot(BASE_URI).expandSelectTree(selectTree).callbacks(callbacksEmployee)
.build();
AtomEntityProvider provider = createAtomEntityProvider();
ODataResponse response =
provider.writeEntry(MockFacade.getMockEdm().getDefaultEntityContainer().getEntitySet("Employees"),
employeeData, properties);
String xmlString = verifyResponse(response);
verifyNavigationProperties(xmlString, F, F, T);
assertXpathNotExists("/a:entry/m:properties", xmlString);
assertXpathExists(teamXPathString + "/m:inline", xmlString);
assertXpathNotExists(teamXPathString + "/m:inline/a:entry", xmlString);
}
@Test
public void expandSelectedRoomsNull() throws Exception {
ExpandSelectTreeNode selectTree = getSelectExpandTree("Buildings('1')", "nb_Rooms", "nb_Rooms");
HashMap<String, ODataCallback> callbacksEmployee = createCallbacks("Buildings");
EntityProviderWriteProperties properties =
EntityProviderWriteProperties.serviceRoot(BASE_URI).expandSelectTree(selectTree).callbacks(callbacksEmployee)
.build();
AtomEntityProvider provider = createAtomEntityProvider();
ODataResponse response =
provider.writeEntry(MockFacade.getMockEdm().getDefaultEntityContainer().getEntitySet("Buildings"),
buildingData, properties);
String xmlString = verifyResponse(response);
assertXpathNotExists("/a:entry/m:properties", xmlString);
assertXpathExists(buildingXPathString + "/m:inline", xmlString);
assertXpathExists(buildingXPathString + "/m:inline/a:feed", xmlString);
assertXpathExists(buildingXPathString + "/m:inline/a:feed/a:id", xmlString);
assertXpathExists(buildingXPathString + "/m:inline/a:feed/a:title", xmlString);
assertXpathExists(buildingXPathString + "/m:inline/a:feed/a:updated", xmlString);
assertXpathExists(buildingXPathString + "/m:inline/a:feed/a:author", xmlString);
assertXpathExists(buildingXPathString + "/m:inline/a:feed/a:author/a:name", xmlString);
assertXpathExists(buildingXPathString + "/m:inline/a:feed/a:link[@href=\"Buildings('1')/nb_Rooms\"]", xmlString);
assertXpathNotExists(buildingXPathString + "/m:inline/a:feed/a:entry", xmlString);
}
@Test(expected = EntityProviderException.class)
public void expandSelectedRoomsWithNullCallback() throws Exception {
ExpandSelectTreeNode selectTree = getSelectExpandTree("Buildings('1')", "nb_Rooms", "nb_Rooms");
HashMap<String, ODataCallback> callbacksBuilding = new HashMap<String, ODataCallback>();
callbacksBuilding.put("nb_Rooms", null);
EntityProviderWriteProperties properties =
EntityProviderWriteProperties.serviceRoot(BASE_URI).expandSelectTree(selectTree).callbacks(callbacksBuilding)
.build();
AtomEntityProvider provider = createAtomEntityProvider();
provider.writeEntry(MockFacade.getMockEdm().getDefaultEntityContainer().getEntitySet("Buildings"), buildingData,
properties);
}
private HashMap<String, ODataCallback> createCallbacks(final String entitySetName) throws EdmException,
ODataException {
HashMap<String, ODataCallback> callbacksEmployee = new HashMap<String, ODataCallback>();
MyCallback callback = new MyCallback(this, inlineBaseUri);
for (String navPropName : MockFacade.getMockEdm().getDefaultEntityContainer().getEntitySet(entitySetName)
.getEntityType().getNavigationPropertyNames()) {
callbacksEmployee.put(navPropName, callback);
}
return callbacksEmployee;
}
@Test
public void expandSelectedRoomWithoutCallback() throws Exception {
ExpandSelectTreeNode selectTree = getSelectExpandTree("Employees('1')", "ne_Room", "ne_Room");
EntityProviderWriteProperties properties =
EntityProviderWriteProperties.serviceRoot(BASE_URI).expandSelectTree(selectTree).build();
AtomEntityProvider provider = createAtomEntityProvider();
ODataResponse response =
provider.writeEntry(MockFacade.getMockEdm().getDefaultEntityContainer().getEntitySet("Employees"),
employeeData, properties);
String xmlString = verifyResponse(response);
verifyNavigationProperties(xmlString, F, T, F);
assertXpathNotExists("/a:entry/m:properties", xmlString);
assertXpathNotExists(roomXPathString + "/m:inline", xmlString);
}
@Test(expected = EntityProviderException.class)
public void expandSelectedRoomWithNullCallback() throws Exception {
ExpandSelectTreeNode selectTree = getSelectExpandTree("Employees('1')", "ne_Room", "ne_Room");
HashMap<String, ODataCallback> callbacksEmployee = new HashMap<String, ODataCallback>();
callbacksEmployee.put("ne_Room", null);
EntityProviderWriteProperties properties =
EntityProviderWriteProperties.serviceRoot(BASE_URI).expandSelectTree(selectTree).callbacks(callbacksEmployee)
.build();
AtomEntityProvider provider = createAtomEntityProvider();
provider.writeEntry(MockFacade.getMockEdm().getDefaultEntityContainer().getEntitySet("Employees"), employeeData,
properties);
}
private void verifyEmployees(final String path, final String xmlString) throws XpathException, IOException,
SAXException {
assertXpathExists(path, xmlString);
assertXpathExists(path + "/m:inline", xmlString);
assertXpathExists(path + "/m:inline/a:feed[@xml:base='" + inlineBaseUri.toString() + "']", xmlString);
assertXpathExists(path + "/m:inline/a:feed/a:entry", xmlString);
assertXpathExists(path + "/m:inline/a:feed/a:entry/a:id", xmlString);
assertXpathExists(path + "/m:inline/a:feed/a:entry/a:title", xmlString);
assertXpathExists(path + "/m:inline/a:feed/a:entry/a:updated", xmlString);
assertXpathExists(path + "/m:inline/a:feed/a:entry/a:category", xmlString);
assertXpathExists(path + "/m:inline/a:feed/a:entry/a:link", xmlString);
assertXpathExists(path + "/m:inline/a:feed/a:entry/a:content", xmlString);
assertXpathExists(path + "/m:inline/a:feed/a:entry/m:properties", xmlString);
assertXpathExists(path + "/m:inline/a:feed/a:entry/m:properties/d:EmployeeId", xmlString);
assertXpathExists(path + "/m:inline/a:feed/a:entry/m:properties/d:EmployeeName", xmlString);
assertXpathExists(path + "/m:inline/a:feed/a:entry/m:properties/d:ManagerId", xmlString);
assertXpathExists(path + "/m:inline/a:feed/a:entry/m:properties/d:TeamId", xmlString);
assertXpathExists(path + "/m:inline/a:feed/a:entry/m:properties/d:RoomId", xmlString);
assertXpathExists(path + "/m:inline/a:feed/a:entry/m:properties/d:Location", xmlString);
assertXpathExists(path + "/m:inline/a:feed/a:entry/m:properties/d:Age", xmlString);
assertXpathExists(path + "/m:inline/a:feed/a:entry/m:properties/d:ImageUrl", xmlString);
}
private void verifyRoom(final String path, final String xmlString) throws XpathException, IOException, SAXException {
assertXpathExists(path, xmlString);
assertXpathExists(path + "/m:inline", xmlString);
assertXpathExists(path + "/m:inline/a:entry", xmlString);
assertXpathExists(path + "/m:inline/a:entry[@xml:base='" + inlineBaseUri.toString() + "']", xmlString);
assertXpathExists(path + "/m:inline/a:entry/a:id", xmlString);
assertXpathExists(path + "/m:inline/a:entry/a:title", xmlString);
assertXpathExists(path + "/m:inline/a:entry/a:updated", xmlString);
assertXpathExists(path + "/m:inline/a:entry/a:category", xmlString);
assertXpathExists(path + "/m:inline/a:entry/a:link", xmlString);
assertXpathExists(path + "/m:inline/a:entry/a:content", xmlString);
assertXpathExists(path + "/m:inline/a:entry/a:content/m:properties", xmlString);
assertXpathExists(path + "/m:inline/a:entry/a:content/m:properties/d:Id", xmlString);
assertXpathExists(path + "/m:inline/a:entry/a:content/m:properties/d:Name", xmlString);
assertXpathExists(path + "/m:inline/a:entry/a:content/m:properties/d:Seats", xmlString);
assertXpathExists(path + "/m:inline/a:entry/a:content/m:properties/d:Version", xmlString);
}
private void verifyNavigationProperties(final String xmlString, final boolean neManager, final boolean neRoom,
final boolean neTeam) throws IOException, SAXException, XpathException {
if (neManager) {
assertXpathExists("/a:entry/a:link[@href=\"Employees('1')/ne_Manager\" and @title='ne_Manager']", xmlString);
} else {
assertXpathNotExists("/a:entry/a:link[@href=\"Employees('1')/ne_Manager\" and @title='ne_Manager']", xmlString);
}
if (neRoom) {
assertXpathExists("/a:entry/a:link[@href=\"Employees('1')/ne_Room\" and @title='ne_Room']", xmlString);
} else {
assertXpathNotExists("/a:entry/a:link[@href=\"Employees('1')/ne_Room\" and @title='ne_Room']", xmlString);
}
if (neTeam) {
assertXpathExists("/a:entry/a:link[@href=\"Employees('1')/ne_Team\" and @title='ne_Team']", xmlString);
} else {
assertXpathNotExists("/a:entry/a:link[@href=\"Employees('1')/ne_Team\" and @title='ne_Team']", xmlString);
}
}
private String verifyResponse(final ODataResponse response) throws IOException {
assertNotNull(response);
assertNotNull(response.getEntity());
assertNull("EntitypProvider should not set content header", response.getContentHeader());
String xmlString = StringHelper.inputStreamToString((InputStream) response.getEntity());
return xmlString;
}
private ExpandSelectTreeNode getSelectExpandTree(final String pathSegment, final String selectString,
final String expandString) throws Exception {
Edm edm = RuntimeDelegate.createEdm(new EdmTestProvider());
UriParserImpl uriParser = new UriParserImpl(edm);
List<PathSegment> pathSegments = new ArrayList<PathSegment>();
pathSegments.add(new ODataPathSegmentImpl(pathSegment, null));
Map<String, String> queryParameters = new HashMap<String, String>();
if (selectString != null) {
queryParameters.put("$select", selectString);
}
if (expandString != null) {
queryParameters.put("$expand", expandString);
}
UriInfo uriInfo = uriParser.parse(pathSegments, queryParameters);
ExpandSelectTreeCreator expandSelectTreeCreator =
new ExpandSelectTreeCreator(uriInfo.getSelect(), uriInfo.getExpand());
ExpandSelectTreeNode expandSelectTree = expandSelectTreeCreator.create();
assertNotNull(expandSelectTree);
return expandSelectTree;
}
}