Function TestAddMutation

posting/list_test.go:166–215

(t *testing.T)

}

func TestAddMutation(t *testing.T) {
	key := x.DataKey(x.AttrInRootNamespace("name"), 2)

	txn := NewTxn(1)
	l, err := txn.Get(key)
	require.NoError(t, err)

	edge := &pb.DirectedEdge{
		ValueId: 9,
		Facets:  []*api.Facet{{Key: "testing"}},
	}
	addMutationHelper(t, l, edge, Set, txn)

	require.Equal(t, listToArray(t, 0, l, 1), []uint64{9})

	p := getFirst(t, l, 1)
	require.NotNil(t, p, "Unable to retrieve posting")
	require.EqualValues(t, "testing", p.Facets[0].Key)

	// Add another edge now.
	edge.ValueId = 81
	addMutationHelper(t, l, edge, Set, txn)
	require.Equal(t, listToArray(t, 0, l, 1), []uint64{9, 81})

	// Add another edge, in between the two above.
	edge.ValueId = 49
	addMutationHelper(t, l, edge, Set, txn)
	require.Equal(t, listToArray(t, 0, l, 1), []uint64{9, 49, 81})

	checkUids(t, l, []uint64{9, 49, 81}, 1)

	// Delete an edge, add an edge, replace an edge
	edge.ValueId = 49
	addMutationHelper(t, l, edge, Del, txn)

	edge.ValueId = 69
	addMutationHelper(t, l, edge, Set, txn)

	edge.ValueId = 9
	edge.Facets = []*api.Facet{{Key: "anti-testing"}}
	addMutationHelper(t, l, edge, Set, txn)
	require.NoError(t, l.commitMutation(1, 2))

	uids := []uint64{9, 69, 81}
	checkUids(t, l, uids, 3)

	p = getFirst(t, l, 3)
	require.NotNil(t, p, "Unable to retrieve posting")
	require.EqualValues(t, "anti-testing", p.Facets[0].Key)
}

func getFirst(t *testing.T, l *List, readTs uint64) (res pb.Posting) {
	require.NoError(t, l.Iterate(readTs, 0, func(p *pb.Posting) error {