ObjectToGsonFunctionTest.java

/*
 * Copyright 2014 mango.jfaster.org
 *
 * The Mango Project licenses this file to you 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.jfaster.mango.invoker.function.json;

import com.google.common.collect.Lists;
import com.google.gson.Gson;
import org.jfaster.mango.annotation.Getter;
import org.jfaster.mango.invoker.FunctionalGetterInvoker;
import org.jfaster.mango.invoker.GetterInvoker;
import org.junit.Test;

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

import static org.hamcrest.CoreMatchers.is;
import static org.hamcrest.MatcherAssert.assertThat;

/**
 * @author ash
 */
public class ObjectToGsonFunctionTest {

  @Test
  public void testApply() throws Exception {
    A a = new A();
    List<Integer> list = Lists.newArrayList(1, 2, 3);
    a.setList(list);
    Method m = A.class.getDeclaredMethod("getList");
    GetterInvoker invoker = FunctionalGetterInvoker.create("list", m);
    String r = (String) invoker.invoke(a);
    assertThat(r, is(new Gson().toJson(list)));

    B b = new B(3, 5);
    a.setB(b);
    Method m2 = A.class.getDeclaredMethod("getB");
    GetterInvoker invoker2 = FunctionalGetterInvoker.create("b", m2);
    String r2 = (String) invoker2.invoke(a);
    assertThat(r2, is(new Gson().toJson(b)));
  }

  static class A {
    private List<Integer> list;
    private B b;

    @Getter(ObjectToGsonFunction.class)
    List<Integer> getList() {
      return list;
    }

    void setList(List<Integer> list) {
      this.list = list;
    }

    @Getter(ObjectToGsonFunction.class)
    B getB() {
      return b;
    }

    void setB(B b) {
      this.b = b;
    }
  }

  public static class B {
    private int x;
    private int y;

    public B(int x, int y) {
      this.x = x;
      this.y = y;
    }

    public int getX() {
      return x;
    }

    public void setX(int x) {
      this.x = x;
    }

    public int getY() {
      return y;
    }

    public void setY(int y) {
      this.y = y;
    }
  }


  @Test
  public void testApplyArray() throws Exception {
    G g = new G();

    int[] a = new int[]{2, 3, 4};
    g.setA(a);
    Method m = G.class.getDeclaredMethod("getA");
    GetterInvoker invoker = FunctionalGetterInvoker.create("a", m);
    String r = (String) invoker.invoke(g);
    assertThat(r, is(new Gson().toJson(a)));

    B[] b = new B[]{new B(4, 5), new B(7, 8)};
    g.setB(b);
    Method m2 = G.class.getDeclaredMethod("getB");
    GetterInvoker invoker2 = FunctionalGetterInvoker.create("b", m2);
    String r2 = (String) invoker2.invoke(g);
    assertThat(r2, is(new Gson().toJson(b)));

    Integer[] c = new Integer[]{1, 9, 5};
    g.setC(c);
    Method m3 = G.class.getDeclaredMethod("getC");
    GetterInvoker invoker3 = FunctionalGetterInvoker.create("c", m3);
    String r3 = (String) invoker3.invoke(g);
    assertThat(r3, is(new Gson().toJson(c)));
  }

  static class G {
    private int[] a;
    private B[] b;
    private Integer[] c;

    @Getter(ObjectToGsonFunction.class)
    int[] getA() {
      return a;
    }

    void setA(int[] a) {
      this.a = a;
    }

    @Getter(ObjectToGsonFunction.class)
    B[] getB() {
      return b;
    }

    void setB(B[] b) {
      this.b = b;
    }

    @Getter(ObjectToGsonFunction.class)
    Integer[] getC() {
      return c;
    }

    void setC(Integer[] c) {
      this.c = c;
    }
  }

}