com.redhat.darcy.ui.internal.ElementListHandler Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of darcy-ui Show documentation

Framework for writing page objects to automate interaction with graphical user interfaces.

The newest version!

/*
 Copyright 2014 Red Hat, Inc. and/or its affiliates.

 This file is part of darcy-ui.

 This program is free software: you can redistribute it and/or modify
 it under the terms of the GNU General Public License as published by
 the Free Software Foundation, either version 3 of the License, or
 (at your option) any later version.

 This program 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 General Public License for more details.

 You should have received a copy of the GNU General Public License
 along with this program.  If not, see .
 */

package com.redhat.darcy.ui.internal;

import com.redhat.darcy.ui.NullContextException;
import com.redhat.darcy.ui.api.ElementContext;
import com.redhat.darcy.ui.api.Locator;
import com.redhat.darcy.ui.api.elements.Element;

import java.lang.reflect.InvocationHandler;
import java.lang.reflect.InvocationTargetException;
import java.lang.reflect.Method;
import java.util.List;

/**
 * Like {@link ElementHandler}, except proxies a List of elements.
 * @see ElementHandler
 */
public class ElementListHandler implements InvocationHandler {
    private Class type;
    private Locator locator;
    private ElementContext context;
    
    private List cachedList;
    
    public ElementListHandler(Class type, Locator locator) {
        this.type = type;
        this.locator = locator;
    }

    @Override
    public Object invoke(Object proxy, Method method, Object[] args) throws Throwable {
        if ("setContext".equals(method.getName())) {
            context = (ElementContext) args[0];
            cachedList = null;
            
            return null;
        }
        
        if (context == null) {
            throw new NullContextException();
        }

        if (cachedList == null) {
            cachedList = locator.findAll(type, context);
        }

        try {
            return method.invoke(cachedList, args);
        } catch (InvocationTargetException e) {
            throw e.getCause();
        }
    }
}