Fix export worker missing @lang and @upsert (#2395)

worker/export.go

@@ -163,6 +163,12 @@ func toSchema(buf *bytes.Buffer, s *skv) {
 	if s.schema.Count {
 		buf.WriteString(" @count")
 	}
+	if s.schema.Lang {
+		buf.WriteString(" @lang")
+	}
+	if s.schema.Upsert {
+		buf.WriteString(" @upsert")
+	}
 	buf.WriteString(" . \n")
 }
 

worker/export_test.go

@@ -9,6 +9,7 @@ package worker
 
 import (
 	"bufio"
+	"bytes"
 	"compress/gzip"
 	"io/ioutil"
 	"math"
@@ -220,6 +221,49 @@ func TestExport(t *testing.T) {
 	require.Equal(t, 1, count)
 }
 
+func TestToSchema(t *testing.T) {
+	testCases := []struct {
+		skv      *skv
+		expected string
+	}{
+		{
+			skv: &skv{
+				attr: "Alice",
+				schema: &intern.SchemaUpdate{
+					Predicate: "mother",
+					ValueType: intern.Posting_STRING,
+					Directive: intern.SchemaUpdate_REVERSE,
+					List:      false,
+					Count:     true,
+					Upsert:    true,
+					Lang:      true,
+				},
+			},
+			expected: "Alice:string @reverse @count @lang @upsert . \n",
+		},
+		{
+			skv: &skv{
+				attr: "Alice:best",
+				schema: &intern.SchemaUpdate{
+					Predicate: "mother",
+					ValueType: intern.Posting_STRING,
+					Directive: intern.SchemaUpdate_REVERSE,
+					List:      false,
+					Count:     false,
+					Upsert:    false,
+					Lang:      true,
+				},
+			},
+			expected: "<Alice:best>:string @reverse @lang . \n",
+		},
+	}
+	for _, testCase := range testCases {
+		buf := new(bytes.Buffer)
+		toSchema(buf, testCase.skv)
+		require.Equal(t, testCase.expected, buf.String())
+	}
+}
+
 // func generateBenchValues() []kv {
 // 	byteInt := make([]byte, 4)
 // 	binary.LittleEndian.PutUint32(byteInt, 123)