com.google.api.tools.framework.model.SymbolTableTest Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of api-compiler Show documentation
Google API Compiler.
There is a newer version: 0.0.8
/*
 * Copyright (C) 2016 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.api.tools.framework.model;

import com.google.common.collect.ImmutableMap;
import com.google.common.collect.ImmutableSet;
import com.google.common.collect.Lists;
import com.google.protobuf.DescriptorProtos.FieldDescriptorProto.Type;

import junit.framework.Assert;

import org.junit.Test;
import org.junit.runner.RunWith;
import org.junit.runners.JUnit4;
import org.mockito.Mockito;

/**
 * Tests for {@link SymbolTable}.
 */
@RunWith(JUnit4.class)
public class SymbolTableTest {

  private final TypeRef m1 = TypeRef.of(Mockito.mock(MessageType.class));
  private final TypeRef m2 = TypeRef.of(Mockito.mock(MessageType.class));
  private final Interface s1 = Mockito.mock(Interface.class);
  private final Interface s2 = Mockito.mock(Interface.class);
  private final Method rpc1 = Mockito.mock(Method.class);

  private final SymbolTable table =
      new SymbolTable(
          ImmutableMap.of("a.b.s1", s1, "a.b.s2", s2),
          ImmutableMap.of("a.b.m", m1, "a.b.m.m", m2),
          ImmutableSet.of("foo", "bar"),
          ImmutableMap.of("rpc1", Lists.newArrayList(rpc1)));

  @Test public void testResolveInterface() {
    Assert.assertSame(s1, table.resolveInterface("a.b",  "s1"));
    Assert.assertSame(s1, table.resolveInterface("a.b",  "b.s1"));
    Assert.assertSame(s1, table.resolveInterface("a.b",  "a.b.s1"));
    Assert.assertSame(s1, table.resolveInterface("a.b",  ".a.b.s1"));
  }

  @Test public void testResolveType() {
    Assert.assertSame(TypeRef.of(Type.TYPE_INT32), table.resolveType("a.b",  "int32"));
    Assert.assertSame(m1, table.resolveType("a.b",  "m"));
    Assert.assertSame(m2, table.resolveType("a.b",  "m.m"));
    Assert.assertSame(m2, table.resolveType("a.b.m",  "m"));
    Assert.assertSame(m1, table.resolveType("a.b.m",  ".a.b.m"));
  }

  @Test public void testContainsFieldName() {
    Assert.assertTrue(table.containsFieldName("foo"));
    Assert.assertTrue(table.containsFieldName("bar"));
    Assert.assertFalse(table.containsFieldName("baz"));
  }

  @Test public void testContainsMethodName() {
    Assert.assertSame(rpc1, table.lookupMethodSimpleName("rpc1").get(0));
    Assert.assertNull(table.lookupMethodSimpleName("rpc2"));
  }
}