add ability to compare envs

main.go

@@ -15,8 +15,13 @@ import (
 
 func extractKeys(fileData []byte, ymlPathArgs []string) []string {
 	keys := make([]string, 0, 50)
-	y, _ := simpleyaml.NewYaml(fileData)
-	data, _ := y.GetPath(ymlPathArgs...).String()
+	data := ""
+	if len(ymlPathArgs) > 0 {
+		y, _ := simpleyaml.NewYaml(fileData)
+		data, _ = y.GetPath(ymlPathArgs...).String()
+	} else {
+		data = string(fileData[:])
+	}
 	lines := strings.Split(data, "\n")
 	for _, line := range lines {
 		keyVals := strings.Split(line, "=")
@@ -81,7 +86,11 @@ func main() {
 
 	app.Action = func(c *cli.Context) error {
 		fileNames := strings.Split(c.Args().Get(0), ",")
-		ymlArgs := strings.Split(c.Args().Get(1), ",")
+		ymlArgs := []string{}
+		cmdArgs := c.Args().Get(1)
+		if cmdArgs != "" {
+			ymlArgs = strings.Split(c.Args().Get(1), ",")
+		}
 		envKeys := ParseFileData(fileNames, ymlArgs)
 		result := CompareEnvArrays(envKeys, fileNames)
 		if result {

main_test.go

@@ -1,8 +1,10 @@
 package main
 
-import "testing"
+import (
+	"testing"
 
-import "github.com/stretchr/testify/assert"
+	"github.com/stretchr/testify/assert"
+)
 
 func TestParseData(t *testing.T) {
 	fileNames := []string{"testfiles/data1.yml", "testfiles/data2.yml"}
@@ -64,3 +66,21 @@ func TestComparingArraysThatArentEqualByLength(t *testing.T) {
 	testArraysAreEqual := CompareEnvArrays(expectedValues, fileNames)
 	assert.Equal(t, false, testArraysAreEqual)
 }
+
+func TestComparingEnvFilesThatAreEqual(t *testing.T) {
+	fileNames := []string{"testfiles/.env1", "testfiles/.env2"}
+
+	ymlArgs := []string{}
+	envKeys := ParseFileData(fileNames, ymlArgs)
+	result := CompareEnvArrays(envKeys, fileNames)
+	assert.Equal(t, true, result)
+}
+
+func TestComparingEnvFilesThatAreNotEqual(t *testing.T) {
+	fileNames := []string{"testfiles/.env1", "testfiles/.env3"}
+
+	ymlArgs := []string{}
+	envKeys := ParseFileData(fileNames, ymlArgs)
+	result := CompareEnvArrays(envKeys, fileNames)
+	assert.Equal(t, false, result)
+}

testfiles/.env1

@@ -0,0 +1,2 @@
+a=b
+b=a

testfiles/.env2

@@ -0,0 +1,2 @@
+a=b
+b=a

testfiles/.env3

@@ -0,0 +1 @@
+a=b