Unit tests

.github/workflows/go.yml

@@ -22,3 +22,6 @@ jobs:
 
       - name: Build
         run: go build -v ./...
+
+      - name: Test 
+        run: go test -v ./...

pkg/display/allcoin/allCoin_test.go

@@ -0,0 +1,77 @@
+/*
+Copyright © 2021 Bhargav SNV [email protected]
+
+Licensed under the Apache License, Version 2.0 (the "License");
+you may not use this file except in compliance with the License.
+You may obtain a copy of the License at
+
+    http://www.apache.org/licenses/LICENSE-2.0
+
+Unless required by applicable law or agreed to in writing, software
+distributed under the License is distributed on an "AS IS" BASIS,
+WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+See the License for the specific language governing permissions and
+limitations under the License.
+*/
+
+package allcoin
+
+import (
+	"sync"
+	"testing"
+
+	"github.com/stretchr/testify/assert"
+)
+
+func TestFilterRows(t *testing.T) {
+	tests := []struct {
+		allRows      [][]string
+		filter       string
+		filteredRows [][]string
+	}{
+		{
+			allRows: [][]string{
+				{"", "BTC", "", "", "", "bitcoin"},
+				{"", "ETH", "", "", "", "ethereum"},
+				{"", "LTC", "", "", "", "litecoin"},
+				{"", "BTC", "", "", "", "bitcoin"},
+				{"", "BTC", "", "", "", "bitcoin"},
+			},
+			filter: "bitcoin",
+			filteredRows: [][]string{
+				{"", "BTC", "", "", "", "bitcoin"},
+				{"", "BTC", "", "", "", "bitcoin"},
+				{"", "BTC", "", "", "", "bitcoin"},
+			},
+		},
+		{
+			allRows: [][]string{
+				{"", "BTC", "", "", "", "bitcoin"},
+				{"", "ETH", "", "", "", "ethereum"},
+				{"", "LTC", "", "", "", "litecoin"},
+				{"", "BTC", "", "", "", "bitcoin"},
+				{"", "BTC", "", "", "", "bitcoin"},
+			},
+			filter: "LTC",
+			filteredRows: [][]string{
+				{"", "LTC", "", "", "", "litecoin"},
+			},
+		},
+		{
+			allRows: [][]string{
+				{"", "XRP", "", "", "", "ripple"},
+				{"", "ETH", "", "", "", "ethereum"},
+				{"", "LTC", "", "", "", "litecoin"},
+				{"", "BTC", "", "", "", "bitcoin"},
+			},
+			filter:       "ABA",
+			filteredRows: [][]string{},
+		},
+	}
+
+	for _, test := range tests {
+		var mutex = &sync.Mutex{}
+		filteredRows := filterRows(test.allRows, test.filter, mutex)
+		assert.Equal(t, test.filteredRows, filteredRows)
+	}
+}

pkg/utils/utils_test.go

@@ -72,3 +72,55 @@ func TestMaxFloat64(t *testing.T) {
 		assert.Equal(t, test.expectedVal, val)
 	}
 }
+
+func TestRoundValues(t *testing.T) {
+	tests := []struct {
+		name           string
+		inputNum1      float64
+		inputNum2      float64
+		expRoundedVals []float64
+		expUnit        string
+	}{
+		{
+			name:           "both values smaller than kilo",
+			inputNum1:      250,
+			inputNum2:      500,
+			expRoundedVals: []float64{250, 500},
+			expUnit:        "",
+		},
+		{
+			name:           "both values kilo",
+			inputNum1:      80000,
+			inputNum2:      9000,
+			expRoundedVals: []float64{80, 9},
+			expUnit:        "K",
+		},
+		{
+			name:           "mega and less than mega",
+			inputNum1:      400000,
+			inputNum2:      7000000,
+			expRoundedVals: []float64{0.4, 7},
+			expUnit:        "M",
+		},
+		{
+			name:           "giga and less than giga",
+			inputNum1:      100000000,
+			inputNum2:      9000000000,
+			expRoundedVals: []float64{0.1, 9},
+			expUnit:        "B",
+		},
+		{
+			name:           "both values tera",
+			inputNum1:      5000000000000000,
+			inputNum2:      100000000000000,
+			expRoundedVals: []float64{5000, 100},
+			expUnit:        "T",
+		},
+	}
+
+	for _, test := range tests {
+		roundedVals, unit := utils.RoundValues(test.inputNum1, test.inputNum2)
+		assert.Equal(t, test.expRoundedVals, roundedVals)
+		assert.Equal(t, test.expUnit, unit)
+	}
+}