Function populateGraphExport

worker/export_test.go:64–122

(t *testing.T)

}

func populateGraphExport(t *testing.T) {
	rdfEdges := []string{
		`<1> <friend> <5> .`,
		`<2> <friend> <5> .`,
		`<3> <friend> <5> .`,
		`<4> <friend> <5> (since=2005-05-02T15:04:05,close=true,` +
			`age=33,game="football",poem="roses are red\nviolets are blue") .`,
		`<1> <name> "pho\ton\u0000" .`,
		`<2> <name> "pho\ton"@en .`,
		`<3> <name> "First Line\nSecondLine" .`,
		"<1> <friend_not_served> <5> .",
		`<5> <name> "" .`,
		`<6> <name> "Ding!\u0007Ding!\u0007Ding!\u0007" .`,
		`<7> <name> "node_to_delete" .`,
		fmt.Sprintf("<8> <dgraph.graphql.schema> \"%s\" .", gqlSchema),
		`<8> <dgraph.graphql.xid> "dgraph.graphql.schema" .`,
		`<8> <dgraph.type> "dgraph.graphql" .`,
		`<9> <name> "ns2" <0x2> .`,
		`<10> <name> "ns2_node_to_delete" <0x2> .`,
	}
	// This triplet will be deleted to ensure deleted nodes do not affect the output of the export.
	edgesToDelete := []string{
		`<7> <name> "node_to_delete" .`,
		`<10> <name> "ns2_node_to_delete" <0x2> .`,
	}

	idMap := map[string]uint64{
		"1": 1,
		"2": 2,
		"3": 3,
		"4": 4,
		"5": 5,
		"6": 6,
		"7": 7,
	}

	l := &lex.Lexer{}
	processEdge := func(edge string, set bool) {
		nq, err := chunker.ParseRDF(edge, l)
		require.NoError(t, err)
		rnq := dql.NQuad{NQuad: &nq}
		require.NoError(t, facets.SortAndValidate(rnq.Facets))
		e, err := rnq.ToEdgeUsing(idMap)
		e.Attr = x.NamespaceAttr(nq.Namespace, e.Attr)
		require.NoError(t, err)
		if set {
			addEdge(t, e, getOrCreate(x.DataKey(e.Attr, e.Entity)))
		} else {
			delEdge(t, e, getOrCreate(x.DataKey(e.Attr, e.Entity)))
		}
	}

	for _, edge := range rdfEdges {
		processEdge(edge, true)
	}
	for _, edge := range edgesToDelete {
		processEdge(edge, false)
	}