Merge pull request #124 from vulcanize/refactor-mcd-configs

Removes _checked from column name query
vulcanize · Jul 31, 2019 · 65e762f · 65e762f
2 parents f2547b5 + dc8d52a
commit 65e762f
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 5 deletions.
diff --git a/libraries/shared/repository/repository.go b/libraries/shared/repository/repository.go
@@ -76,7 +76,8 @@ func GetCheckedColumnNames(db *postgres.DB) ([]string, error) {
 		`SELECT column_name FROM information_schema.columns
 		WHERE table_schema = 'public'
 			AND table_name = 'checked_headers'
-			AND column_name ~ '_checked';`
+			AND column_name <> 'id'
+			AND column_name <> 'header_id';`
 
 	err := db.Select(&queryResult, columnNamesQuery)
 	if err != nil {

diff --git a/libraries/shared/repository/repository_test.go b/libraries/shared/repository/repository_test.go
@@ -295,9 +295,9 @@ var _ = Describe("Repository", func() {
 
 func getExpectedColumnNames() []string {
 	return []string{
-		"column_1_checked",
-		"column_2_checked",
-		"column_3_checked",
-		"column_4_checked",
+		"column_1",
+		"column_2",
+		"column_3",
+		"column_4",
 	}
 }