de.unioninvestment.eai.portal.support.vaadin.junit.LiferayContext.java Source code

Introduction

Here is the source code for de.unioninvestment.eai.portal.support.vaadin.junit.LiferayContext.java

Source

/*
    
 * 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 de.unioninvestment.eai.portal.support.vaadin.junit;

import static org.mockito.Matchers.argThat;
import static org.mockito.Mockito.never;
import static org.mockito.Mockito.verify;
import static org.mockito.Mockito.when;

import javax.portlet.PortletContext;
import javax.portlet.PortletPreferences;
import javax.portlet.PortletRequest;
import javax.portlet.PortletResponse;
import javax.portlet.PortletSession;
import javax.portlet.ResourceRequest;
import javax.portlet.ResourceResponse;

import org.junit.rules.TestWatchman;
import org.junit.runners.model.FrameworkMethod;
import org.mockito.Mock;
import org.mockito.MockitoAnnotations;

import com.vaadin.data.util.converter.DefaultConverterFactory;
import com.vaadin.server.Page;
import com.vaadin.server.VaadinPortletRequest;
import com.vaadin.server.VaadinPortletResponse;
import com.vaadin.server.VaadinPortletSession;
import com.vaadin.server.VaadinRequest;
import com.vaadin.server.VaadinResponse;
import com.vaadin.server.VaadinService;
import com.vaadin.server.VaadinSession;
import com.vaadin.ui.ConnectorTracker;
import com.vaadin.ui.JavaScript;
import com.vaadin.ui.Notification.Type;
import com.vaadin.ui.UI;
import com.vaadin.util.CurrentInstance;

import de.unioninvestment.eai.portal.support.vaadin.CrudVaadinPortletService;
import de.unioninvestment.eai.portal.support.vaadin.LiferayUI;

@SuppressWarnings("deprecation")
public class LiferayContext extends TestWatchman {

    private String portletId;
    private long communityId;

    @Mock
    private LiferayUI uiMock;
    @Mock
    private Page pageMock;
    @Mock
    private VaadinPortletResponse vaadinPortletResponseMock;
    @Mock
    private ResourceResponse portletResponseMock;
    @Mock
    private VaadinPortletRequest vaadinPortletRequestMock;
    @Mock
    private ResourceRequest portletRequestMock;
    @Mock
    private PortletSession portletSessionMock;
    @Mock
    private PortletContext portletContextMock;
    @Mock
    private PortletPreferences portletPreferencesMock;
    @Mock
    private VaadinPortletSession vaadinPortletSessionMock;
    @Mock
    private CrudVaadinPortletService vaadinPortletServiceMock;
    @Mock
    private ConnectorTracker connectorTrackerMock;
    @Mock
    private JavaScript javascriptMock;

    public LiferayContext() {
        initialize();
    }

    public LiferayContext(String portletId, long communityId) {
        this.portletId = portletId;
        this.communityId = communityId;

        initialize();
    }

    private void initialize() {
        MockitoAnnotations.initMocks(this);

        UI.setCurrent(uiMock);
        if (portletId != null) {
            when(uiMock.getPortletId()).thenReturn(portletId);
            when(uiMock.getCommunityId()).thenReturn(communityId);
        }

        when(uiMock.getPage()).thenReturn(pageMock);
        when(uiMock.getConnectorTracker()).thenReturn(connectorTrackerMock);

        when(pageMock.getJavaScript()).thenReturn(javascriptMock);

        CurrentInstance.set(VaadinResponse.class, vaadinPortletResponseMock);
        when(vaadinPortletResponseMock.getPortletResponse()).thenReturn(portletResponseMock);

        CurrentInstance.set(VaadinRequest.class, vaadinPortletRequestMock);
        when(vaadinPortletRequestMock.getPortletRequest()).thenReturn(portletRequestMock);
        when(portletRequestMock.getPortletSession()).thenReturn(portletSessionMock);
        when(portletSessionMock.getPortletContext()).thenReturn(portletContextMock);
        when(portletRequestMock.getPreferences()).thenReturn(portletPreferencesMock);
        when(vaadinPortletSessionMock.getConverterFactory()).thenReturn(new DefaultConverterFactory());
        when(vaadinPortletSessionMock.getPortletSession()).thenReturn(portletSessionMock);

        CurrentInstance.set(VaadinSession.class, vaadinPortletSessionMock);
        CurrentInstance.set(VaadinService.class, vaadinPortletServiceMock);
    }

    @Override
    public void finished(FrameworkMethod method) {
        CurrentInstance.clearAll();
    }

    public void noCurrentRequest() {
        finished(null);
    }

    public void shouldShowNotification(String caption, String description, Type type) {
        verify(pageMock).showNotification(argThat(new NotificationMatcher(caption, description)));
    }

    public void shouldNotShowNotification(String caption, String description, Type type) {
        verify(pageMock, never()).showNotification(argThat(new NotificationMatcher(caption, description)));
    }

    public LiferayUI getUiMock() {
        return uiMock;
    }

    public Page getPageMock() {
        return pageMock;
    }

    public VaadinPortletResponse getVaadinPortletResponseMock() {
        return vaadinPortletResponseMock;
    }

    public PortletResponse getPortletResponseMock() {
        return portletResponseMock;
    }

    public VaadinPortletRequest getVaadinPortletRequestMock() {
        return vaadinPortletRequestMock;
    }

    public PortletRequest getPortletRequestMock() {
        return portletRequestMock;
    }

    public PortletSession getPortletSessionMock() {
        return portletSessionMock;
    }

    public PortletContext getPortletContextMock() {
        return portletContextMock;
    }

    public PortletPreferences getPortletPreferencesMock() {
        return portletPreferencesMock;
    }

    public VaadinPortletSession getVaadinSessionMock() {
        return vaadinPortletSessionMock;
    }

    public CrudVaadinPortletService getVaadinPortletServiceMock() {
        return vaadinPortletServiceMock;
    }
}