// Copyright (c) HashiCorp, Inc.
// SPDX-License-Identifier: BUSL-1.1

package dbassert

import (
	"database/sql"

	"github.com/hashicorp/dbassert"
	gormAssert "github.com/hashicorp/dbassert/gorm"
	"github.com/stretchr/testify/assert"
)

// DbAsserts provides database assertion methods.
type DbAsserts struct {
	asserts *gormAssert.GormAsserts
}

// New creates a new DbAsserts.
func New(t dbassert.TestingT, r *sql.DB) *DbAsserts {
	assert.NotNil(t, r, "db.Reader is nill")
	return &DbAsserts{
		asserts: gormAssert.New(t, r, "postgres"),
	}
}

// Log enable/disable log of database queries.
func (a *DbAsserts) Log(enable bool) {
	a.asserts.DbLog(enable)
}

// IsNull asserts that the resource fieldName is null in the db.
func (a *DbAsserts) IsNull(resource any, fieldName string) bool {
	return a.asserts.IsNull(resource, fieldName)
}

// NotNull asserts that the resource fieldName is not null in the db.
func (a *DbAsserts) NotNull(resource any, fieldName string) bool {
	return a.asserts.NotNull(resource, fieldName)
}

// Nullable asserts that the resource fieldName is nullable in the db.
func (a *DbAsserts) Nullable(resource any, fieldName string) bool {
	return a.asserts.Nullable(resource, fieldName)
}

// Domain asserts that the resource fieldName is the domainName in the db.
func (a *DbAsserts) Domain(resource any, fieldName, domainName string) bool {
	return a.asserts.Domain(resource, fieldName, domainName)
}