/*
 * Copyright (C) 2011 eXo Platform SAS.
 *
 * This is free software; you can redistribute it and/or modify it
 * under the terms of the GNU Lesser General Public License as
 * published by the Free Software Foundation; either version 2.1 of
 * the License, or (at your option) any later version.
 *
 * This software is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
 * Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public
 * License along with this software; if not, write to the Free
 * Software Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA
 * 02110-1301 USA, or see the FSF site: http://www.fsf.org.
 */

package org.exoplatform.portal.config;

import org.exoplatform.container.PortalContainer;
import org.exoplatform.portal.config.model.*;
import org.exoplatform.portal.mop.SiteType;
import org.exoplatform.portal.mop.importer.ImportMode;
import org.exoplatform.portal.mop.page.*;
import org.exoplatform.portal.mop.service.LayoutService;
import org.exoplatform.portal.mop.storage.PageStorage;
import org.exoplatform.portal.pom.spi.portlet.Portlet;

/**
 * @author <a href="trongtt@gmail.com">Trong Tran</a>
 * @version $Revision$
 */
public class TestSiteDataImportOverwrite extends AbstractSiteDataImportTest {

  @Override
  protected ImportMode getMode() {
    return ImportMode.OVERWRITE;
  }

  @Override
  protected void afterOneBootWithExtention(PortalContainer container) throws Exception {
    begin();
    // Test portal
    LayoutService layoutService = container.getComponentInstanceOfType(LayoutService.class);
    PageStorage pageStorage = container.getComponentInstanceOfType(PageStorage.class);
    PortalConfig portal = layoutService.getPortalConfig("classic");
    Container layout = portal.getPortalLayout();
    assertEquals(1, layout.getChildren().size());
    Application<Portlet> layoutPortlet = (Application<Portlet>) layout.getChildren().get(0);
    assertEquals("site2/layout", layoutService.getId(layoutPortlet.getState()));
    PageContext page = pageStorage.loadPage(PageKey.parse("portal::classic::home"));
    assertNull(page);
    page = pageStorage.loadPage(PageKey.parse("portal::classic::page1"));
    assertNotNull(page);
    assertEquals("site 2", page.getState().getDisplayName());
    page = pageStorage.loadPage(PageKey.parse("portal::classic::page2"));
    assertNotNull(page);
    assertEquals("site 2", page.getState().getDisplayName());
    // Test group
    portal = layoutService.getPortalConfig(SiteType.GROUP.getName(), "/platform/administrators");
    layout = portal.getPortalLayout();
    assertEquals(1, layout.getChildren().size());
    layoutPortlet = (Application<Portlet>) layout.getChildren().get(0);
    assertEquals("site1/layout", layoutService.getId(layoutPortlet.getState()));
    page = pageStorage.loadPage(PageKey.parse("group::/platform/administrators::page1"));
    assertNotNull(page);
    assertEquals("site 2", page.getState().getDisplayName());

    end();
  }

  @Override
  protected void afterSecondBootWithOverride(PortalContainer container) throws Exception {
    begin();

    LayoutService layoutService = container.getComponentInstanceOfType(LayoutService.class);
    PageStorage pageStorage = container.getComponentInstanceOfType(PageStorage.class);
    PortalConfig portal = layoutService.getPortalConfig("classic");
    Container layout = portal.getPortalLayout();
    assertEquals(1, layout.getChildren().size());
    Application<Portlet> layoutPortlet = (Application<Portlet>) layout.getChildren().get(0);
    assertEquals("site2/layout", layoutService.getId(layoutPortlet.getState()));

    //
    PageContext home = pageStorage.loadPage(PageKey.parse("portal::classic::home"));
    assertNull(home);

    PageContext page1 = pageStorage.loadPage(PageKey.parse("portal::classic::page1"));
    assertNotNull(page1);
    assertEquals("site 2", page1.getState().getDisplayName());

    PageContext page2 = pageStorage.loadPage(PageKey.parse("portal::classic::page2"));
    assertNotNull(page2);
    assertEquals("site 2", page2.getState().getDisplayName());

    end();
  }
}
