Added some unit tests along with Travis CI config

Author: Persata

.travis.yml

@@ -0,0 +1,20 @@
+language: go
+
+go:
+    - 1.9.x
+    - release
+
+script:
+ - go test -v ./...
+
+env:
+    - DEP_VERSION="0.3.2"
+
+before_install:
+    # Download the binary to bin folder in $GOPATH
+    - curl -L -s https://github.com/golang/dep/releases/download/v${DEP_VERSION}/dep-linux-amd64 -o $GOPATH/bin/dep
+    # Make the binary executable
+    - chmod +x $GOPATH/bin/dep
+
+install:
+    - dep ensure

Gopkg.lock

@@ -28,3 +28,7 @@
 [[constraint]]
   branch = "v2"
   name = "gopkg.in/yaml.v2"
+
+[[constraint]]
+  name = "github.com/stretchr/testify"
+  version = "1.2.0"

Gopkg.toml

@@ -2,29 +2,41 @@ package commands
 
 import (
 	"fmt"
-	"github.com/Persata/mysqldumptablepurger/io"
+	"github.com/Persata/mysqldumptablepurger/iowrapper"
 	"github.com/Persata/mysqldumptablepurger/parser"
 )
 
 func ListTables(path string) int {
-	s, f := io.GetScanner(path)
+	fmt.Println(fmt.Sprintf("Scanning for tables in %s...\n", path))
+
+	tables := GetTablesList(path)
+
+	for _, table := range tables {
+		fmt.Println(table)
+	}
+
+	fmt.Println("\nScan complete!")
+
+	return 0
+}
+
+func GetTablesList(path string) []string {
+	s, f := iowrapper.GetScanner(path)
 	defer f.Close()
 
-	fmt.Println(fmt.Sprintf("Scanning for tables in %s...\n", path))
+	var tables []string
 
 	for s.Scan() {
 		match := parser.TableStructureRegex.FindStringSubmatch(s.Text())
 
 		if len(match) > 0 {
-			fmt.Println(match[1])
+			tables = append(tables, match[1])
 		}
 	}
 
 	if s.Err() != nil {
 		fmt.Println(fmt.Sprintf("An error occurred scanning the file: %s", s.Err()))
 	}
 
-	fmt.Println("\nScan complete!")
-
-	return 0
+	return tables
 }

commands/list_tables.go

@@ -0,0 +1,14 @@
+package commands_test
+
+import (
+	"testing"
+	"github.com/stretchr/testify/assert"
+	"github.com/Persata/mysqldumptablepurger/commands"
+)
+
+func TestListTables(t *testing.T) {
+	tables := commands.GetTablesList("../test/data/test_dump.sql")
+	assert.Len(t, tables, 2)
+	assert.Contains(t, tables, "test_table_1")
+	assert.Contains(t, tables, "test_table_2")
+}

commands/list_tables_test.go

@@ -5,13 +5,14 @@ import (
 	"os"
 	"bufio"
 	"strings"
-	"github.com/Persata/mysqldumptablepurger/io"
+	"github.com/Persata/mysqldumptablepurger/iowrapper"
 	"github.com/Persata/mysqldumptablepurger/parser"
 	"regexp"
 )
 
 func RemoveTables(inputPath string, outputPath string, tables []string) int {
-	fmt.Println("The following tables will be removed: \n")
+	fmt.Println("The following tables will be removed:")
+	fmt.Println()
 
 	for _, table := range tables {
 		fmt.Println(fmt.Sprintf("- %s", table))
@@ -32,13 +33,14 @@ func RemoveTables(inputPath string, outputPath string, tables []string) int {
 		fmt.Println("Exiting, no action performed")
 		return 0
 	} else {
-		fmt.Println("Processing...\n")
+		fmt.Println("Processing...")
+		fmt.Println()
 	}
 
-	s, fr := io.GetScanner(inputPath)
+	s, fr := iowrapper.GetScanner(inputPath)
 	defer fr.Close()
 
-	w, fw := io.GetWriter(outputPath)
+	w, fw := iowrapper.GetWriter(outputPath)
 	defer fw.Close()
 
 	replaceRegex := regexp.MustCompile(fmt.Sprintf(parser.TableStructureRegexReplace, strings.Join(tables, "|")))

commands/remove_tables.go

@@ -12,6 +12,23 @@ type RemoveConf struct {
 }
 
 func RemoveTablesByConf(inputPath string, configFilePath string) int {
+
+	configFileData := GetConfigFileData(configFilePath)
+
+	if len(configFileData.OutputFile) == 0 {
+		fmt.Println(fmt.Sprintf("No output file found in the given config file - got '%s'.", configFileData.OutputFile))
+		return 1
+	}
+
+	if len(configFileData.Tables) == 0 {
+		fmt.Println("Specified config file does not contain any tables to remove.")
+		return 1
+	}
+
+	return RemoveTables(inputPath, configFileData.OutputFile, configFileData.Tables)
+}
+
+func GetConfigFileData(configFilePath string) RemoveConf {
 	configFileBuffer, err := ioutil.ReadFile(configFilePath)
 
 	if err != nil {
@@ -26,15 +43,5 @@ func RemoveTablesByConf(inputPath string, configFilePath string) int {
 		panic(err)
 	}
 
-	if len(configFileData.OutputFile) == 0 {
-		fmt.Println(fmt.Sprintf("No output file found in the given config file - got '%s'.", configFileData.OutputFile))
-		return 1
-	}
-
-	if len(configFileData.Tables) == 0 {
-		fmt.Println("Specified config file does not contain any tables to remove.")
-		return 1
-	}
-
-	return RemoveTables(inputPath, configFileData.OutputFile, configFileData.Tables)
+	return configFileData
 }

commands/remove_tables_by_conf.go

@@ -0,0 +1,20 @@
+package commands_test
+
+import (
+	"testing"
+	"github.com/stretchr/testify/assert"
+	"github.com/Persata/mysqldumptablepurger/commands"
+)
+
+func TestGetConfigFileDataSingle(t *testing.T) {
+	configFileData := commands.GetConfigFileData("../test/conf/test_single_table.yml")
+	assert.Equal(t, configFileData.OutputFile, "test_dump_cleaned.sql")
+	assert.Len(t, configFileData.Tables, 1)
+}
+
+
+func TestGetConfigFileDataTwo(t *testing.T) {
+	configFileData := commands.GetConfigFileData("../test/conf/test_two_tables.yml")
+	assert.Equal(t, configFileData.OutputFile, "test_dump_cleaned.sql")
+	assert.Len(t, configFileData.Tables, 2)
+}

commands/remove_tables_by_conf_test.go

@@ -1,4 +1,4 @@
-package io
+package iowrapper
 
 import (
 	"bufio"

io/get_scanner.go renamed to iowrapper/get_scanner.go

@@ -1,4 +1,4 @@
-package io
+package iowrapper
 
 import (
 	"bufio"