/*
* $Id$
* JBoss, Home of Professional Open Source
* Copyright 2010, Red Hat, Inc. and individual contributors
* by the @authors tag. See the copyright.txt in the distribution for a
* full listing of individual contributors.
*
* 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.jboss.test.faces.mockito.component;
import static org.mockito.Mockito.when;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.Iterator;
import java.util.List;
import java.util.Map;
import javax.faces.component.UIComponent;
import org.mockito.ArgumentCaptor;
import org.mockito.invocation.InvocationOnMock;
import org.mockito.stubbing.Answer;
/**
* <p class="changed_added_4_0">
* This class builds tree of mock components
* </p>
*
* @author asmirnov@exadel.com
*
*/
public class TreeBuilderImpl<C extends UIComponent> implements TreeBuilder<C> {
private final C component;
private final List<TreeBuilder<?>> childrenBuilders;
private final List<UIComponent> children;
private final Map<String, UIComponent> facets;
protected TreeBuilderImpl(C component) {
this.component = component;
this.childrenBuilders = new ArrayList<TreeBuilder<?>>();
this.children = new ArrayList<UIComponent>();
this.facets = new HashMap<String, UIComponent>();
when(component.getChildren()).thenReturn(children);
when(component.getFacets()).thenReturn(facets);
when(component.getChildCount()).thenAnswer(new Answer<Integer>() {
public Integer answer(InvocationOnMock invocation) throws Throwable {
return children.size();
}
});
final ArgumentCaptor<String> argument = ArgumentCaptor.forClass(String.class);
when(component.getFacet(argument.capture())).thenAnswer(new Answer<UIComponent>() {
public UIComponent answer(InvocationOnMock invocation) throws Throwable {
return facets.get(argument.getValue());
}
});
when(component.getFacetCount()).thenAnswer(new Answer<Integer>() {
public Integer answer(InvocationOnMock invocation) throws Throwable {
return facets.size();
}
});
when(component.getFacetsAndChildren()).thenAnswer(new Answer<Iterator<UIComponent>>() {
public Iterator<UIComponent> answer(InvocationOnMock invocation) throws Throwable {
ArrayList<UIComponent> facetsAndChildren = new ArrayList<UIComponent>(facets.values());
facetsAndChildren.addAll(children);
return facetsAndChildren.iterator();
}
});
}
/*
* (non-Javadoc)
*
* @see org.jboss.test.faces.mock.component.TreeBuilder#setId(java.lang.String)
*/
public TreeBuilder<C> id(String id) {
when(component.getId()).thenReturn(id);
return this;
}
public void visitTree(TreeVisitor visitor) {
visitor.visit(this);
for (TreeBuilder<?> child : childrenBuilders) {
child.visitTree(visitor);
}
}
/*
* (non-Javadoc)
*
* @see org.jboss.test.faces.mock.component.TreeBuilder#getComponent()
*/
public C getComponent() {
return this.component;
}
public TreeBuilder<C> children(TreeBuilder<?>... builders) {
for (TreeBuilder<?> treeBuilder : builders) {
children.add(treeBuilder.getComponent());
addChildBuilder(treeBuilder);
}
return this;
}
private void addChildBuilder(TreeBuilder<?> treeBuilder) {
childrenBuilders.add(treeBuilder);
when(treeBuilder.getComponent().getParent()).thenReturn(component);
}
public TreeBuilder<C> facets(Facet<?>... facets) {
for (Facet<?> facet : facets) {
this.facets.put(facet.getName(), facet.getBuilder().getComponent());
addChildBuilder(facet.getBuilder());
}
return null;
}
}