package main

import (
	"testing"

	"github.com/stretchr/testify/assert"
)

func TestBatchProgressTracker(t *testing.T) {
	tracker := newBatchProgressTracker(100)
	tracker.markFileProcessed(1024, 10, "a.log.gz")
	assert.Equal(t, int64(1), tracker.processedFiles.Load())
	assert.Equal(t, int64(1024), tracker.downloadedBytes.Load())
	assert.Equal(t, int64(10), tracker.parsedRecords.Load())

	tracker.markUploaded(2048, 400, "statement_part1.gz")
	assert.Equal(t, int64(2048), tracker.uploadedBytes.Load())
	assert.Equal(t, int64(400), tracker.uploadedRecords.Load())
	assert.Equal(t, int64(1), tracker.uploadParts.Load())
}