Test for dropbox integration

cmd/backup/config.go

@@ -70,6 +70,7 @@ type Config struct {
 	AzureStorageContainerName     string          `split_words:"true"`
 	AzureStoragePath              string          `split_words:"true"`
 	AzureStorageEndpoint          string          `split_words:"true" default:"https://{{ .AccountName }}.blob.core.windows.net/"`
+	DropboxEndpoint               string          `split_words:"true" default:"https://api.dropbox.com/"`
 	DropboxRefreshToken           string          `split_words:"true"`
 	DropboxAppKey                 string          `split_words:"true"`
 	DropboxAppSecret              string          `split_words:"true"`

cmd/backup/script.go

@@ -222,6 +222,7 @@ func newScript() (*script, error) {
 
 	if s.c.DropboxRefreshToken != "" && s.c.DropboxAppKey != "" && s.c.DropboxAppSecret != "" {
 		dropboxConfig := dropbox.Config{
+			Endpoint:         s.c.DropboxEndpoint,
 			RefreshToken:     s.c.DropboxRefreshToken,
 			AppKey:           s.c.DropboxAppKey,
 			AppSecret:        s.c.DropboxAppSecret,

internal/storage/dropbox/dropbox.go

@@ -4,6 +4,7 @@ import (
 	"bytes"
 	"context"
 	"fmt"
+	"net/url"
 	"os"
 	"path"
 	"path/filepath"
@@ -26,6 +27,7 @@ type dropboxStorage struct {
 
 // Config allows to configure a Dropbox storage backend.
 type Config struct {
+	Endpoint         string
 	RefreshToken     string
 	AppKey           string
 	AppSecret        string
@@ -35,24 +37,41 @@ type Config struct {
 
 // NewStorageBackend creates and initializes a new Dropbox storage backend.
 func NewStorageBackend(opts Config, logFunc storage.Log) (storage.Backend, error) {
+	tokenUrl, _ := url.JoinPath(opts.Endpoint, "oauth2/token")
+
 	conf := &oauth2.Config{
 		ClientID:     opts.AppKey,
 		ClientSecret: opts.AppSecret,
 		Endpoint: oauth2.Endpoint{
-			TokenURL: "https://api.dropbox.com/oauth2/token",
+			TokenURL: tokenUrl,
 		},
 	}
 
+	isCITest := opts.Endpoint != "https://api.dropbox.com/"
+
 	logFunc(storage.LogLevelInfo, "Dropbox", "Fetching fresh access token for Dropbox storage backend.")
-	tkSource := conf.TokenSource(context.TODO(), &oauth2.Token{RefreshToken: opts.RefreshToken})
-	token, err := tkSource.Token()
-	if err != nil {
-		return nil, fmt.Errorf("(*dropboxStorage).NewStorageBackend: Error refreshing token: %w", err)
+	token := &oauth2.Token{RefreshToken: opts.RefreshToken}
+	if !isCITest {
+		tkSource := conf.TokenSource(context.TODO(), &oauth2.Token{RefreshToken: opts.RefreshToken})
+		var err error
+		token, err = tkSource.Token()
+		if err != nil {
+			return nil, fmt.Errorf("(*dropboxStorage).NewStorageBackend: Error refreshing token: %w", err)
+		}
 	}
 
-	client := files.New(dropbox.Config{
-		Token: token.AccessToken,
-	})
+	dbxConfig := dropbox.Config{}
+
+	if isCITest {
+		dbxConfig.Token = opts.RefreshToken
+		dbxConfig.URLGenerator = func(hostType string, namespace string, route string) string {
+			return fmt.Sprintf("%s/%d/%s/%s", opts.Endpoint, 2, namespace, route)
+		}
+	} else {
+		dbxConfig.Token = token.AccessToken
+	}
+
+	client := files.New(dbxConfig)
 
 	if opts.ConcurrencyLevel < 1 {
 		logFunc(storage.LogLevelWarning, "Dropbox", "Concurrency level must be at least 1! Using 1 instead of %d.", opts.ConcurrencyLevel)

test/dropbox/openapi-mock.yaml

@@ -0,0 +1,25 @@
+# OpenAPI specification options
+openapi:
+  specification_url: '/etc/openapi/user_v2.yaml'
+
+# web server options
+http:
+  cors_enabled: false
+  port: 8080
+  response_timeout: 1.0
+
+# application specific options
+application:
+  debug: false
+  log_format: tty
+  log_level: info
+
+# options to control generation process
+generation:
+  default_min_float: -1.073741823e+09
+  default_max_float: 1.073741823e+09
+  default_min_int: 0
+  default_max_int: 2147483647
+  null_probability: 0
+  suppress_errors: false
+  use_examples: 'if_present'

test/dropbox/run.sh

@@ -0,0 +1,50 @@
+#!/bin/sh
+
+set -e
+
+cd "$(dirname "$0")"
+. ../util.sh
+current_test=$(basename $(pwd))
+
+docker compose up -d
+sleep 5
+
+docker compose exec backup backup
+
+sleep 5
+
+expect_running_containers "3"
+
+dvb_logs=$(docker logs backup-1 2>&1)
+if $dvb_logs | grep "ERROR"
+then
+  fail "Backup failed, errors reported: $dvb_logs"
+else
+  pass "Backup succeeded, no errors reported."
+fi
+
+dbx_logs=$(docker logs openapi_mock-1 2>&1)
+if $dbx_logs | grep "ERROR"
+then
+  fail "Backup failed, errors reported: $dvb_logs"
+else
+  pass "Backup succeeded, no errors reported."
+fi
+
+# The second part of this test checks if backups get deleted when the retention
+# is set to 0 days (which it should not as it would mean all backups get deleted)
+# TODO: find out if we can test actual deletion without having to wait for a day
+BACKUP_RETENTION_DAYS="0" docker compose up -d
+sleep 5
+
+docker compose exec backup backup
+
+dvb_logs=$(docker logs backup-1 2>&1)
+if $dvb_logs | grep "Refusing to do so, please check your configuratio"
+then
+  pass "Remote backups have not been deleted."
+else
+  fail "Remote backups would have been deleted: $dvb_logs"
+fi
+
+docker compose down --volumes