mirror of
https://github.com/rclone/rclone.git
synced 2024-11-22 14:39:24 +08:00
3f8d286a75
Previous to this fix the vfs cache wasn't being cleaned properly on startup as the atimes of the existing files were being ignored.
328 lines
8.5 KiB
Go
328 lines
8.5 KiB
Go
package vfs
|
|
|
|
import (
|
|
"fmt"
|
|
"io/ioutil"
|
|
"os"
|
|
"path/filepath"
|
|
"sort"
|
|
"strings"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/djherbis/times"
|
|
"github.com/ncw/rclone/fstest"
|
|
"github.com/spf13/pflag"
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
"golang.org/x/net/context" // switch to "context" when we stop supporting go1.6
|
|
)
|
|
|
|
// Check CacheMode it satisfies the pflag interface
|
|
var _ pflag.Value = (*CacheMode)(nil)
|
|
|
|
func TestCacheModeString(t *testing.T) {
|
|
assert.Equal(t, "off", CacheModeOff.String())
|
|
assert.Equal(t, "full", CacheModeFull.String())
|
|
assert.Equal(t, "CacheMode(17)", CacheMode(17).String())
|
|
}
|
|
|
|
func TestCacheModeSet(t *testing.T) {
|
|
var m CacheMode
|
|
|
|
err := m.Set("full")
|
|
assert.NoError(t, err)
|
|
assert.Equal(t, CacheModeFull, m)
|
|
|
|
err = m.Set("potato")
|
|
assert.Error(t, err, "Unknown cache mode level")
|
|
|
|
err = m.Set("")
|
|
assert.Error(t, err, "Unknown cache mode level")
|
|
}
|
|
|
|
func TestCacheModeType(t *testing.T) {
|
|
var m CacheMode
|
|
assert.Equal(t, "string", m.Type())
|
|
}
|
|
|
|
// convert c.item to a string
|
|
func itemAsString(c *cache) string {
|
|
c.itemMu.Lock()
|
|
defer c.itemMu.Unlock()
|
|
var out []string
|
|
for name, item := range c.item {
|
|
out = append(out, fmt.Sprintf("name=%q isFile=%v opens=%d", name, item.isFile, item.opens))
|
|
}
|
|
sort.Strings(out)
|
|
return strings.Join(out, "\n")
|
|
}
|
|
|
|
func TestCacheNew(t *testing.T) {
|
|
r := fstest.NewRun(t)
|
|
defer r.Finalise()
|
|
|
|
ctx, cancel := context.WithCancel(context.Background())
|
|
defer cancel()
|
|
|
|
c, err := newCache(ctx, r.Fremote, &DefaultOpt)
|
|
require.NoError(t, err)
|
|
|
|
assert.Contains(t, c.root, "vfs")
|
|
assert.Contains(t, c.f.Root(), filepath.Base(r.Fremote.Root()))
|
|
assert.Equal(t, "", itemAsString(c))
|
|
|
|
// mkdir
|
|
p, err := c.mkdir("potato")
|
|
require.NoError(t, err)
|
|
assert.Equal(t, "potato", filepath.Base(p))
|
|
assert.Equal(t, `name="" isFile=false opens=0`, itemAsString(c))
|
|
|
|
fi, err := os.Stat(filepath.Dir(p))
|
|
require.NoError(t, err)
|
|
assert.True(t, fi.IsDir())
|
|
|
|
// get
|
|
item := c.get("potato")
|
|
item2 := c.get("potato")
|
|
assert.Equal(t, item, item2)
|
|
assert.WithinDuration(t, time.Now(), item.atime, time.Second)
|
|
|
|
// updateTime
|
|
//.. before
|
|
t1 := time.Now().Add(-60 * time.Minute)
|
|
c.updateTime("potato", t1)
|
|
item = c.get("potato")
|
|
assert.NotEqual(t, t1, item.atime)
|
|
assert.Equal(t, 0, item.opens)
|
|
//..after
|
|
t2 := time.Now().Add(60 * time.Minute)
|
|
c.updateTime("potato", t2)
|
|
item = c.get("potato")
|
|
assert.Equal(t, t2, item.atime)
|
|
assert.Equal(t, 0, item.opens)
|
|
|
|
// open
|
|
assert.Equal(t, `name="" isFile=false opens=0
|
|
name="potato" isFile=true opens=0`, itemAsString(c))
|
|
c.open("potato")
|
|
assert.Equal(t, `name="" isFile=false opens=1
|
|
name="potato" isFile=true opens=1`, itemAsString(c))
|
|
item = c.get("potato")
|
|
assert.WithinDuration(t, time.Now(), item.atime, time.Second)
|
|
assert.Equal(t, 1, item.opens)
|
|
|
|
// write the file
|
|
err = ioutil.WriteFile(p, []byte("hello"), 0600)
|
|
require.NoError(t, err)
|
|
|
|
// read its atime
|
|
fi, err = os.Stat(p)
|
|
assert.NoError(t, err)
|
|
atime := times.Get(fi).AccessTime()
|
|
|
|
// updateAtimes
|
|
item = c.get("potato")
|
|
item.atime = time.Now().Add(-24 * time.Hour)
|
|
err = c.updateAtimes()
|
|
require.NoError(t, err)
|
|
assert.Equal(t, `name="" isFile=false opens=1
|
|
name="potato" isFile=true opens=1`, itemAsString(c))
|
|
item = c.get("potato")
|
|
assert.Equal(t, atime, item.atime)
|
|
|
|
// updateAtimes - not in the cache
|
|
oldItem := item
|
|
delete(c.item, "potato") // remove from cache
|
|
err = c.updateAtimes()
|
|
require.NoError(t, err)
|
|
assert.Equal(t, `name="" isFile=false opens=1
|
|
name="potato" isFile=true opens=0`, itemAsString(c))
|
|
item = c.get("potato")
|
|
assert.Equal(t, atime, item.atime)
|
|
c.item["potato"] = oldItem // restore to cache
|
|
|
|
// try purging with file open
|
|
c.purgeOld(10 * time.Second)
|
|
_, err = os.Stat(p)
|
|
assert.NoError(t, err)
|
|
|
|
// close
|
|
assert.Equal(t, `name="" isFile=false opens=1
|
|
name="potato" isFile=true opens=1`, itemAsString(c))
|
|
c.updateTime("potato", t2)
|
|
assert.Equal(t, `name="" isFile=false opens=1
|
|
name="potato" isFile=true opens=1`, itemAsString(c))
|
|
c.close("potato")
|
|
assert.Equal(t, `name="" isFile=false opens=0
|
|
name="potato" isFile=true opens=0`, itemAsString(c))
|
|
item = c.get("potato")
|
|
assert.WithinDuration(t, time.Now(), item.atime, time.Second)
|
|
assert.Equal(t, 0, item.opens)
|
|
|
|
// try purging with file closed
|
|
c.purgeOld(10 * time.Second)
|
|
// ...nothing should happend
|
|
_, err = os.Stat(p)
|
|
assert.NoError(t, err)
|
|
|
|
//.. purge again with -ve age
|
|
c.purgeOld(-10 * time.Second)
|
|
_, err = os.Stat(p)
|
|
assert.True(t, os.IsNotExist(err))
|
|
|
|
// clean - have tested the internals already
|
|
c.clean()
|
|
|
|
// cleanup
|
|
err = c.cleanUp()
|
|
require.NoError(t, err)
|
|
_, err = os.Stat(c.root)
|
|
assert.True(t, os.IsNotExist(err))
|
|
}
|
|
|
|
func TestCacheOpens(t *testing.T) {
|
|
r := fstest.NewRun(t)
|
|
defer r.Finalise()
|
|
|
|
ctx, cancel := context.WithCancel(context.Background())
|
|
defer cancel()
|
|
|
|
c, err := newCache(ctx, r.Fremote, &DefaultOpt)
|
|
require.NoError(t, err)
|
|
|
|
assert.Equal(t, "", itemAsString(c))
|
|
c.open("potato")
|
|
assert.Equal(t, `name="" isFile=false opens=1
|
|
name="potato" isFile=true opens=1`, itemAsString(c))
|
|
c.open("potato")
|
|
assert.Equal(t, `name="" isFile=false opens=2
|
|
name="potato" isFile=true opens=2`, itemAsString(c))
|
|
c.close("potato")
|
|
assert.Equal(t, `name="" isFile=false opens=1
|
|
name="potato" isFile=true opens=1`, itemAsString(c))
|
|
c.close("potato")
|
|
assert.Equal(t, `name="" isFile=false opens=0
|
|
name="potato" isFile=true opens=0`, itemAsString(c))
|
|
|
|
c.open("potato")
|
|
c.open("a/b/c/d/one")
|
|
c.open("a/b/c/d/e/two")
|
|
c.open("a/b/c/d/e/f/three")
|
|
assert.Equal(t, `name="" isFile=false opens=4
|
|
name="a" isFile=false opens=3
|
|
name="a/b" isFile=false opens=3
|
|
name="a/b/c" isFile=false opens=3
|
|
name="a/b/c/d" isFile=false opens=3
|
|
name="a/b/c/d/e" isFile=false opens=2
|
|
name="a/b/c/d/e/f" isFile=false opens=1
|
|
name="a/b/c/d/e/f/three" isFile=true opens=1
|
|
name="a/b/c/d/e/two" isFile=true opens=1
|
|
name="a/b/c/d/one" isFile=true opens=1
|
|
name="potato" isFile=true opens=1`, itemAsString(c))
|
|
|
|
c.close("potato")
|
|
c.close("a/b/c/d/one")
|
|
c.close("a/b/c/d/e/two")
|
|
c.close("a/b/c/d/e/f/three")
|
|
assert.Equal(t, `name="" isFile=false opens=0
|
|
name="a" isFile=false opens=0
|
|
name="a/b" isFile=false opens=0
|
|
name="a/b/c" isFile=false opens=0
|
|
name="a/b/c/d" isFile=false opens=0
|
|
name="a/b/c/d/e" isFile=false opens=0
|
|
name="a/b/c/d/e/f" isFile=false opens=0
|
|
name="a/b/c/d/e/f/three" isFile=true opens=0
|
|
name="a/b/c/d/e/two" isFile=true opens=0
|
|
name="a/b/c/d/one" isFile=true opens=0
|
|
name="potato" isFile=true opens=0`, itemAsString(c))
|
|
|
|
}
|
|
|
|
func TestCacheCacheDir(t *testing.T) {
|
|
r := fstest.NewRun(t)
|
|
defer r.Finalise()
|
|
|
|
ctx, cancel := context.WithCancel(context.Background())
|
|
defer cancel()
|
|
|
|
c, err := newCache(ctx, r.Fremote, &DefaultOpt)
|
|
require.NoError(t, err)
|
|
|
|
assert.Equal(t, "", itemAsString(c))
|
|
|
|
c.cacheDir("dir")
|
|
assert.Equal(t, `name="" isFile=false opens=0
|
|
name="dir" isFile=false opens=0`, itemAsString(c))
|
|
|
|
c.cacheDir("dir/sub")
|
|
assert.Equal(t, `name="" isFile=false opens=0
|
|
name="dir" isFile=false opens=0
|
|
name="dir/sub" isFile=false opens=0`, itemAsString(c))
|
|
|
|
c.cacheDir("dir/sub2/subsub2")
|
|
assert.Equal(t, `name="" isFile=false opens=0
|
|
name="dir" isFile=false opens=0
|
|
name="dir/sub" isFile=false opens=0
|
|
name="dir/sub2" isFile=false opens=0
|
|
name="dir/sub2/subsub2" isFile=false opens=0`, itemAsString(c))
|
|
}
|
|
|
|
func TestCachePurgeOld(t *testing.T) {
|
|
r := fstest.NewRun(t)
|
|
defer r.Finalise()
|
|
|
|
ctx, cancel := context.WithCancel(context.Background())
|
|
defer cancel()
|
|
|
|
c, err := newCache(ctx, r.Fremote, &DefaultOpt)
|
|
require.NoError(t, err)
|
|
|
|
// Test funcs
|
|
var removed []string
|
|
removedDir := true
|
|
removeFile := func(name string) {
|
|
removed = append(removed, name)
|
|
}
|
|
removeDir := func(name string) bool {
|
|
if removedDir {
|
|
removed = append(removed, name+"/")
|
|
}
|
|
return removedDir
|
|
}
|
|
|
|
removed = nil
|
|
c._purgeOld(-10*time.Second, removeFile, removeDir)
|
|
assert.Equal(t, []string(nil), removed)
|
|
|
|
c.open("sub/dir/potato")
|
|
c.close("sub/dir/potato")
|
|
|
|
assert.Equal(t, `name="" isFile=false opens=0
|
|
name="sub" isFile=false opens=0
|
|
name="sub/dir" isFile=false opens=0
|
|
name="sub/dir/potato" isFile=true opens=0`, itemAsString(c))
|
|
|
|
removed = nil
|
|
removedDir = false
|
|
c._purgeOld(10*time.Second, removeFile, removeDir)
|
|
assert.Equal(t, []string(nil), removed)
|
|
|
|
assert.Equal(t, `name="" isFile=false opens=0
|
|
name="sub" isFile=false opens=0
|
|
name="sub/dir" isFile=false opens=0
|
|
name="sub/dir/potato" isFile=true opens=0`, itemAsString(c))
|
|
|
|
removed = nil
|
|
removedDir = true
|
|
c._purgeOld(-10*time.Second, removeFile, removeDir)
|
|
assert.Equal(t, []string{
|
|
"sub/dir/potato",
|
|
"sub/dir/",
|
|
"sub/",
|
|
"/",
|
|
}, removed)
|
|
|
|
assert.Equal(t, ``, itemAsString(c))
|
|
}
|