Function TestBulkLoadMixedPredicates

systest/vector/load_test.go:355–515

TestBulkLoadMixedPredicates tests bulk loading vector data alongside other predicate types (string with index, int with index, uid edges) to ensure vector indexing doesn't break existing functionality.

(t *testing.T)

// predicate types (string with index, int with index, uid edges) to ensure
// vector indexing doesn't break existing functionality.
func TestBulkLoadMixedPredicates(t *testing.T) {
	// Schema with vectors AND other indexed predicates
	mixedSchema := `
		project_description_v: float32vector @index(hnsw(exponent: "5", metric: "euclidean")) .
		name: string @index(term, fulltext) .
		age: int @index(int) .
		score: float .
		friend: [uid] @reverse .
		dgraph.type: [string] @index(exact) .
	`

	numVectors := 500
	vectorDim := 10

	// Step 1: Create source cluster and load mixed data
	sourceConf := dgraphtest.NewClusterConfig().
		WithNumAlphas(1).
		WithNumZeros(1).
		WithReplicas(1).
		WithACL(time.Hour)
	sourceCluster, err := dgraphtest.NewLocalCluster(sourceConf)
	require.NoError(t, err)
	defer func() { sourceCluster.Cleanup(t.Failed()) }()
	require.NoError(t, sourceCluster.Start())

	gc, cleanup, err := sourceCluster.Client()
	require.NoError(t, err)
	defer cleanup()
	require.NoError(t, gc.LoginIntoNamespace(context.Background(),
		dgraphapi.DefaultUser, dgraphapi.DefaultPassword, x.RootNamespace))

	hc, err := sourceCluster.HTTPClient()
	require.NoError(t, err)
	require.NoError(t, hc.LoginIntoNamespace(dgraphapi.DefaultUser,
		dgraphapi.DefaultPassword, x.RootNamespace))

	require.NoError(t, gc.SetupSchema(mixedSchema))

	// Generate mixed RDF data: vectors + strings + ints + edges
	var rdfBuilder strings.Builder
	vectors := make([][]float32, numVectors)

	for i := 0; i < numVectors; i++ {
		uid := i + 10
		// Generate random vector
		vec := dgraphapi.GenerateRandomVector(vectorDim)
		vectors[i] = vec
		vecStr := fmt.Sprintf(`"[%s]"`, strings.Trim(strings.Join(strings.Fields(fmt.Sprint(vec)), ", "), "[]"))

		// Add vector predicate
		rdfBuilder.WriteString(fmt.Sprintf("<0x%x> <project_description_v> %s .\n", uid, vecStr))
		// Add string predicate
		rdfBuilder.WriteString(fmt.Sprintf("<0x%x> <name> \"Person %d\" .\n", uid, i))
		// Add int predicate
		rdfBuilder.WriteString(fmt.Sprintf("<0x%x> <age> \"%d\"^^<xs:int> .\n", uid, 20+i%50))
		// Add float predicate
		rdfBuilder.WriteString(fmt.Sprintf("<0x%x> <score> \"%f\"^^<xs:float> .\n", uid, float64(i)*1.5))
		// Add dgraph.type