/*
 * 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 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 or the Server
 * Side Public License, v 1.
 */

package org.elasticsearch.painless;

/** Tests for unary operators across different types */
public class UnaryTests extends ScriptTestCase {

    /** basic tests */
    public void testBasics() {
        assertEquals(false, exec("return !true;"));
        assertEquals(true, exec("boolean x = false; return !x;"));
        assertEquals(-2, exec("return ~1;"));
        assertEquals(-2, exec("byte x = 1; return ~x;"));
        assertEquals(1, exec("return +1;"));
        assertEquals(1.0, exec("double x = 1; return +x;"));
        assertEquals(-1, exec("return -1;"));
        assertEquals(-2, exec("short x = 2; return -x;"));
        assertEquals(-1.0, exec("def x = (def)-1.0; return +x"));
        expectScriptThrows(IllegalArgumentException.class, () -> exec("double x = (Double)-1.0; return +x"));
        expectScriptThrows(IllegalArgumentException.class, () -> exec("double x = (ArrayList)-1.0; return +x"));
    }

    public void testNegationInt() throws Exception {
        assertEquals(-1, exec("return -1;"));
        assertEquals(1, exec("return -(-1);"));
        assertEquals(0, exec("return -0;"));
    }

    public void testPlus() {
        assertEquals(-1, exec("byte x = (byte)-1; return +x"));
        assertEquals(-1, exec("short x = (short)-1; return +x"));
        assertEquals(65535, exec("char x = (char)-1; return +x"));
        assertEquals(-1, exec("int x = -1; return +x"));
        assertEquals(-1L, exec("long x = -1L; return +x"));
        assertEquals(-1.0F, exec("float x = -1F; return +x"));
        assertEquals(-1.0, exec("double x = -1.0; return +x"));
    }

    public void testDefNot() {
        assertEquals(~1, exec("def x = (byte)1; return ~x"));
        assertEquals(~1, exec("def x = (short)1; return ~x"));
        assertEquals(~1, exec("def x = (char)1; return ~x"));
        assertEquals(~1, exec("def x = 1; return ~x"));
        assertEquals(~1L, exec("def x = 1L; return ~x"));
    }

    public void testDefNotTypedRet() {
        assertEquals((double) ~1, exec("def x = (byte)1; double y = ~x; return y;"));
        assertEquals((float) ~1, exec("def x = (short)1; float y = ~x; return y;"));
        assertEquals((long) ~1, exec("def x = (char)1; long y = ~x; return y;"));
        assertEquals(~1, exec("def x = 1; int y = ~x; return y;"));
    }

    public void testDefNeg() {
        assertEquals(-1, exec("def x = (byte)1; return -x"));
        assertEquals(-1, exec("def x = (short)1; return -x"));
        assertEquals(-1, exec("def x = (char)1; return -x"));
        assertEquals(-1, exec("def x = 1; return -x"));
        assertEquals(-1L, exec("def x = 1L; return -x"));
        assertEquals(-1.0F, exec("def x = 1F; return -x"));
        assertEquals(-1.0, exec("def x = 1.0; return -x"));
    }

    public void testDefNegTypedRet() {
        assertEquals((double) -1, exec("def x = (byte)1; double y = -x; return y;"));
        assertEquals((float) -1, exec("def x = (short)1; float y = -x; return y;"));
        assertEquals((long) -1, exec("def x = (char)1; long y = -x; return y;"));
        assertEquals(-1, exec("def x = 1; int y = -x; return y;"));
    }

    public void testDefPlus() {
        assertEquals(-1, exec("def x = (byte)-1; return +x"));
        assertEquals(-1, exec("def x = (short)-1; return +x"));
        assertEquals(65535, exec("def x = (char)-1; return +x"));
        assertEquals(-1, exec("def x = -1; return +x"));
        assertEquals(-1L, exec("def x = -1L; return +x"));
        assertEquals(-1.0F, exec("def x = -1F; return +x"));
        assertEquals(-1.0D, exec("def x = -1.0; return +x"));
    }

    public void testDefPlusTypedRet() {
        assertEquals((double) -1, exec("def x = (byte)-1; double y = +x; return y;"));
        assertEquals((float) -1, exec("def x = (short)-1; float y = +x; return y;"));
        assertEquals((long) 65535, exec("def x = (char)-1; long y = +x; return y;"));
        assertEquals(-1, exec("def x = -1; int y = +x; return y;"));
    }
}
