Spaces:
Running
Running
File size: 2,537 Bytes
b110593 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 |
// _ _
// __ _____ __ ___ ___ __ _| |_ ___
// \ \ /\ / / _ \/ _` \ \ / / |/ _` | __/ _ \
// \ V V / __/ (_| |\ V /| | (_| | || __/
// \_/\_/ \___|\__,_| \_/ |_|\__,_|\__\___|
//
// Copyright © 2016 - 2024 Weaviate B.V. All rights reserved.
//
// CONTACT: [email protected]
//
package batch_request_endpoints
import (
"testing"
"time"
clschema "github.com/weaviate/weaviate/client/schema"
"github.com/weaviate/weaviate/entities/models"
"github.com/weaviate/weaviate/entities/schema"
"github.com/weaviate/weaviate/test/helper"
)
func Test_Batch(t *testing.T) {
// there is no gql provider if there is no schema, so we need some sort of a schema
t.Run("setup", func(t *testing.T) {
createObjectClass(t, &models.Class{
Class: "BulkTest",
Properties: []*models.Property{
{
Name: "name",
DataType: schema.DataTypeText.PropString(),
Tokenization: models.PropertyTokenizationWhitespace,
},
},
})
createObjectClass(t, &models.Class{
Class: "BulkTestSource",
Properties: []*models.Property{
{
Name: "name",
DataType: schema.DataTypeText.PropString(),
Tokenization: models.PropertyTokenizationWhitespace,
},
{
Name: "ref",
DataType: []string{"BulkTest"},
},
},
})
createObjectClass(t, &models.Class{
Class: "BulkTestTarget",
Properties: []*models.Property{
{
Name: "intProp",
DataType: []string{"int"},
},
{
Name: "fromSource",
DataType: []string{"BulkTestSource"},
},
},
})
})
time.Sleep(2000 * time.Millisecond)
t.Run("gql results order", batchJourney)
t.Run("gql results order", gqlResultsOrder)
t.Run("gql malformed request", gqlMalformedRequest)
t.Run("batch delete", batchDeleteJourney)
deleteObjectClass(t, "BulkTest")
deleteObjectClass(t, "BulkTestSource")
deleteObjectClass(t, "BulkTestTarget")
}
func createObjectClass(t *testing.T, class *models.Class) {
params := clschema.NewSchemaObjectsCreateParams().WithObjectClass(class)
resp, err := helper.Client(t).Schema.SchemaObjectsCreate(params, nil)
helper.AssertRequestOk(t, resp, err, nil)
}
func deleteObjectClass(t *testing.T, class string) {
delParams := clschema.NewSchemaObjectsDeleteParams().WithClassName(class)
delRes, err := helper.Client(t).Schema.SchemaObjectsDelete(delParams, nil)
helper.AssertRequestOk(t, delRes, err, nil)
}
|