/
restore_test.go
104 lines (83 loc) · 2.92 KB
/
restore_test.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
package tests
import (
"strconv"
"testing"
"time"
"github.com/couchbase/backup/archive"
"github.com/couchbase/backup/value"
)
func TestBackupRestore(t *testing.T) {
defer cleanup()
defer deleteAllBuckets(testHost, t)
cleanup()
deleteAllBuckets(testHost, t)
createCouchbaseBucket(testHost, "default", "", t)
backupName := "restore-test"
loadData(testHost, rbacUsername, rbacPassword, "default", 5000, "full", false, t)
config := value.CreateBackupConfig("", "", make([]string, 0),
make([]string, 0), make([]string, 0), make([]string, 0),
false, false, false, false, false, false, false, false, []int{})
a, err := archive.MountArchive(testDir, true)
checkError(err, t)
checkError(a.CreateRepo(backupName, config), t)
// Test that restoring data when none exists gives an error
err = executeRestore(a, backupName, testHost, rbacUsername, rbacPassword, "name",
"name", 4, false, config)
if err == nil {
t.Fatal(err.Error())
} else if _, ok := err.(archive.EmptyRangeError); !ok {
t.Fatal(err.Error())
}
// Backup the data
name, err := executeBackup(a, backupName, "archive", testHost, rbacUsername, rbacPassword,
4, false, false)
checkError(err, t)
info, err := a.BackupInfo(backupName, name)
checkError(err, t)
count := info["default"].NumDocs
if count != 5000 {
t.Fatal("Expected to backup 5000 items, got " + strconv.Itoa(count))
}
deleteBucket(testHost, "default", t, true)
createCouchbaseBucket(testHost, "default", "", t)
// Check that using an invalid start point causes an error
err = executeRestore(a, backupName, testHost, rbacUsername, rbacPassword, "name",
name, 4, false, config)
if err == nil {
t.Fatal(err.Error())
} else if _, ok := err.(archive.RangePointError); !ok {
t.Fatal(err.Error())
}
// Check that using an invalid end point causes an error
err = executeRestore(a, backupName, testHost, rbacUsername, rbacPassword, name,
"end", 4, false, config)
if err == nil {
t.Fatal(err.Error())
} else if _, ok := err.(archive.RangePointError); !ok {
t.Fatal(err.Error())
}
// Restore the data using explicit start/end specification
err = executeRestore(a, backupName, testHost, rbacUsername, rbacPassword, name,
name, 4, false, config)
checkError(err, t)
time.Sleep(5 * time.Second)
items, err := getNumItems(testHost, rbacUsername, rbacPassword, "default")
if err != nil {
t.Fatalf("Error getting item count: %s", err.Error())
}
if items != 5000 {
t.Fatalf("Expected 5000 items, got %d", items)
}
deleteBucket(testHost, "default", t, true)
createCouchbaseBucket(testHost, "default", "", t)
// Restore the data without explicitly setting the start/end point
err = executeRestore(a, backupName, testHost, rbacUsername, rbacPassword, "",
"", 4, false, config)
checkError(err, t)
time.Sleep(5 * time.Second)
items, err = getNumItems(testHost, rbacUsername, rbacPassword, "default")
checkError(err, t)
if items != 5000 {
t.Fatalf("Expected 5000 items, got %d", items)
}
}