Michael McGuinness
56 lines
1.3 KiB
Go
56 lines
1.3 KiB
Go
package document_test
|
|
|
|
import (
|
|
"queryorchestration/internal/document"
|
|
"queryorchestration/internal/serviceconfig"
|
|
"testing"
|
|
|
|
"github.com/google/uuid"
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestIsValidCleanVersion(t *testing.T) {
|
|
cfg := &serviceconfig.BaseConfig{}
|
|
svc := document.New(cfg)
|
|
|
|
assert.Nil(t, svc.IsValidCleanVersion(1))
|
|
assert.Error(t, svc.IsValidCleanVersion(2))
|
|
assert.Error(t, svc.IsValidCleanVersion(0))
|
|
assert.Error(t, svc.IsValidCleanVersion(-1))
|
|
}
|
|
|
|
func TestIsValidTextVersion(t *testing.T) {
|
|
cfg := &serviceconfig.BaseConfig{}
|
|
svc := document.New(cfg)
|
|
|
|
assert.Nil(t, svc.IsValidTextVersion(1))
|
|
assert.Error(t, svc.IsValidTextVersion(2))
|
|
assert.Error(t, svc.IsValidTextVersion(0))
|
|
assert.Error(t, svc.IsValidTextVersion(-1))
|
|
}
|
|
|
|
func TestGetCleanVersion(t *testing.T) {
|
|
cfg := &serviceconfig.BaseConfig{}
|
|
svc := document.New(cfg)
|
|
|
|
assert.Equal(t, int32(1), svc.GetCleanVersion())
|
|
}
|
|
|
|
func TestGetTextVersion(t *testing.T) {
|
|
cfg := &serviceconfig.BaseConfig{}
|
|
svc := document.New(cfg)
|
|
|
|
assert.Equal(t, int32(1), svc.GetTextVersion())
|
|
}
|
|
|
|
func TestIsTextExtracted(t *testing.T) {
|
|
cfg := &serviceconfig.BaseConfig{}
|
|
svc := document.New(cfg)
|
|
|
|
assert.Nil(t, svc.IsTextExtracted(&document.IsTextExtractedParams{
|
|
DocumentID: uuid.New(),
|
|
MinCleanVersion: int32(1),
|
|
MinTextVersion: int32(1),
|
|
}))
|
|
}
|