first commit

server/internal/storage/googledrive/provider_test.go

@@ -0,0 +1,75 @@
+package googledrive
+
+import (
+	"context"
+	"io"
+	"strings"
+	"testing"
+	"time"
+
+	"backupx/server/internal/storage"
+)
+
+type fakeClient struct{ data map[string]string }
+
+func (c *fakeClient) TestConnection(context.Context, string) error { return nil }
+func (c *fakeClient) Upload(_ context.Context, _ string, objectKey string, reader io.Reader) error {
+	content, _ := io.ReadAll(reader)
+	c.data[objectKey] = string(content)
+	return nil
+}
+func (c *fakeClient) Download(_ context.Context, _ string, objectKey string) (io.ReadCloser, error) {
+	return io.NopCloser(strings.NewReader(c.data[objectKey])), nil
+}
+func (c *fakeClient) Delete(_ context.Context, _ string, objectKey string) error {
+	delete(c.data, objectKey)
+	return nil
+}
+func (c *fakeClient) List(_ context.Context, _ string, prefix string) ([]storage.ObjectInfo, error) {
+	items := make([]storage.ObjectInfo, 0)
+	for key, value := range c.data {
+		if prefix == "" || strings.HasPrefix(key, prefix) {
+			items = append(items, storage.ObjectInfo{Key: key, Size: int64(len(value)), UpdatedAt: time.Now().UTC()})
+		}
+	}
+	return items, nil
+}
+func (c *fakeClient) EnsureFolder(_ context.Context, _, name string) (string, error) {
+	return "fake-folder-" + name, nil
+}
+
+func TestGoogleDriveProviderCRUD(t *testing.T) {
+	factory := Factory{newClient: func(context.Context, storage.GoogleDriveConfig) (client, error) {
+		return &fakeClient{data: make(map[string]string)}, nil
+	}}
+	providerAny, err := factory.New(context.Background(), map[string]any{"clientId": "id", "clientSecret": "secret", "refreshToken": "refresh"})
+	if err != nil {
+		t.Fatalf("Factory.New returned error: %v", err)
+	}
+	provider := providerAny.(*Provider)
+	if err := provider.TestConnection(context.Background()); err != nil {
+		t.Fatalf("TestConnection returned error: %v", err)
+	}
+	if err := provider.Upload(context.Background(), "backup.tar.gz", strings.NewReader("payload"), 7, nil); err != nil {
+		t.Fatalf("Upload returned error: %v", err)
+	}
+	reader, err := provider.Download(context.Background(), "backup.tar.gz")
+	if err != nil {
+		t.Fatalf("Download returned error: %v", err)
+	}
+	defer reader.Close()
+	content, _ := io.ReadAll(reader)
+	if string(content) != "payload" {
+		t.Fatalf("unexpected content: %s", string(content))
+	}
+	items, err := provider.List(context.Background(), "backup")
+	if err != nil {
+		t.Fatalf("List returned error: %v", err)
+	}
+	if len(items) != 1 || items[0].Key != "backup.tar.gz" {
+		t.Fatalf("unexpected list result: %#v", items)
+	}
+	if err := provider.Delete(context.Background(), "backup.tar.gz"); err != nil {
+		t.Fatalf("Delete returned error: %v", err)
+	}
+}