/*
 * Copyright Elasticsearch B.V. and/or licensed to Elasticsearch B.V. under one
 * or more contributor license agreements. Licensed under the "Elastic License
 * 2.0", the "GNU Affero General Public License v3.0 only", and the "Server Side
 * Public License v 1"; you may not use this file except in compliance with, at
 * your election, the "Elastic License 2.0", the "GNU Affero General Public
 * License v3.0 only", or the "Server Side Public License, v 1".
 */

package org.elasticsearch.painless;

/** Tests for and operator across all types */
public class AndTests extends ScriptTestCase {

    public void testBasics() throws Exception {
        assertEquals(5 & 3, exec("return 5 & 3;"));
        assertEquals(5 & 3L, exec("return 5 & 3L;"));
        assertEquals(5L & 3, exec("return 5L & 3;"));
        assertEquals(1L, exec("int x = 5; long y = 3; return x & y;"));
    }

    public void testInt() throws Exception {
        assertEquals(5 & 12, exec("int x = 5; int y = 12; return x & y;"));
        assertEquals(5 & -12, exec("int x = 5; int y = -12; return x & y;"));
        assertEquals(7 & 15 & 3, exec("int x = 7; int y = 15; int z = 3; return x & y & z;"));
    }

    public void testIntConst() throws Exception {
        assertEquals(5 & 12, exec("return 5 & 12;"));
        assertEquals(5 & -12, exec("return 5 & -12;"));
        assertEquals(7 & 15 & 3, exec("return 7 & 15 & 3;"));
    }

    public void testLong() throws Exception {
        assertEquals(5L & 12L, exec("long x = 5; long y = 12; return x & y;"));
        assertEquals(5L & -12L, exec("long x = 5; long y = -12; return x & y;"));
        assertEquals(7L & 15L & 3L, exec("long x = 7; long y = 15; long z = 3; return x & y & z;"));
    }

    public void testLongConst() throws Exception {
        assertEquals(5L & 12L, exec("return 5L & 12L;"));
        assertEquals(5L & -12L, exec("return 5L & -12L;"));
        assertEquals(7L & 15L & 3L, exec("return 7L & 15L & 3L;"));
    }

    public void testIllegal() throws Exception {
        expectScriptThrows(ClassCastException.class, () -> { exec("float x = (float)4; int y = 1; return x & y"); });
        expectScriptThrows(ClassCastException.class, () -> { exec("double x = (double)4; int y = 1; return x & y"); });
    }

