Fixed the math tests. Added query.FindAllCountries() as requested in #15

pariz · Oct 19, 2017 · e7715c9 · e7715c9
1 parent a6cd508
commit e7715c9
Show file tree

Hide file tree

Showing 3 changed files with 13 additions and 2 deletions.
diff --git a/math_test.go b/math_test.go
@@ -2,7 +2,7 @@ package gountries
 
 import "fmt"
 
-func ExamplePythagorasEquirectangular() {
+func ExampleCalculatePythagorasEquirectangular() {
 
 	se, _ := query.FindCountryByAlpha("SWE")
 	de, _ := query.FindCountryByAlpha("DEU")
@@ -22,7 +22,7 @@ func ExamplePythagorasEquirectangular() {
 	// 2667.2283097795016
 }
 
-func ExampleHaversine() {
+func ExampleCalculateHaversine() {
 
 	se, _ := query.FindCountryByAlpha("SWE")
 	de, _ := query.FindCountryByAlpha("DEU")

diff --git a/query.go b/query.go
@@ -47,6 +47,11 @@ func (q *Query) FindCountryByAlpha(code string) (result Country, err error) {
 	}
 }
 
+// FindAllCountries returns a list of all countries
+func (q *Query) FindAllCountries() (countries map[string]Country) {
+	return q.Countries
+}
+
 // FindCountries finds a Country based on the given struct data
 func (q Query) FindCountries(c Country) (countries []Country) {
 

diff --git a/query_test.go b/query_test.go
@@ -149,6 +149,12 @@ func TestFindCountryByAlpha(t *testing.T) {
 
 }
 
+func TestFindAllCountries(t *testing.T) {
+
+	assert.Len(t, query.FindAllCountries(), 247)
+
+}
+
 func TestFindCountries(t *testing.T) {
 
 	country := Country{}