Merge pull request #110 from heetch/sixstone-001/add-parse-compat

Add avro.ParseCompatMode

Author: sixstone-qq

compat.go

@@ -37,3 +37,25 @@ func (m CompatMode) String() string {
 	}
 	return s
 }
+
+// ParseCompatMode returns the CompatMode from a string.
+// It returns -1 if no matches are found.
+func ParseCompatMode(s string) CompatMode {
+	switch s {
+	case "BACKWARD":
+		return Backward
+	case "FORWARD":
+		return Forward
+	case "FULL":
+		return Full
+	case "BACKWARD_TRANSITIVE":
+		return BackwardTransitive
+	case "FORWARD_TRANSITIVE":
+		return ForwardTransitive
+	case "FULL_TRANSITIVE":
+		return FullTransitive
+	case "NONE":
+		return 0
+	}
+	return -1
+}

compat_test.go

@@ -30,3 +30,17 @@ func TestCompatString(t *testing.T) {
 		})
 	}
 }
+
+func TestCompatParse(t *testing.T) {
+	c := qt.New(t)
+	for _, test := range compatStringTests {
+		c.Run(test.s, func(c *qt.C) {
+			if test.s == "UNKNOWN" {
+				// We can't return same data we don't know
+				c.Assert(avro.ParseCompatMode(test.s), qt.Equals, avro.CompatMode(-1))
+			} else {
+				c.Assert(avro.ParseCompatMode(test.s), qt.Equals, test.m)
+			}
+		})
+	}
+}