package batchimport

import (
	"context"
	"strings"
	"testing"
	"time"
)

func TestNormalizeBatchNoPrefix(t *testing.T) {
	prefix, err := normalizeBatchNoPrefix(" inb ")
	if err != nil {
		t.Fatalf("normalize prefix: %v", err)
	}
	if prefix != "INB" {
		t.Fatalf("unexpected prefix: %s", prefix)
	}

	if _, err = normalizeBatchNoPrefix("IN"); err == nil {
		t.Fatal("expected error for non-three-character prefix")
	}
}

func TestFormatBatchNo(t *testing.T) {
	now := time.Date(2026, 8, 10, 12, 34, 56, 789000000, time.Local)
	batchNo, err := formatBatchNo("REN", now, 7)
	if err != nil {
		t.Fatalf("format batch no: %v", err)
	}
	if batchNo != "REN20260810123456007" {
		t.Fatalf("unexpected batch no: %s", batchNo)
	}
	if len(batchNo) != 20 {
		t.Fatalf("unexpected batch no length: %d", len(batchNo))
	}
	if strings.Contains(batchNo, "789") {
		t.Fatalf("batch no should not include milliseconds: %s", batchNo)
	}
}

func TestFormatBatchNoRejectsOverflowSequence(t *testing.T) {
	_, err := formatBatchNo("REN", time.Now(), 1000)
	if err == nil {
		t.Fatal("expected overflow sequence error")
	}
}

func TestBatchNoBoundaries(t *testing.T) {
	now := time.Date(2026, 10, 12, 23, 59, 59, 0, time.FixedZone("CST", 8*3600))
	for _, sequence := range []int64{-1, 0, 1000} {
		if _, err := formatBatchNo("REN", now, sequence); err == nil {
			t.Fatalf("sequence %d should be rejected", sequence)
		}
	}
	for _, sequence := range []int64{1, 999} {
		number, err := formatBatchNo("REN", now, sequence)
		if err != nil || len(number) != 20 {
			t.Fatalf("sequence %d: number=%q err=%v", sequence, number, err)
		}
	}
	if batchNoSequenceKey(now) != "iotlcm:batch:no:seq:20261012" || batchNoSequenceKey(now.Add(time.Second)) != "iotlcm:batch:no:seq:20261013" {
		t.Fatal("daily sequence key does not roll over at midnight")
	}
}

func TestNextBatchNoRejectsInvalidPrefixBeforeRedis(t *testing.T) {
	for _, prefix := range []string{"", " ", "RE", "RENEW"} {
		for _, ctx := range []context.Context{nil, context.Background()} {
			number, err := NextBatchNoContext(ctx, prefix)
			if err == nil || number != "" {
				t.Fatalf("prefix %q: number=%q err=%v", prefix, number, err)
			}
		}
	}
}