/*
 * Copyright (C) 2012 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.monkey;

import com.android.tradefed.util.ArrayUtil;

import junit.framework.TestCase;

import java.util.ArrayList;
import java.util.Collection;

/**
 * Unit tests for {@link MonkeyBase}
 */
public class MonkeyBaseTest extends TestCase {

    /**
     * Test that {@link MonkeyBase#setSubtract(Collection, Collection)} returns same object if
     * exclude is empty.
     */
    public void testSetSubtract_noExclude() {
        Collection<String> haystack = ArrayUtil.list("a", "b", "c");
        Collection<String> needles = new ArrayList<String>();
        // double-checking comparison assumptions
        assertFalse(haystack == needles);
        Collection<String> output = MonkeyBase.setSubtract(haystack, needles);
        assertTrue(haystack == output);
    }

    /**
     * Test that {@link MonkeyBase#setSubtract(Collection, Collection)} returns the set subtraction
     * if exclude is not empty.
     */
    public void testSetSubtract() {
        Collection<String> haystack = ArrayUtil.list("a", "b", "c");
        Collection<String> needles = ArrayUtil.list("b");
        Collection<String> output = MonkeyBase.setSubtract(haystack, needles);
        assertEquals(2, output.size());
        assertTrue(output.contains("a"));
        assertFalse(output.contains("b"));
        assertTrue(output.contains("c"));
    }
}