forked from kahing/fusego
/
in_process.go
147 lines (122 loc) · 3.6 KB
/
in_process.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
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
// Copyright 2015 Google Inc. All Rights Reserved.
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
package samples
import (
"context"
"fmt"
"io"
"io/ioutil"
"log"
"os"
"time"
"github.com/jacobsa/fuse"
"github.com/jacobsa/ogletest"
"github.com/jacobsa/timeutil"
)
// A struct that implements common behavior needed by tests in the samples/
// directory. Use it as an embedded field in your test fixture, calling its
// SetUp method from your SetUp method after setting the Server field.
type SampleTest struct {
// The server under test and the configuration with which it should be
// mounted. These must be set by the user of this type before calling SetUp;
// all the other fields below are set by SetUp itself.
Server fuse.Server
MountConfig fuse.MountConfig
// A context object that can be used for long-running operations.
Ctx context.Context
// A clock with a fixed initial time. The test's set up method may use this
// to wire the server with a clock, if desired.
Clock timeutil.SimulatedClock
// The directory at which the file system is mounted.
Dir string
// Anothing non-nil in this slice will be closed by TearDown. The test will
// fail if closing fails.
ToClose []io.Closer
mfs *fuse.MountedFileSystem
}
// Mount t.Server and initialize the other exported fields of the struct.
// Panics on error.
//
// REQUIRES: t.Server has been set.
func (t *SampleTest) SetUp(ti *ogletest.TestInfo) {
cfg := t.MountConfig
if *fDebug {
cfg.DebugLogger = log.New(os.Stderr, "fuse: ", 0)
}
err := t.initialize(ti.Ctx, t.Server, &cfg)
if err != nil {
panic(err)
}
}
// Like SetUp, but doens't panic.
func (t *SampleTest) initialize(
ctx context.Context,
server fuse.Server,
config *fuse.MountConfig) error {
// Initialize the context used by the test.
t.Ctx = ctx
// Make the server share that context, if the test hasn't already set some
// other one.
if config.OpContext == nil {
config.OpContext = ctx
}
// Initialize the clock.
t.Clock.SetTime(time.Date(2012, 8, 15, 22, 56, 0, 0, time.Local))
// Set up a temporary directory.
var err error
t.Dir, err = ioutil.TempDir("", "sample_test")
if err != nil {
return fmt.Errorf("TempDir: %v", err)
}
// Mount the file system.
t.mfs, err = fuse.Mount(t.Dir, server, config)
if err != nil {
return fmt.Errorf("Mount: %v", err)
}
return nil
}
// Unmount the file system and clean up. Panics on error.
func (t *SampleTest) TearDown() {
err := t.destroy()
if err != nil {
panic(err)
}
}
// Like TearDown, but doesn't panic.
func (t *SampleTest) destroy() (err error) {
// Close what is necessary.
for _, c := range t.ToClose {
if c == nil {
continue
}
ogletest.ExpectEq(nil, c.Close())
}
// Was the file system mounted?
if t.mfs == nil {
return nil
}
// Unmount the file system.
if err := unmount(t.Dir); err != nil {
return fmt.Errorf("unmount: %v", err)
}
// Unlink the mount point.
if err := os.Remove(t.Dir); err != nil {
return fmt.Errorf("Unlinking mount point: %v", err)
}
// Join the file system.
if err := t.mfs.Join(t.Ctx); err != nil {
return fmt.Errorf("mfs.Join: %v", err)
}
return nil
}