/*
 * Copyright 2017-2018 JetBrains s.r.o. Use of this source code is governed by the Apache 2.0 license.
 */

package kotlinx.atomicfu.test

import kotlinx.atomicfu.*
import kotlin.test.Test
import kotlin.test.assertEquals

private val a = atomic(0)
private val b = atomic(2424920024888888848)
private val c = atomic(true)
private val abcNode = atomic(ANode(BNode(CNode(8))))
private val any = atomic<Any?>(null)

private val intArr = AtomicIntArray(3)
private val longArr = AtomicLongArray(5)
private val booleanArr = AtomicBooleanArray(4)
private val refArr = AtomicArray<ANode<BNode<CNode>>>(5)
private val anyRefArr = AtomicArray<Any>(10)

private val stringAtomicNullArr = atomicArrayOfNulls<String>(10)

internal val internalTopLevelField = atomic(0)
public val publicTopLevelField = atomic(0)

class TopLevelPrimitiveTest {
    @Test
    fun testTopLevelInt() {
        check(a.value == 0)
        check(a.getAndSet(3) == 0)
        check(a.compareAndSet(3, 8))
        a.lazySet(5)
        check(a.value == 5)
        a.value = 1
        check(a.value == 1)
        check(a.getAndSet(2) == 1)
        check(a.value == 2)
        check(a.getAndIncrement() == 2)
        check(a.value == 3)
        check(a.getAndDecrement() == 3)
        check(a.value == 2)
        check(a.getAndAdd(2) == 2)
        check(a.value == 4)
        check(a.addAndGet(3) == 7)
        check(a.value == 7)
        check(a.incrementAndGet() == 8)
        check(a.value == 8)
        check(a.decrementAndGet() == 7)
        check(a.value == 7)
        a.compareAndSet(7, 10)
    }

    @Test
    fun testTopLevelLong() {
        check(b.value == 2424920024888888848)
        b.lazySet(8424920024888888848)
        check(b.value == 8424920024888888848)
        b.value = 8424920024888888833
        check(b.value == 8424920024888888833)
        check(b.getAndSet(8924920024888888848) == 8424920024888888833)
        check(b.value == 8924920024888888848)
        check(b.incrementAndGet() == 8924920024888888849)
        check(b.value == 8924920024888888849)
        check(b.getAndDecrement() == 8924920024888888849)
        check(b.value == 8924920024888888848)
        check(b.getAndAdd(100000000000000000) == 8924920024888888848)
        check(b.value == 9024920024888888848)
        check(b.addAndGet(-9223372036854775807) == -198452011965886959)
        check(b.value == -198452011965886959)
        check(b.incrementAndGet() == -198452011965886958)
        check(b.value == -198452011965886958)
        check(b.decrementAndGet() == -198452011965886959)
        check(b.value == -198452011965886959)
    }

    @Test
    fun testTopLevelBoolean() {
        check(c.value)
        c.lazySet(false)
        check(!c.value)
        c.value = false
        check(!c.value)
        check(!c.getAndSet(true))
        check(c.compareAndSet(true, false))
        check(!c.value)
    }

    @Test
    fun testTopLevelRef() {
        check(abcNode.value.b.c.d == 8)
        val newNode = ANode(BNode(CNode(76)))
        check(abcNode.getAndSet(newNode).b.c.d == 8)
        check(abcNode.value.b.c.d == 76)
        val l = IntArray(4){i -> i}
        any.lazySet(l)
        check((any.value as IntArray)[2] == 2)
        abcNode.value = ANode(BNode(CNode(88)))
        check(abcNode.value.b.c.d == 88)
    }
}

class TopLevelArrayTest {
    @Test
    fun testIntArray() {
        check(intArr[0].compareAndSet(0, 3))
        check(intArr[1].value == 0)
        intArr[0].lazySet(5)
        check(intArr[0].value + intArr[1].value + intArr[2].value == 5)
        check(intArr[0].compareAndSet(5, 10))
        check(intArr[0].getAndDecrement() == 10)
        check(intArr[0].value == 9)
        intArr[2].value = 2
        check(intArr[2].value == 2)
        check(intArr[2].compareAndSet(2, 34))
        check(intArr[2].value == 34)
        assertEquals(8, intArr.size + longArr.size)
    }

    @Test
    fun testLongArray() {
        longArr[0].value = 2424920024888888848
        check(longArr[0].value == 2424920024888888848)
        longArr[0].lazySet(8424920024888888848)
        check(longArr[0].value == 8424920024888888848)
        val ac = longArr[0].value
        longArr[3].value = ac
        check(longArr[3].getAndSet(8924920024888888848) == 8424920024888888848)
        check(longArr[3].value == 8924920024888888848)
        val ac1 = longArr[3].value
        longArr[4].value = ac1
        check(longArr[4].incrementAndGet() == 8924920024888888849)
        check(longArr[4].value == 8924920024888888849)
        check(longArr[4].getAndDecrement() == 8924920024888888849)
        check(longArr[4].value == 8924920024888888848)
        longArr[4].value = 8924920024888888848
        check(longArr[4].getAndAdd(100000000000000000) == 8924920024888888848)
        val ac2 = longArr[4].value
        longArr[1].value = ac2
        check(longArr[1].value == 9024920024888888848)
        check(longArr[1].addAndGet(-9223372036854775807) == -198452011965886959)
        check(longArr[1].value == -198452011965886959)
        check(longArr[1].incrementAndGet() == -198452011965886958)
        check(longArr[1].value == -198452011965886958)
        check(longArr[1].decrementAndGet() == -198452011965886959)
        check(longArr[1].value == -198452011965886959)
    }

    @Test
    fun testBooleanArray() {
        check(!booleanArr[1].value)
        booleanArr[1].compareAndSet(false, true)
        booleanArr[0].lazySet(true)
        check(!booleanArr[2].getAndSet(true))
        check(booleanArr[0].value && booleanArr[1].value && booleanArr[2].value)
    }
    @Test
    fun testTopLevelArrayOfNulls() {
        check(stringAtomicNullArr[0].value == null)
        check(stringAtomicNullArr[0].compareAndSet(null, "aa"))
        stringAtomicNullArr[1].lazySet("aa")
        check(stringAtomicNullArr[0].value == stringAtomicNullArr[1].value)
    }

    @Suppress("UNCHECKED_CAST")
    @Test
    fun testRefArray() {
        val a2 = ANode(BNode(CNode(2)))
        val a3 = ANode(BNode(CNode(3)))
        refArr[0].value = a2
        check(refArr[0].value!!.b.c.d == 2)
        check(refArr[0].compareAndSet(a2, a3))
        check(refArr[0].value!!.b.c.d == 3)
        val r0 = refArr[0].value
        refArr[3].value = r0
        check(refArr[3].value!!.b.c.d == 3)
        val a = abcNode.value
        check(refArr[3].compareAndSet(a3, a))

        val l = listOf(1, 2, 3)
        anyRefArr[5].lazySet(l)
        check((anyRefArr[5].value as List<Int>)[1] == 2)
        check(anyRefArr[5].compareAndSet(l, a2))
        check((anyRefArr[5].value as ANode<BNode<CNode>>).b.c.d == 2)
    }
}

data class ANode<T>(val b: T)
data class BNode<T>(val c: T)
data class CNode(val d: Int)