minor: Adds method to list organisation and improve tests

renatoaraujo · Oct 7, 2023 · a50f169 · a50f169
1 parent 1f3aa6c
commit a50f169
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 2 deletions.
diff --git a/internal/sponsors/handler_test.go b/internal/sponsors/handler_test.go
@@ -7,7 +7,6 @@ import (
 	"renatoaraujo/uk-visa-sponsors/internal/sponsors"
 	mocks "renatoaraujo/uk-visa-sponsors/internal/sponsors/mocks"
 
-	"github.com/stretchr/testify/assert"
 	"github.com/stretchr/testify/mock"
 	"github.com/stretchr/testify/require"
 )
@@ -110,7 +109,11 @@ func TestNewHandler(t *testing.T) {
 			}
 
 			if handler != nil {
-				assert.IsType(t, &sponsors.Handler{}, handler)
+				require.IsType(t, &sponsors.Handler{}, handler)
+			}
+
+			if tt.load && !tt.wantErr {
+				require.NotEmpty(t, handler.Organisations)
 			}
 
 			mock.AssertExpectationsForObjects(t, fetcherMock)

diff --git a/internal/sponsors/organisation.go b/internal/sponsors/organisation.go
@@ -18,6 +18,10 @@ type Organisation struct {
 	Description string
 }
 
+func (o *Organisations) List() []Organisation {
+	return o.list
+}
+
 func (o *Organisations) SearchOrganisationsByName(name string) []Organisation {
 	var found []Organisation