# Nim-RocksDB
# Copyright 2024 Status Research & Development GmbH
# Licensed under either of
#
#  * Apache License, version 2.0, ([LICENSE-APACHE](LICENSE-APACHE) or http://www.apache.org/licenses/LICENSE-2.0)
#  * GPL license, version 2.0, ([LICENSE-GPLv2](LICENSE-GPLv2) or https://www.gnu.org/licenses/old-licenses/gpl-2.0.en.html)
#
# at your option. This file may not be copied, modified, or distributed except according to those terms.

{.used.}

import
  std/os,
  tempfile,
  unittest2,
  ../../rocksdb/lib/librocksdb,
  ../../rocksdb/columnfamily/cfhandle

suite "ColFamilyHandleRef Tests":
  const TEST_CF_NAME = "test"

  setup:
    let
      dbPath = mkdtemp() / "data"
      dbOpts = rocksdb_options_create()
      cfOpts = rocksdb_options_create()

    var errors: cstring

    rocksdb_options_set_create_if_missing(dbOpts, 1)

    let db = rocksdb_open(dbOpts, dbPath.cstring, cast[cstringArray](errors.addr))
    doAssert errors.isNil()
    doAssert not db.isNil()

    let cfHandlePtr = rocksdb_create_column_family(
      db, cfOpts, TEST_CF_NAME.cstring, cast[cstringArray](errors.addr)
    )
    doAssert errors.isNil()
    doAssert not cfHandlePtr.isNil()

  teardown:
    rocksdb_close(db)
    removeDir($dbPath)

  test "Test newColFamilyHandle":
    var cfHandle = newColFamilyHandle(cfHandlePtr)

    check:
      not cfHandle.cPtr.isNil()
      cfHandle.cPtr == cfHandlePtr

    cfHandle.close()

  test "Test close":
    var cfHandle = newColFamilyHandle(cfHandlePtr)

    check not cfHandle.isClosed()
    cfHandle.close()
    check cfHandle.isClosed()
    cfHandle.close()
    check cfHandle.isClosed()