    public void testDef() {
        expectScriptThrows(ClassCastException.class, () -> { exec("def x = (float)4; def y = (byte)1; return x & y"); });
        expectScriptThrows(ClassCastException.class, () -> { exec("def x = (double)4; def y = (byte)1; return x & y"); });
        assertEquals(0, exec("def x = (byte)4; def y = (byte)1; return x & y"));
        assertEquals(0, exec("def x = (short)4; def y = (byte)1; return x & y"));
        assertEquals(0, exec("def x = (char)4; def y = (byte)1; return x & y"));
        assertEquals(0, exec("def x = (int)4; def y = (byte)1; return x & y"));
        assertEquals(0L, exec("def x = (long)4; def y = (byte)1; return x & y"));

        assertEquals(0, exec("def x = (byte)4; def y = (short)1; return x & y"));
        assertEquals(0, exec("def x = (short)4; def y = (short)1; return x & y"));
        assertEquals(0, exec("def x = (char)4; def y = (short)1; return x & y"));
        assertEquals(0, exec("def x = (int)4; def y = (short)1; return x & y"));
        assertEquals(0L, exec("def x = (long)4; def y = (short)1; return x & y"));

        assertEquals(0, exec("def x = (byte)4; def y = (char)1; return x & y"));
        assertEquals(0, exec("def x = (short)4; def y = (char)1; return x & y"));
        assertEquals(0, exec("def x = (char)4; def y = (char)1; return x & y"));
        assertEquals(0, exec("def x = (int)4; def y = (char)1; return x & y"));
        assertEquals(0L, exec("def x = (long)4; def y = (char)1; return x & y"));

        assertEquals(0, exec("def x = (byte)4; def y = (int)1; return x & y"));
        assertEquals(0, exec("def x = (short)4; def y = (int)1; return x & y"));
        assertEquals(0, exec("def x = (char)4; def y = (int)1; return x & y"));
        assertEquals(0, exec("def x = (int)4; def y = (int)1; return x & y"));
        assertEquals(0L, exec("def x = (long)4; def y = (int)1; return x & y"));

        assertEquals(0L, exec("def x = (byte)4; def y = (long)1; return x & y"));
        assertEquals(0L, exec("def x = (short)4; def y = (long)1; return x & y"));
        assertEquals(0L, exec("def x = (char)4; def y = (long)1; return x & y"));
        assertEquals(0L, exec("def x = (int)4; def y = (long)1; return x & y"));
        assertEquals(0L, exec("def x = (long)4; def y = (long)1; return x & y"));

        assertEquals(0, exec("def x = (byte)4; def y = (byte)1; return x & y"));
        assertEquals(0, exec("def x = (short)4; def y = (short)1; return x & y"));
        assertEquals(0, exec("def x = (char)4; def y = (char)1; return x & y"));
        assertEquals(0, exec("def x = (int)4; def y = (int)1; return x & y"));
        assertEquals(0L, exec("def x = (long)4; def y = (long)1; return x & y"));

        assertEquals(true, exec("def x = true;  def y = true; return x & y"));
        assertEquals(false, exec("def x = true;  def y = false; return x & y"));
        assertEquals(false, exec("def x = false; def y = true; return x & y"));
        assertEquals(false, exec("def x = false; def y = false; return x & y"));
    }

    public void testDefTypedLHS() {
        expectScriptThrows(ClassCastException.class, () -> { exec("float x = (float)4; def y = (byte)1; return x & y"); });
        expectScriptThrows(ClassCastException.class, () -> { exec("double x = (double)4; def y = (byte)1; return x & y"); });
        assertEquals(0, exec("byte x = (byte)4; def y = (byte)1; return x & y"));
        assertEquals(0, exec("short x = (short)4; def y = (byte)1; return x & y"));
        assertEquals(0, exec("char x = (char)4; def y = (byte)1; return x & y"));
        assertEquals(0, exec("int x = (int)4; def y = (byte)1; return x & y"));
        assertEquals(0L, exec("long x = (long)4; def y = (byte)1; return x & y"));

        assertEquals(0, exec("byte x = (byte)4; def y = (short)1; return x & y"));
        assertEquals(0, exec("short x = (short)4; def y = (short)1; return x & y"));
        assertEquals(0, exec("char x = (char)4; def y = (short)1; return x & y"));
        assertEquals(0, exec("int x = (int)4; def y = (short)1; return x & y"));
        assertEquals(0L, exec("long x = (long)4; def y = (short)1; return x & y"));

        assertEquals(0, exec("byte x = (byte)4; def y = (char)1; return x & y"));
        assertEquals(0, exec("short x = (short)4; def y = (char)1; return x & y"));
        assertEquals(0, exec("char x = (char)4; def y = (char)1; return x & y"));
        assertEquals(0, exec("int x = (int)4; def y = (char)1; return x & y"));
        assertEquals(0L, exec("long x = (long)4; def y = (char)1; return x & y"));

        assertEquals(0, exec("byte x = (byte)4; def y = (int)1; return x & y"));
        assertEquals(0, exec("short x = (short)4; def y = (int)1; return x & y"));
        assertEquals(0, exec("char x = (char)4; def y = (int)1; return x & y"));
        assertEquals(0, exec("int x = (int)4; def y = (int)1; return x & y"));
        assertEquals(0L, exec("long x = (long)4; def y = (int)1; return x & y"));

        assertEquals(0L, exec("byte x = (byte)4; def y = (long)1; return x & y"));
        assertEquals(0L, exec("short x = (short)4; def y = (long)1; return x & y"));
        assertEquals(0L, exec("char x = (char)4; def y = (long)1; return x & y"));
        assertEquals(0L, exec("int x = (int)4; def y = (long)1; return x & y"));
        assertEquals(0L, exec("long x = (long)4; def y = (long)1; return x & y"));

        assertEquals(0, exec("byte x = (byte)4; def y = (byte)1; return x & y"));
        assertEquals(0, exec("short x = (short)4; def y = (short)1; return x & y"));
        assertEquals(0, exec("char x = (char)4; def y = (char)1; return x & y"));
        assertEquals(0, exec("int x = (int)4; def y = (int)1; return x & y"));
        assertEquals(0L, exec("long x = (long)4; def y = (long)1; return x & y"));

        assertEquals(true, exec("boolean x = true;  def y = true; return x & y"));
        assertEquals(false, exec("boolean x = true;  def y = false; return x & y"));
        assertEquals(false, exec("boolean x = false; def y = true; return x & y"));
        assertEquals(false, exec("boolean x = false; def y = false; return x & y"));
    }

