| /* |
| * Copyright 2014 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.gwt.resources.gss; |
| |
| import static org.mockito.Mockito.doAnswer; |
| import static org.mockito.Mockito.mock; |
| import static org.mockito.Mockito.when; |
| |
| import com.google.gwt.core.ext.typeinfo.JClassType; |
| import com.google.gwt.resources.client.CssResource.ImportedWithPrefix; |
| import com.google.gwt.thirdparty.common.css.compiler.ast.CssClassSelectorNode; |
| import com.google.gwt.thirdparty.common.css.compiler.ast.CssTree; |
| import com.google.gwt.thirdparty.common.css.compiler.ast.VisitController; |
| |
| import junit.framework.TestCase; |
| |
| import org.mockito.invocation.InvocationOnMock; |
| import org.mockito.stubbing.Answer; |
| |
| import java.util.ArrayList; |
| import java.util.HashSet; |
| import java.util.List; |
| import java.util.Set; |
| |
| /** |
| * Test for {@link ClassNamesCollector}. |
| */ |
| public class ClassNamesCollectorTest extends TestCase { |
| private CssClassSelectorNode cssClassSelectorNode; |
| private CssTree cssTree; |
| private VisitController visitController; |
| private ClassNamesCollector classNamesCollector; |
| |
| @Override |
| protected void setUp() { |
| classNamesCollector = new ClassNamesCollector(); |
| cssClassSelectorNode = mock(CssClassSelectorNode.class); |
| cssTree = mock(CssTree.class); |
| visitController = mock(VisitController.class); |
| |
| when(cssTree.getVisitController()).thenReturn(visitController); |
| } |
| |
| public void testGetClassNames_noImport_collectAllClasses() { |
| // Given |
| List<CssClassSelectorNode> classSelectorNodes = createClassSelectorNodes("class1", "class2", |
| "class3"); |
| whenVisitorControllerVisitTreeThenVisitAllNodes(classSelectorNodes); |
| |
| // When |
| Set<String> classNames = classNamesCollector.getClassNames(cssTree, new HashSet<JClassType>()); |
| |
| // Then |
| assertEquals(3, classNames.size()); |
| assertTrue(classNames.contains("class1")); |
| assertTrue(classNames.contains("class2")); |
| assertTrue(classNames.contains("class3")); |
| } |
| |
| public void testGetClassNames_withImport_dontCollectImportedClasses() { |
| // Given |
| List<CssClassSelectorNode> classSelectorNodes = createClassSelectorNodes("class1", "class2", |
| "class3", "Imported-class", "ImportedFakeClass", "otherImport-class"); |
| whenVisitorControllerVisitTreeThenVisitAllNodes(classSelectorNodes); |
| |
| Set<JClassType> importedType = new HashSet<JClassType>(); |
| |
| // mock a class without @ImportedWithPrefix annotation |
| JClassType importedClassType = mock(JClassType.class); |
| when(importedClassType.getSimpleSourceName()).thenReturn("Imported"); |
| importedType.add(importedClassType); |
| |
| // mock a class with a @ImportedWithPrefix annotation |
| JClassType importedWithPrefixClassType = mock(JClassType.class); |
| when(importedWithPrefixClassType.getSimpleSourceName()).thenReturn("ImportedWithPrefix"); |
| ImportedWithPrefix importedWithPrefixAnnotation = mock(ImportedWithPrefix.class); |
| when(importedWithPrefixAnnotation.value()).thenReturn("otherImport"); |
| when(importedWithPrefixClassType.getAnnotation(ImportedWithPrefix.class)) |
| .thenReturn(importedWithPrefixAnnotation); |
| importedType.add(importedWithPrefixClassType); |
| |
| // When |
| Set<String> classNames = classNamesCollector.getClassNames(cssTree, importedType); |
| |
| // Then |
| assertEquals(4, classNames.size()); |
| assertTrue(classNames.contains("class1")); |
| assertTrue(classNames.contains("class2")); |
| assertTrue(classNames.contains("class3")); |
| assertTrue(classNames.contains("ImportedFakeClass")); |
| assertFalse(classNames.contains("otherImport-class")); |
| assertFalse(classNames.contains("Imported-class")); |
| } |
| |
| private List<CssClassSelectorNode> createClassSelectorNodes(String... classNames) { |
| List<CssClassSelectorNode> classSelectorNodes = new ArrayList<CssClassSelectorNode>(classNames |
| .length); |
| |
| for (String className : classNames) { |
| CssClassSelectorNode node = mock(CssClassSelectorNode.class); |
| when(node.getRefinerName()).thenReturn(className); |
| classSelectorNodes.add(node); |
| } |
| |
| return classSelectorNodes; |
| } |
| |
| private void whenVisitorControllerVisitTreeThenVisitAllNodes(final List<CssClassSelectorNode> |
| nodes) { |
| doAnswer(new Answer() { |
| @Override |
| public Object answer(InvocationOnMock invocation) throws Throwable { |
| for (CssClassSelectorNode node : nodes) { |
| classNamesCollector.enterClassSelector(node); |
| } |
| return null; |
| } |
| }).when(visitController).startVisit(classNamesCollector); |
| } |
| } |