/*
 * Copyright 2007-2012 Amazon Technologies, Inc. or its affiliates.
 * Amazon, Amazon.com and Carbonado are trademarks or registered trademarks
 * of Amazon Technologies, Inc. or its affiliates.  All rights reserved.
 *
 * 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.amazon.carbonado.cursor;

import junit.framework.TestCase;
import junit.framework.TestSuite;

import com.amazon.carbonado.filter.Filter;

import com.amazon.carbonado.stored.*;

/**
 * 
 *
 * @author Brian S O'Neill
 */
public class TestShortCircuitOptimizer extends TestCase {
    public static void main(String[] args) {
        junit.textui.TestRunner.run(suite());
    }

    public static TestSuite suite() {
        return new TestSuite(TestShortCircuitOptimizer.class);
    }

    public TestShortCircuitOptimizer(String name) {
        super(name);
    }

    protected void setUp() {
    }

    protected void tearDown() {
    }

    public void testBasic() throws Exception {
        {
            Filter<Order> filter = Filter.getOpenFilter(Order.class);
            Filter<Order> optimized = ShortCircuitOptimizer.optimize(filter);

            // No change.
            assertEquals(filter, optimized);
        }

        {
            Filter<Order> filter = Filter.getClosedFilter(Order.class);
            Filter<Order> optimized = ShortCircuitOptimizer.optimize(filter);

            // No change.
            assertEquals(filter, optimized);
        }

        {
            Filter<Order> filter = Filter.filterFor(Order.class, "orderNumber = ?");
            Filter<Order> optimized = ShortCircuitOptimizer.optimize(filter);

            // No change.
            assertEquals(filter, optimized);
        }
    }

    public void testNoJoins() throws Exception {
        {
            Filter<Order> filter = Filter
                .filterFor(Order.class, "orderNumber = ? & orderTotal = ?");
            Filter<Order> optimized = ShortCircuitOptimizer.optimize(filter);

            // No change.
            assertEquals(filter, optimized);
        }

        {
            Filter<Order> filter = Filter
                .filterFor(Order.class, "orderNumber = ? | orderTotal = ?");
            Filter<Order> optimized = ShortCircuitOptimizer.optimize(filter);

            // No change.
            assertEquals(filter, optimized);
        }

        {
            Filter<Order> filter = Filter
                .filterFor(Order.class, "(orderNumber = ? | orderTotal = ?) & addressID != ?");
            Filter<Order> optimized = ShortCircuitOptimizer.optimize(filter);

            // No change.
            assertEquals(filter, optimized);
        }
    }

    public void testManyToOneJoins() throws Exception {
        {
            Filter<Order> filter = Filter
                .filterFor(Order.class, "orderNumber = ? & address.addressState = ?");
            Filter<Order> optimized = ShortCircuitOptimizer.optimize(filter);

            // No change.
            assertEquals(filter, optimized);
        }

        {
            Filter<Order> filter = Filter
                .filterFor(Order.class, "address.addressState = ? & orderNumber = ?");
            Filter<Order> optimized = ShortCircuitOptimizer.optimize(filter);

            assertEquals(Filter.filterFor
                         (Order.class, "orderNumber = ? & address.addressState = ?"),
                         optimized);
        }

        {
            Filter<Order> filter = Filter
                .filterFor(Order.class,
                           "(address.addressState = ? & orderNumber = ?) | orderTotal < ?");
            Filter<Order> optimized = ShortCircuitOptimizer.optimize(filter);

            assertEquals(Filter.filterFor
                         (Order.class,
                          "orderTotal < ? | (orderNumber = ? & address.addressState = ?)"),
                         optimized);
        }

        {
            Filter<Shipment> filter = Filter
                .filterFor(Shipment.class,
                           "order.address.addressState = ? | shipper.shipperName = ?");
            Filter<Shipment> optimized = ShortCircuitOptimizer.optimize(filter);

            assertEquals(Filter.filterFor
                         (Shipment.class,
                          "shipper.shipperName = ? | order.address.addressState = ?"),
                         optimized);
        }

        {
            Filter<Shipment> filter = Filter
                .filterFor(Shipment.class,
                           "order.address.addressState = ? | shipper.address.addressCity = ?");
            Filter<Shipment> optimized = ShortCircuitOptimizer.optimize(filter);

            // No change.
            assertEquals(filter, optimized);
        }
    }

    public void testOneToManyJoins() throws Exception {
        {
            Filter<Order> filter = Filter
                .filterFor(Order.class, "orderNumber = ? & orderItems(itemPrice = ?)");
            Filter<Order> optimized = ShortCircuitOptimizer.optimize(filter);

            // No change.
            assertEquals(filter, optimized);
        }

        {
            Filter<Order> filter = Filter
                .filterFor(Order.class, "orderItems(itemPrice = ?) & orderNumber = ?");
            Filter<Order> optimized = ShortCircuitOptimizer.optimize(filter);

            assertEquals(Filter.filterFor
                         (Order.class, "orderNumber = ? & orderItems(itemPrice = ?)"),
                         optimized);
        }

        {
            Filter<Order> filter = Filter
                .filterFor(Order.class,
                           "(orderItems(itemPrice = ?) & orderNumber = ?) | orderTotal < ?");
            Filter<Order> optimized = ShortCircuitOptimizer.optimize(filter);

            assertEquals(Filter.filterFor
                         (Order.class,
                          "orderTotal < ? | (orderNumber = ? & orderItems(itemPrice = ?))"),
                         optimized);
        }

        {
            Filter<Order> filter = Filter
                .filterFor(Order.class,
                           "orderItems(shipment.shipmentDate > ?) | shipments(shipmentDate < ?)");
            Filter<Order> optimized = ShortCircuitOptimizer.optimize(filter);

            // No change because optimizer doesn't consider complexity of sub-filters.
            assertEquals(filter, optimized);
        }

        {
            Filter<Order> filter = Filter
                .filterFor
                (Order.class,
                 "orderItems(shipment.shipmentDate > ?) | shipments(shipper.shipperName != ?)");
            Filter<Order> optimized = ShortCircuitOptimizer.optimize(filter);

            // No change.
            assertEquals(filter, optimized);
        }
    }
}