org.assertj.db.api.ProxifyPositionResult 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.MethodInterceptor;
import net.sf.cglib.proxy.MethodProxy;
import org.assertj.core.util.Arrays;
import org.assertj.db.type.Change;
import org.assertj.db.type.Column;
import org.assertj.db.type.Row;
import org.assertj.db.type.Value;

import java.lang.reflect.Method;

import static org.assertj.db.util.Proxies.isProxified;
import static org.assertj.db.util.Proxies.unProxy;

/**
 * Method interceptor that proxify result of assertions methods.
 * Useful for navigation assertion methods like ( column(...) , row(...) ).
 *
 * @author Julien Roy
 */
class ProxifyPositionResult implements MethodInterceptor {
  private final SoftProxies proxies;

  ProxifyPositionResult(SoftProxies proxies) {
    this.proxies = proxies;
  }

  public Object intercept(Object obj, Method method, Object[] args, MethodProxy proxy) throws Throwable {
    Object result = proxy.invokeSuper(obj, args);
    if (isProxified(result.getClass()) || Arrays.isNullOrEmpty(actual(result))) {
      return result;
    }
    return this.proxies.create(result.getClass(), actualClass(result), actual(result));
  }

  private static Class[] actualClass(Object result) {

    if (result instanceof AbstractColumnAssert) {
      return Arrays.array(
          unProxy(((AbstractColumnAssert) result).origin.getClass()),
          Column.class
      );
    } else if (result instanceof AbstractRowAssert) {
      return Arrays.array(
          unProxy(((AbstractRowAssert) result).origin.getClass()),
          Row.class
      );
    } else if (result instanceof AbstractValueAssert) {
      return Arrays.array(
          unProxy(((AbstractValueAssert) result).origin.getClass()),
          Value.class
      );
    } else if (result instanceof ChangeAssert) {
      return Arrays.array(
          unProxy(((ChangeAssert) result).origin.getClass()),
          Change.class
      );
    } else if (result instanceof ChangeColumnAssert) {
      return Arrays.array(
          unProxy(((ChangeColumnAssert) result).origin.getClass()),
          String.class,
          Value.class,
          Value.class
      );
    } else {
      return Arrays.array();
    }
  }

  private static Object[] actual(Object result) {
    if (result instanceof AbstractColumnAssert) {
      return Arrays.array(
          ((AbstractColumnAssert) result).origin,
          ((AbstractColumnAssert) result).column
      );
    } else if (result instanceof AbstractRowAssert) {
      return Arrays.array(
          ((AbstractRowAssert) result).origin,
          ((AbstractRowAssert) result).row
      );
    } else if (result instanceof AbstractValueAssert) {
      return Arrays.array(
          ((AbstractValueAssert) result).origin,
          ((AbstractValueAssert) result).value
      );
    } else if (result instanceof ChangeAssert) {
      return Arrays.array(
          ((ChangeAssert) result).origin,
          ((ChangeAssert) result).change
      );
    } else if (result instanceof ChangeColumnAssert) {
      return Arrays.array(
          ((ChangeColumnAssert) result).origin,
          ((ChangeColumnAssert) result).columnName,
          ((ChangeColumnAssert) result).valueAtStartPoint,
          ((ChangeColumnAssert) result).valueAtEndPoint
      );
    } else {
      return Arrays.array();
    }
  }
}