Fix tests

Signed-off-by: Rohit Nayak <[email protected]>

go/dbinfo/dbinfo.go

@@ -48,9 +48,9 @@ func run(out io.Writer, cfg Config) error {
 type TableColumn struct {
 	Name       string `json:"name"`
 	Type       string `json:"type"`
-	KeyType    string `json:"keyType"`
+	KeyType    string `json:"keyType,omitempty"`
 	IsNullable bool   `json:"isNullable"`
-	Extra      string `json:"extra"`
+	Extra      string `json:"extra,omitempty"`
 }
 
 type TableInfo struct {
@@ -110,7 +110,8 @@ func Get(cfg Config) (*Info, error) {
 	}
 
 	dbInfo := &Info{
-		Tables: tableInfo,
+		FileType: "dbinfo",
+		Tables:   tableInfo,
 	}
 	return dbInfo, nil
 }

go/dbinfo/dbinfo_test.go

@@ -23,11 +23,11 @@ import (
 )
 
 func TestDBInfoRowCount(t *testing.T) {
-	si, err := Load("../testdata/sakila-schema-info.json")
+	si, err := Load("../testdata/sakila-dbinfo.json")
 	require.NoError(t, err)
 	require.NotNil(t, si)
 	require.NotEmpty(t, si.Tables)
-	require.Len(t, si.Tables, 16)
+	require.Len(t, si.Tables, 23)
 	var tables []string
 	for _, table := range si.Tables {
 		tables = append(tables, table.Name)
@@ -41,8 +41,6 @@ func TestDBInfoRowCount(t *testing.T) {
 			require.Equal(t, 6, table.Rows)
 		case "film":
 			require.Equal(t, 1000, table.Rows)
-		default:
-			require.Positive(t, table.Rows, "table %s has no rows", table.Name)
 		}
 	}
 }

go/summarize/utils_test.go

@@ -36,7 +36,7 @@ func TestGetFileType(t *testing.T) {
 			expectedType: unknownFile,
 		},
 		{
-			filename:     "../testdata/sakila-schema-info.json",
+			filename:     "../testdata/sakila-dbinfo.json",
 			expectedType: dbInfoFile,
 		},
 		{

go/testdata/sakila-dbinfo.json

@@ -1,5 +1,5 @@
 {
-  "fileType": "",
+  "fileType": "dbinfo",
   "tables": [
     {
       "name": "inventory",