Add TestCreateOrganizationMap test

pkg/modules/migration/trello/trello_test.go

@@ -32,14 +32,12 @@ import (
 	"github.com/stretchr/testify/require"
 )
 
-func TestConvertTrelloToVikunja(t *testing.T) {
+func GetBoard(t *testing.T) ([]*trello.Board, time.Time) {
 
 	config.InitConfig()
 
 	time1, err := time.Parse(time.RFC3339Nano, "2014-09-26T08:25:05Z")
 	require.NoError(t, err)
-	exampleFile, err := os.ReadFile(config.ServiceRootpath.GetString() + "/pkg/modules/migration/testimage.jpg")
-	require.NoError(t, err)
 
 	trelloData := []*trello.Board{
 		{
@@ -215,6 +213,15 @@ func TestConvertTrelloToVikunja(t *testing.T) {
 	}
 	trelloData[0].Prefs.BackgroundImage = "https://vikunja.io/testimage.jpg" // Using an image which we are hosting, so it'll still be up
 
+	return trelloData, time1
+}
+
+func TestConvertTrelloToVikunja(t *testing.T) {
+	trelloData, time1 := GetBoard(t)
+
+	exampleFile, err := os.ReadFile(config.ServiceRootpath.GetString() + "/pkg/modules/migration/testimage.jpg")
+	require.NoError(t, err)
+
 	expectedHierarchyOrg := map[string][]*models.ProjectWithTasksAndBuckets{
 		"orgid": {
 			{
@@ -423,3 +430,21 @@ func TestConvertTrelloToVikunja(t *testing.T) {
 		}
 	}
 }
+
+func TestCreateOrganizationMap(t *testing.T) {
+	trelloData, _ := GetBoard(t)
+
+	organizationMap := createOrganizationMap(trelloData)
+	expectedMap := map[string][]*trello.Board{
+		"orgid": {
+			trelloData[0],
+			trelloData[2],
+		},
+		"orgid2": {
+			trelloData[1],
+		},
+	}
+	if diff, equal := messagediff.PrettyDiff(organizationMap, expectedMap); !equal {
+		t.Errorf("converted trello data = %v,\nwant %v,\ndiff: %v", organizationMap, expectedMap, diff)
+	}
+}