EXTCDI-302 use CodiFacesContextWrapper also for PartialViewContext

git-svn-id: https://svn.apache.org/repos/asf/myfaces/extensions/cdi/trunk@1412002 13f79535-47bb-0310-9956-ffa450edef68
diff --git a/jee-modules/jsf20-module/impl/src/main/java/org/apache/myfaces/extensions/cdi/jsf2/impl/listener/request/CodiFacesContextWrapper.java b/jee-modules/jsf20-module/impl/src/main/java/org/apache/myfaces/extensions/cdi/jsf2/impl/listener/request/CodiFacesContextWrapper.java
index c507335..fb9d234 100644
--- a/jee-modules/jsf20-module/impl/src/main/java/org/apache/myfaces/extensions/cdi/jsf2/impl/listener/request/CodiFacesContextWrapper.java
+++ b/jee-modules/jsf20-module/impl/src/main/java/org/apache/myfaces/extensions/cdi/jsf2/impl/listener/request/CodiFacesContextWrapper.java
@@ -28,12 +28,15 @@
 import org.apache.myfaces.extensions.cdi.jsf2.impl.security.SecurityAwareViewHandler;

 import org.apache.myfaces.extensions.cdi.message.api.Message;

 

+import javax.faces.FactoryFinder;

 import javax.faces.application.FacesMessage;

 import javax.faces.application.Application;

 import javax.faces.context.ExceptionHandler;

 import javax.faces.context.ExternalContext;

 import javax.faces.context.FacesContext;

 import javax.faces.context.FacesContextWrapper;

+import javax.faces.context.PartialViewContext;

+import javax.faces.context.PartialViewContextFactory;

 import java.util.List;

 import java.util.Map;

 import java.util.concurrent.CopyOnWriteArrayList;

@@ -55,6 +58,8 @@
 

     private boolean defaultErrorViewExceptionHandlerActivated;

 

+    private PartialViewContext partialViewContext;

+

     CodiFacesContextWrapper(FacesContext wrappedFacesContext)

     {

         this.wrappedFacesContext = wrappedFacesContext;

@@ -153,6 +158,18 @@
         }

     }

 

+    @Override

+    public PartialViewContext getPartialViewContext()

+    {

+        if (this.partialViewContext == null)

+        {

+            PartialViewContextFactory partialViewContextFactory = (PartialViewContextFactory)

+                    FactoryFinder.getFactory(FactoryFinder.PARTIAL_VIEW_CONTEXT_FACTORY);

+            this.partialViewContext = partialViewContextFactory.getPartialViewContext(this);

+        }

+        return this.partialViewContext;

+    }

+

     /**

      * {@inheritDoc}

      */