Function initTestExport

worker/export_test.go:124–159

(t *testing.T, schemaStr string)

}

func initTestExport(t *testing.T, schemaStr string) {
	require.NoError(t, schema.ParseBytes([]byte(schemaStr), 1))

	val, err := proto.Marshal(&pb.SchemaUpdate{ValueType: pb.Posting_UID})
	require.NoError(t, err)

	txn := pstore.NewTransactionAt(math.MaxUint64, true)
	require.NoError(t, txn.Set(testutil.RootNsSchemaKey("friend"), val))
	// Schema is always written at timestamp 1
	require.NoError(t, txn.CommitAt(1, nil))

	require.NoError(t, err)
	val, err = proto.Marshal(&pb.SchemaUpdate{ValueType: pb.Posting_UID})
	require.NoError(t, err)

	txn = pstore.NewTransactionAt(math.MaxUint64, true)
	require.NoError(t, txn.Set(testutil.RootNsSchemaKey("http://www.w3.org/2000/01/rdf-schema#range"), val))
	require.NoError(t, txn.Set(testutil.RootNsSchemaKey("friend_not_served"), val))
	require.NoError(t, txn.Set(testutil.RootNsSchemaKey("age"), val))
	require.NoError(t, txn.CommitAt(1, nil))

	val, err = proto.Marshal(personType)
	require.NoError(t, err)

	txn = pstore.NewTransactionAt(math.MaxUint64, true)
	require.NoError(t, txn.Set(testutil.RootNsTypeKey("Person"), val))
	require.NoError(t, txn.CommitAt(1, nil))

	populateGraphExport(t)

	// Drop age predicate after populating DB.
	// age should not exist in the exported schema.
	txn = pstore.NewTransactionAt(math.MaxUint64, true)
	require.NoError(t, txn.Delete(testutil.RootNsSchemaKey("age")))
	require.NoError(t, txn.CommitAt(1, nil))
}

func getExportFileList(t *testing.T, bdir string) (dataFiles, schemaFiles, gqlSchema []string) {
	searchDir := bdir