add enum support for modelgen

Fix #133

cmd/modelgen/table_test.go

@@ -14,7 +14,7 @@ func TestNewTableGenerator(t *testing.T) {
 		"name": "AtomicDB",
 		"version": "0.0.0",
 		"tables": {
-			"atomicTable": {
+			"AtomicTable": {
 				"columns": {
 					"str": {
 						"type": "string"
@@ -24,7 +24,13 @@ func TestNewTableGenerator(t *testing.T) {
 					},
 					"float": {
 						"type": "real"
-					}
+					},
+					"protocol": {
+						"type": {"key": {"type": "string",
+								 "enum": ["set", ["tcp", "udp", "sctp"]]},
+								 "min": 0, "max": 1}},
+					"event_type": {"type": {"key": {"type": "string",
+													"enum": ["set", ["empty_lb_backends"]]}}}
 				}
 			}
 		}
@@ -35,12 +41,28 @@ func TestNewTableGenerator(t *testing.T) {
 
 package test
 
-// test defines an object in test table
-type test struct {
-	UUID  string  ` + "`" + `ovs:"_uuid"` + "`" + `
-	Float float64 ` + "`" + `ovs:"float"` + "`" + `
-	Int   int     ` + "`" + `ovs:"int"` + "`" + `
-	Str   string  ` + "`" + `ovs:"str"` + "`" + `
+type (
+	AtomicTableEventType string
+
+	AtomicTableProtocol string
+)
+
+const (
+	AtomicTableEventTypeEmptyLbBackends AtomicTableEventType = "empty_lb_backends"
+
+	AtomicTableProtocolTCP  AtomicTableProtocol = "tcp"
+	AtomicTableProtocolUDP  AtomicTableProtocol = "udp"
+	AtomicTableProtocolSCTP AtomicTableProtocol = "sctp"
+)
+
+// AtomicTable defines an object in AtomicTable table
+type AtomicTable struct {
+	UUID      string                ` + "`" + `ovs:"_uuid"` + "`" + `
+	EventType AtomicTableEventType  ` + "`" + `ovs:"event_type"` + "`" + `
+	Float     float64               ` + "`" + `ovs:"float"` + "`" + `
+	Int       int                   ` + "`" + `ovs:"int"` + "`" + `
+	Protocol  []AtomicTableProtocol ` + "`" + `ovs:"protocol"` + "`" + `
+	Str       string                ` + "`" + `ovs:"str"` + "`" + `
 }
 `
 
@@ -49,10 +71,11 @@ type test struct {
 	if err != nil {
 		t.Fatal(err)
 	}
-	table := schema.Tables["atomicTable"]
+	fakeTable := "AtomicTable"
+	table := schema.Tables[fakeTable]
 	gen := NewTableGenerator(
 		"test",
-		"test",
+		fakeTable,
 		&table,
 	)