// *****************************************************************************
// Copyright 2013-2024 Aerospike, Inc.
//
// Licensed under the Apache License, Version 2.0 (the "License")
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//     http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
// *****************************************************************************

'use strict'

/* global context, it, expect */

const path = require('path')

import Aerospike, { AerospikeError, Job } from 'aerospike';

import { expect } from 'chai'; 
import * as helper from './test_helper';

context('registering/unregistering UDF modules', function () {
  const client = helper.client
  const module = 'udf.lua'
  const filename = path.join(__dirname, module)

  it('should register and then remove a module', function (done) {
    client.udfRegister(filename, function (err?: AerospikeError, registerJob?: Job) {
      if (err) throw err
      registerJob?.wait(10, function (err?: AerospikeError) {
        if (err) throw err
        client.udfRemove(module, function (err?: AerospikeError, removeJob?: Job) {
          if (err) throw err
          removeJob?.wait(10, done)
        })
      })
    })
  })

  it('should register a module as Lua language', function (done) {
    client.udfRegister(filename, Aerospike.language.LUA, function (err?: AerospikeError, registerJob?: Job) {
      if (err) throw err
      registerJob?.wait(10, function (err?: AerospikeError) {
        if (err) throw err
        client.udfRemove(module, function (err?: AerospikeError, removeJob?: Job) {
          if (err) throw err
          removeJob?.wait(10, done)
        })
      })
    })
  })

  it('should register a module with an info policy', function (done) {
    const policy = new Aerospike.InfoPolicy({
      timeout: 1000,
      sendAsIs: true,
      checkBounds: false
    })

    client.udfRegister(filename, policy, function (err?: AerospikeError, registerJob?: Job) {
      if (err) throw err
      registerJob?.wait(10, function (err?: AerospikeError) {
        if (err) throw err
        client.udfRemove(module, function (err?: AerospikeError, removeJob?: Job) {
          if (err) throw err
          removeJob?.wait(10, done)
        })
      })
    })
  })

  it('should register a module as Lua language with an info policy', function (done) {
    const policy = new Aerospike.InfoPolicy({
      timeout: 1000,
      sendAsIs: true,
      checkBounds: false
    })

    client.udfRegister(filename, Aerospike.language.LUA, policy, function (err?: AerospikeError, registerJob?: Job) {
      if (err) throw err
      registerJob?.wait(10, function (err?: AerospikeError) {
        if (err) throw err
        client.udfRemove(module, function (err?: AerospikeError, removeJob?: Job) {
          if (err) throw err
          removeJob?.wait(10, done)
        })
      })
    })
  })

  it('returns a Promise if no callback function is passed', function () {
    return client.udfRegister(filename)
      .then((job: any) => job.wait(10))
      .then(() => client.udfRemove(module))
      .then((job: any) => job.wait(10))
  })

  context('error handling', function () {
    it('should fail to register an non-existent module', function (done) {
      client.udfRegister('no-such-udf.lua', function (err?: AerospikeError) {
        expect(err?.code).to.equal(Aerospike.status.ERR_CLIENT)
        done()
      })
    })

    it('should fail to register module with invalid language', function (done) {
      client.udfRegister(filename, -99, function (err?: AerospikeError) {
        expect(err?.code).to.equal(Aerospike.status.ERR_PARAM)
        done()
      })
    })

    context('removing a non-existent module', function () {
      context('server version 4.5.1 and later', function () {
        helper.skipUnlessVersion('>= 4.5.1', this)

        it('should not fail when removing a non-existent module', function () {
          return client.udfRemove('no-such-udf.lua').then((job: Job) => job.waitUntilDone())
        })
      })

      context('server version 4.5.0 and earlier', function () {
        helper.skipUnlessVersion('< 4.5.1', this)

        it('should return an error when removing a non-existent module', function (done) {
          client.udfRemove('no-such-udf.lua', function (error?: AerospikeError) {
            expect(error).to.exist.and.have.property('code', Aerospike.status.ERR_UDF)
            done()
          })
        })
      })
    })
  })
})