config.target = 'core'

utils = require('utils')

function with_temp_ref(...)

local args = {...}

local dtype = 'int32_t'

if type(args[1]) == 'string' then

dtype = table.remove(args, 1)

end

local count = 1

if type(args[1]) == 'number' then

count = table.remove(args, 1)

end

local func = table.remove(args, 1)

assert(type(func) == 'function')

dfhack.with_temp_object(df.new(dtype, count), func, table.unpack(args))

end

function test.zero_initialize()

with_temp_ref(function(x)

expect.eq(x.value, 0)

end)

end

function test.zero_initialize_array()

with_temp_ref(2, function(x)

expect.eq(x:_displace(1).value, 0)

end)

end

function test.displace_zero()

with_temp_ref(function(x)

expect.eq(x, x:_displace(0))

end)

end

function test.displace_nonzero()

with_temp_ref(2, function(x)

expect.ne(x, x:_displace(1))

expect.eq(utils.addressof(x) + x:sizeof(), utils.addressof(x:_displace(1)))

end)

end

function test.displace_negative()

with_temp_ref(2, function(x)

expect.true_(x:_displace(1):_displace(-1).value)

end)

end

function test.index_read()

with_temp_ref(function(x)

expect.eq(x.value, x[0])

end)

end

function test.index_write()

with_temp_ref(function(x)

x[0] = 1

expect.eq(x.value, 1)

expect.eq(x[0], 1)

x.value = 2

expect.eq(x.value, 2)

expect.eq(x[0], 2)

end)

end

function test.index_write_multi()

local len = 3

with_temp_ref(len, function(x)

for i = 0, len - 1 do

x[i] = i * i

end

for i = 0, len - 1 do

expect.eq(x[i], i * i, i)

end

end)

end

function test.index_read_negative()

with_temp_ref(function(x)

expect.error_match('negative index', function()

expect.true_(x:_displace(1)[-1])

end)

end)

end

function test.index_write_negative()

with_temp_ref(function(x)

expect.error_match('negative index', function()

x:_displace(1)[-1] = 7

end)

end)

end