    public void testDefTypedRHS() {
        expectScriptThrows(ClassCastException.class, () -> { exec("def x = (float)4; byte y = (byte)1; return x & y"); });
        expectScriptThrows(ClassCastException.class, () -> { exec("def x = (double)4; byte y = (byte)1; return x & y"); });
        assertEquals(0, exec("def x = (byte)4; byte y = (byte)1; return x & y"));
        assertEquals(0, exec("def x = (short)4; byte y = (byte)1; return x & y"));
        assertEquals(0, exec("def x = (char)4; byte y = (byte)1; return x & y"));
        assertEquals(0, exec("def x = (int)4; byte y = (byte)1; return x & y"));
        assertEquals(0L, exec("def x = (long)4; byte y = (byte)1; return x & y"));

        assertEquals(0, exec("def x = (byte)4; short y = (short)1; return x & y"));
        assertEquals(0, exec("def x = (short)4; short y = (short)1; return x & y"));
        assertEquals(0, exec("def x = (char)4; short y = (short)1; return x & y"));
        assertEquals(0, exec("def x = (int)4; short y = (short)1; return x & y"));
        assertEquals(0L, exec("def x = (long)4; short y = (short)1; return x & y"));

        assertEquals(0, exec("def x = (byte)4; char y = (char)1; return x & y"));
        assertEquals(0, exec("def x = (short)4; char y = (char)1; return x & y"));
        assertEquals(0, exec("def x = (char)4; char y = (char)1; return x & y"));
        assertEquals(0, exec("def x = (int)4; char y = (char)1; return x & y"));
        assertEquals(0L, exec("def x = (long)4; char y = (char)1; return x & y"));

        assertEquals(0, exec("def x = (byte)4; int y = (int)1; return x & y"));
        assertEquals(0, exec("def x = (short)4; int y = (int)1; return x & y"));
        assertEquals(0, exec("def x = (char)4; int y = (int)1; return x & y"));
        assertEquals(0, exec("def x = (int)4; int y = (int)1; return x & y"));
        assertEquals(0L, exec("def x = (long)4; int y = (int)1; return x & y"));

        assertEquals(0L, exec("def x = (byte)4; long y = (long)1; return x & y"));
        assertEquals(0L, exec("def x = (short)4; long y = (long)1; return x & y"));
        assertEquals(0L, exec("def x = (char)4; long y = (long)1; return x & y"));
        assertEquals(0L, exec("def x = (int)4; long y = (long)1; return x & y"));
        assertEquals(0L, exec("def x = (long)4; long y = (long)1; return x & y"));

        assertEquals(0, exec("def x = (byte)4; byte y = (byte)1; return x & y"));
        assertEquals(0, exec("def x = (short)4; short y = (short)1; return x & y"));
        assertEquals(0, exec("def x = (char)4; char y = (char)1; return x & y"));
        assertEquals(0, exec("def x = (int)4; int y = (int)1; return x & y"));
        assertEquals(0L, exec("def x = (long)4; long y = (long)1; return x & y"));

        assertEquals(true, exec("def x = true;  boolean y = true; return x & y"));
        assertEquals(false, exec("def x = true;  boolean y = false; return x & y"));
        assertEquals(false, exec("def x = false; boolean y = true; return x & y"));
        assertEquals(false, exec("def x = false; boolean y = false; return x & y"));
    }

