tapestry/tapestry-framework/src/test/org/apache/tapestry/enhance/TestDispatchToInjectWorker.java

// Copyright 2005 The Apache Software Foundation
//
// Licensed 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 org.apache.tapestry.enhance;

import org.apache.hivemind.ErrorLog;
import org.apache.hivemind.Location;
import org.apache.tapestry.BaseComponentTestCase;
import org.apache.tapestry.html.BasePage;
import org.apache.tapestry.spec.IComponentSpecification;
import org.apache.tapestry.spec.InjectSpecification;
import org.apache.tapestry.spec.InjectSpecificationImpl;
import static org.easymock.EasyMock.expect;
import static org.easymock.EasyMock.expectLastCall;
import org.testng.annotations.Test;

import java.util.Collections;
import java.util.Map;

/**
 * Tests for {@link org.apache.tapestry.enhance.DispatchToInjectWorker}.
 *
 * @author Howard M. Lewis Ship
 * @since 4.0
 */
@Test
public class TestDispatchToInjectWorker extends BaseComponentTestCase
{
    private InjectSpecification newInjectSpecification(String propertyName, String type,
                                                       String object)
    {
        return newInjectSpecification(propertyName, type, object, null);
    }

    private InjectSpecification newInjectSpecification(String propertyName, String type,
                                                       String object, Location location)
    {
        InjectSpecification result = new InjectSpecificationImpl();
        result.setProperty(propertyName);
        result.setType(type);
        result.setObject(object);
        result.setLocation(location);

        return result;
    }

    private IComponentSpecification newSpec(InjectSpecification injectSpec)
    {
        IComponentSpecification spec = newSpec();

        expect(spec.getInjectSpecifications()).andReturn(Collections.singletonList(injectSpec));

        return spec;
    }

    private Map newMap(String key, Object value)
    {
        return Collections.singletonMap(key, value);
    }

    public void test_Success()
    {
        EnhancementOperation op = newOp();
        InjectSpecification is = newInjectSpecification("property", "object", "service:Foo");
        InjectEnhancementWorker worker = newWorker();
        Map map = newMap("object", worker);
        IComponentSpecification spec = newSpec(is);

        worker.performEnhancement(op, is);

        replay();

        DispatchToInjectWorker d = new DispatchToInjectWorker();
        d.setInjectWorkers(map);

        d.performEnhancement(op, spec);

        verify();
    }

    public void test_Unknown_Type()
    {
        Location l = newLocation();
        EnhancementOperation op = newOp();
        InjectSpecification is = newInjectSpecification(
          "injectedProperty",
          "object",
          "service:Foo",
          l);
        IComponentSpecification spec = newSpec(is);
        ErrorLog log = newErrorLog();

        log.error(EnhanceMessages.unknownInjectType("injectedProperty", "object"), l, null);

        replay();

        DispatchToInjectWorker d = new DispatchToInjectWorker();
        d.setInjectWorkers(Collections.EMPTY_MAP);
        d.setErrorLog(log);

        d.performEnhancement(op, spec);

        verify();
    }

    public void test_Failure()
    {
        Location l = newLocation();

        EnhancementOperation op = newOp();
        InjectSpecification is = newInjectSpecification("myProperty", "object", "service:Foo", l);
        InjectEnhancementWorker worker = newMock(InjectEnhancementWorker.class);

        Map map = newMap("object", worker);
        IComponentSpecification spec = newSpec(is);

        Throwable t = new RuntimeException("Simulated failure.");
        ErrorLog log = newErrorLog();

        worker.performEnhancement(op, is);
        expectLastCall().andThrow(t);

        expect(op.getBaseClass()).andReturn(BasePage.class);

        log
          .error(
            "Error adding property myProperty to class org.apache.tapestry.html.BasePage: Simulated failure.",
            l,
            t);

        replay();

        DispatchToInjectWorker d = new DispatchToInjectWorker();
        d.setInjectWorkers(map);
        d.setErrorLog(log);

        d.performEnhancement(op, spec);

        verify();
    }

    private InjectEnhancementWorker newWorker()
    {
        return newMock(InjectEnhancementWorker.class);
    }

    private ErrorLog newErrorLog()
    {
        return newMock(ErrorLog.class);
    }

    private EnhancementOperation newOp()
    {
        return newMock(EnhancementOperation.class);
    }
}