restic-scheduler/tasks_test.go

package main_test

import (
	"bytes"
	"errors"
	"log"
	"testing"

	main "git.iamthefij.com/iamthefij/restic-scheduler"
)

func NewBufferedLogger(prefix string) (*bytes.Buffer, *log.Logger) {
	outputBuffer := bytes.Buffer{}
	logger := log.New(&outputBuffer, prefix, 0)

	return &outputBuffer, logger
}

func TestJobTaskScript(t *testing.T) {
	t.Parallel()

	cases := []struct {
		name           string
		script         main.JobTaskScript
		config         main.TaskConfig
		expectedErr    error
		expectedOutput string
	}{
		{
			name: "simple",
			config: main.TaskConfig{
				BackupPaths: nil,
				Env:         nil,
				Logger:      nil,
				Restic:      nil,
			},
			script: main.JobTaskScript{
				Cwd:       "./test",
				OnBackup:  "echo yass",
				OnRestore: "echo yass",
			},
			expectedErr:    nil,
			expectedOutput: "t yass\nt \n",
		},
		{
			name: "check from job dir",
			config: main.TaskConfig{
				BackupPaths: nil,
				Env:         nil,
				Logger:      nil,
				Restic:      nil,
			},
			script: main.JobTaskScript{
				Cwd:       "./test",
				OnBackup:  "basename `pwd`",
				OnRestore: "basename `pwd`",
			},
			expectedErr:    nil,
			expectedOutput: "t test\nt \n",
		},
		{
			name: "check env",
			config: main.TaskConfig{
				BackupPaths: nil,
				Env:         map[string]string{"TEST": "OK"},
				Logger:      nil,
				Restic:      nil,
			},
			script: main.JobTaskScript{
				Cwd:       "./test",
				OnBackup:  "echo $TEST",
				OnRestore: "echo $TEST",
			},
			expectedErr:    nil,
			expectedOutput: "t OK\nt \n",
		},
	}

	for _, c := range cases {
		testCase := c

		buf, logger := NewBufferedLogger("t")
		testCase.config.Logger = logger

		t.Run(testCase.name, func(t *testing.T) {
			t.Parallel()

			actual := testCase.script.RunBackup(testCase.config)

			if !errors.Is(actual, testCase.expectedErr) {
				t.Errorf("expected error to wrap %v but found %v", testCase.expectedErr, actual)
			}

			output := buf.String()

			if testCase.expectedOutput != output {
				t.Errorf("Unexpected output. expected: %s actual: %s", testCase.expectedOutput, output)
			}
		})
	}
}

func TestJobTaskSql(t *testing.T) {
	t.Parallel()

	type TaskGenerator interface {
		Validate() error
		GetPreTask() main.ExecutableTask
		GetPostTask() main.ExecutableTask
	}

	cases := []struct {
		name          string
		task          TaskGenerator
		validationErr error
		preBackup     string
		postBackup    string
		preRestore    string
		postRestore   string
	}{
		{
			name: "mysql simple",
			//nolint:exhaustruct
			task: main.JobTaskMySQL{
				Name:       "simple",
				DumpToPath: "./simple.sql",
			},
			validationErr: nil,
			preBackup:     "mysqldump --result-file ./simple.sql --all-databases",
			postBackup:    "",
			preRestore:    "",
			postRestore:   "mysql < ./simple.sql",
		},
		{
			name: "mysql tables no database",
			//nolint:exhaustruct
			task: main.JobTaskMySQL{
				Name:       "name",
				Tables:     []string{"table1", "table2"},
				DumpToPath: "./simple.sql",
			},
			validationErr: main.ErrMissingField,
			preBackup:     "",
			postBackup:    "",
			preRestore:    "",
			postRestore:   "",
		},
		{
			name: "mysql all options",
			task: main.JobTaskMySQL{
				Name:          "simple",
				Hostname:      "host",
				Port:          3306,
				Username:      "user",
				Password:      "pass",
				Database:      "db",
				NoTablespaces: true,
				Tables:        []string{"table1", "table2"},
				DumpToPath:    "./simple.sql",
			},
			validationErr: nil,
			preBackup: "mysqldump --result-file ./simple.sql --host host --port 3306" +
				" --user user --password=pass --no-tablespaces db table1 table2",
			postBackup:  "",
			preRestore:  "",
			postRestore: "mysql --host host --port 3306 --user user --password=pass db < ./simple.sql",
		},
		{
			name: "psql all",
			task: main.JobTaskPostgres{
				Name:          "simple",
				Hostname:      "host",
				Port:          6543,
				Username:      "user",
				Password:      "pass",
				Database:      "db",
				NoTablespaces: true,
				Create:        true,
				Clean:         true,
				Tables:        []string{"table1", "table2"},
				DumpToPath:    "./simple.sql",
			},
			validationErr: nil,
			preBackup: "pg_dump --file ./simple.sql --host host --port 6543 --username user --no-tablespaces" +
				" --clean --create --table table1 --table table2 db",
			postBackup:  "",
			preRestore:  "",
			postRestore: "psql --host host --port 6543 --username user db < ./simple.sql",
		},
		// Sqlite
		{
			name: "sqlite simple",

			task: main.JobTaskSqlite{
				Name:       "simple",
				Path:       "database.db",
				DumpToPath: "./simple.db.bak",
			},
			validationErr: nil,
			preBackup:     "sqlite3 'database.db' '.backup ./simple.db.bak'",
			postBackup:    "",
			preRestore:    "",
			postRestore:   "cp './simple.db.bak' 'database.db'",
		},
	}

	for _, c := range cases {
		testCase := c

		t.Run(testCase.name, func(t *testing.T) {
			t.Parallel()

			validateErr := testCase.task.Validate()
			if !errors.Is(validateErr, testCase.validationErr) {
				t.Errorf("unexpected validation result. expected: %v, actual: %v", testCase.validationErr, validateErr)
			}

			if validateErr != nil {
				return
			}

			if preTask, ok := testCase.task.GetPreTask().(main.JobTaskScript); ok {
				AssertEqual(t, "incorrect pre-backup", testCase.preBackup, preTask.OnBackup)
				AssertEqual(t, "incorrect pre-restore", testCase.preRestore, preTask.OnRestore)
			} else {
				t.Error("pre task was not a JobTaskScript")
			}

			if postTask, ok := testCase.task.GetPostTask().(main.JobTaskScript); ok {
				AssertEqual(t, "incorrect post-backup", testCase.postBackup, postTask.OnBackup)
				AssertEqual(t, "incorrect post-restore", testCase.postRestore, postTask.OnRestore)
			} else {
				t.Error("post task was not a JobTaskScript")
			}
		})
	}
}