1114 lines
36 KiB
Go
1114 lines
36 KiB
Go
package service
|
|
|
|
import (
|
|
"bytes"
|
|
"context"
|
|
"encoding/json"
|
|
"errors"
|
|
"io"
|
|
"os"
|
|
"strings"
|
|
"testing"
|
|
"time"
|
|
|
|
"gis/internal/domain"
|
|
|
|
"github.com/google/uuid"
|
|
)
|
|
|
|
// stubDatasetRepo is an in-memory DatasetRepository for tests.
|
|
type stubDatasetRepo struct {
|
|
store map[uuid.UUID]domain.Dataset
|
|
observations map[uuid.UUID][]domain.Observation
|
|
districts map[string]domain.District
|
|
createErr error
|
|
deleted []uuid.UUID
|
|
lastLimit, lastOffset int
|
|
lastFilter domain.DatasetFilter
|
|
}
|
|
|
|
func newStubDatasetRepo() *stubDatasetRepo {
|
|
return &stubDatasetRepo{
|
|
store: map[uuid.UUID]domain.Dataset{},
|
|
observations: map[uuid.UUID][]domain.Observation{},
|
|
districts: map[string]domain.District{},
|
|
}
|
|
}
|
|
|
|
func (r *stubDatasetRepo) Create(_ context.Context, d domain.Dataset) (domain.Dataset, error) {
|
|
if r.createErr != nil {
|
|
return domain.Dataset{}, r.createErr
|
|
}
|
|
if d.ID == uuid.Nil {
|
|
d.ID = uuid.New()
|
|
}
|
|
r.store[d.ID] = d
|
|
return d, nil
|
|
}
|
|
|
|
func (r *stubDatasetRepo) GetByID(_ context.Context, id uuid.UUID) (domain.Dataset, error) {
|
|
d, ok := r.store[id]
|
|
if !ok {
|
|
return domain.Dataset{}, domain.ErrNotFound
|
|
}
|
|
return d, nil
|
|
}
|
|
|
|
func (r *stubDatasetRepo) ListSummaries(_ context.Context, filter domain.DatasetFilter, limit, offset int) ([]domain.DatasetSummary, error) {
|
|
r.lastFilter = filter
|
|
r.lastLimit = limit
|
|
r.lastOffset = offset
|
|
return nil, nil
|
|
}
|
|
|
|
func (r *stubDatasetRepo) Count(_ context.Context, _ domain.DatasetFilter) (int, error) {
|
|
return len(r.store), nil
|
|
}
|
|
|
|
func (r *stubDatasetRepo) Delete(_ context.Context, id uuid.UUID) error {
|
|
r.deleted = append(r.deleted, id)
|
|
delete(r.store, id)
|
|
return nil
|
|
}
|
|
|
|
func (r *stubDatasetRepo) MarkParsed(_ context.Context, id uuid.UUID, cols []domain.AttributeColumn) error {
|
|
d, ok := r.store[id]
|
|
if !ok {
|
|
return domain.ErrNotFound
|
|
}
|
|
d.AttributeColumns = cols
|
|
d.Status = domain.DatasetStatusAwaitingMapping
|
|
r.store[id] = d
|
|
return nil
|
|
}
|
|
|
|
func (r *stubDatasetRepo) MarkParseFailed(_ context.Context, id uuid.UUID, reason string) error {
|
|
d, ok := r.store[id]
|
|
if !ok {
|
|
return domain.ErrNotFound
|
|
}
|
|
d.Status = domain.DatasetStatusFailed
|
|
d.ParseError = &reason
|
|
r.store[id] = d
|
|
return nil
|
|
}
|
|
|
|
func (r *stubDatasetRepo) SaveMapping(_ context.Context, id uuid.UUID, kato string, years []domain.YearColumn) (domain.Dataset, error) {
|
|
d, ok := r.store[id]
|
|
if !ok {
|
|
return domain.Dataset{}, domain.ErrNotFound
|
|
}
|
|
d.KatoColumn = &kato
|
|
d.YearColumns = years
|
|
d.Status = domain.DatasetStatusExtracting
|
|
r.store[id] = d
|
|
return d, nil
|
|
}
|
|
|
|
func (r *stubDatasetRepo) MarkReady(_ context.Context, id uuid.UUID, geometry []byte) error {
|
|
d, ok := r.store[id]
|
|
if !ok {
|
|
return domain.ErrNotFound
|
|
}
|
|
if len(geometry) > 0 {
|
|
d.Geometry = geometry
|
|
}
|
|
d.Status = domain.DatasetStatusReady
|
|
r.store[id] = d
|
|
return nil
|
|
}
|
|
|
|
func (r *stubDatasetRepo) MarkConverted(_ context.Context, id uuid.UUID, cogKey string, footprint []byte) error {
|
|
d, ok := r.store[id]
|
|
if !ok {
|
|
return domain.ErrNotFound
|
|
}
|
|
d.CogStorageKey = &cogKey
|
|
if len(footprint) > 0 {
|
|
d.Geometry = footprint
|
|
}
|
|
d.Status = domain.DatasetStatusReady
|
|
r.store[id] = d
|
|
return nil
|
|
}
|
|
|
|
func (r *stubDatasetRepo) SetProperties(_ context.Context, id uuid.UUID, properties, geometry []byte) error {
|
|
d, ok := r.store[id]
|
|
if !ok {
|
|
return domain.ErrNotFound
|
|
}
|
|
d.Properties = properties
|
|
if len(geometry) > 0 {
|
|
d.Geometry = geometry
|
|
}
|
|
d.Status = domain.DatasetStatusReady
|
|
r.store[id] = d
|
|
return nil
|
|
}
|
|
|
|
func (r *stubDatasetRepo) ReplaceObservations(_ context.Context, id uuid.UUID, obs []domain.Observation) error {
|
|
r.observations[id] = obs
|
|
return nil
|
|
}
|
|
|
|
func (r *stubDatasetRepo) ListObservations(_ context.Context, id uuid.UUID, _ *string, _, _ int) ([]domain.Observation, error) {
|
|
return r.observations[id], nil
|
|
}
|
|
|
|
func (r *stubDatasetRepo) CountObservations(_ context.Context, id uuid.UUID, _ *string) (int, error) {
|
|
return len(r.observations[id]), nil
|
|
}
|
|
|
|
func (r *stubDatasetRepo) ListAllObservations(_ context.Context, id uuid.UUID) ([]domain.Observation, error) {
|
|
return r.observations[id], nil
|
|
}
|
|
|
|
func (r *stubDatasetRepo) DistrictGeometriesByKato(_ context.Context, katos []string) (map[string]domain.District, error) {
|
|
out := make(map[string]domain.District)
|
|
for _, k := range katos {
|
|
if d, ok := r.districts[k]; ok {
|
|
out[k] = d
|
|
}
|
|
}
|
|
return out, nil
|
|
}
|
|
|
|
// stubEnqueuer records parse, properties, extract, and convert enqueues.
|
|
type stubEnqueuer struct {
|
|
enqueued []uuid.UUID
|
|
properties []uuid.UUID
|
|
extracted []uuid.UUID
|
|
converted []uuid.UUID
|
|
err error
|
|
}
|
|
|
|
func (s *stubEnqueuer) EnqueueParse(_ context.Context, id uuid.UUID) error {
|
|
if s.err != nil {
|
|
return s.err
|
|
}
|
|
s.enqueued = append(s.enqueued, id)
|
|
return nil
|
|
}
|
|
|
|
func (s *stubEnqueuer) EnqueueProperties(_ context.Context, id uuid.UUID) error {
|
|
if s.err != nil {
|
|
return s.err
|
|
}
|
|
s.properties = append(s.properties, id)
|
|
return nil
|
|
}
|
|
|
|
func (s *stubEnqueuer) EnqueueExtract(_ context.Context, id uuid.UUID) error {
|
|
if s.err != nil {
|
|
return s.err
|
|
}
|
|
s.extracted = append(s.extracted, id)
|
|
return nil
|
|
}
|
|
|
|
func (s *stubEnqueuer) EnqueueConvert(_ context.Context, id uuid.UUID) error {
|
|
if s.err != nil {
|
|
return s.err
|
|
}
|
|
s.converted = append(s.converted, id)
|
|
return nil
|
|
}
|
|
|
|
// stubConverter records raster conversions.
|
|
type stubConverter struct {
|
|
cogCalls int
|
|
toCOGErr error
|
|
footprint []byte
|
|
footprintFn func(src string) ([]byte, error)
|
|
vectorGeom []byte
|
|
vectorGeomFn func(src string) ([]byte, error)
|
|
}
|
|
|
|
func (c *stubConverter) ToCOG(_ context.Context, _, dst string) error {
|
|
c.cogCalls++
|
|
if c.toCOGErr != nil {
|
|
return c.toCOGErr
|
|
}
|
|
return os.WriteFile(dst, []byte("COG"), 0o600)
|
|
}
|
|
|
|
func (c *stubConverter) Footprint(_ context.Context, src string) ([]byte, error) {
|
|
if c.footprintFn != nil {
|
|
return c.footprintFn(src)
|
|
}
|
|
return c.footprint, nil
|
|
}
|
|
|
|
func (c *stubConverter) VectorGeometry(_ context.Context, src string) ([]byte, error) {
|
|
if c.vectorGeomFn != nil {
|
|
return c.vectorGeomFn(src)
|
|
}
|
|
return c.vectorGeom, nil
|
|
}
|
|
|
|
var (
|
|
noopParser ColumnParser = func(string, []byte) ([]domain.AttributeColumn, error) { return nil, nil }
|
|
noopRowParser RowParser = func(string, []byte) ([]map[string]string, error) { return nil, nil }
|
|
)
|
|
|
|
// stubStore records object-store interactions.
|
|
type stubStore struct {
|
|
put []string
|
|
removed []string
|
|
putErr error
|
|
}
|
|
|
|
func (s *stubStore) Put(_ context.Context, key string, _ io.Reader, _ int64, _ string) error {
|
|
if s.putErr != nil {
|
|
return s.putErr
|
|
}
|
|
s.put = append(s.put, key)
|
|
return nil
|
|
}
|
|
|
|
func (s *stubStore) Get(_ context.Context, _ string) (io.ReadCloser, error) {
|
|
return io.NopCloser(strings.NewReader("")), nil
|
|
}
|
|
|
|
func (s *stubStore) Remove(_ context.Context, key string) error {
|
|
s.removed = append(s.removed, key)
|
|
return nil
|
|
}
|
|
|
|
// stubCategoryReader satisfies the unexported categoryReader dependency.
|
|
type stubCategoryReader struct {
|
|
exists bool
|
|
}
|
|
|
|
func (s stubCategoryReader) GetByID(_ context.Context, id uuid.UUID) (domain.Category, error) {
|
|
if !s.exists {
|
|
return domain.Category{}, domain.ErrNotFound
|
|
}
|
|
return domain.Category{ID: id}, nil
|
|
}
|
|
|
|
func (s stubCategoryReader) GetByCode(_ context.Context, code string) (domain.Category, error) {
|
|
if !s.exists {
|
|
return domain.Category{}, domain.ErrNotFound
|
|
}
|
|
return domain.Category{ID: uuid.New(), Code: code}, nil
|
|
}
|
|
|
|
func validUpload() UploadInput {
|
|
body := `{"type":"FeatureCollection","features":[]}`
|
|
return UploadInput{
|
|
CategoryID: uuid.New(),
|
|
Name: "Population",
|
|
Filename: "data.geojson",
|
|
FileType: domain.FileTypeVector,
|
|
Size: int64(len(body)),
|
|
Reader: strings.NewReader(body),
|
|
}
|
|
}
|
|
|
|
func newDatasetService(repo *stubDatasetRepo, store *stubStore, catExists bool) *DatasetService {
|
|
return NewDatasetService(repo, store, stubCategoryReader{exists: catExists}, &stubEnqueuer{}, noopParser, noopRowParser, &stubConverter{})
|
|
}
|
|
|
|
func TestDatasetService_Upload_Validation(t *testing.T) {
|
|
ctx := context.Background()
|
|
|
|
tests := []struct {
|
|
name string
|
|
mutate func(*UploadInput)
|
|
}{
|
|
{"invalid file type", func(in *UploadInput) { in.FileType = "bogus" }},
|
|
{"unknown extension", func(in *UploadInput) { in.Filename = "data.txt" }},
|
|
{"extension/type mismatch", func(in *UploadInput) { in.Filename = "data.tif" }}, // .tif is raster
|
|
}
|
|
|
|
for _, tt := range tests {
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
repo := newStubDatasetRepo()
|
|
store := &stubStore{}
|
|
svc := newDatasetService(repo, store, true)
|
|
|
|
in := validUpload()
|
|
tt.mutate(&in)
|
|
|
|
_, err := svc.Upload(ctx, in)
|
|
if !errors.Is(err, domain.ErrValidation) {
|
|
t.Fatalf("want ErrValidation, got %v", err)
|
|
}
|
|
if len(store.put) != 0 {
|
|
t.Fatalf("nothing should be uploaded on validation failure, got %v", store.put)
|
|
}
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestDatasetService_Upload_RejectsMismatchedContent(t *testing.T) {
|
|
repo := newStubDatasetRepo()
|
|
store := &stubStore{}
|
|
svc := newDatasetService(repo, store, true)
|
|
|
|
// Declared raster .tif but the bytes are JSON, not a TIFF.
|
|
in := validUpload()
|
|
in.FileType = domain.FileTypeRaster
|
|
in.Filename = "fake.tif"
|
|
in.Reader = strings.NewReader(`{"type":"FeatureCollection"}`)
|
|
|
|
_, err := svc.Upload(context.Background(), in)
|
|
if !errors.Is(err, domain.ErrValidation) {
|
|
t.Fatalf("want ErrValidation, got %v", err)
|
|
}
|
|
if len(store.put) != 0 {
|
|
t.Fatalf("mismatched file should not be stored, got %v", store.put)
|
|
}
|
|
}
|
|
|
|
func TestDatasetService_Upload_MissingCategory(t *testing.T) {
|
|
svc := newDatasetService(newStubDatasetRepo(), &stubStore{}, false)
|
|
_, err := svc.Upload(context.Background(), validUpload())
|
|
if !errors.Is(err, domain.ErrValidation) {
|
|
t.Fatalf("want ErrValidation, got %v", err)
|
|
}
|
|
}
|
|
|
|
func TestDatasetService_Upload_Success(t *testing.T) {
|
|
repo := newStubDatasetRepo()
|
|
store := &stubStore{}
|
|
svc := newDatasetService(repo, store, true)
|
|
|
|
in := validUpload()
|
|
in.Name = "" // should fall back to filename
|
|
|
|
got, err := svc.Upload(context.Background(), in)
|
|
if err != nil {
|
|
t.Fatalf("unexpected error: %v", err)
|
|
}
|
|
if got.Name != in.Filename {
|
|
t.Fatalf("name should default to filename, got %q", got.Name)
|
|
}
|
|
if len(store.put) != 1 {
|
|
t.Fatalf("want one stored object, got %v", store.put)
|
|
}
|
|
if got.StorageKey != store.put[0] {
|
|
t.Fatalf("dataset storage key %q != stored key %q", got.StorageKey, store.put[0])
|
|
}
|
|
}
|
|
|
|
func TestDatasetService_Upload_CompensatesOnDBFailure(t *testing.T) {
|
|
repo := newStubDatasetRepo()
|
|
repo.createErr = errors.New("insert failed")
|
|
store := &stubStore{}
|
|
svc := newDatasetService(repo, store, true)
|
|
|
|
_, err := svc.Upload(context.Background(), validUpload())
|
|
if err == nil {
|
|
t.Fatal("expected an error")
|
|
}
|
|
if len(store.put) != 1 || len(store.removed) != 1 {
|
|
t.Fatalf("orphaned object not cleaned up: put=%v removed=%v", store.put, store.removed)
|
|
}
|
|
if store.put[0] != store.removed[0] {
|
|
t.Fatalf("removed key %q != stored key %q", store.removed[0], store.put[0])
|
|
}
|
|
}
|
|
|
|
func TestDatasetService_Upload_VectorWithKato_EnqueuesParse(t *testing.T) {
|
|
repo := newStubDatasetRepo()
|
|
enq := &stubEnqueuer{}
|
|
svc := NewDatasetService(repo, &stubStore{}, stubCategoryReader{exists: true}, enq, noopParser, noopRowParser, &stubConverter{})
|
|
|
|
in := validUpload()
|
|
in.FileType = domain.FileTypeVectorWithKato
|
|
in.Filename = "regions.geojson"
|
|
|
|
got, err := svc.Upload(context.Background(), in)
|
|
if err != nil {
|
|
t.Fatalf("unexpected error: %v", err)
|
|
}
|
|
if got.Status != domain.DatasetStatusParsing {
|
|
t.Fatalf("want status parsing, got %q", got.Status)
|
|
}
|
|
if len(enq.enqueued) != 1 || enq.enqueued[0] != got.ID {
|
|
t.Fatalf("parse not enqueued for dataset: %v", enq.enqueued)
|
|
}
|
|
}
|
|
|
|
func TestDatasetService_Upload_Vector_EnqueuesProperties(t *testing.T) {
|
|
repo := newStubDatasetRepo()
|
|
enq := &stubEnqueuer{}
|
|
svc := NewDatasetService(repo, &stubStore{}, stubCategoryReader{exists: true}, enq, noopParser, noopRowParser, &stubConverter{})
|
|
|
|
got, err := svc.Upload(context.Background(), validUpload()) // plain vector .geojson
|
|
if err != nil {
|
|
t.Fatalf("unexpected error: %v", err)
|
|
}
|
|
if got.Status != domain.DatasetStatusProcessing {
|
|
t.Fatalf("want status processing, got %q", got.Status)
|
|
}
|
|
if len(enq.properties) != 1 || enq.properties[0] != got.ID {
|
|
t.Fatalf("properties extraction not enqueued: %v", enq.properties)
|
|
}
|
|
}
|
|
|
|
func TestDatasetService_ExtractProperties(t *testing.T) {
|
|
id := uuid.New()
|
|
repo := newStubDatasetRepo()
|
|
repo.store[id] = domain.Dataset{
|
|
ID: id, FileType: domain.FileTypeVector,
|
|
Filename: "d.geojson", StorageKey: "k", Status: domain.DatasetStatusProcessing,
|
|
}
|
|
rows := []map[string]string{
|
|
{"name": "Astana", "pop": "1000"},
|
|
{"name": "Almaty", "pop": "2000"},
|
|
}
|
|
rp := RowParser(func(string, []byte) ([]map[string]string, error) { return rows, nil })
|
|
geom := []byte(`{"type":"GeometryCollection","geometries":[]}`)
|
|
conv := &stubConverter{vectorGeom: geom}
|
|
svc := NewDatasetService(repo, &stubStore{}, stubCategoryReader{exists: true}, &stubEnqueuer{}, noopParser, rp, conv)
|
|
|
|
if err := svc.ExtractProperties(context.Background(), id); err != nil {
|
|
t.Fatalf("unexpected error: %v", err)
|
|
}
|
|
got := repo.store[id]
|
|
if got.Status != domain.DatasetStatusReady {
|
|
t.Fatalf("want ready, got %q", got.Status)
|
|
}
|
|
var parsed []map[string]string
|
|
if err := json.Unmarshal(got.Properties, &parsed); err != nil {
|
|
t.Fatalf("properties not valid JSON: %v (%s)", err, got.Properties)
|
|
}
|
|
if len(parsed) != 2 {
|
|
t.Fatalf("want 2 rows in properties, got %d", len(parsed))
|
|
}
|
|
if string(got.Geometry) != string(geom) {
|
|
t.Fatalf("want geometry %s, got %s", geom, got.Geometry)
|
|
}
|
|
}
|
|
|
|
func TestDatasetService_ExtractProperties_GeometryBestEffort(t *testing.T) {
|
|
id := uuid.New()
|
|
repo := newStubDatasetRepo()
|
|
repo.store[id] = domain.Dataset{
|
|
ID: id, FileType: domain.FileTypeVector,
|
|
Filename: "d.geojson", StorageKey: "k", Status: domain.DatasetStatusProcessing,
|
|
}
|
|
rp := RowParser(func(string, []byte) ([]map[string]string, error) {
|
|
return []map[string]string{{"name": "Astana"}}, nil
|
|
})
|
|
// Geometry extraction fails; the job must still succeed with properties set.
|
|
conv := &stubConverter{vectorGeomFn: func(string) ([]byte, error) {
|
|
return nil, errors.New("ogr2ogr boom")
|
|
}}
|
|
svc := NewDatasetService(repo, &stubStore{}, stubCategoryReader{exists: true}, &stubEnqueuer{}, noopParser, rp, conv)
|
|
|
|
if err := svc.ExtractProperties(context.Background(), id); err != nil {
|
|
t.Fatalf("unexpected error: %v", err)
|
|
}
|
|
got := repo.store[id]
|
|
if got.Status != domain.DatasetStatusReady {
|
|
t.Fatalf("want ready, got %q", got.Status)
|
|
}
|
|
if got.Geometry != nil {
|
|
t.Fatalf("expected nil geometry on extraction failure, got %s", got.Geometry)
|
|
}
|
|
if got.Properties == nil {
|
|
t.Fatalf("expected properties to be set despite geometry failure")
|
|
}
|
|
}
|
|
|
|
func TestDatasetService_ExtractProperties_NoTable(t *testing.T) {
|
|
id := uuid.New()
|
|
repo := newStubDatasetRepo()
|
|
repo.store[id] = domain.Dataset{
|
|
ID: id, FileType: domain.FileTypeVector,
|
|
Filename: "d.geojson", StorageKey: "k", Status: domain.DatasetStatusProcessing,
|
|
}
|
|
// features with no attributes
|
|
rp := RowParser(func(string, []byte) ([]map[string]string, error) {
|
|
return []map[string]string{{}, {}}, nil
|
|
})
|
|
svc := NewDatasetService(repo, &stubStore{}, stubCategoryReader{exists: true}, &stubEnqueuer{}, noopParser, rp, &stubConverter{})
|
|
|
|
if err := svc.ExtractProperties(context.Background(), id); err != nil {
|
|
t.Fatalf("unexpected error: %v", err)
|
|
}
|
|
got := repo.store[id]
|
|
if got.Status != domain.DatasetStatusReady {
|
|
t.Fatalf("want ready, got %q", got.Status)
|
|
}
|
|
if got.Properties != nil {
|
|
t.Fatalf("expected nil properties for empty table, got %s", got.Properties)
|
|
}
|
|
}
|
|
|
|
func TestDatasetService_Upload_Raster_EnqueuesConvert(t *testing.T) {
|
|
repo := newStubDatasetRepo()
|
|
enq := &stubEnqueuer{}
|
|
svc := NewDatasetService(repo, &stubStore{}, stubCategoryReader{exists: true}, enq, noopParser, noopRowParser, &stubConverter{})
|
|
|
|
in := validUpload()
|
|
in.FileType = domain.FileTypeRaster
|
|
in.Filename = "dem.tif"
|
|
in.Reader = bytes.NewReader([]byte("II*\x00\x08\x00\x00\x00")) // TIFF magic
|
|
|
|
got, err := svc.Upload(context.Background(), in)
|
|
if err != nil {
|
|
t.Fatalf("unexpected error: %v", err)
|
|
}
|
|
if got.Status != domain.DatasetStatusProcessing {
|
|
t.Fatalf("want status processing, got %q", got.Status)
|
|
}
|
|
if len(enq.converted) != 1 || enq.converted[0] != got.ID {
|
|
t.Fatalf("conversion not enqueued: %v", enq.converted)
|
|
}
|
|
}
|
|
|
|
func TestDatasetService_ConvertToCOG(t *testing.T) {
|
|
id := uuid.New()
|
|
repo := newStubDatasetRepo()
|
|
repo.store[id] = domain.Dataset{
|
|
ID: id, FileType: domain.FileTypeRaster,
|
|
Filename: "dem.tif", StorageKey: "uid/dem.tif",
|
|
Status: domain.DatasetStatusProcessing,
|
|
}
|
|
store := &stubStore{}
|
|
footprint := []byte(`{"type":"Polygon","coordinates":[[[70,50],[72,50],[72,52],[70,52],[70,50]]]}`)
|
|
conv := &stubConverter{footprint: footprint}
|
|
svc := NewDatasetService(repo, store, stubCategoryReader{exists: true}, &stubEnqueuer{}, noopParser, noopRowParser, conv)
|
|
|
|
if err := svc.ConvertToCOG(context.Background(), id); err != nil {
|
|
t.Fatalf("unexpected error: %v", err)
|
|
}
|
|
got := repo.store[id]
|
|
if got.Status != domain.DatasetStatusReady {
|
|
t.Fatalf("want ready, got %q", got.Status)
|
|
}
|
|
if got.CogStorageKey == nil || *got.CogStorageKey != "uid/cog/dem.tif" {
|
|
t.Fatalf("unexpected cog key: %v", got.CogStorageKey)
|
|
}
|
|
if string(got.Geometry) != string(footprint) {
|
|
t.Fatalf("footprint not stored: %s", got.Geometry)
|
|
}
|
|
if conv.cogCalls != 1 {
|
|
t.Fatalf("ToCOG calls = %d, want 1", conv.cogCalls)
|
|
}
|
|
if len(store.put) != 1 || store.put[0] != "uid/cog/dem.tif" {
|
|
t.Fatalf("cog not uploaded: %v", store.put)
|
|
}
|
|
}
|
|
|
|
func TestDatasetService_ConvertToCOG_RecordsFailure(t *testing.T) {
|
|
id := uuid.New()
|
|
repo := newStubDatasetRepo()
|
|
repo.store[id] = domain.Dataset{
|
|
ID: id, FileType: domain.FileTypeRaster,
|
|
Filename: "dem.tif", StorageKey: "uid/dem.tif",
|
|
Status: domain.DatasetStatusProcessing,
|
|
}
|
|
conv := &stubConverter{toCOGErr: errors.New("gdal failed")}
|
|
svc := NewDatasetService(repo, &stubStore{}, stubCategoryReader{exists: true}, &stubEnqueuer{}, noopParser, noopRowParser, conv)
|
|
|
|
if err := svc.ConvertToCOG(context.Background(), id); err != nil {
|
|
t.Fatalf("conversion failure should be recorded, not returned: %v", err)
|
|
}
|
|
if repo.store[id].Status != domain.DatasetStatusFailed {
|
|
t.Fatalf("want failed, got %q", repo.store[id].Status)
|
|
}
|
|
}
|
|
|
|
func TestDatasetService_Parse(t *testing.T) {
|
|
id := uuid.New()
|
|
repo := newStubDatasetRepo()
|
|
repo.store[id] = domain.Dataset{
|
|
ID: id, FileType: domain.FileTypeVectorWithKato,
|
|
Filename: "r.geojson", StorageKey: "k", Status: domain.DatasetStatusParsing,
|
|
}
|
|
cols := []domain.AttributeColumn{{Name: "като"}, {Name: "F_2023"}}
|
|
parser := ColumnParser(func(string, []byte) ([]domain.AttributeColumn, error) { return cols, nil })
|
|
svc := NewDatasetService(repo, &stubStore{}, stubCategoryReader{exists: true}, &stubEnqueuer{}, parser, noopRowParser, &stubConverter{})
|
|
|
|
if err := svc.Parse(context.Background(), id); err != nil {
|
|
t.Fatalf("unexpected error: %v", err)
|
|
}
|
|
got := repo.store[id]
|
|
if got.Status != domain.DatasetStatusAwaitingMapping {
|
|
t.Fatalf("want awaiting_mapping, got %q", got.Status)
|
|
}
|
|
if len(got.AttributeColumns) != 2 {
|
|
t.Fatalf("columns not stored: %v", got.AttributeColumns)
|
|
}
|
|
}
|
|
|
|
func TestDatasetService_Parse_RecordsFailure(t *testing.T) {
|
|
id := uuid.New()
|
|
repo := newStubDatasetRepo()
|
|
repo.store[id] = domain.Dataset{
|
|
ID: id, FileType: domain.FileTypeVectorWithKato,
|
|
Filename: "r.zip", StorageKey: "k", Status: domain.DatasetStatusParsing,
|
|
}
|
|
parser := ColumnParser(func(string, []byte) ([]domain.AttributeColumn, error) {
|
|
return nil, errors.New("corrupt archive")
|
|
})
|
|
svc := NewDatasetService(repo, &stubStore{}, stubCategoryReader{exists: true}, &stubEnqueuer{}, parser, noopRowParser, &stubConverter{})
|
|
|
|
if err := svc.Parse(context.Background(), id); err != nil {
|
|
t.Fatalf("parse failure should be recorded, not returned: %v", err)
|
|
}
|
|
got := repo.store[id]
|
|
if got.Status != domain.DatasetStatusFailed {
|
|
t.Fatalf("want failed, got %q", got.Status)
|
|
}
|
|
if got.ParseError == nil || *got.ParseError == "" {
|
|
t.Fatal("expected parse error to be recorded")
|
|
}
|
|
}
|
|
|
|
func TestDatasetService_SaveMapping(t *testing.T) {
|
|
ctx := context.Background()
|
|
id := uuid.New()
|
|
base := domain.Dataset{
|
|
ID: id, FileType: domain.FileTypeVectorWithKato,
|
|
Status: domain.DatasetStatusAwaitingMapping,
|
|
AttributeColumns: []domain.AttributeColumn{{Name: "като"}, {Name: "F_2023"}},
|
|
}
|
|
validYears := []domain.YearColumn{{Column: "F_2023", Date: "2023-01-01"}}
|
|
|
|
newSvc := func() (*stubDatasetRepo, *DatasetService) {
|
|
repo := newStubDatasetRepo()
|
|
repo.store[id] = base
|
|
return repo, newDatasetService(repo, &stubStore{}, true)
|
|
}
|
|
|
|
t.Run("unknown kato column", func(t *testing.T) {
|
|
_, svc := newSvc()
|
|
_, err := svc.SaveMapping(ctx, id, MappingInput{KatoColumn: "missing", YearColumns: validYears})
|
|
if !errors.Is(err, domain.ErrValidation) {
|
|
t.Fatalf("want ErrValidation, got %v", err)
|
|
}
|
|
})
|
|
|
|
t.Run("unknown year column", func(t *testing.T) {
|
|
_, svc := newSvc()
|
|
_, err := svc.SaveMapping(ctx, id, MappingInput{KatoColumn: "като", YearColumns: []domain.YearColumn{{Column: "X", Date: "2023-01-01"}}})
|
|
if !errors.Is(err, domain.ErrValidation) {
|
|
t.Fatalf("want ErrValidation, got %v", err)
|
|
}
|
|
})
|
|
|
|
t.Run("bad date", func(t *testing.T) {
|
|
_, svc := newSvc()
|
|
_, err := svc.SaveMapping(ctx, id, MappingInput{KatoColumn: "като", YearColumns: []domain.YearColumn{{Column: "F_2023", Date: "2023"}}})
|
|
if !errors.Is(err, domain.ErrValidation) {
|
|
t.Fatalf("want ErrValidation, got %v", err)
|
|
}
|
|
})
|
|
|
|
t.Run("no year columns", func(t *testing.T) {
|
|
_, svc := newSvc()
|
|
_, err := svc.SaveMapping(ctx, id, MappingInput{KatoColumn: "като"})
|
|
if !errors.Is(err, domain.ErrValidation) {
|
|
t.Fatalf("want ErrValidation, got %v", err)
|
|
}
|
|
})
|
|
|
|
t.Run("wrong state is a conflict", func(t *testing.T) {
|
|
repo := newStubDatasetRepo()
|
|
d := base
|
|
d.Status = domain.DatasetStatusParsing
|
|
repo.store[id] = d
|
|
svc := newDatasetService(repo, &stubStore{}, true)
|
|
_, err := svc.SaveMapping(ctx, id, MappingInput{KatoColumn: "като", YearColumns: validYears})
|
|
if !errors.Is(err, domain.ErrConflict) {
|
|
t.Fatalf("want ErrConflict, got %v", err)
|
|
}
|
|
})
|
|
|
|
t.Run("success moves to extracting and enqueues extraction", func(t *testing.T) {
|
|
repo := newStubDatasetRepo()
|
|
repo.store[id] = base
|
|
enq := &stubEnqueuer{}
|
|
svc := NewDatasetService(repo, &stubStore{}, stubCategoryReader{exists: true}, enq, noopParser, noopRowParser, &stubConverter{})
|
|
|
|
got, err := svc.SaveMapping(ctx, id, MappingInput{KatoColumn: "като", YearColumns: validYears})
|
|
if err != nil {
|
|
t.Fatalf("unexpected error: %v", err)
|
|
}
|
|
if got.Status != domain.DatasetStatusExtracting {
|
|
t.Fatalf("want extracting, got %q", got.Status)
|
|
}
|
|
if got.KatoColumn == nil || *got.KatoColumn != "като" {
|
|
t.Fatalf("kato column not saved: %+v", got.KatoColumn)
|
|
}
|
|
if len(enq.extracted) != 1 || enq.extracted[0] != id {
|
|
t.Fatalf("extraction not enqueued: %v", enq.extracted)
|
|
}
|
|
})
|
|
}
|
|
|
|
func TestBuildObservations(t *testing.T) {
|
|
id := uuid.New()
|
|
years := []domain.YearColumn{
|
|
{Column: "F_2023", Date: "2023-01-01"},
|
|
{Column: "D_2025", Date: "2025-01-01"},
|
|
}
|
|
rows := []map[string]string{
|
|
{"като": "751010000", "F_2023": "100", "D_2025": "n/a"},
|
|
{"като": "751020000", "F_2023": "150", "D_2025": "250"},
|
|
{"като": "", "F_2023": "999"}, // skipped: no KATO code
|
|
}
|
|
|
|
obs := buildObservations(id, "като", years, rows)
|
|
if len(obs) != 4 { // 2 valid rows x 2 years
|
|
t.Fatalf("want 4 observations, got %d", len(obs))
|
|
}
|
|
|
|
byKey := map[string]domain.Observation{}
|
|
for _, o := range obs {
|
|
byKey[o.KatoCode+"|"+o.Date] = o
|
|
}
|
|
if o := byKey["751010000|2023-01-01"]; o.Value == nil || *o.Value != 100 {
|
|
t.Errorf("numeric cell not stored as value: %+v", o)
|
|
}
|
|
if o := byKey["751010000|2025-01-01"]; o.ValueText == nil || *o.ValueText != "n/a" {
|
|
t.Errorf("non-numeric cell not stored as value_text: %+v", o)
|
|
}
|
|
}
|
|
|
|
func TestDatasetService_Extract(t *testing.T) {
|
|
id := uuid.New()
|
|
repo := newStubDatasetRepo()
|
|
kato := "като"
|
|
repo.store[id] = domain.Dataset{
|
|
ID: id, FileType: domain.FileTypeVectorWithKato,
|
|
Filename: "r.geojson", StorageKey: "k",
|
|
Status: domain.DatasetStatusExtracting,
|
|
KatoColumn: &kato,
|
|
YearColumns: []domain.YearColumn{{Column: "F_2023", Date: "2023-01-01"}},
|
|
}
|
|
rows := []map[string]string{{"като": "751010000", "F_2023": "100"}}
|
|
rp := RowParser(func(string, []byte) ([]map[string]string, error) { return rows, nil })
|
|
geom := []byte(`{"type":"GeometryCollection","geometries":[]}`)
|
|
conv := &stubConverter{vectorGeom: geom}
|
|
svc := NewDatasetService(repo, &stubStore{}, stubCategoryReader{exists: true}, &stubEnqueuer{}, noopParser, rp, conv)
|
|
|
|
if err := svc.Extract(context.Background(), id); err != nil {
|
|
t.Fatalf("unexpected error: %v", err)
|
|
}
|
|
if repo.store[id].Status != domain.DatasetStatusReady {
|
|
t.Fatalf("want ready, got %q", repo.store[id].Status)
|
|
}
|
|
if string(repo.store[id].Geometry) != string(geom) {
|
|
t.Fatalf("want geometry %s, got %s", geom, repo.store[id].Geometry)
|
|
}
|
|
got := repo.observations[id]
|
|
if len(got) != 1 || got[0].KatoCode != "751010000" || got[0].Value == nil || *got[0].Value != 100 {
|
|
t.Fatalf("unexpected observations: %+v", got)
|
|
}
|
|
}
|
|
|
|
func TestDatasetService_GeoJSON_DistrictJoin(t *testing.T) {
|
|
ctx := context.Background()
|
|
id := uuid.New()
|
|
repo := newStubDatasetRepo()
|
|
repo.store[id] = domain.Dataset{ID: id, FileType: domain.FileTypeVectorWithKato, Status: domain.DatasetStatusReady}
|
|
|
|
v2020, v2021 := 100.0, 150.0
|
|
repo.observations[id] = []domain.Observation{
|
|
{KatoCode: "710000000", Date: "2020-01-01", Value: &v2020},
|
|
{KatoCode: "710000000", Date: "2021-01-01", Value: &v2021},
|
|
{KatoCode: "999999999", Date: "2020-01-01", Value: &v2020}, // no district -> skipped
|
|
}
|
|
repo.districts["710000000"] = domain.District{
|
|
Kato: "710000000", Name: "Astana",
|
|
Geometry: json.RawMessage(`{"type":"Polygon","coordinates":[[[71,51],[72,51],[72,52],[71,51]]]}`),
|
|
}
|
|
|
|
svc := newDatasetService(repo, &stubStore{}, true)
|
|
fc, err := svc.GeoJSON(ctx, id)
|
|
if err != nil {
|
|
t.Fatalf("unexpected error: %v", err)
|
|
}
|
|
if fc.Type != domain.GeoJSONFeatureCollection {
|
|
t.Fatalf("type = %q", fc.Type)
|
|
}
|
|
if len(fc.Features) != 1 {
|
|
t.Fatalf("want 1 feature (the unmatched KATO is skipped), got %d", len(fc.Features))
|
|
}
|
|
f := fc.Features[0]
|
|
if f.Type != domain.GeoJSONFeature {
|
|
t.Fatalf("feature type = %q", f.Type)
|
|
}
|
|
if len(f.Geometry) == 0 {
|
|
t.Fatal("feature geometry should come from the district")
|
|
}
|
|
if f.Properties["kato"] != "710000000" || f.Properties["name"] != "Astana" {
|
|
t.Fatalf("unexpected properties: %+v", f.Properties)
|
|
}
|
|
data, ok := f.Properties["data"].(map[string]any)
|
|
if !ok {
|
|
t.Fatalf("observations not nested under data: %+v", f.Properties)
|
|
}
|
|
if data["2020-01-01"] != 100.0 || data["2021-01-01"] != 150.0 {
|
|
t.Fatalf("year values not nested under data: %+v", data)
|
|
}
|
|
|
|
// The whole thing must marshal to valid GeoJSON.
|
|
if _, err := json.Marshal(fc); err != nil {
|
|
t.Fatalf("feature collection not valid JSON: %v", err)
|
|
}
|
|
}
|
|
|
|
func TestDatasetService_GeoJSON_IgnoresDatasetGeometry(t *testing.T) {
|
|
ctx := context.Background()
|
|
id := uuid.New()
|
|
repo := newStubDatasetRepo()
|
|
// Dataset HAS its own geometry, which GeoJSON must ignore entirely for a
|
|
// vector_with_kato dataset, joining the districts table on KATO code instead.
|
|
repo.store[id] = domain.Dataset{
|
|
ID: id, FileType: domain.FileTypeVectorWithKato, Status: domain.DatasetStatusReady,
|
|
Name: "Population", Geometry: json.RawMessage(`{"type":"MultiPolygon","coordinates":[[[[0,0]]]]}`),
|
|
}
|
|
v := 100.0
|
|
repo.observations[id] = []domain.Observation{
|
|
{KatoCode: "710000000", Date: "2020-01-01", Value: &v},
|
|
{KatoCode: "999999999", Date: "2020-01-01", Value: &v}, // no district -> skipped
|
|
}
|
|
district := json.RawMessage(`{"type":"Polygon","coordinates":[[[71,51],[72,51],[72,52],[71,51]]]}`)
|
|
repo.districts["710000000"] = domain.District{Kato: "710000000", Name: "Astana", Geometry: district}
|
|
|
|
svc := newDatasetService(repo, &stubStore{}, true)
|
|
fc, err := svc.GeoJSON(ctx, id)
|
|
if err != nil {
|
|
t.Fatalf("unexpected error: %v", err)
|
|
}
|
|
if len(fc.Features) != 1 {
|
|
t.Fatalf("want 1 feature (unmatched KATO skipped), got %d", len(fc.Features))
|
|
}
|
|
f := fc.Features[0]
|
|
if string(f.Geometry) != string(district) {
|
|
t.Fatalf("geometry must come from the district, not the dataset: %s", f.Geometry)
|
|
}
|
|
if f.Properties["kato"] != "710000000" || f.Properties["name"] != "Astana" {
|
|
t.Fatalf("unexpected properties: %+v", f.Properties)
|
|
}
|
|
data, ok := f.Properties["data"].(map[string]any)
|
|
if !ok {
|
|
t.Fatalf("observations must be nested under data: %+v", f.Properties)
|
|
}
|
|
if data["2020-01-01"] != 100.0 {
|
|
t.Fatalf("observation value not mapped under data: %+v", data)
|
|
}
|
|
if _, flat := f.Properties["2020-01-01"]; flat {
|
|
t.Fatalf("observation dates must not be flat in properties: %+v", f.Properties)
|
|
}
|
|
}
|
|
|
|
func TestDatasetService_GeoJSON_ConflictWhenNotReady(t *testing.T) {
|
|
id := uuid.New()
|
|
repo := newStubDatasetRepo()
|
|
repo.store[id] = domain.Dataset{ID: id, FileType: domain.FileTypeVectorWithKato, Status: domain.DatasetStatusExtracting}
|
|
svc := newDatasetService(repo, &stubStore{}, true)
|
|
|
|
if _, err := svc.GeoJSON(context.Background(), id); !errors.Is(err, domain.ErrConflict) {
|
|
t.Fatalf("want ErrConflict for non-ready dataset, got %v", err)
|
|
}
|
|
}
|
|
|
|
func TestDatasetService_GeoJSON_RejectsRaster(t *testing.T) {
|
|
id := uuid.New()
|
|
repo := newStubDatasetRepo()
|
|
repo.store[id] = domain.Dataset{ID: id, FileType: domain.FileTypeRaster, Status: domain.DatasetStatusReady}
|
|
svc := newDatasetService(repo, &stubStore{}, true)
|
|
|
|
if _, err := svc.GeoJSON(context.Background(), id); !errors.Is(err, domain.ErrValidation) {
|
|
t.Fatalf("want ErrValidation, got %v", err)
|
|
}
|
|
}
|
|
|
|
func TestDatasetService_GeoJSON_Vector_GeometryOnly(t *testing.T) {
|
|
id := uuid.New()
|
|
repo := newStubDatasetRepo()
|
|
geom := json.RawMessage(`{"type":"MultiPolygon","coordinates":[[[[1,1],[2,2],[3,1],[1,1]]]]}`)
|
|
repo.store[id] = domain.Dataset{
|
|
ID: id, FileType: domain.FileTypeVector, Status: domain.DatasetStatusReady,
|
|
Name: "Roads", Geometry: geom,
|
|
}
|
|
svc := newDatasetService(repo, &stubStore{}, true)
|
|
|
|
fc, err := svc.GeoJSON(context.Background(), id)
|
|
if err != nil {
|
|
t.Fatalf("unexpected error: %v", err)
|
|
}
|
|
if len(fc.Features) != 1 {
|
|
t.Fatalf("want a single geometry-only feature, got %d", len(fc.Features))
|
|
}
|
|
f := fc.Features[0]
|
|
if string(f.Geometry) != string(geom) {
|
|
t.Fatalf("feature should reuse the dataset geometry, got %s", f.Geometry)
|
|
}
|
|
if len(f.Properties) != 0 {
|
|
t.Fatalf("vector feature should have empty properties, got %+v", f.Properties)
|
|
}
|
|
if _, err := json.Marshal(fc); err != nil {
|
|
t.Fatalf("feature collection not valid JSON: %v", err)
|
|
}
|
|
}
|
|
|
|
func TestDatasetService_GeoJSON_Vector_TableDataAsProperties(t *testing.T) {
|
|
id := uuid.New()
|
|
repo := newStubDatasetRepo()
|
|
geom := json.RawMessage(`{"type":"Polygon","coordinates":[[[1,1],[2,2],[3,1],[1,1]]]}`)
|
|
repo.store[id] = domain.Dataset{
|
|
ID: id, FileType: domain.FileTypeVector, Status: domain.DatasetStatusReady,
|
|
Geometry: geom,
|
|
Properties: json.RawMessage(`[{"name":"Astana","pop":"1000"}]`), // single gpkg row
|
|
}
|
|
svc := newDatasetService(repo, &stubStore{}, true)
|
|
|
|
fc, err := svc.GeoJSON(context.Background(), id)
|
|
if err != nil {
|
|
t.Fatalf("unexpected error: %v", err)
|
|
}
|
|
if len(fc.Features) != 1 {
|
|
t.Fatalf("want 1 feature, got %d", len(fc.Features))
|
|
}
|
|
props := fc.Features[0].Properties
|
|
if props["name"] != "Astana" || props["pop"] != "1000" {
|
|
t.Fatalf("table data not exposed as top-level properties: %+v", props)
|
|
}
|
|
}
|
|
|
|
func TestDatasetService_GeoJSON_Vector_MultiRowTableData(t *testing.T) {
|
|
id := uuid.New()
|
|
repo := newStubDatasetRepo()
|
|
geom := json.RawMessage(`{"type":"Polygon","coordinates":[[[1,1],[2,2],[3,1],[1,1]]]}`)
|
|
repo.store[id] = domain.Dataset{
|
|
ID: id, FileType: domain.FileTypeVector, Status: domain.DatasetStatusReady,
|
|
Geometry: geom,
|
|
Properties: json.RawMessage(`[{"name":"Astana"},{"name":"Almaty"}]`),
|
|
}
|
|
svc := newDatasetService(repo, &stubStore{}, true)
|
|
|
|
fc, err := svc.GeoJSON(context.Background(), id)
|
|
if err != nil {
|
|
t.Fatalf("unexpected error: %v", err)
|
|
}
|
|
rows, ok := fc.Features[0].Properties["rows"].([]map[string]any)
|
|
if !ok || len(rows) != 2 {
|
|
t.Fatalf("multi-row table data not kept under \"rows\": %+v", fc.Features[0].Properties)
|
|
}
|
|
}
|
|
|
|
func TestDatasetService_GeoJSON_Vector_NoGeometry(t *testing.T) {
|
|
id := uuid.New()
|
|
repo := newStubDatasetRepo()
|
|
repo.store[id] = domain.Dataset{ID: id, FileType: domain.FileTypeVector, Status: domain.DatasetStatusReady}
|
|
svc := newDatasetService(repo, &stubStore{}, true)
|
|
|
|
fc, err := svc.GeoJSON(context.Background(), id)
|
|
if err != nil {
|
|
t.Fatalf("unexpected error: %v", err)
|
|
}
|
|
if fc.Type != domain.GeoJSONFeatureCollection {
|
|
t.Fatalf("type = %q", fc.Type)
|
|
}
|
|
if len(fc.Features) != 0 {
|
|
t.Fatalf("want empty feature collection when vector has no geometry, got %d", len(fc.Features))
|
|
}
|
|
}
|
|
|
|
func TestDatasetService_ListSummaries_ClampsPaging(t *testing.T) {
|
|
repo := newStubDatasetRepo()
|
|
repo.store[uuid.New()] = domain.Dataset{}
|
|
svc := newDatasetService(repo, &stubStore{}, true)
|
|
|
|
// page < 1 -> 1, pageSize > max -> MaxPageSize, offset = 0.
|
|
res, err := svc.ListSummaries(context.Background(), domain.DatasetFilter{}, nil, 0, 10_000)
|
|
if err != nil {
|
|
t.Fatalf("unexpected error: %v", err)
|
|
}
|
|
if res.Page != 1 || res.PageSize != MaxPageSize {
|
|
t.Fatalf("clamp failed: page=%d pageSize=%d", res.Page, res.PageSize)
|
|
}
|
|
if repo.lastLimit != MaxPageSize || repo.lastOffset != 0 {
|
|
t.Fatalf("repo got limit=%d offset=%d", repo.lastLimit, repo.lastOffset)
|
|
}
|
|
if res.Total != 1 {
|
|
t.Fatalf("total = %d, want 1", res.Total)
|
|
}
|
|
|
|
// page 3, pageSize 20 -> offset 40.
|
|
if _, err := svc.ListSummaries(context.Background(), domain.DatasetFilter{}, nil, 3, 20); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if repo.lastOffset != 40 || repo.lastLimit != 20 {
|
|
t.Fatalf("expected limit=20 offset=40, got limit=%d offset=%d", repo.lastLimit, repo.lastOffset)
|
|
}
|
|
}
|
|
|
|
func TestDatasetService_WaitForStatus(t *testing.T) {
|
|
ctx := context.Background()
|
|
id := uuid.New()
|
|
repo := newStubDatasetRepo()
|
|
repo.store[id] = domain.Dataset{ID: id, Status: domain.DatasetStatusReady}
|
|
svc := newDatasetService(repo, &stubStore{}, true)
|
|
|
|
t.Run("returns immediately when status differs from current", func(t *testing.T) {
|
|
info, err := svc.WaitForStatus(ctx, id, domain.DatasetStatusProcessing, time.Minute)
|
|
if err != nil {
|
|
t.Fatalf("unexpected error: %v", err)
|
|
}
|
|
if info.Status != domain.DatasetStatusReady || info.ID != id {
|
|
t.Fatalf("unexpected info: %+v", info)
|
|
}
|
|
})
|
|
|
|
t.Run("returns immediately when no current is given", func(t *testing.T) {
|
|
info, err := svc.WaitForStatus(ctx, id, "", time.Minute)
|
|
if err != nil {
|
|
t.Fatalf("unexpected error: %v", err)
|
|
}
|
|
if info.Status != domain.DatasetStatusReady {
|
|
t.Fatalf("status = %q", info.Status)
|
|
}
|
|
})
|
|
|
|
t.Run("times out returning the unchanged status", func(t *testing.T) {
|
|
start := time.Now()
|
|
info, err := svc.WaitForStatus(ctx, id, domain.DatasetStatusReady, 30*time.Millisecond)
|
|
if err != nil {
|
|
t.Fatalf("unexpected error: %v", err)
|
|
}
|
|
if info.Status != domain.DatasetStatusReady {
|
|
t.Fatalf("status = %q", info.Status)
|
|
}
|
|
if elapsed := time.Since(start); elapsed > time.Second {
|
|
t.Fatalf("timed out too slowly: %v", elapsed)
|
|
}
|
|
})
|
|
|
|
t.Run("not found", func(t *testing.T) {
|
|
_, err := svc.WaitForStatus(ctx, uuid.New(), "", time.Minute)
|
|
if !errors.Is(err, domain.ErrNotFound) {
|
|
t.Fatalf("want ErrNotFound, got %v", err)
|
|
}
|
|
})
|
|
}
|
|
|
|
func TestDatasetService_Delete_RemovesObject(t *testing.T) {
|
|
repo := newStubDatasetRepo()
|
|
id := uuid.New()
|
|
repo.store[id] = domain.Dataset{ID: id, StorageKey: "key/data.geojson"}
|
|
store := &stubStore{}
|
|
svc := newDatasetService(repo, store, true)
|
|
|
|
if err := svc.Delete(context.Background(), id); err != nil {
|
|
t.Fatalf("unexpected error: %v", err)
|
|
}
|
|
if len(repo.deleted) != 1 || repo.deleted[0] != id {
|
|
t.Fatalf("row not deleted: %v", repo.deleted)
|
|
}
|
|
if len(store.removed) != 1 || store.removed[0] != "key/data.geojson" {
|
|
t.Fatalf("object not removed: %v", store.removed)
|
|
}
|
|
}
|