Function TestLargeStringIndex

dgraph/cmd/alpha/upsert_test.go:2950–3000

(t *testing.T)

}

func TestLargeStringIndex(t *testing.T) {
	require.NoError(t, dropAll())

	// Exact Index
	require.NoError(t, alterSchemaWithRetry(`name_exact: string @index(exact) .`))
	bigval := strings.Repeat("biggest value", 7000)
	mu := fmt.Sprintf(`{ set { <0x01> <name_exact> "%v" . } }`, bigval)
	require.Contains(t, runMutation(mu).Error(), "value in the mutation is too large for the index")

	// Term Index
	require.NoError(t, alterSchemaWithRetry(`name_term: string @index(term) .`))
	bigval = strings.Repeat("biggestvalue", 7000)
	mu = fmt.Sprintf(`{ set { <0x01> <name_term> "%v" . } }`, bigval)
	require.Contains(t, runMutation(mu).Error(), "value in the mutation is too large for the index")
	// while the value is large, term index is fine
	bigval = strings.Repeat("biggestvalue", 3000)
	mu = fmt.Sprintf(`{ set { <0x01> <name_term> "%v %v" . } }`, bigval, bigval)
	require.NoError(t, runMutation(mu))

	// FullText Index
	require.NoError(t, alterSchemaWithRetry(`name_fulltext: string @index(fulltext) .`))
	bigval = strings.Repeat("biggestvalue", 7000)
	mu = fmt.Sprintf(`{ set { <0x01> <name_fulltext> "%v" . } }`, bigval)
	require.Contains(t, runMutation(mu).Error(), "value in the mutation is too large for the index")
	// while the value is large, fulltext index is fine
	bigval = strings.Repeat("biggestvalue", 3000)
	mu = fmt.Sprintf(`{ set { <0x01> <name_fulltext> "%v %v" . } }`, bigval, bigval)
	require.NoError(t, runMutation(mu))

	// Trigram Index
	require.NoError(t, alterSchemaWithRetry(`name_trigram: string @index(trigram) .`))
	bigval = strings.Repeat("biggestvalue", 7000)
	mu = fmt.Sprintf(`{ set { <0x01> <name_trigram> "%v" . } }`, bigval)
	require.NoError(t, runMutation(mu))

	// Large Predicate
	bigpred := strings.Repeat("large-predicate", 2000)
	require.NoError(t, alterSchemaWithRetry(fmt.Sprintf(`%v: string @index(exact) .`, bigpred)))
	bigval = strings.Repeat("biggestvalue", 3000)
	mu = fmt.Sprintf(`{ set { <0x01> <%v> "%v" . } }`, bigpred, bigval)
	require.Contains(t, runMutation(mu).Error(), "value in the mutation is too large for the index")

	// name_term has index term. Now, if we create an exact index, that will fail.
	// This is because one of the value has small terms but the whole value is bigger
	// than the limit. In such a case, indexing will fail and the schema should not change.
	require.NoError(t, alterSchemaWithRetry(`name_term: string @index(exact) .`))
	dqlSchema, err := runGraphqlQuery(`schema{}`)
	require.NoError(t, err)
	require.Contains(t, dqlSchema,
		`{"predicate":"name_term","type":"string","index":true,"tokenizer":["term"]}`)
}

// TestDQLInjectionViaCondField is a security regression test for LEAD-001.
// It verifies that a crafted cond field containing an injected DQL query block