CompoundTermTest.java

/**
 * Copyright 1999-2009 The Pegadi Team
 *
 * 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 org.pegadi.sqlsearch;

import junit.framework.TestCase;

import java.util.Arrays;
import java.util.Iterator;
import java.util.List;

public class CompoundTermTest extends TestCase {

    SearchTerm one;
    SearchTerm two;

    public void setUp() {
        one = new SearchTerm() {
            public String whereClause() {
                return "one";
            }

            public List<String> getTables() {
                return Arrays.asList("Article");
            }

        };

        two = new SearchTerm() {
            public String whereClause() {
                return "two";
            }

            public List<String> getTables() {
                return Arrays.asList("Section");
            }

        };

    }

    public void testAddTerm() {

        CompoundTerm comp = new CompoundTerm(one) {
            public String whereClause() {
                return "one";
            }
        };

        List<SearchTerm> returnTerms = comp.getTerms();

        assertEquals(returnTerms.size(), 1);
        assertSame(one, comp.getTerms().iterator().next());


        comp.addTerm(two);

        returnTerms = comp.getTerms();

        assertEquals(returnTerms.size(), 2);
        Iterator<SearchTerm> iterator = comp.getTerms().iterator();
        assertSame(one, iterator.next());
        assertSame(two, iterator.next());

    }

    public void testGetTables() {

        String tables[] = {"Article", "Section"};

        CompoundTerm compTerm = new CompoundTerm(one, two) {
            public String whereClause() {
                return "one";
            }
        };


        List<String> returnTables = compTerm.getTables();

        assertEquals(returnTables.size(), 2);
        assertSame(returnTables.get(0), tables[0]);
        assertSame(returnTables.get(1), tables[1]);

    }

    public void testGetIngoreTables() {

        SearchTerm one = new SearchTerm() {
            public String whereClause() {
                return "one";
            }

            public List<String> getTables() {
                return Arrays.asList("Article");
            }

            public List<String> getIgnoreTables() {
                return Arrays.asList("Article");
            }
        };

        SearchTerm two = new SearchTerm() {
            public String whereClause() {
                return "two";
            }

            public List<String> getTables() {
                return Arrays.asList("Section");
            }

        };

        CompoundTerm compTerm = new CompoundTerm(one, two) {
            public String whereClause() {
                return "one";
            }
        };

        List<String> ignoreTables = compTerm.getIgnoreTables();

        assertEquals(1, ignoreTables.size());

        assertTrue(ignoreTables.contains("Article"));
    }

}