パーティションテーブル対応 (#46)

* パーティションテーブル対応

* テスト修正

* コメント追加

Author: tamiyamoto

compare.xlsx

@@ -2,14 +2,15 @@ package exceltesting
 
 import (
 	"database/sql"
-	"github.com/future-architect/go-exceltesting/testonly"
 	"net"
 	"os"
 	"path/filepath"
 	"reflect"
 	"testing"
 	"time"
 
+	"github.com/future-architect/go-exceltesting/testonly"
+
 	"github.com/google/go-cmp/cmp"
 	"github.com/google/go-cmp/cmp/cmpopts"
 	"github.com/jackc/pgtype"
@@ -176,10 +177,11 @@ func Test_exceltesing_Compare(t *testing.T) {
 	mockT := new(testing.T)
 
 	tests := []struct {
-		name     string
-		input    func(t *testing.T)
-		wantFile string
-		equal    bool
+		name      string
+		input     func(t *testing.T)
+		wantFile  string
+		wantSheet string
+		equal     bool
 	}{
 		{
 			name: "equal",
@@ -195,8 +197,9 @@ func Test_exceltesing_Compare(t *testing.T) {
 					t.Fatal(err)
 				}
 			},
-			wantFile: filepath.Join("testdata", "compare.xlsx"),
-			equal:    true,
+			wantFile:  filepath.Join("testdata", "compare.xlsx"),
+			wantSheet: "会社",
+			equal:     true,
 		},
 		{
 			name: "equal on exceltesing version 2.0 sheet",
@@ -212,8 +215,9 @@ func Test_exceltesing_Compare(t *testing.T) {
 					t.Fatal(err)
 				}
 			},
-			wantFile: filepath.Join("testdata", "compare_v2.xlsx"),
-			equal:    true,
+			wantFile:  filepath.Join("testdata", "compare_v2.xlsx"),
+			wantSheet: "会社",
+			equal:     true,
 		},
 		{
 			name: "diff",
@@ -229,8 +233,9 @@ func Test_exceltesing_Compare(t *testing.T) {
 					t.Fatal(err)
 				}
 			},
-			wantFile: filepath.Join("testdata", "compare.xlsx"),
-			equal:    false,
+			wantFile:  filepath.Join("testdata", "compare.xlsx"),
+			wantSheet: "会社",
+			equal:     false,
 		},
 		{
 			name: "fewer records of results",
@@ -246,8 +251,9 @@ func Test_exceltesing_Compare(t *testing.T) {
 					t.Fatal(err)
 				}
 			},
-			wantFile: filepath.Join("testdata", "compare.xlsx"),
-			equal:    false,
+			wantFile:  filepath.Join("testdata", "compare.xlsx"),
+			wantSheet: "会社",
+			equal:     false,
 		},
 		{
 			name: "many records of results",
@@ -263,8 +269,27 @@ func Test_exceltesing_Compare(t *testing.T) {
 					t.Fatal(err)
 				}
 			},
-			wantFile: filepath.Join("testdata", "compare.xlsx"),
-			equal:    false,
+			wantFile:  filepath.Join("testdata", "compare.xlsx"),
+			wantSheet: "会社",
+			equal:     false,
+		},
+		{
+			name: "partition table",
+			input: func(t *testing.T) {
+				t.Helper()
+				tdb := testonly.OpenTestDB(t)
+				defer tdb.Close()
+				if _, err := tdb.Exec(`TRUNCATE temperature;`); err != nil {
+					t.Fatal(err)
+				}
+				if _, err := tdb.Exec(`INSERT INTO temperature (ymd,value)
+						VALUES ('20210228',-2.0),('20210831',38.5);`); err != nil {
+					t.Fatal(err)
+				}
+			},
+			wantFile:  filepath.Join("testdata", "compare.xlsx"),
+			wantSheet: "気温",
+			equal:     true,
 		},
 	}
 	for _, tt := range tests {
@@ -274,7 +299,7 @@ func Test_exceltesing_Compare(t *testing.T) {
 			e := New(conn)
 			got := e.Compare(mockT, CompareRequest{
 				TargetBookPath: filepath.Join("testdata", "compare.xlsx"),
-				SheetPrefix:    "",
+				SheetPrefix:    tt.wantSheet,
 				IgnoreSheet:    nil,
 				IgnoreColumns:  []string{"created_at", "updated_at"},
 			})

exceltesting_test.go

@@ -16,7 +16,7 @@ AND	i.oid			=	ix.indexrelid
 AND	ix.indisprimary	=	TRUE
 AND	A.attrelid		=	T.oid
 AND	A.attnum		=	ANY(ix.indkey)
-AND	T.relkind		=	'r'
+AND	T.relkind		IN	('r', 'p') -- TODO: 将来的には他の relkind にも対応する予定
 AND	T.relname		=	ta.tablename
 AND	ta.schemaname	=	CURRENT_SCHEMA()
 AND	T.relname		=	$1

query.go

@@ -43,3 +43,16 @@ CREATE TABLE test_x(
     CONSTRAINT test_x_pkc PRIMARY KEY(id)
 )
 ;
+
+DROP TABLE IF EXISTS temperature
+;
+CREATE TABLE temperature(
+    ymd varchar(8) NOT NULL,
+    value numeric(4,1) NOT NULL,
+    CONSTRAINT temperature_pkc PRIMARY KEY(ymd)
+) PARTITION BY RANGE (ymd)
+;
+DROP TABLE IF EXISTS temperature_2021_2022
+;
+CREATE TABLE temperature_2021_2022 PARTITION OF temperature FOR VALUES FROM ('20210101') TO ('20220101')
+;