org.assertj.db.api.SoftProxies Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of assertj-db Show documentation

AssertJ-DB - Rich and fluent assertions for testing with database

There is a newer version: 2.0.2

/**
 * 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.
 *
 * Copyright 2012-2016 the original author or authors.
 */
package org.assertj.db.api;

import net.sf.cglib.proxy.Callback;
import net.sf.cglib.proxy.CallbackFilter;
import net.sf.cglib.proxy.Enhancer;
import org.assertj.core.util.Arrays;

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

/**
 * Proxy implementation utilities.
 *
 * @author Julien Roy
 */
class SoftProxies {
  private final ErrorCollector collector = new ErrorCollector();

  SoftProxies() {
  }

  List errorsCollected() {
    return this.collector.errors();
  }

  @SuppressWarnings("unchecked")
   V create(Class assertClass, Class actualClass, T actual) {
    Enhancer enhancer = new Enhancer();
    enhancer.setSuperclass(assertClass);
    enhancer.setCallbackFilter(SoftProxies.CollectErrorsOrCreateExtractedProxy.FILTER);
    enhancer.setCallbacks(new Callback[] { this.collector, new ProxifyPositionResult(this) });
    return (V) enhancer.create((Class[]) Arrays.array(new Class[] { actualClass }), Arrays.array(new Object[] { actual }));
  }

  @SuppressWarnings("unchecked")
   V create(Class assertClass, Class[] paramClass, Object[] params) {
    Enhancer enhancer = new Enhancer();
    enhancer.setSuperclass(assertClass);
    enhancer.setCallbackFilter(SoftProxies.CollectErrorsOrCreateExtractedProxy.FILTER);
    enhancer.setCallbacks(new Callback[] { this.collector, new ProxifyPositionResult(this) });
    return (V) enhancer.create(paramClass, params);
  }

  public boolean wasSuccess() {
    return this.collector.wasSuccess();
  }

  private enum CollectErrorsOrCreateExtractedProxy implements CallbackFilter {
    FILTER;

    public int accept(Method method) {
      return this.isPositionMethod(method) ? 1 : 0;
    }

    private boolean isPositionMethod(Method method) {
      String methodName = method.getName();
      return
          java.util.Arrays.asList("change", "column", "row", "value").contains(methodName) ||
          methodName.startsWith("changeOf") ||
          methodName.startsWith("rowAt") ||
          methodName.startsWith("of");
    }
  }
}