test/com/facebook/buck/android/MergeAndroidResourcesStepTest.java - buck - Git at Google

 /*
  * Copyright 2012-present Facebook, 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.facebook.buck.android;

 import static org.junit.Assert.assertEquals;
 import static org.junit.Assert.assertFalse;

 import com.facebook.buck.android.MergeAndroidResourcesStep.Resource;
 import com.facebook.buck.util.BuckConstant;
 import com.google.common.base.Function;
 import com.google.common.collect.ImmutableMap;
 import com.google.common.collect.Sets;
 import com.google.common.collect.SortedSetMultimap;

 import org.junit.Test;

 import java.io.IOException;
 import java.io.StringReader;
 import java.util.Iterator;
 import java.util.Map;
 import java.util.Set;
 import java.util.SortedSet;

 public class MergeAndroidResourcesStepTest {
   @Test
   public void testGenerateRDotJavaForMultipleSymbolsFiles() throws IOException {
     RDotTxtEntryBuilder entriesBuilder = new RDotTxtEntryBuilder();

     // Merge everything into the same package space.
     String sharedPackageName = "com.facebook.abc";
     entriesBuilder.add(new RDotTxtEntry(sharedPackageName, "a-R.txt",
         "int id a1 0x7f010001\n" +
         "int id a2 0x7f010002\n" +
         "int string a1 0x7f020001\n"));

     entriesBuilder.add(new RDotTxtEntry(sharedPackageName, "b-R.txt",
         "int id b1 0x7f010001\n" +
         "int id b2 0x7f010002\n"));

     entriesBuilder.add(new RDotTxtEntry(sharedPackageName, "c-R.txt",
         "int attr c1 0x7f010001\n" +
         "int[] styleable c1 { 0x7f010001 }\n"));

     SortedSetMultimap<String, Resource> packageNameToResources =
         MergeAndroidResourcesStep.sortSymbols(
             entriesBuilder.buildReadableMapper(),
             entriesBuilder.buildFilePathToPackageNameSet(),
             true /* reenumerate */);

     assertEquals(1, packageNameToResources.keySet().size());
     SortedSet<Resource> resources = packageNameToResources.get(sharedPackageName);
     assertEquals(7, resources.size());

     Set<String> uniqueEntries = Sets.newHashSet();
     for (Resource resource : resources) {
       if (!resource.type.equals("styleable")) {
         assertFalse("Duplicate ids should be fixed by renumerate=true; duplicate was: " +
             resource.idValueToWrite, uniqueEntries.contains(resource.idValueToWrite));
         uniqueEntries.add(resource.idValueToWrite);
       }
     }

     assertEquals(6, uniqueEntries.size());

     // All good, no need to further test whether we can write the Java file correctly...
   }

   @Test
   public void testGenerateRDotJavaForOneSymbolsFile() {
     String symbolsFile = BuckConstant.BIN_DIR +
         "/android_res/com/facebook/http/__res_resources_text_symbols__/R.txt";
     String rDotJavaPackage = "com.facebook";
     final String outputTextSymbols =
         "int id placeholder 0x7f020000\n" +
         "int string debug_http_proxy_dialog_title 0x7f030004\n" +
         "int string debug_http_proxy_hint 0x7f030005\n" +
         "int string debug_http_proxy_summary 0x7f030003\n" +
         "int string debug_http_proxy_title 0x7f030002\n" +
         "int string debug_ssl_cert_check_summary 0x7f030001\n" +
         "int string debug_ssl_cert_check_title 0x7f030000\n" +
         "int styleable SherlockMenuItem_android_visible 4\n" +
         "int[] styleable SherlockMenuView { 0x7f010026, 0x7f010027, 0x7f010028, 0x7f010029, 0x7f01002a, 0x7f01002b, 0x7f01002c, 0x7f01002d }\n";
     RDotTxtEntryBuilder entriesBuilder = new RDotTxtEntryBuilder();
     entriesBuilder.add(new RDotTxtEntry(rDotJavaPackage, symbolsFile, outputTextSymbols));
     SortedSetMultimap<String, Resource> rDotJavaPackageToResources =
         MergeAndroidResourcesStep.sortSymbols(
             entriesBuilder.buildReadableMapper(),
             entriesBuilder.buildFilePathToPackageNameSet(),
             false /* reenumerate */);

     assertEquals(1, rDotJavaPackageToResources.keySet().size());

     // Verify all of the values in the resources collection.
     SortedSet<Resource> resources = rDotJavaPackageToResources.get(rDotJavaPackage);
     assertEquals(9, resources.size());
     Iterator<Resource> iter = resources.iterator();
     assertResource("int", "id", "placeholder", "0x7f020000", iter.next());
     assertResource("int", "string", "debug_http_proxy_dialog_title", "0x7f030004", iter.next());
     assertResource("int", "string", "debug_http_proxy_hint", "0x7f030005", iter.next());
     assertResource("int", "string", "debug_http_proxy_summary", "0x7f030003", iter.next());
     assertResource("int", "string", "debug_http_proxy_title", "0x7f030002", iter.next());
     assertResource("int", "string", "debug_ssl_cert_check_summary", "0x7f030001", iter.next());
     assertResource("int", "string", "debug_ssl_cert_check_title", "0x7f030000", iter.next());
     assertResource("int", "styleable", "SherlockMenuItem_android_visible", "4", iter.next());
     assertResource("int[]",
         "styleable",
         "SherlockMenuView",
         "{ 0x7f010026, 0x7f010027, 0x7f010028, 0x7f010029, 0x7f01002a, 0x7f01002b, 0x7f01002c, 0x7f01002d }",
         iter.next());

     // Verify that the correct Java code is generated.
     String javaCode = MergeAndroidResourcesStep.generateJavaCodeForPackageAndResources(
         rDotJavaPackage, resources);
     assertEquals(
         "package com.facebook;\n" +
         "\n" +
         "public class R {\n" +
         "\n" +
         "  public static class id {\n" +
         "    public static int placeholder=0x7f020000;\n" +
         "  }\n" +
         "\n" +
         "  public static class string {\n" +
         "    public static int debug_http_proxy_dialog_title=0x7f030004;\n" +
         "    public static int debug_http_proxy_hint=0x7f030005;\n" +
         "    public static int debug_http_proxy_summary=0x7f030003;\n" +
         "    public static int debug_http_proxy_title=0x7f030002;\n" +
         "    public static int debug_ssl_cert_check_summary=0x7f030001;\n" +
         "    public static int debug_ssl_cert_check_title=0x7f030000;\n" +
         "  }\n" +
         "\n" +
         "  public static class styleable {\n" +
         "    public static int SherlockMenuItem_android_visible=4;\n" +
         "    public static int[] SherlockMenuView={ 0x7f010026, 0x7f010027, 0x7f010028, 0x7f010029, 0x7f01002a, 0x7f01002b, 0x7f01002c, 0x7f01002d };\n" +
         "  }\n" +
         "\n" +
         "}\n",
         javaCode);
   }

   @Test
   public void testGenerateRDotJavaForEmptyResources() {
     String packageName = "com.facebook";
     String rDotJava = MergeAndroidResourcesStep.generateJavaCodeForPackageWithoutResources(
         packageName);
     assertEquals(
         "package com.facebook;\n" +
         "\n" +
         "public class R {\n" +
         "\n" +
         "}\n",
         rDotJava);
   }

   /**
    * A special comparison for two {@link Resource} objects because {@link Resource#equals(Object)}
    * does not compare all of the fields.
    */
   private void assertResource(
       String expectedIdType,
       String expectedType,
       String expectedName,
       String expectedIdValue,
       Resource observedResource) {
     assertEquals(expectedIdType, observedResource.idType);
     assertEquals(expectedType, observedResource.type);
     assertEquals(expectedName, observedResource.name);
     assertEquals(expectedIdValue, observedResource.originalIdValue);
   }

   // sortSymbols has a goofy API.  This will help.
   private static class RDotTxtEntryBuilder {
     private final ImmutableMap.Builder<String, String> filePathToContents =
         ImmutableMap.builder();
     private final ImmutableMap.Builder<String, String> filePathToPackageName =
         ImmutableMap.builder();

     public RDotTxtEntryBuilder() {
     }

     public void add(RDotTxtEntry entry) {
       filePathToContents.put(entry.filePath, entry.contents);
       filePathToPackageName.put(entry.filePath, entry.packageName);
     }

     public Function<String, Readable> buildReadableMapper() {
       final ImmutableMap<String, String> builtInstance = filePathToContents.build();
       return new Function<String, Readable>() {
         @Override
         public Readable apply(String filePath) {
           String content = builtInstance.get(filePath);
           if (content == null) {
             throw new RuntimeException("No content for " + filePath);
           } else {
             return new StringReader(content);
           }
         }
       };
     }

     public Map<String, String> buildFilePathToPackageNameSet() {
       return filePathToPackageName.build();
     }
   }

   private static class RDotTxtEntry {
     public String packageName;
     public String filePath;
     public String contents;

     public RDotTxtEntry(String packageName, String filePath, String contents) {
       this.packageName = packageName;
       this.filePath = filePath;
       this.contents = contents;
     }
   }
 }
	/*
	* Copyright 2012-present Facebook, 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.facebook.buck.android;

	import static org.junit.Assert.assertEquals;
	import static org.junit.Assert.assertFalse;

	import com.facebook.buck.android.MergeAndroidResourcesStep.Resource;
	import com.facebook.buck.util.BuckConstant;
	import com.google.common.base.Function;
	import com.google.common.collect.ImmutableMap;
	import com.google.common.collect.Sets;
	import com.google.common.collect.SortedSetMultimap;

	import org.junit.Test;

	import java.io.IOException;
	import java.io.StringReader;
	import java.util.Iterator;
	import java.util.Map;
	import java.util.Set;
	import java.util.SortedSet;

	public class MergeAndroidResourcesStepTest {
	@Test
	public void testGenerateRDotJavaForMultipleSymbolsFiles() throws IOException {
	RDotTxtEntryBuilder entriesBuilder = new RDotTxtEntryBuilder();

	// Merge everything into the same package space.
	String sharedPackageName = "com.facebook.abc";
	entriesBuilder.add(new RDotTxtEntry(sharedPackageName, "a-R.txt",
	"int id a1 0x7f010001\n" +
	"int id a2 0x7f010002\n" +
	"int string a1 0x7f020001\n"));

	entriesBuilder.add(new RDotTxtEntry(sharedPackageName, "b-R.txt",
	"int id b1 0x7f010001\n" +
	"int id b2 0x7f010002\n"));

	entriesBuilder.add(new RDotTxtEntry(sharedPackageName, "c-R.txt",
	"int attr c1 0x7f010001\n" +
	"int[] styleable c1 { 0x7f010001 }\n"));

	SortedSetMultimap<String, Resource> packageNameToResources =
	MergeAndroidResourcesStep.sortSymbols(
	entriesBuilder.buildReadableMapper(),
	entriesBuilder.buildFilePathToPackageNameSet(),
	true /* reenumerate */);

	assertEquals(1, packageNameToResources.keySet().size());
	SortedSet<Resource> resources = packageNameToResources.get(sharedPackageName);
	assertEquals(7, resources.size());

	Set<String> uniqueEntries = Sets.newHashSet();
	for (Resource resource : resources) {
	if (!resource.type.equals("styleable")) {
	assertFalse("Duplicate ids should be fixed by renumerate=true; duplicate was: " +
	resource.idValueToWrite, uniqueEntries.contains(resource.idValueToWrite));
	uniqueEntries.add(resource.idValueToWrite);
	}
	}

	assertEquals(6, uniqueEntries.size());

	// All good, no need to further test whether we can write the Java file correctly...
	}

	@Test
	public void testGenerateRDotJavaForOneSymbolsFile() {
	String symbolsFile = BuckConstant.BIN_DIR +
	"/android_res/com/facebook/http/__res_resources_text_symbols__/R.txt";
	String rDotJavaPackage = "com.facebook";
	final String outputTextSymbols =
	"int id placeholder 0x7f020000\n" +
	"int string debug_http_proxy_dialog_title 0x7f030004\n" +
	"int string debug_http_proxy_hint 0x7f030005\n" +
	"int string debug_http_proxy_summary 0x7f030003\n" +
	"int string debug_http_proxy_title 0x7f030002\n" +
	"int string debug_ssl_cert_check_summary 0x7f030001\n" +
	"int string debug_ssl_cert_check_title 0x7f030000\n" +
	"int styleable SherlockMenuItem_android_visible 4\n" +
	"int[] styleable SherlockMenuView { 0x7f010026, 0x7f010027, 0x7f010028, 0x7f010029, 0x7f01002a, 0x7f01002b, 0x7f01002c, 0x7f01002d }\n";
	RDotTxtEntryBuilder entriesBuilder = new RDotTxtEntryBuilder();
	entriesBuilder.add(new RDotTxtEntry(rDotJavaPackage, symbolsFile, outputTextSymbols));
	SortedSetMultimap<String, Resource> rDotJavaPackageToResources =
	MergeAndroidResourcesStep.sortSymbols(
	entriesBuilder.buildReadableMapper(),
	entriesBuilder.buildFilePathToPackageNameSet(),
	false /* reenumerate */);

	assertEquals(1, rDotJavaPackageToResources.keySet().size());

	// Verify all of the values in the resources collection.
	SortedSet<Resource> resources = rDotJavaPackageToResources.get(rDotJavaPackage);
	assertEquals(9, resources.size());
	Iterator<Resource> iter = resources.iterator();
	assertResource("int", "id", "placeholder", "0x7f020000", iter.next());
	assertResource("int", "string", "debug_http_proxy_dialog_title", "0x7f030004", iter.next());
	assertResource("int", "string", "debug_http_proxy_hint", "0x7f030005", iter.next());
	assertResource("int", "string", "debug_http_proxy_summary", "0x7f030003", iter.next());
	assertResource("int", "string", "debug_http_proxy_title", "0x7f030002", iter.next());
	assertResource("int", "string", "debug_ssl_cert_check_summary", "0x7f030001", iter.next());
	assertResource("int", "string", "debug_ssl_cert_check_title", "0x7f030000", iter.next());
	assertResource("int", "styleable", "SherlockMenuItem_android_visible", "4", iter.next());
	assertResource("int[]",
	"styleable",
	"SherlockMenuView",
	"{ 0x7f010026, 0x7f010027, 0x7f010028, 0x7f010029, 0x7f01002a, 0x7f01002b, 0x7f01002c, 0x7f01002d }",
	iter.next());

	// Verify that the correct Java code is generated.
	String javaCode = MergeAndroidResourcesStep.generateJavaCodeForPackageAndResources(
	rDotJavaPackage, resources);
	assertEquals(
	"package com.facebook;\n" +
	"\n" +
	"public class R {\n" +
	"\n" +
	" public static class id {\n" +
	" public static int placeholder=0x7f020000;\n" +
	" }\n" +
	"\n" +
	" public static class string {\n" +
	" public static int debug_http_proxy_dialog_title=0x7f030004;\n" +
	" public static int debug_http_proxy_hint=0x7f030005;\n" +
	" public static int debug_http_proxy_summary=0x7f030003;\n" +
	" public static int debug_http_proxy_title=0x7f030002;\n" +
	" public static int debug_ssl_cert_check_summary=0x7f030001;\n" +
	" public static int debug_ssl_cert_check_title=0x7f030000;\n" +
	" }\n" +
	"\n" +
	" public static class styleable {\n" +
	" public static int SherlockMenuItem_android_visible=4;\n" +
	" public static int[] SherlockMenuView={ 0x7f010026, 0x7f010027, 0x7f010028, 0x7f010029, 0x7f01002a, 0x7f01002b, 0x7f01002c, 0x7f01002d };\n" +
	" }\n" +
	"\n" +
	"}\n",
	javaCode);
	}

	@Test
	public void testGenerateRDotJavaForEmptyResources() {
	String packageName = "com.facebook";
	String rDotJava = MergeAndroidResourcesStep.generateJavaCodeForPackageWithoutResources(
	packageName);
	assertEquals(
	"package com.facebook;\n" +
	"\n" +
	"public class R {\n" +
	"\n" +
	"}\n",
	rDotJava);
	}

	/**
	* A special comparison for two {@link Resource} objects because {@link Resource#equals(Object)}
	* does not compare all of the fields.
	*/
	private void assertResource(
	String expectedIdType,
	String expectedType,
	String expectedName,
	String expectedIdValue,
	Resource observedResource) {
	assertEquals(expectedIdType, observedResource.idType);
	assertEquals(expectedType, observedResource.type);
	assertEquals(expectedName, observedResource.name);
	assertEquals(expectedIdValue, observedResource.originalIdValue);
	}

	// sortSymbols has a goofy API. This will help.
	private static class RDotTxtEntryBuilder {
	private final ImmutableMap.Builder<String, String> filePathToContents =
	ImmutableMap.builder();
	private final ImmutableMap.Builder<String, String> filePathToPackageName =
	ImmutableMap.builder();

	public RDotTxtEntryBuilder() {
	}

	public void add(RDotTxtEntry entry) {
	filePathToContents.put(entry.filePath, entry.contents);
	filePathToPackageName.put(entry.filePath, entry.packageName);
	}

	public Function<String, Readable> buildReadableMapper() {
	final ImmutableMap<String, String> builtInstance = filePathToContents.build();
	return new Function<String, Readable>() {
	@Override
	public Readable apply(String filePath) {
	String content = builtInstance.get(filePath);
	if (content == null) {
	throw new RuntimeException("No content for " + filePath);
	} else {
	return new StringReader(content);
	}
	}
	};
	}

	public Map<String, String> buildFilePathToPackageNameSet() {
	return filePathToPackageName.build();
	}
	}

	private static class RDotTxtEntry {
	public String packageName;
	public String filePath;
	public String contents;

	public RDotTxtEntry(String packageName, String filePath, String contents) {
	this.packageName = packageName;
	this.filePath = filePath;
	this.contents = contents;
	}
	}
	}