1 /* 2 * Copyright (C) 2013 The Android Open Source Project 3 * 4 * Licensed under the Apache License, Version 2.0 (the "License"); 5 * you may not use this file except in compliance with the License. 6 * You may obtain a copy of the License at 7 * 8 * http://www.apache.org/licenses/LICENSE-2.0 9 * 10 * Unless required by applicable law or agreed to in writing, software 11 * distributed under the License is distributed on an "AS IS" BASIS, 12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 13 * See the License for the specific language governing permissions and 14 * limitations under the License. 15 */ 16 package com.android.tradefed.targetprep; 17 18 import com.android.tradefed.build.BuildInfo; 19 import com.android.tradefed.build.IBuildInfo; 20 import com.android.tradefed.config.OptionSetter; 21 22 import junit.framework.TestCase; 23 24 import java.util.Map; 25 26 /** 27 * Unit tests for {@link BuildInfoAttributePreparer} 28 */ 29 public class BuildInfoAttributePreparerTest extends TestCase { 30 private BuildInfoAttributePreparer mPrep = null; 31 32 /** 33 * {@inheritDoc} 34 */ 35 @Override 36 public void setUp() throws Exception { 37 super.setUp(); 38 mPrep = new BuildInfoAttributePreparer(); 39 } 40 41 public void testSimple() throws Exception { 42 final IBuildInfo build = new BuildInfo(); 43 44 OptionSetter opt = new OptionSetter(mPrep); 45 opt.setOptionValue("build-attribute", "key", "value"); 46 mPrep.setUp(null, build); 47 48 Map<String, String> map = build.getBuildAttributes(); 49 assertTrue(map.containsKey("key")); 50 assertEquals("value", map.get("key")); 51 } 52 } 53