Function TestMultipleMutationMerge

dgraph/cmd/alpha/upsert_test.go:2500–2609

(t *testing.T)

}

func TestMultipleMutationMerge(t *testing.T) {
	require.NoError(t, dropAll())
	require.NoError(t, alterSchema(`
name: string @index(term) .
email: [string] @index(exact) @upsert .`))

	m1 := `
{
  set {
    _:user1 <name> "user1" .
    _:user1 <email> "user_email1@company1.io" .
    _:user2 <name> "user2" .
    _:user2 <email> "user_email2@company1.io" .
  }
}`
	mr, err := mutationWithTs(mutationInp{body: m1, typ: "application/rdf", commitNow: true})
	require.NoError(t, err)
	require.True(t, len(mr.keys) == 0)
	require.Equal(t, []string{"email", "name"}, splitPreds(mr.preds))

	q1 := `
{
  q(func: has(name)) {
    uid
  }
}`
	res, _, err := queryWithTs(queryInp{body: q1, typ: "application/dql"})
	require.NoError(t, err)
	var result struct {
		Data struct {
			Q []struct {
				UID string `json:"uid"`
			} `json:"q"`
		} `json:"data"`
	}
	require.NoError(t, json.Unmarshal([]byte(res), &result))
	require.Equal(t, 2, len(result.Data.Q))

	m2 := `
upsert {
  query {
    # filter is needed to ensure that we do not get same UIDs in u1 and u2
    q1(func: eq(email, "user_email1@company1.io")) @filter(not(eq(email, "user_email2@company1.io"))) {
      u1 as uid
    }

    q2(func: eq(email, "user_email2@company1.io")) @filter(not(eq(email, "user_email1@company1.io"))) {
      u2 as uid
    }

    q3(func: eq(email, "user_email1@company1.io")) @filter(eq(email, "user_email2@company1.io")) {
      u3 as uid
    }
  }

  # case when both emails do not exist
  mutation @if(eq(len(u1), 0) AND eq(len(u2), 0) AND eq(len(u3), 0)) {
    set {