when(photoContainer.getEntitySet("Photos")).thenReturn(photoEntitySet);
when(photoContainer.getName()).thenReturn("Container2");
when(photoEntitySet.getEntityContainer()).thenReturn(photoContainer);
Edm edm = mock(Edm.class);
EdmServiceMetadata serviceMetadata = mock(EdmServiceMetadata.class);
when(serviceMetadata.getDataServiceVersion()).thenReturn("MockEdm");
when(edm.getServiceMetadata()).thenReturn(serviceMetadata);
when(edm.getDefaultEntityContainer()).thenReturn(defaultContainer);
when(edm.getEntityContainer("Container1")).thenReturn(specificContainer);
when(edm.getEntityContainer("Container2")).thenReturn(photoContainer);
when(edm.getEntityType("RefScenario", "Employee")).thenReturn(employeeType);
when(edm.getEntityType("RefScenario", "Team")).thenReturn(teamType);
when(edm.getEntityType("RefScenario", "Room")).thenReturn(roomType);
when(edm.getEntityType("RefScenario", "Building")).thenReturn(buildingType);
when(edm.getComplexType("RefScenario", "c_Location")).thenReturn(locationComplexType);
when(edm.getEntityType("RefScenario2", "Photo")).thenReturn(photoEntityType);
return edm;
}