e2e_upload_folder_test.go

//go:build e2e
// +build e2e

package cmdupload

import (
	"context"
	"errors"
	"fmt"
	"immich-go/immich"
	"immich-go/logger"
	"testing"
	"time"

	"github.com/joho/godotenv"
)

func TestE2eUpload(t *testing.T) {

	var myEnv map[string]string
	myEnv, err := godotenv.Read("../.env")
	if err != nil {
		t.Errorf("cant initialize environment variables: %s", err)
		return
	}

	host := myEnv["IMMICH_HOST"]
	if host == "" {
		host = "http://localhost:2283"
	}
	key := myEnv["IMMICH_KEY"]
	if key == "" {
		t.Fatal("you must provide the IMMICH's API KEY in the environnement variable DEBUG_IMMICH_KEY")
	}

	tests := []struct {
		name        string
		args        []string
		resetImmich bool
		expectError bool
	}{
		{
			name: "upload google photos",
			args: []string{
				"-google-photos",
				"../../test-data/low_high/Takeout",
			},
			resetImmich: true,
			expectError: false,
		},
		{
			name: "upload folder",
			args: []string{
				"../../test-data/low_high/high",
			},
			// resetImmich: true,

			expectError: false,
		},
		{
			name: "upload folder",
			args: []string{
				"../../test-data/low_high/high",
			},

			// resetImmich: true,
			expectError: false,
		},
		{
			name: "upload folder *.jpg",
			args: []string{
				"-google-photos",
				"../../test-data/test_folder/*.jpg",
			},

			resetImmich: true,
			expectError: true,
		},
		{
			name: "upload folder *.jpg",
			args: []string{
				"../../test-data/test_folder/*/*.jpg",
			},

			// resetImmich: true,
			expectError: false,
		},

		{
			name: "upload folder *.jpg - dry run",
			args: []string{
				"-dry-run",
				"../../test-data/full_takeout (copy)/Takeout/Google Photos/Photos from 2023",
			},

			// resetImmich: true,
			expectError: false,
		},
	}

	logger := logger.NewLogger(logger.Debug, true, false)
	ic, err := immich.NewImmichClient(host, key)

	if err != nil {
		t.Error(err)
		return
	}

	ctx := context.Background()
	for _, tc := range tests {
		t.Run(tc.name, func(t *testing.T) {
			if tc.resetImmich {
				err := resetImmich(ic, "debug@example.com")
				if err != nil {
					t.Error(err)
					return
				}
			}
			err = UploadCommand(ctx, ic, logger, tc.args)
			if (tc.expectError && err == nil) || (!tc.expectError && err != nil) {
				t.Errorf("unexpected err: %v", err)
				return
			}
		})
	}
}

// ResetImmich
// ⛔: will remove the content of the server.‼️
// Give the user of the connection to confirm the server instance: debug@example.com
//

func resetImmich(ic *immich.ImmichClient, user string) error {
	u, err := ic.ValidateConnection(context.Background())
	if err != nil {
		return err
	}
	if u.Email != user {
		return fmt.Errorf("Not the debug server")
	}

	albums, err := ic.GetAllAlbums(context.Background())
	if err != nil {
		return err
	}

	for _, album := range albums {
		err = ic.DeleteAlbum(context.Background(), album.ID)
		if err != nil {
			return err
		}
	}

	assets, err := ic.GetAllAssets(context.Background(), nil)
	if err != nil {
		return err
	}
	ids := []string{}
	for _, a := range assets {
		ids = append(ids, a.ID)
	}
	err = ic.DeleteAssets(context.Background(), ids, true)
	if err != nil {
		return nil
	}

	attempts := 5
	for attempts > 0 {
		assets, err := ic.GetAllAssets(context.Background(), nil)
		if err != nil {
			return err
		}
		if len(assets) == 0 {
			return nil
		}
		time.Sleep(5 * time.Second)
		attempts--
	}

	return errors.New("can't reset immich")
}