| /* |
| * Copyright (C) 2014 The Android Open Source Project |
| * |
| * 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.android.tools.idea.fileTypes; |
| |
| import com.android.tools.idea.rendering.FlagManager; |
| import com.intellij.openapi.vfs.VirtualFile; |
| import com.intellij.psi.PsiFile; |
| import com.intellij.psi.PsiManager; |
| import org.jetbrains.android.AndroidTestCase; |
| import org.jetbrains.annotations.NotNull; |
| import org.jetbrains.annotations.Nullable; |
| |
| import javax.swing.*; |
| |
| public class AndroidIconProviderTest extends AndroidTestCase { |
| public void test() { |
| VirtualFile vFile = myFixture.copyFileToProject("R.java", "gen/p1/p2/R.java"); |
| PsiFile file = PsiManager.getInstance(getProject()).findFile(vFile); |
| assertNotNull(file); |
| |
| checkIcon("AndroidManifest.xml", null); |
| checkIcon("gen/p1/p2/R.java", null); |
| checkIcon("res/wrong/path.xml", null); |
| checkIcon("res/layout/file.xml", null); |
| checkIcon("res/layout-land/file.xml", null); |
| checkIcon("res/values-no/strings.xml", "NO"); |
| checkIcon("res/values-en-rUS/strings.xml", "US"); |
| checkIcon("res/values-en-rGB/strings.xml", "GB"); |
| } |
| |
| private void checkIcon(@NotNull String path, @Nullable String region) { |
| AndroidIconProvider provider = new AndroidIconProvider(); |
| VirtualFile file = myFixture.copyFileToProject("R.java", path); // file content does not matter |
| PsiFile psiFile = PsiManager.getInstance(getProject()).findFile(file); |
| assertNotNull(psiFile); |
| int flags = 0; |
| Icon icon = provider.getIcon(psiFile, flags); |
| if (region == null) { |
| assertNull(icon); |
| } else { |
| assertSame(FlagManager.get().getFlag(region), icon); |
| } |
| } |
| } |