| /* |
| * Copyright 2016 The Bazel Authors. All rights reserved. |
| * |
| * 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.idea.blaze.android.sync.importer.model.idea; |
| |
| import static com.google.common.truth.Truth.THROW_ASSERTION_ERROR; |
| import static com.google.common.truth.Truth.assertThat; |
| import static org.mockito.Mockito.mock; |
| |
| import com.google.idea.blaze.android.sync.model.AndroidResourceModule; |
| import com.google.idea.blaze.android.sync.model.AndroidResourceModuleRegistry; |
| import com.google.idea.blaze.base.BlazeTestCase; |
| import com.google.idea.blaze.base.ideinfo.TargetKey; |
| import com.google.idea.blaze.base.model.primitives.Label; |
| import com.intellij.openapi.module.Module; |
| import org.jetbrains.annotations.NotNull; |
| import org.junit.Test; |
| import org.junit.runner.RunWith; |
| import org.junit.runners.JUnit4; |
| |
| /** Unit tests for {@link AndroidResourceModuleRegistry}. */ |
| @RunWith(JUnit4.class) |
| public class AndroidResourceModuleRegistryTest extends BlazeTestCase { |
| private AndroidResourceModuleRegistry registry; |
| |
| @Override |
| protected void initTest( |
| @NotNull Container applicationServices, @NotNull Container projectServices) { |
| super.initTest(applicationServices, projectServices); |
| projectServices.register( |
| AndroidResourceModuleRegistry.class, new AndroidResourceModuleRegistry()); |
| |
| registry = AndroidResourceModuleRegistry.getInstance(getProject()); |
| } |
| |
| @Test |
| public void testPutAndGet() { |
| Module moduleOne = mock(Module.class); |
| Module moduleTwo = mock(Module.class); |
| Module moduleThree = mock(Module.class); |
| AndroidResourceModule resourceModuleOne = |
| AndroidResourceModule.builder(TargetKey.forPlainTarget(Label.create("//foo/bar:one"))) |
| .build(); |
| AndroidResourceModule resourceModuleTwo = |
| AndroidResourceModule.builder(TargetKey.forPlainTarget(Label.create("//foo/bar:two"))) |
| .build(); |
| AndroidResourceModule resourceModuleThree = |
| AndroidResourceModule.builder(TargetKey.forPlainTarget(Label.create("//foo/bar:three"))) |
| .build(); |
| registry.put(moduleOne, resourceModuleOne); |
| registry.put(moduleTwo, resourceModuleTwo); |
| registry.put(moduleThree, resourceModuleThree); |
| |
| assertThat(registry.get(moduleOne)).isEqualTo(resourceModuleOne); |
| assertThat(registry.get(moduleTwo)).isEqualTo(resourceModuleTwo); |
| assertThat(registry.get(moduleThree)).isEqualTo(resourceModuleThree); |
| assertThat(registry.getTargetKey(moduleOne)).isEqualTo(resourceModuleOne.targetKey); |
| assertThat(registry.getTargetKey(moduleTwo)).isEqualTo(resourceModuleTwo.targetKey); |
| assertThat(registry.getTargetKey(moduleThree)).isEqualTo(resourceModuleThree.targetKey); |
| assertThat(registry.getLabel(moduleOne)).isEqualTo(resourceModuleOne.targetKey.label); |
| assertThat(registry.getLabel(moduleTwo)).isEqualTo(resourceModuleTwo.targetKey.label); |
| assertThat(registry.getLabel(moduleThree)).isEqualTo(resourceModuleThree.targetKey.label); |
| } |
| |
| @Test |
| public void testPutSameKeyDifferentValues() { |
| Module module = mock(Module.class); |
| AndroidResourceModule resourceModuleOne = |
| AndroidResourceModule.builder(TargetKey.forPlainTarget(Label.create("//foo/bar:one"))) |
| .build(); |
| AndroidResourceModule resourceModuleTwo = |
| AndroidResourceModule.builder(TargetKey.forPlainTarget(Label.create("//foo/bar:two"))) |
| .build(); |
| registry.put(module, resourceModuleOne); |
| registry.put(module, resourceModuleTwo); |
| assertThat(registry.get(module)).isEqualTo(resourceModuleTwo); |
| } |
| |
| @Test |
| public void testPutDifferentKeysSameValue() { |
| Module moduleOne = mock(Module.class); |
| Module moduleTwo = mock(Module.class); |
| AndroidResourceModule resourceModule = |
| AndroidResourceModule.builder(TargetKey.forPlainTarget(Label.create("//foo/bar:one"))) |
| .build(); |
| registry.put(moduleOne, resourceModule); |
| try { |
| registry.put(moduleTwo, resourceModule); |
| THROW_ASSERTION_ERROR.fail("Expected IllegalArgumentException"); |
| } catch (IllegalArgumentException ignored) { |
| // ignored |
| } |
| assertThat(registry.get(moduleOne)).isEqualTo(resourceModule); |
| assertThat(registry.get(moduleTwo)).isNull(); |
| } |
| |
| @Test |
| public void testGetNull() { |
| assertThat(registry.get(null)).isNull(); |
| assertThat(registry.getTargetKey(null)).isNull(); |
| assertThat(registry.getLabel(null)).isNull(); |
| } |
| |
| @Test |
| public void testGetWithoutPut() { |
| assertThat(registry.get(mock(Module.class))).isNull(); |
| assertThat(registry.getTargetKey(mock(Module.class))).isNull(); |
| assertThat(registry.getLabel(mock(Module.class))).isNull(); |
| } |
| } |