// Copyright 2016 Yahoo Inc. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
package com.yahoo.document.predicate;

import org.junit.Test;

import java.util.Arrays;

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

/**
 * @author <a href="mailto:simon@yahoo-inc.com">Simon Thoresen Hult</a>
 */
public class DisjunctionTest {

    @Test
    public void requireThatDisjunctionIsAnOperator() {
        assertTrue(PredicateOperator.class.isAssignableFrom(Disjunction.class));
    }

    @Test
    public void requireThatAccessorsWork() {
        Disjunction node = new Disjunction();
        Predicate a = SimplePredicates.newString("a");
        node.addOperand(a);
        assertEquals(Arrays.asList(a), node.getOperands());
        Predicate b = SimplePredicates.newString("b");
        node.addOperand(b);
        assertEquals(Arrays.asList(a, b), node.getOperands());
        Predicate c = SimplePredicates.newString("c");
        Predicate d = SimplePredicates.newString("d");
        node.addOperands(Arrays.asList(c, d));
        assertEquals(Arrays.asList(a, b, c, d), node.getOperands());
        Predicate e = SimplePredicates.newString("e");
        Predicate f = SimplePredicates.newString("f");
        node.setOperands(Arrays.asList(e, f));
        assertEquals(Arrays.asList(e, f), node.getOperands());
    }

    @Test
    public void requireThatConstructorsWork() {
        Predicate foo = SimplePredicates.newString("foo");
        Predicate bar = SimplePredicates.newString("bar");
        Disjunction node = new Disjunction(foo, bar);
        assertEquals(Arrays.asList(foo, bar), node.getOperands());

        node = new Disjunction(Arrays.asList(foo, bar));
        assertEquals(Arrays.asList(foo, bar), node.getOperands());
    }

    @Test
    public void requireThatCloneIsImplemented() throws CloneNotSupportedException {
        Disjunction node1 = new Disjunction(SimplePredicates.newString("a"), SimplePredicates.newString("b"));
        Disjunction node2 = node1.clone();
        assertEquals(node1, node2);
        assertNotSame(node1, node2);
        assertNotSame(node1.getOperands(), node2.getOperands());
    }

    @Test
    public void requireThatHashCodeIsImplemented() {
        assertEquals(new Disjunction().hashCode(), new Disjunction().hashCode());
    }

    @Test
    public void requireThatEqualsIsImplemented() {
        Disjunction lhs = new Disjunction(SimplePredicates.newString("foo"),
                                          SimplePredicates.newString("bar"));
        assertTrue(lhs.equals(lhs));
        assertFalse(lhs.equals(new Object()));

        Disjunction rhs = new Disjunction();
        assertFalse(lhs.equals(rhs));
        rhs.addOperand(SimplePredicates.newString("foo"));
        assertFalse(lhs.equals(rhs));
        rhs.addOperand(SimplePredicates.newString("bar"));
        assertTrue(lhs.equals(rhs));
    }

    @Test
    public void requireThatNodeDelimiterIsOR() {
        assertEquals("", newDisjunction().toString());
        assertEquals("foo", newDisjunction("foo").toString());
        assertEquals("foo or bar", newDisjunction("foo", "bar").toString());
        assertEquals("foo or bar or baz", newDisjunction("foo", "bar", "baz").toString());
    }

    @Test
    public void requireThatSimpleDisjunctionsArePrettyPrinted() {
        assertEquals("foo or bar",
                     new Disjunction(SimplePredicates.newString("foo"),
                                     SimplePredicates.newString("bar")).toString());
    }

    @Test
    public void requireThatComplexDisjunctionsArePrintedAsGroup() {
        assertEquals("foo or bar or baz",
                     new Disjunction(SimplePredicates.newString("foo"),
                                     new Disjunction(SimplePredicates.newString("bar"),
                                                     SimplePredicates.newString("baz"))).toString());
        assertEquals("foo or (bar and baz)",
                     new Disjunction(SimplePredicates.newString("foo"),
                                     new Conjunction(SimplePredicates.newString("bar"),
                                                     SimplePredicates.newString("baz"))).toString());
    }

    private static Disjunction newDisjunction(String... operands) {
        return new Disjunction(SimplePredicates.newStrings(operands));
    }

}