package bootstrapimpl

import (
	"regexp"
	"testing"

	"github.com/DATA-DOG/go-sqlmock"
	"github.com/nexuscrm/shared/pkg/constants"
)

func TestManagedSystemObjectVisibilityPodReadinessAllowsFullyAbsentConnectedDeliveryPriorState(t *testing.T) {
	db, mock, err := sqlmock.New()
	if err != nil {
		t.Fatalf("sqlmock: %v", err)
	}
	defer db.Close()

	expectManagedVisibilityRowsWithoutConnectedDelivery(mock)
	mock.ExpectQuery(regexp.QuoteMeta("SELECT COUNT(*) FROM INFORMATION_SCHEMA.TABLES WHERE TABLE_SCHEMA = DATABASE() AND TABLE_NAME IN (?,?,?,?,?)")).
		WithArgs(
			constants.TableConnectedPrincipal,
			constants.TableConnectedAuthorizationRequest,
			constants.TableConnectedPrincipalToken,
			constants.TableConnectedDeliveryJob,
			constants.TableConnectedDeliveryPayload,
		).
		WillReturnRows(sqlmock.NewRows([]string{"COUNT(*)"}).AddRow(0))

	result := &AssertionResult{}
	assertManagedSystemObjectVisibilityWithPriorStatePolicy(db, result, true)
	if len(result.Violations) != 0 {
		t.Fatalf("violations = %+v, want fully absent connected-delivery prior state deferred", result.Violations)
	}
	if err := mock.ExpectationsWereMet(); err != nil {
		t.Fatalf("sql expectations: %v", err)
	}
}

func TestManagedSystemObjectVisibilityPodReadinessRejectsPartialConnectedDeliveryState(t *testing.T) {
	db, mock, err := sqlmock.New()
	if err != nil {
		t.Fatalf("sqlmock: %v", err)
	}
	defer db.Close()

	expectManagedVisibilityRowsWithoutConnectedDelivery(mock)
	mock.ExpectQuery(regexp.QuoteMeta("SELECT COUNT(*) FROM INFORMATION_SCHEMA.TABLES WHERE TABLE_SCHEMA = DATABASE() AND TABLE_NAME IN (?,?,?,?,?)")).
		WithArgs(
			constants.TableConnectedPrincipal,
			constants.TableConnectedAuthorizationRequest,
			constants.TableConnectedPrincipalToken,
			constants.TableConnectedDeliveryJob,
			constants.TableConnectedDeliveryPayload,
		).
		WillReturnRows(sqlmock.NewRows([]string{"COUNT(*)"}).AddRow(1))

	result := &AssertionResult{}
	assertManagedSystemObjectVisibilityWithPriorStatePolicy(db, result, true)
	assertConnectedDeliveryVisibilityViolationCount(t, result, len(connectedDeliveryTableNames))
	if err := mock.ExpectationsWereMet(); err != nil {
		t.Fatalf("sql expectations: %v", err)
	}
}

func TestManagedSystemObjectVisibilityOperatorPathRejectsAbsentConnectedDeliveryCatalog(t *testing.T) {
	db, mock, err := sqlmock.New()
	if err != nil {
		t.Fatalf("sqlmock: %v", err)
	}
	defer db.Close()

	expectManagedVisibilityRowsWithoutConnectedDelivery(mock)
	result := &AssertionResult{}
	assertManagedSystemObjectVisibility(db, result)
	assertConnectedDeliveryVisibilityViolationCount(t, result, len(connectedDeliveryTableNames))
	if err := mock.ExpectationsWereMet(); err != nil {
		t.Fatalf("sql expectations: %v", err)
	}
}

func expectManagedVisibilityRowsWithoutConnectedDelivery(mock sqlmock.Sqlmock) {
	rows := sqlmock.NewRows([]string{constants.FieldAPIName, constants.FieldIsDeleted})
	for _, def := range GetManagedTableDefinitions() {
		if !isConnectedDeliveryManagedTable(def.TableName) {
			rows.AddRow(def.TableName, false)
		}
	}
	mock.ExpectQuery(regexp.QuoteMeta("SELECT `ApiName`, `IsDeleted` FROM `_System_Object`")).
		WillReturnRows(rows)
}

func assertConnectedDeliveryVisibilityViolationCount(t *testing.T, result *AssertionResult, want int) {
	t.Helper()
	got := 0
	for _, violation := range result.Violations {
		if isConnectedDeliveryManagedTable(violation.Object) {
			got++
		}
	}
	if got != want {
		t.Fatalf("connected-delivery visibility violations = %d, want %d; all violations: %+v", got, want, result.Violations)
	}
}