Add test

k1LoW · Nov 18, 2018 · f534ddb · f534ddb
1 parent dcb6c13
commit f534ddb
Show file tree

Hide file tree

Showing 6 changed files with 30 additions and 7 deletions.
diff --git a/output/dot/dot_test.go b/output/dot/dot_test.go
@@ -12,8 +12,12 @@ import (
 
 func TestOutputSchema(t *testing.T) {
 	s := newTestSchema()
+	err := s.LoadAdditionalData(filepath.Join(testdataDir(), "md_test_additional_data.yml"))
+	if err != nil {
+		t.Error(err)
+	}
 	buf := &bytes.Buffer{}
-	err := OutputSchema(buf, s)
+	err = OutputSchema(buf, s)
 	if err != nil {
 		t.Error(err)
 	}
@@ -26,6 +30,10 @@ func TestOutputSchema(t *testing.T) {
 
 func TestOutputTable(t *testing.T) {
 	s := newTestSchema()
+	err := s.LoadAdditionalData(filepath.Join(testdataDir(), "md_test_additional_data.yml"))
+	if err != nil {
+		t.Error(err)
+	}
 	ta := s.Tables[0]
 
 	buf := &bytes.Buffer{}

diff --git a/output/md/md_test.go b/output/md/md_test.go
@@ -24,12 +24,16 @@ var tests = []struct {
 func TestOutput(t *testing.T) {
 	for _, tt := range tests {
 		s := newTestSchema()
+		err := s.LoadAdditionalData(filepath.Join(testdataDir(), "md_test_additional_data.yml"))
+		if err != nil {
+			t.Error(err)
+		}
 		tempDir, _ := ioutil.TempDir("", "tbls")
 		force := true
 		adjust := tt.adjust
 		erFormat := "png"
 		defer os.RemoveAll(tempDir)
-		err := Output(s, tempDir, force, adjust, erFormat)
+		err = Output(s, tempDir, force, adjust, erFormat)
 		if err != nil {
 			t.Error(err)
 		}
@@ -50,12 +54,16 @@ func TestOutput(t *testing.T) {
 func TestDiff(t *testing.T) {
 	for _, tt := range tests {
 		s := newTestSchema()
+		err := s.LoadAdditionalData(filepath.Join(testdataDir(), "md_test_additional_data.yml"))
+		if err != nil {
+			t.Error(err)
+		}
 		tempDir, _ := ioutil.TempDir("", "tbls")
 		force := true
 		adjust := tt.adjust
 		erFormat := "png"
 		defer os.RemoveAll(tempDir)
-		err := Output(s, tempDir, force, adjust, erFormat)
+		err = Output(s, tempDir, force, adjust, erFormat)
 		if err != nil {
 			t.Error(err)
 		}

diff --git a/testdata/md_test_README.md.adjust.golden b/testdata/md_test_README.md.adjust.golden
@@ -4,7 +4,7 @@
 
 | Name      | Columns | Comment | Type |
 | --------- | ------- | ------- | ---- |
-| [a](a.md) | 2       | table a |      |
+| [a](a.md) | 2       | TABLE A |      |
 | [b](b.md) | 2       | table b |      |
 
 ---

diff --git a/testdata/md_test_README.md.golden b/testdata/md_test_README.md.golden
@@ -4,7 +4,7 @@
 
 | Name | Columns | Comment | Type |
 | ---- | ------- | ------- | ---- |
-| [a](a.md) | 2 | table a |  |
+| [a](a.md) | 2 | TABLE A |  |
 | [b](b.md) | 2 | table b |  |
 
 ---

diff --git a/testdata/md_test_a.md.golden b/testdata/md_test_a.md.golden
@@ -2,13 +2,13 @@
 
 ## Description
 
-table a
+TABLE A
 
 ## Columns
 
 | Name | Type | Default | Nullable | Children | Parents | Comment |
 | ---- | ---- | ------- | -------- | -------- | ------- | ------- |
-| a |  |  | false |  | [b](b.md) | column a |
+| a |  |  | false |  | [b](b.md) | COLUMN A |
 | a2 |  |  | false |  |  | column a2 |
 
 ---

diff --git a/testdata/md_test_additional_data.yml b/testdata/md_test_additional_data.yml
@@ -0,0 +1,7 @@
+---
+comments:
+  -
+    table: a
+    tableComment: TABLE A
+    columnComments:
+      a: COLUMN A