TestInjectMockResourceFactory.java

/*
   Copyright (c) 2012 LinkedIn Corp.

   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.
*/

/**
 * $Id: $
 */


package com.linkedin.restli.server.mock;

import com.linkedin.data.ByteString;
import com.linkedin.common.callback.Callback;
import com.linkedin.r2.message.RequestContext;
import com.linkedin.r2.message.rest.RestRequest;
import com.linkedin.r2.message.rest.RestRequestBuilder;
import com.linkedin.r2.message.rest.RestResponse;
import com.linkedin.restli.server.RestLiConfig;
import com.linkedin.restli.server.RestLiServer;
import com.linkedin.restli.server.annotations.Action;
import com.linkedin.restli.server.annotations.RestLiActions;
import org.testng.Assert;
import org.testng.annotations.Test;

import javax.inject.Inject;
import javax.inject.Named;
import java.net.URI;
import java.util.Collections;
import java.util.List;
import java.util.Map;


/**
 * @author Josh Walker
 * @version $Revision: $
 */

public class TestInjectMockResourceFactory
{
  public static class CounterBean
  {
    public int _counter = 0;
  }

  public static class SpecialBean
  {

  }

  public static class DerivedBean extends SpecialBean
  {

  }

  @RestLiActions(name="test")
  public static class Resource
  {
    @Inject @Named("counterBean")
    private CounterBean _counter;

    @Inject
    private SpecialBean _myOtherBean;

    @Action(name="validateBeans")
    public void validateBeans()
    {
      Assert.assertNotNull(_counter);
      Assert.assertNotNull(_myOtherBean);
      ++_counter._counter;
    }
  }

  @Test
  public void testMockInjection()
  {
    InjectMockResourceFactory factory =
            new InjectMockResourceFactory(new SimpleBeanProvider()
                                                  .add("counterBean", new CounterBean())
                                                  .add("mySpecialBean", new SpecialBean()),
                                          Resource.class);
    Resource instance = factory.create(Resource.class);
    instance.validateBeans();
  }

  @Test
  public void testMockInjectionSubClass()
  {
    InjectMockResourceFactory factory =
            new InjectMockResourceFactory(new SimpleBeanProvider()
                                                  .add("counterBean", new CounterBean())
                                                  .add("mySpecialBean", new DerivedBean()),
                                          Resource.class);

    Resource instance = factory.create(Resource.class);
    instance.validateBeans();
  }

  public static class MockRequest implements RestRequest
  {
    @Override
    public String getMethod()
    {
      return "POST";
    }

    @Override
    public RestRequestBuilder builder()
    {
      return null;
    }

    @Override
    public URI getURI()
    {
      return URI.create("/test?action=validateBeans");
    }

    @Override
    public String getHeader(String name)
    {
      return null;
    }

    @Override
    public List<String> getHeaderValues(String name)
    {
      return Collections.emptyList();
    }

    @Override
    public Map<String, String> getHeaders()
    {
      return Collections.emptyMap();
    }

    @Override
    public List<String> getCookies()
    {
      return Collections.emptyList();
    }

    @Override
    public ByteString getEntity()
    {
      return ByteString.copy(new byte[0]);
    }
  }

  @Test
  public void testMockInjectionViaServer()
  {
    final CounterBean counter = new CounterBean();

    InjectMockResourceFactory factory =
            new InjectMockResourceFactory(new SimpleBeanProvider()
                                                  .add("counterBean", counter)
                                                  .add("mySpecialBean", new SpecialBean()));
    RestLiConfig config = new RestLiConfig();
    config.addResourcePackageNames("com.linkedin.restli.server.mock");
    RestLiServer server = new RestLiServer(config, factory);
    server.handleRequest(new MockRequest(), new RequestContext(), new Callback<RestResponse>()
    {
      @Override
      public void onError(Throwable e)
      {
        Assert.fail("expected success response");
      }

      @Override
      public void onSuccess(RestResponse result)
      {
        Assert.assertEquals(counter._counter, 1);
      }
    });
  }
}