Merged in feature/docclean (pull request #55)

Doc Clean Structure

* outline

* isdocclean

* placeholder for clean log

* versionplustesting

* versionplustesting

* testspassed
This commit is contained in:
Michael McGuinness
2025-02-07 12:12:51 +00:00
parent e146725825
commit 90353d8161
56 changed files with 1106 additions and 316 deletions
@@ -5,7 +5,7 @@ import (
"fmt"
"os"
"path"
"queryorchestration/internal/server/runner"
"queryorchestration/internal/serviceconfig"
"queryorchestration/internal/serviceconfig/objectstore"
documentcleanc "queryorchestration/internal/serviceconfig/queue/documentclean"
"queryorchestration/internal/test"
@@ -19,19 +19,19 @@ import (
"github.com/stretchr/testify/assert"
)
type DocInitConfig struct {
runner.BaseConfig
type ProcessConfig struct {
serviceconfig.BaseConfig
documentcleanc.DocCleanConfig
objectstore.ObjectStoreConfig
}
func TestDocInitRunner(t *testing.T) {
func TestProcess(t *testing.T) {
t.SkipNow()
ctx := context.Background()
cfg := &DocInitConfig{}
cfg := &ProcessConfig{}
test.SetCfgProvider(t, cfg)
cfg.SetBasePath(path.Join(os.Getenv("PWD"), "../.."))
cfg.SetBasePath(path.Join(os.Getenv("PWD"), ".."))
network, ncleanup := test.CreateNetwork(t, ctx)
defer ncleanup()
@@ -46,7 +46,6 @@ func TestDocInitRunner(t *testing.T) {
assert.NoError(t, err)
test.CreateStoreClient(t, ctx, cfg, acfg.ExternalEndpoint)
doccleanurl := test.CreateQueue(t, ctx, cfg, "docclean")
bucketName := "docinitbucket"
test.CreateBucket(t, ctx, cfg, bucketName)
arn := fmt.Sprintf("arn:aws:sqs:%s:000000000000:%s", cfg.AWSRegion, test.DocInitRunner)
@@ -65,6 +64,9 @@ func TestDocInitRunner(t *testing.T) {
})
assert.NoError(t, err)
doccleanurl := test.CreateQueue(t, ctx, cfg, test.DocCleanRunner)
doctexturl := test.CreateQueue(t, ctx, cfg, "doctextrunner")
net, cleanup := test.CreateRunnersAndServicesNetwork(t, ctx, &test.EcosystemNetworkConfig{
Cfg: cfg,
Network: network,
@@ -75,6 +77,12 @@ func TestDocInitRunner(t *testing.T) {
"DOCUMENT_CLEAN_URL": doccleanurl,
},
},
{
Name: test.DocCleanRunner,
Env: map[string]string{
"DOCUMENT_TEXT_URL": doctexturl,
},
},
},
Services: []*test.ServiceNetworkConfig{
{
@@ -116,5 +124,5 @@ func TestDocInitRunner(t *testing.T) {
assert.NoError(t, err)
resRegex := regexp.MustCompile(`{"id": "[0-9a-f]{8}-[0-9a-f]{4}-[0-9a-f]{4}-[0-9a-f]{4}-[0-9a-f]{12}"`)
test.AssertMessageBody(t, ctx, cfg, doccleanurl, resRegex)
test.AssertMessageBody(t, ctx, cfg, doctexturl, resRegex)
}
-41
View File
@@ -1,41 +0,0 @@
package integration_test
import (
"context"
"queryorchestration/internal/query"
"queryorchestration/internal/serviceconfig"
"queryorchestration/internal/serviceconfig/queue"
"queryorchestration/internal/test"
"testing"
"github.com/google/uuid"
"github.com/stretchr/testify/assert"
)
func TestQueryRunner(t *testing.T) {
ctx := context.Background()
cfg := &serviceconfig.BaseConfig{}
test.SetCfgProvider(t, cfg)
c, cleanup := test.CreateRunnerNetwork(t, ctx, &test.RunnerNetworkConfig{
Cfg: cfg,
Name: test.QueryRunner,
})
defer cleanup()
document := query.Document{
ID: uuid.New(),
JobID: uuid.New(),
CleanVersion: int32(1),
TextVersion: int32(1),
}
err := cfg.SendToQueue(ctx, &queue.SendParams{
QueueURL: c.URI,
Body: document,
})
assert.NoError(t, err)
// TODO - check document output
}