tests_mgrapp/src/com/google/codereview/manager/unpack/ReceivedBundleUnpackerTest.java

// Copyright 2008 Google Inc.
//
// 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.

package com.google.codereview.manager.unpack;

import com.google.codereview.TrashTestCase;
import com.google.codereview.internal.NextReceivedBundle.NextReceivedBundleRequest;
import com.google.codereview.internal.NextReceivedBundle.NextReceivedBundleResponse;
import com.google.codereview.internal.UpdateReceivedBundle.UpdateReceivedBundleRequest;
import com.google.codereview.internal.UpdateReceivedBundle.UpdateReceivedBundleResponse;
import com.google.codereview.manager.Backend;
import com.google.codereview.manager.RepositoryCache;
import com.google.codereview.rpc.MockRpcChannel;
import com.google.protobuf.ByteString;
import com.google.protobuf.Message;
import com.google.protobuf.RpcCallback;
import com.google.protobuf.RpcChannel;
import com.google.protobuf.RpcController;
import com.google.protobuf.Descriptors.MethodDescriptor;

import org.spearce.jgit.lib.Repository;

import java.io.File;

public class ReceivedBundleUnpackerTest extends TrashTestCase {
  private MockRpcChannel rpc;
  private RepositoryCache repoCache;
  private Backend server;

  @Override
  protected void setUp() throws Exception {
    super.setUp();

    rpc = new MockRpcChannel();
    repoCache = new RepositoryCache(tempRoot);
    server = new Backend(repoCache, rpc, null, null);
    new Repository(new File(tempRoot, "foo.git")).create();
  }

  public void testNextReceivedBundle_EmptyQueue() throws Exception {
    final ReceivedBundleUnpacker rbu = newRBU();
    rpc.add(new RpcChannel() {
      public void callMethod(final MethodDescriptor method,
          final RpcController controller, final Message request,
          final Message responsePrototype, final RpcCallback<Message> done) {
        assertEquals("NextReceivedBundle", method.getName());
        assertSame(NextReceivedBundleRequest.getDefaultInstance(), request);

        final NextReceivedBundleResponse.Builder r =
            NextReceivedBundleResponse.newBuilder();
        r.setStatusCode(NextReceivedBundleResponse.CodeType.QUEUE_EMPTY);
        done.run(r.build());
      }
    });
    rbu.run();
    rpc.assertAllCallsMade();
  }

  public void testNextReceivedBundle_GetBundle() throws Exception {
    final String bundleKey = "bundle-key-abcd123efg";
    final NextReceivedBundleResponse nrb[] = new NextReceivedBundleResponse[1];
    final UpdateReceivedBundleRequest urbr[] =
        new UpdateReceivedBundleRequest[1];

    final ReceivedBundleUnpacker rbu =
        new ReceivedBundleUnpacker(server) {
          @Override
          protected UpdateReceivedBundleRequest unpackImpl(
              final NextReceivedBundleResponse rsp) {
            assertNotNull("unpackImpl only after bundle available", nrb[0]);
            assertSame(nrb[0], rsp);

            final UpdateReceivedBundleRequest.Builder r =
                UpdateReceivedBundleRequest.newBuilder();
            r.setBundleKey(bundleKey);
            r.setStatusCode(UpdateReceivedBundleRequest.CodeType.UNPACKED_OK);
            urbr[0] = r.build();
            return urbr[0];
          }
        };

    rpc.add(new RpcChannel() {
      public void callMethod(final MethodDescriptor method,
          final RpcController controller, final Message request,
          final Message responsePrototype, final RpcCallback<Message> done) {
        assertEquals("NextReceivedBundle", method.getName());
        assertSame(NextReceivedBundleRequest.getDefaultInstance(), request);

        final NextReceivedBundleResponse.Builder r =
            NextReceivedBundleResponse.newBuilder();
        r.setStatusCode(NextReceivedBundleResponse.CodeType.BUNDLE_AVAILABLE);
        r.setBundleKey(bundleKey);
        r.setBundleData(ByteString.EMPTY);
        r.setDestProject("foo.git");
        r.setDestProjectKey("project:foo.git");
        r.setDestBranchKey("branch:refs/heads/master");
        r.setOwner("author@example.com");
        nrb[0] = r.build();
        done.run(nrb[0]);
      }
    });

    rpc.add(new RpcChannel() {
      public void callMethod(final MethodDescriptor method,
          final RpcController controller, final Message request,
          final Message responsePrototype, final RpcCallback<Message> done) {
        assertEquals("UpdateReceivedBundle", method.getName());
        assertSame(urbr[0], request);

        final UpdateReceivedBundleResponse.Builder r =
            UpdateReceivedBundleResponse.newBuilder();
        r.setStatusCode(UpdateReceivedBundleResponse.CodeType.UPDATED);
        done.run(r.build());
      }
    });

    rpc.add(new RpcChannel() {
      public void callMethod(final MethodDescriptor method,
          final RpcController controller, final Message request,
          final Message responsePrototype, final RpcCallback<Message> done) {
        assertEquals("NextReceivedBundle", method.getName());
        assertSame(NextReceivedBundleRequest.getDefaultInstance(), request);

        final NextReceivedBundleResponse.Builder r =
            NextReceivedBundleResponse.newBuilder();
        r.setStatusCode(NextReceivedBundleResponse.CodeType.QUEUE_EMPTY);
        done.run(r.build());
      }
    });

    rbu.run();
    assertNotNull(nrb[0]);
    assertNotNull(urbr[0]);
    rpc.assertAllCallsMade();
  }

  public void testNextReceivedBundle_RpcFailure() throws Exception {
    final ReceivedBundleUnpacker rbu = newRBU();
    rpc.add(new RpcChannel() {
      public void callMethod(final MethodDescriptor method,
          final RpcController controller, final Message request,
          final Message responsePrototype, final RpcCallback<Message> done) {
        assertEquals("NextReceivedBundle", method.getName());
        controller.setFailed("mock failure");
      }
    });
    rbu.run();
    rpc.assertAllCallsMade();
  }

  public void testNextReceivedBundle_RuntimeException() throws Exception {
    final String msg = "test-a-message-win-a-prize";
    final ReceivedBundleUnpacker rbu = newRBU();
    rpc.add(new RpcChannel() {
      public void callMethod(final MethodDescriptor method,
          final RpcController controller, final Message request,
          final Message responsePrototype, final RpcCallback<Message> done) {
        assertEquals("NextReceivedBundle", method.getName());
        throw new RuntimeException(msg);
      }
    });
    try {
      rbu.run();
      fail("Unpacker did not rethrow an unexpected runtime exception");
    } catch (RuntimeException re) {
      assertEquals(msg, re.getMessage());
    }
    rpc.assertAllCallsMade();
  }

  public void testNextReceivedBundle_RuntimeError() throws Exception {
    final String msg = "test-a-message-win-a-prize";
    final ReceivedBundleUnpacker rbu = newRBU();
    rpc.add(new RpcChannel() {
      public void callMethod(final MethodDescriptor method,
          final RpcController controller, final Message request,
          final Message responsePrototype, final RpcCallback<Message> done) {
        assertEquals("NextReceivedBundle", method.getName());
        throw new OutOfMemoryError(msg);
      }
    });
    try {
      rbu.run();
      fail("Unpacker did not rethrow an unexpected OutOfMemoryError");
    } catch (OutOfMemoryError re) {
      assertEquals(msg, re.getMessage());
    }
    rpc.assertAllCallsMade();
  }

  private ReceivedBundleUnpacker newRBU() {
    return new ReceivedBundleUnpacker(server);
  }
}