    public void testCompoundAssignment() {
        // boolean
        assertEquals(true, exec("boolean x = true; x &= true; return x;"));
        assertEquals(false, exec("boolean x = true; x &= false; return x;"));
        assertEquals(false, exec("boolean x = false; x &= true; return x;"));
        assertEquals(false, exec("boolean x = false; x &= false; return x;"));
        assertEquals(true, exec("boolean[] x = new boolean[1]; x[0] = true; x[0] &= true; return x[0];"));
        assertEquals(false, exec("boolean[] x = new boolean[1]; x[0] = true; x[0] &= false; return x[0];"));
        assertEquals(false, exec("boolean[] x = new boolean[1]; x[0] = false; x[0] &= true; return x[0];"));
        assertEquals(false, exec("boolean[] x = new boolean[1]; x[0] = false; x[0] &= false; return x[0];"));

        // byte
        assertEquals((byte) (13 & 14), exec("byte x = 13; x &= 14; return x;"));
        // short
        assertEquals((short) (13 & 14), exec("short x = 13; x &= 14; return x;"));
        // char
        assertEquals((char) (13 & 14), exec("char x = 13; x &= 14; return x;"));
        // int
        assertEquals(13 & 14, exec("int x = 13; x &= 14; return x;"));
        // long
        assertEquals((long) (13 & 14), exec("long x = 13L; x &= 14; return x;"));
    }

    public void testBogusCompoundAssignment() {
        expectScriptThrows(ClassCastException.class, () -> { exec("float x = 4; int y = 1; x &= y"); });
        expectScriptThrows(ClassCastException.class, () -> { exec("double x = 4; int y = 1; x &= y"); });
        expectScriptThrows(ClassCastException.class, () -> { exec("int x = 4; float y = 1; x &= y"); });
        expectScriptThrows(ClassCastException.class, () -> { exec("int x = 4; double y = 1; x &= y"); });
    }

    public void testDefCompoundAssignment() {
        // boolean
        assertEquals(true, exec("def x = true; x &= true; return x;"));
        assertEquals(false, exec("def x = true; x &= false; return x;"));
        assertEquals(false, exec("def x = false; x &= true; return x;"));
        assertEquals(false, exec("def x = false; x &= false; return x;"));
        assertEquals(true, exec("def[] x = new def[1]; x[0] = true; x[0] &= true; return x[0];"));
        assertEquals(false, exec("def[] x = new def[1]; x[0] = true; x[0] &= false; return x[0];"));
        assertEquals(false, exec("def[] x = new def[1]; x[0] = false; x[0] &= true; return x[0];"));
        assertEquals(false, exec("def[] x = new def[1]; x[0] = false; x[0] &= false; return x[0];"));

        // byte
        assertEquals((byte) (13 & 14), exec("def x = (byte)13; x &= 14; return x;"));
        // short
        assertEquals((short) (13 & 14), exec("def x = (short)13; x &= 14; return x;"));
        // char
        assertEquals((char) (13 & 14), exec("def x = (char)13; x &= 14; return x;"));
        // int
        assertEquals(13 & 14, exec("def x = 13; x &= 14; return x;"));
        // long
        assertEquals((long) (13 & 14), exec("def x = 13L; x &= 14; return x;"));
    }

    public void testDefBogusCompoundAssignment() {
        expectScriptThrows(ClassCastException.class, () -> { exec("def x = 4F; int y = 1; x &= y"); });
        expectScriptThrows(ClassCastException.class, () -> { exec("def x = 4D; int y = 1; x &= y"); });
        expectScriptThrows(ClassCastException.class, () -> { exec("int x = 4; def y = 1F; x &= y"); });
        expectScriptThrows(ClassCastException.class, () -> { exec("int x = 4; def y = 1D; x &= y"); });
    }
}