mirror of
https://github.com/tickstep/aliyunpan.git
synced 2025-01-23 22:42:15 +08:00
286 lines
8.1 KiB
Go
286 lines
8.1 KiB
Go
package syncdrive
|
||
|
||
import (
|
||
"fmt"
|
||
"github.com/tickstep/aliyunpan/internal/utils"
|
||
"io/ioutil"
|
||
"os"
|
||
"path"
|
||
"path/filepath"
|
||
"strings"
|
||
"testing"
|
||
)
|
||
|
||
func TestGet(t *testing.T) {
|
||
b := NewPanSyncDb("D:\\smb\\feny\\goprojects\\dev\\pan.db")
|
||
b.Open()
|
||
defer b.Close()
|
||
|
||
fmt.Println(b.Get("/Parallels Desktop/v17/部分电脑安装v17可能有问题，请退回v16版本.txt"))
|
||
}
|
||
|
||
func TestGetFileList(t *testing.T) {
|
||
b := NewPanSyncDb("D:\\smb\\feny\\goprojects\\dev\\pan.db")
|
||
b.Open()
|
||
defer b.Close()
|
||
|
||
fmt.Println(b.GetFileList("/Parallels Desktop/v17"))
|
||
}
|
||
|
||
func WalkAllFile(dirPath string, walkFn filepath.WalkFunc) error {
|
||
dirPath = strings.ReplaceAll(dirPath, "\\", "/")
|
||
info, err := os.Lstat(dirPath)
|
||
if err != nil {
|
||
err = walkFn(dirPath, nil, err)
|
||
} else {
|
||
err = walkAllFile(dirPath, info, walkFn)
|
||
}
|
||
return err
|
||
}
|
||
|
||
func walkAllFile(dirPath string, info os.FileInfo, walkFn filepath.WalkFunc) error {
|
||
if !info.IsDir() {
|
||
return walkFn(dirPath, info, nil)
|
||
}
|
||
|
||
files, err := ioutil.ReadDir(dirPath)
|
||
if err != nil {
|
||
return walkFn(dirPath, nil, err)
|
||
}
|
||
for _, fi := range files {
|
||
subFilePath := dirPath + "/" + fi.Name()
|
||
err = walkFn(subFilePath, fi, err)
|
||
if err != nil {
|
||
return err
|
||
}
|
||
if fi.IsDir() {
|
||
err = walkAllFile(subFilePath, fi, walkFn)
|
||
if err != nil {
|
||
return err
|
||
}
|
||
}
|
||
}
|
||
return nil
|
||
}
|
||
|
||
func TestLocalSyncDb(t *testing.T) {
|
||
b := NewLocalSyncDb("D:\\smb\\feny\\goprojects\\dev\\local.db")
|
||
b.Open()
|
||
defer b.Close()
|
||
|
||
var walkFunc filepath.WalkFunc
|
||
walkFunc = func(file string, fi os.FileInfo, err error) error {
|
||
fmt.Println(file)
|
||
fileType := "file"
|
||
if fi.IsDir() {
|
||
fileType = "folder"
|
||
}
|
||
item := &LocalFileItem{
|
||
FileName: path.Base(file),
|
||
FileSize: fi.Size(),
|
||
FileType: fileType,
|
||
CreatedAt: fi.ModTime().Format("2006-01-02 15:04:05"),
|
||
UpdatedAt: fi.ModTime().Format("2006-01-02 15:04:05"),
|
||
FileExtension: "",
|
||
Sha1Hash: "",
|
||
Path: file,
|
||
}
|
||
if _, e := b.Add(item); e != nil {
|
||
fmt.Println(e)
|
||
}
|
||
return nil
|
||
}
|
||
WalkAllFile("D:\\smb\\feny\\goprojects\\dl\\a761171495", walkFunc)
|
||
}
|
||
|
||
func TestLocalSyncDbAddFileList(t *testing.T) {
|
||
b := NewLocalSyncDb("D:\\smb\\feny\\goprojects\\dev\\tmp.db")
|
||
b.Open()
|
||
defer b.Close()
|
||
|
||
fl := LocalFileList{}
|
||
fl = append(fl, &LocalFileItem{
|
||
FileName: "aliyunpan_command_history.txt",
|
||
FileSize: 1542,
|
||
FileType: "file",
|
||
CreatedAt: "2020-12-12 12:51:12",
|
||
UpdatedAt: "2020-12-12 12:51:12",
|
||
FileExtension: "",
|
||
Sha1Hash: "",
|
||
Path: "D:\\smb\\feny\\goprojects\\dev\\aliyunpan_command_history.txt",
|
||
})
|
||
fl = append(fl, &LocalFileItem{
|
||
FileName: "f1.txt",
|
||
FileSize: 1542,
|
||
FileType: "file",
|
||
CreatedAt: "2020-12-12 12:51:12",
|
||
UpdatedAt: "2020-12-12 12:51:12",
|
||
FileExtension: "",
|
||
Sha1Hash: "",
|
||
Path: "D:\\smb\\feny\\goprojects\\dev\\f1.txt",
|
||
})
|
||
fl = append(fl, &LocalFileItem{
|
||
FileName: "f2.txt",
|
||
FileSize: 1542,
|
||
FileType: "file",
|
||
CreatedAt: "2020-12-12 12:51:12",
|
||
UpdatedAt: "2020-12-12 12:51:12",
|
||
FileExtension: "",
|
||
Sha1Hash: "",
|
||
Path: "D:\\smb\\feny\\goprojects\\dev\\f2.txt",
|
||
})
|
||
fl = append(fl, &LocalFileItem{
|
||
FileName: "f1.txt",
|
||
FileSize: 1542,
|
||
FileType: "file",
|
||
CreatedAt: "2020-12-12 12:51:12",
|
||
UpdatedAt: "2020-12-12 12:51:12",
|
||
FileExtension: "",
|
||
Sha1Hash: "",
|
||
Path: "D:\\smb\\feny\\goprojects\\dev\\fo\\f1.txt",
|
||
})
|
||
fmt.Println(b.AddFileList(fl))
|
||
}
|
||
|
||
func TestLocalGet(t *testing.T) {
|
||
b := NewLocalSyncDb("/Volumes/Downloads/dev/sync_drive/840f28af799747848c0b3155e0bdfeab/local.bolt")
|
||
b.Open()
|
||
defer b.Close()
|
||
v, _ := b.Get("/Volumes/Downloads/dev/upload/未命名文件夹/[HAIDAN.VIDEO].绣春刀.2014.mp4.torrent")
|
||
fmt.Println(utils.ObjectToJsonStr(v, true))
|
||
}
|
||
|
||
func TestLocalGetFileList(t *testing.T) {
|
||
b := NewLocalSyncDb("D:\\smb\\feny\\goprojects\\dev\\local.db")
|
||
b.Open()
|
||
defer b.Close()
|
||
|
||
fmt.Println(b.GetFileList("D:/smb/feny/goprojects/dl/a761171495"))
|
||
}
|
||
|
||
func TestSyncDbAdd(t *testing.T) {
|
||
b := NewSyncFileDb("D:\\smb\\feny\\goprojects\\dev\\sync_drive\\sync.db")
|
||
b.Open()
|
||
defer b.Close()
|
||
|
||
b.Add(&SyncFileItem{
|
||
Action: SyncFileActionDownload,
|
||
Status: SyncFileStatusCreate,
|
||
LocalFile: nil,
|
||
PanFile: &PanFileItem{
|
||
Category: "others",
|
||
Crc64Hash: "16173291050517323365",
|
||
CreatedAt: "2021-07-09 23:17:52",
|
||
DomainId: "bj29",
|
||
DriveId: "19519221",
|
||
FileExtension: "apk",
|
||
FileId: "60e868a0a005315fe2b149b4ade47b2df8bdccee",
|
||
FileName: "10000996@yunpan-release.apk",
|
||
FileSize: 61029089,
|
||
FileType: "file",
|
||
ParentFileId: "60f29e3ac420ae8c08e645db8b80b881e2a3633c",
|
||
Path: "/sync_drive/10000996@yunpan-release.apk",
|
||
Sha1Hash: "05A7597967C3C947D84564A7D55309DEA56ED985",
|
||
UpdatedAt: "2021-07-21 17:05:49",
|
||
UploadId: "rapid-9141d56d-21c3-4fcc-b41d-5df7bf47540f",
|
||
},
|
||
StatusUpdateTime: "",
|
||
})
|
||
b.Add(&SyncFileItem{
|
||
Action: SyncFileActionDownload,
|
||
Status: SyncFileStatusCreate,
|
||
LocalFile: nil,
|
||
PanFile: &PanFileItem{
|
||
Category: "others",
|
||
Crc64Hash: "16173291050517323365",
|
||
CreatedAt: "2021-07-09 23:17:52",
|
||
DomainId: "bj29",
|
||
DriveId: "19519221",
|
||
FileExtension: "apk",
|
||
FileId: "60e868a0a005315fe2b149b4ade47b2df8bdccee",
|
||
FileName: "10000996@yunpan-release-1.apk",
|
||
FileSize: 61029089,
|
||
FileType: "file",
|
||
ParentFileId: "60f29e3ac420ae8c08e645db8b80b881e2a3633c",
|
||
Path: "/sync_drive/10000996@yunpan-release-1.apk",
|
||
Sha1Hash: "05A7597967C3C947D84564A7D55309DEA56ED985",
|
||
UpdatedAt: "2021-07-21 17:05:49",
|
||
UploadId: "rapid-9141d56d-21c3-4fcc-b41d-5df7bf47540f",
|
||
},
|
||
StatusUpdateTime: "",
|
||
})
|
||
b.Add(&SyncFileItem{
|
||
Action: SyncFileActionUpload,
|
||
Status: SyncFileStatusCreate,
|
||
LocalFile: &LocalFileItem{
|
||
FileName: "f1.txt",
|
||
FileSize: 1542,
|
||
FileType: "file",
|
||
CreatedAt: "2020-12-12 12:51:12",
|
||
UpdatedAt: "2020-12-12 12:51:12",
|
||
FileExtension: ".txt",
|
||
Sha1Hash: "",
|
||
Path: "D:\\smb\\feny\\goprojects\\dev\\fo\\f1.txt",
|
||
},
|
||
PanFile: nil,
|
||
StatusUpdateTime: "",
|
||
})
|
||
//b.AddUnique(&SyncFileItem{
|
||
// Action: SyncFileActionDownload,
|
||
// Status: SyncFileStatusCreate,
|
||
// LocalFile: nil,
|
||
// PanFile: &PanFileItem{
|
||
// Category: "others",
|
||
// Crc64Hash: "16173291050517323365",
|
||
// CreatedAt: "2021-07-09 23:17:52",
|
||
// DomainId: "bj29",
|
||
// DriveId: "19519221",
|
||
// FileExtension: "apk",
|
||
// FileId: "60e868a0a005315fe2b149b4ade47b2df8bdccee",
|
||
// FileName: "10000996@yunpan-release-1.apk",
|
||
// FileSize: 61029089,
|
||
// FileType: "file",
|
||
// ParentFileId: "60f29e3ac420ae8c08e645db8b80b881e2a3633c",
|
||
// Path: "/sync_drive/10000996@yunpan-release-1.apk",
|
||
// Sha1Hash: "05A7597967C3C947D84564A7D55309DEA56ED985",
|
||
// UpdatedAt: "2021-07-21 17:05:49",
|
||
// UploadId: "rapid-9141d56d-21c3-4fcc-b41d-5df7bf47540f",
|
||
// },
|
||
// StatusUpdateTime: "",
|
||
//})
|
||
//b.AddUnique(&SyncFileItem{
|
||
// Action: SyncFileActionDownload,
|
||
// Status: SyncFileStatusCreate,
|
||
// LocalFile: nil,
|
||
// PanFile: &PanFileItem{
|
||
// Category: "others",
|
||
// Crc64Hash: "16173291050517323365",
|
||
// CreatedAt: "2021-07-09 23:17:52",
|
||
// DomainId: "bj29",
|
||
// DriveId: "19519221",
|
||
// FileExtension: "apk",
|
||
// FileId: "60e868a0a005315fe2b149b4ade47b2df8bdccee",
|
||
// FileName: "10000996@yunpan-release-2.apk",
|
||
// FileSize: 61029089,
|
||
// FileType: "file",
|
||
// ParentFileId: "60f29e3ac420ae8c08e645db8b80b881e2a3633c",
|
||
// Path: "/sync_drive/10000996@yunpan-release-2.apk",
|
||
// Sha1Hash: "05A7597967C3C947D84564A7D55309DEA56ED985",
|
||
// UpdatedAt: "2021-07-21 17:05:49",
|
||
// UploadId: "rapid-9141d56d-21c3-4fcc-b41d-5df7bf47540f",
|
||
// },
|
||
// StatusUpdateTime: "",
|
||
//})
|
||
}
|
||
|
||
func TestSyncDbClear(t *testing.T) {
|
||
b := NewSyncFileDb("D:\\smb\\feny\\goprojects\\dev\\sync.bolt")
|
||
b.Open()
|
||
defer b.Close()
|
||
|
||
files, _ := b.GetFileList(SyncFileStatusSuccess)
|
||
for _, file := range files {
|
||
b.Delete(file.Id())
|
||
}
|
||
}
|