123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395139613971398139914001401140214031404140514061407140814091410141114121413141414151416141714181419142014211422142314241425142614271428142914301431143214331434143514361437143814391440144114421443144414451446144714481449145014511452145314541455145614571458145914601461146214631464146514661467146814691470147114721473147414751476147714781479148014811482148314841485148614871488148914901491149214931494149514961497149814991500150115021503150415051506150715081509151015111512151315141515151615171518151915201521152215231524152515261527152815291530153115321533153415351536153715381539154015411542154315441545154615471548154915501551155215531554155515561557155815591560156115621563156415651566156715681569157015711572157315741575157615771578157915801581158215831584158515861587158815891590159115921593159415951596159715981599160016011602160316041605160616071608160916101611161216131614161516161617161816191620162116221623162416251626162716281629163016311632163316341635163616371638163916401641164216431644164516461647164816491650165116521653165416551656165716581659166016611662166316641665166616671668166916701671167216731674167516761677167816791680168116821683168416851686168716881689169016911692169316941695169616971698169917001701170217031704170517061707170817091710171117121713171417151716171717181719 |
- /*
- * Rufs - Remote Union File System
- *
- * Copyright 2017 Matthias Ladkau. All rights reserved.
- *
- * This Source Code Form is subject to the terms of the MIT
- * License, If a copy of the MIT License was not distributed with this
- * file, You can obtain one at https://opensource.org/licenses/MIT.
- */
- package rufs
- import (
- "bytes"
- "fmt"
- "io"
- "io/ioutil"
- "os"
- "strings"
- "testing"
- "devt.de/krotik/common/bitutil"
- "devt.de/krotik/common/errorutil"
- "devt.de/krotik/common/pools"
- "devt.de/krotik/rufs/config"
- "devt.de/krotik/rufs/node"
- )
- func TestRefresh(t *testing.T) {
- // Build up a tree from one branch
- cfg := map[string]interface{}{
- config.TreeSecret: "123",
- }
- tree, _ := NewTree(cfg, clientCert)
- branchRPC := fmt.Sprintf("%v:%v", branchConfigs["footest"][config.RPCHost], branchConfigs["footest"][config.RPCPort])
- if fp, err := tree.PingBranch("footest", branchRPC); err != nil || fp != footest.SSLFingerprint() {
- t.Error("Unexpected result:", fp, err)
- return
- }
- tree.AddBranch("footest", branchRPC, "")
- tree.AddMapping("/", "footest", true)
- paths, infos, err := tree.Dir("/", "", true, false)
- if res := DirResultToString(paths, infos); err != nil || res != `
- /
- drwxrwxrwx 4.0 KiB sub1
- -rw-rw-rw- 10 B test1
- -rw-rw-rw- 10 B test2
- /sub1
- -rw-rw-rw- 17 B test3
- `[1:] {
- t.Error("Unexpected result:", res, err)
- return
- }
- tree.Refresh()
- paths, infos, err = tree.Dir("/", "", true, false)
- if res := DirResultToString(paths, infos); err != nil || res != `
- /
- drwxrwxrwx 4.0 KiB sub1
- -rw-rw-rw- 10 B test1
- -rw-rw-rw- 10 B test2
- /sub1
- -rw-rw-rw- 17 B test3
- `[1:] {
- t.Error("Unexpected result:", res, err)
- return
- }
- // Simulate that the branch is no longer reachable
- tree.client.RemovePeer("footest")
- tree.client.RegisterPeer("footest", ":9095", "")
- if _, _, err = tree.Dir("/", "", true, false); err == nil {
- t.Error("Dir call should fail.")
- return
- }
- tree.Refresh()
- if res := tree.NotReachableBranches(); len(res) != 1 {
- t.Error("Unexpected result:", res)
- return
- }
- if res, _ := tree.ActiveBranches(); len(res) != 0 {
- t.Error("Unexpected result:", res)
- return
- }
- if res := tree.KnownBranches(); len(res) != 1 {
- t.Error("Unexpected result:", res)
- return
- }
- // Restore the branch again
- tree.client.RemovePeer("footest")
- tree.client.RegisterPeer("footest", branchRPC, "")
- tree.Refresh()
- paths, infos, err = tree.Dir("/", "", true, false)
- if res := DirResultToString(paths, infos); err != nil || res != `
- /
- drwxrwxrwx 4.0 KiB sub1
- -rw-rw-rw- 10 B test1
- -rw-rw-rw- 10 B test2
- /sub1
- -rw-rw-rw- 17 B test3
- `[1:] {
- t.Error("Unexpected result:", res, err)
- return
- }
- }
- func TestItemOpWrite(t *testing.T) {
- // Build up a tree from one branch
- cfg := map[string]interface{}{
- config.TreeSecret: "123",
- }
- tree, _ := NewTree(cfg, clientCert)
- branchRPC := fmt.Sprintf("%v:%v", branchConfigs["footest"][config.RPCHost], branchConfigs["footest"][config.RPCPort])
- if fp, err := tree.PingBranch("footest", branchRPC); err != nil || fp != footest.SSLFingerprint() {
- t.Error("Unexpected result:", fp, err)
- return
- }
- tree.AddBranch("footest", branchRPC, "")
- if res := fmt.Sprint(tree.ActiveBranches()); res != "[footest] ["+footest.SSLFingerprint()+"]" {
- t.Error("Unexpected result:", res)
- return
- }
- if err := tree.AddMapping("/", "bla", true); err == nil || err.Error() != "Unknown target node" {
- t.Error("Unexpected result:", err)
- return
- }
- tree.AddMapping("/", "footest", true)
- tree.AddMapping("/sub1", "footest", false)
- paths, infos, err := tree.Dir("/", "", true, false)
- if res := DirResultToString(paths, infos); err != nil || res != `
- /
- drwxrwxrwx 4.0 KiB sub1
- -rw-rw-rw- 10 B test1
- -rw-rw-rw- 10 B test2
- /sub1
- drwxrwxrwx 4.0 KiB sub1
- -rw-rw-rw- 10 B test1
- -rw-rw-rw- 10 B test2
- -rw-rw-rw- 17 B test3
- /sub1/sub1
- -rw-rw-rw- 17 B test3
- `[1:] {
- t.Error("Unexpected result:", res, err)
- return
- }
- // Make the byteSlicePool byte slices very small
- oldDefaultReadBufferSize := DefaultReadBufferSize
- defer func() {
- DefaultReadBufferSize = oldDefaultReadBufferSize
- byteSlicePool = pools.NewByteSlicePool(DefaultReadBufferSize)
- }()
- DefaultReadBufferSize = 3
- byteSlicePool = pools.NewByteSlicePool(DefaultReadBufferSize)
- bb := bytes.Buffer{}
- n, err := tree.WriteFile("/test99", []byte("Test"), 5)
- defer func() {
- // Make sure the written files are deleted
- footest.ItemOp("/", map[string]string{
- ItemOpAction: ItemOpActDelete,
- ItemOpName: "test99",
- })
- }()
- if err != nil || n != 4 {
- t.Error(n, err)
- return
- }
- if footest.ReadFileToBuffer("/test99", &bb); bb.String() !=
- string([]byte{0, 0, 0, 0, 0, 84, 101, 115, 116}) {
- t.Error("Unexpected result:", bb.Bytes())
- return
- }
- bb.Reset()
- n, err = footest.WriteFile("/test99", []byte("oo"), 1)
- if err != nil || n != 2 {
- t.Error(n, err)
- return
- }
- tree.WriteFile("/test99", []byte("oo"), 11)
- if footest.ReadFileToBuffer("/test99", &bb); bb.String() !=
- string([]byte{0, 111, 111, 0, 0, 84, 101, 115, 116, 0, 0, 111, 111}) {
- t.Error("Unexpected result:", bb.Bytes())
- return
- }
- bb.Reset()
- n, err = footest.WriteFile("/test98", []byte("Test"), 0)
- defer func() {
- // Make sure the written files are deleted
- footest.ItemOp("/", map[string]string{
- ItemOpAction: ItemOpActDelete,
- ItemOpName: "test98",
- })
- }()
- if err != nil || n != 4 {
- t.Error(n, err)
- return
- }
- if footest.ReadFileToBuffer("/test98", &bb); bb.String() != string("Test") {
- t.Error("Unexpected result:", bb.Bytes())
- return
- }
- bb.Reset()
- bb = *bytes.NewBuffer([]byte("Test buffer file"))
- err = footest.WriteFileFromBuffer("/test97", &bb)
- defer func() {
- // Make sure the written files are deleted
- footest.ItemOp("/", map[string]string{
- ItemOpAction: ItemOpActDelete,
- ItemOpName: "test97",
- })
- }()
- if err != nil {
- t.Error(err)
- return
- }
- if footest.ReadFileToBuffer("/test97", &bb); bb.String() != string("Test buffer file") {
- t.Error("Unexpected result:", bb.Bytes())
- return
- }
- bb.Reset()
- bb = *bytes.NewBuffer([]byte("Test buffer file2"))
- err = tree.WriteFileFromBuffer("/test96", &bb)
- defer func() {
- // Make sure the written files are deleted
- footest.ItemOp("/", map[string]string{
- ItemOpAction: ItemOpActDelete,
- ItemOpName: "test96",
- })
- }()
- if err != nil {
- t.Error(err)
- return
- }
- if footest.ReadFileToBuffer("/test96", &bb); bb.String() != string("Test buffer file2") {
- t.Error("Unexpected result:", bb.Bytes())
- return
- }
- bb.Reset()
- // Create an empty file
- tree.WriteFileFromBuffer("/test95", &bb)
- defer func() {
- // Make sure the written files are deleted
- footest.ItemOp("/", map[string]string{
- ItemOpAction: ItemOpActDelete,
- ItemOpName: "test95",
- })
- }()
- if res := dirLocal("./foo"); res != `
- sub1(dir)
- test1(10)
- test2(10)
- test95(0)
- test96(17)
- test97(16)
- test98(4)
- test99(13)
- `[1:] {
- t.Error("Unexpected result:", res)
- return
- }
- }
- func TestItemOpRead(t *testing.T) {
- // Build up a tree from two brances
- cfg := map[string]interface{}{
- config.TreeSecret: "123",
- }
- tree, _ := NewTree(cfg, clientCert)
- branchRPC := fmt.Sprintf("%v:%v", branchConfigs["footest"][config.RPCHost], branchConfigs["footest"][config.RPCPort])
- tree.AddBranch("footest", branchRPC, "")
- tree.AddMapping("/1", "footest", false)
- tree.AddMapping("/1/in/in", "footest", true)
- // We should now have the following structure:
- //
- // /1/test1
- // /1/test2
- // /1/sub1/test3
- // /1/in/in/test1
- // /1/in/in/test2
- // /1/in/in/sub1/test3
- paths, infos, err := tree.Dir("/", "", true, false)
- if res := DirResultToString(paths, infos); err != nil || res != `
- /
- drwxrwxrwx 0 B 1
- /1
- drwxrwxrwx 0 B in
- drwxrwxrwx 4.0 KiB sub1
- -rw-rw-rw- 10 B test1
- -rw-rw-rw- 10 B test2
- /1/in
- drwxrwxrwx 0 B in
- /1/in/in
- drwxrwxrwx 4.0 KiB sub1
- -rw-rw-rw- 10 B test1
- -rw-rw-rw- 10 B test2
- /1/in/in/sub1
- -rw-rw-rw- 17 B test3
- /1/sub1
- -rw-rw-rw- 17 B test3
- `[1:] {
- t.Error("Unexpected result:", res, err)
- return
- }
- paths, infos, err = tree.Dir("/1", "", false, false)
- if res := DirResultToString(paths, infos); err != nil || res != `
- /1
- drwxrwxrwx 0 B in
- drwxrwxrwx 4.0 KiB sub1
- -rw-rw-rw- 10 B test1
- -rw-rw-rw- 10 B test2
- `[1:] {
- t.Error("Unexpected result:", res, err)
- return
- }
- // Make the byteSlicePool byte slices very small
- oldDefaultReadBufferSize := DefaultReadBufferSize
- defer func() {
- DefaultReadBufferSize = oldDefaultReadBufferSize
- byteSlicePool = pools.NewByteSlicePool(DefaultReadBufferSize)
- }()
- DefaultReadBufferSize = 3
- byteSlicePool = pools.NewByteSlicePool(DefaultReadBufferSize)
- b := make([]byte, 20)
- n, err := tree.ReadFile("/1/test1", b, 0)
- if err != nil {
- t.Error(err)
- return
- }
- // Despite the small buffer size we should have read everything!
- if n != 10 || string(b[:n]) != "Test1 file" {
- t.Error("Unexpected result:", n, string(b[:n]))
- return
- }
- b = make([]byte, 3)
- n, err = tree.ReadFile("/1/test1", b, 0)
- if err != nil {
- t.Error(err)
- return
- }
- if n != 3 || string(b[:n]) != "Tes" {
- t.Error("Unexpected result:", n, string(b))
- return
- }
- // Try to read outside of root path
- _, err = tree.ReadFile("/1/../../test1", b, 0)
- if err == nil || err.Error() != "RufsError: Remote error (Requested path ../../test1 is outside of the branch)" {
- t.Error(err)
- return
- }
- // Do offset reading
- n, err = tree.ReadFile("/1/test1", b, 3)
- if err != nil {
- t.Error(err)
- return
- }
- if n != 3 || string(b[:n]) != "t1 " {
- t.Error("Unexpected result:", n, string(b))
- return
- }
- _, err = tree.ReadFile("/1/test1", b, 10)
- if err.(*node.Error).Detail != io.EOF.Error() {
- t.Error(err)
- return
- }
- byteSlicePool = pools.NewByteSlicePool(DefaultReadBufferSize)
- // Now read the same file using a reader
- bb := bytes.Buffer{}
- if err = tree.ReadFileToBuffer("/1/test2", &bb); err != nil {
- t.Error(err)
- return
- }
- if bb.String() != "Test2 file" {
- t.Error("Unexpected result: ", bb.String())
- return
- }
- bb.Reset()
- if err = footest.ReadFileToBuffer("test2", &bb); err != nil {
- t.Error(err)
- return
- }
- if bb.String() != "Test2 file" {
- t.Error("Unexpected result: ", bb.String())
- return
- }
- // Test error return
- _, err = tree.ReadFile("/1/sub1", []byte{0}, 0)
- if err == nil || err.Error() != "RufsError: Remote error (read /sub1: is a directory)" {
- t.Error("Unexpected result:", err)
- return
- }
- err = tree.ReadFileToBuffer("/1/sub1", &bb)
- if err == nil || err.Error() != "RufsError: Remote error (read /sub1: is a directory)" {
- t.Error("Unexpected result:", err)
- return
- }
- }
- func TestTreeDir(t *testing.T) {
- // Test the branch output
- paths, infos, err := footest.Dir("/", "", false, false)
- if res := DirResultToString(paths, infos); err != nil || res != `
- /
- drwxrwxrwx 4.0 KiB sub1
- -rw-rw-rw- 10 B test1
- -rw-rw-rw- 10 B test2
- `[1:] {
- t.Error("Unexpected result:", res, err)
- return
- }
- paths, infos, err = footest.Dir("/", "", true, false)
- if res := DirResultToString(paths, infos); err != nil || res != `
- /
- drwxrwxrwx 4.0 KiB sub1
- -rw-rw-rw- 10 B test1
- -rw-rw-rw- 10 B test2
- /sub1
- -rw-rw-rw- 17 B test3
- `[1:] {
- t.Error("Unexpected result:", res, err)
- return
- }
- // Build up a tree from one branch
- cfg := map[string]interface{}{
- config.TreeSecret: "123",
- }
- tree, _ := NewTree(cfg, clientCert)
- branchRPC := fmt.Sprintf("%v:%v", branchConfigs["footest"][config.RPCHost], branchConfigs["footest"][config.RPCPort])
- // Test errors for adding branches
- if err := tree.AddBranch("footest", branchRPC, "123"); err == nil ||
- !strings.HasPrefix(err.Error(), "Remote branch has an unexpected fingerprint") {
- t.Error(err)
- return
- }
- if err := tree.AddBranch("my", branchRPC, ""); err == nil ||
- err.Error() != "RufsError: Remote error (Unknown target node)" {
- t.Error(err)
- return
- }
- if err := tree.AddBranch("footest", branchRPC, ""); err != nil {
- t.Error(err)
- return
- }
- tree.AddMapping("/1", "footest", false)
- tree.AddMapping("/1/sub1", "footest", false)
- // We should now have the following structure:
- //
- // /1/test1
- // /1/test2
- // /1/sub1/test3
- // /1/sub1/test1
- // /1/sub1/test2
- // /1/sub1/sub1/test3
- paths, infos, err = tree.Dir("/", "", true, false)
- if res := DirResultToString(paths, infos); err != nil || res != `
- /
- drwxrwxrwx 0 B 1
- /1
- drwxrwxrwx 4.0 KiB sub1
- -rw-rw-rw- 10 B test1
- -rw-rw-rw- 10 B test2
- /1/sub1
- drwxrwxrwx 4.0 KiB sub1
- -rw-rw-rw- 10 B test1
- -rw-rw-rw- 10 B test2
- -rw-rw-rw- 17 B test3
- /1/sub1/sub1
- -rw-rw-rw- 17 B test3
- `[1:] {
- t.Error("Unexpected result:", res, err)
- return
- }
- // Add another mapping
- branchRPC = fmt.Sprintf("%v:%v", branchConfigs["bartest"][config.RPCHost], branchConfigs["bartest"][config.RPCPort])
- tree.AddBranch("bartest", branchRPC, "")
- tree.AddMapping("/2/sub2", "bartest", false)
- paths, infos, err = tree.Dir("/", "", true, false)
- if res := DirResultToString(paths, infos); err != nil || res != `
- /
- drwxrwxrwx 0 B 1
- drwxrwxrwx 0 B 2
- /1
- drwxrwxrwx 4.0 KiB sub1
- -rw-rw-rw- 10 B test1
- -rw-rw-rw- 10 B test2
- /1/sub1
- drwxrwxrwx 4.0 KiB sub1
- -rw-rw-rw- 10 B test1
- -rw-rw-rw- 10 B test2
- -rw-rw-rw- 17 B test3
- /1/sub1/sub1
- -rw-rw-rw- 17 B test3
- /2
- drwxrwxrwx 0 B sub2
- /2/sub2
- -rw-rw-rw- 10 B test1
- `[1:] {
- t.Error("Unexpected result:", res, err)
- return
- }
- // Now overlay a files - test1 of bartest should be overlaid by the
- // existing test1 from footest
- tree.AddMapping("/1", "bartest", false)
- paths, infos, err = tree.Dir("/", "", true, false)
- if res := DirResultToString(paths, infos); err != nil || res != `
- /
- drwxrwxrwx 0 B 1
- drwxrwxrwx 0 B 2
- /1
- drwxrwxrwx 4.0 KiB sub1
- -rw-rw-rw- 10 B test1
- -rw-rw-rw- 10 B test2
- /1/sub1
- drwxrwxrwx 4.0 KiB sub1
- -rw-rw-rw- 10 B test1
- -rw-rw-rw- 10 B test2
- -rw-rw-rw- 17 B test3
- /1/sub1/sub1
- -rw-rw-rw- 17 B test3
- /2
- drwxrwxrwx 0 B sub2
- /2/sub2
- -rw-rw-rw- 10 B test1
- `[1:] {
- t.Error("Unexpected result:", res, err)
- return
- }
- paths, infos, err = tree.Dir("/1", "", false, false)
- if res := DirResultToString(paths, infos); err != nil || res != `
- /1
- drwxrwxrwx 4.0 KiB sub1
- -rw-rw-rw- 10 B test1
- -rw-rw-rw- 10 B test2
- `[1:] {
- t.Error("Unexpected result:", res, err)
- return
- }
- paths, infos, err = tree.Dir("/1/sub1", "", false, false)
- if res := DirResultToString(paths, infos); err != nil || res != `
- /1/sub1
- drwxrwxrwx 4.0 KiB sub1
- -rw-rw-rw- 10 B test1
- -rw-rw-rw- 10 B test2
- -rw-rw-rw- 17 B test3
- `[1:] {
- t.Error("Unexpected result:", res, err)
- return
- }
- }
- func TestItemOpRename(t *testing.T) {
- // Build up a tree from one branch
- cfg := map[string]interface{}{
- config.TreeSecret: "123",
- }
- tree, _ := NewTree(cfg, clientCert)
- branchRPC := fmt.Sprintf("%v:%v", branchConfigs["footest"][config.RPCHost], branchConfigs["footest"][config.RPCPort])
- tree.AddBranch("footest", branchRPC, "")
- tree.AddMapping("/1", "footest", false)
- tree.AddMapping("/1/sub1", "footest", true)
- // We should now have the following structure:
- //
- // /1/test1
- // /1/test2
- // /1/sub1/test3
- // /1/sub1/test1
- // /1/sub1/test2
- // /1/sub1/sub1/test3
- if res := fmt.Sprint(tree); res != `
- /:
- 1/: footest(r)
- sub1/: footest(w)
- `[1:] {
- t.Error("Unexpected result:", res)
- return
- }
- paths, infos, err := footest.Dir("/", "", true, false)
- if res := DirResultToString(paths, infos); err != nil || res != `
- /
- drwxrwxrwx 4.0 KiB sub1
- -rw-rw-rw- 10 B test1
- -rw-rw-rw- 10 B test2
- /sub1
- -rw-rw-rw- 17 B test3
- `[1:] {
- t.Error("Unexpected result:", res, err)
- return
- }
- paths, infos, err = tree.Dir("/", "", true, false)
- if res := DirResultToString(paths, infos); err != nil || res != `
- /
- drwxrwxrwx 0 B 1
- /1
- drwxrwxrwx 4.0 KiB sub1
- -rw-rw-rw- 10 B test1
- -rw-rw-rw- 10 B test2
- /1/sub1
- drwxrwxrwx 4.0 KiB sub1
- -rw-rw-rw- 10 B test1
- -rw-rw-rw- 10 B test2
- -rw-rw-rw- 17 B test3
- /1/sub1/sub1
- -rw-rw-rw- 17 B test3
- `[1:] {
- t.Error("Unexpected result:", res, err)
- return
- }
- // Rename test1 to test111
- tree.ItemOp("/1/sub1", map[string]string{
- ItemOpAction: ItemOpActRename,
- ItemOpName: "test1",
- ItemOpNewName: "test111",
- })
- tree.ItemOp("/1/sub1", map[string]string{
- ItemOpAction: ItemOpActRename,
- ItemOpName: "sub1",
- ItemOpNewName: "sub99",
- })
- paths, infos, err = footest.Dir("/", "", true, false)
- if res := DirResultToString(paths, infos); err != nil || res != `
- /
- drwxrwxrwx 4.0 KiB sub99
- -rw-rw-rw- 10 B test111
- -rw-rw-rw- 10 B test2
- /sub99
- -rw-rw-rw- 17 B test3
- `[1:] {
- t.Error("Unexpected result:", res, err)
- return
- }
- paths, infos, err = tree.Dir("/", "", true, false)
- if res := DirResultToString(paths, infos); err != nil || res != `
- /
- drwxrwxrwx 0 B 1
- /1
- drwxrwxrwx 0 B sub1
- drwxrwxrwx 4.0 KiB sub99
- -rw-rw-rw- 10 B test111
- -rw-rw-rw- 10 B test2
- /1/sub1
- drwxrwxrwx 4.0 KiB sub99
- -rw-rw-rw- 10 B test111
- -rw-rw-rw- 10 B test2
- /1/sub1/sub99
- -rw-rw-rw- 17 B test3
- /1/sub99
- -rw-rw-rw- 17 B test3
- `[1:] {
- t.Error("Unexpected result:", res, err)
- return
- }
- // Rename test111 back to test1
- ok, err := tree.ItemOp("/1/sub1", map[string]string{
- ItemOpAction: ItemOpActRename,
- ItemOpName: "test111",
- ItemOpNewName: "test1",
- })
- if !ok || err != nil {
- t.Error(ok, err)
- return
- }
- ok, err = tree.ItemOp("/1/sub1", map[string]string{
- ItemOpAction: ItemOpActRename,
- ItemOpName: "sub99",
- ItemOpNewName: "sub1",
- })
- if !ok || err != nil {
- t.Error(ok, err)
- return
- }
- // Rename non existing file
- ok, err = tree.ItemOp("/1/sub1", map[string]string{
- ItemOpAction: ItemOpActRename,
- ItemOpName: "foobar",
- ItemOpNewName: "test99",
- })
- if ok || err == nil ||
- (err.Error() != "RufsError: Remote error (rename /foobar /test99: no such file or directory)" &&
- err.Error() != "RufsError: Remote error (rename \\foobar \\test99: The system cannot find the file specified.)") {
- t.Error(ok, err)
- return
- }
- paths, infos, err = footest.Dir("/", "", true, false)
- if res := DirResultToString(paths, infos); err != nil || res != `
- /
- drwxrwxrwx 4.0 KiB sub1
- -rw-rw-rw- 10 B test1
- -rw-rw-rw- 10 B test2
- /sub1
- -rw-rw-rw- 17 B test3
- `[1:] {
- t.Error("Unexpected result:", res, err)
- return
- }
- paths, infos, err = tree.Dir("/", "", true, false)
- if res := DirResultToString(paths, infos); err != nil || res != `
- /
- drwxrwxrwx 0 B 1
- /1
- drwxrwxrwx 4.0 KiB sub1
- -rw-rw-rw- 10 B test1
- -rw-rw-rw- 10 B test2
- /1/sub1
- drwxrwxrwx 4.0 KiB sub1
- -rw-rw-rw- 10 B test1
- -rw-rw-rw- 10 B test2
- -rw-rw-rw- 17 B test3
- /1/sub1/sub1
- -rw-rw-rw- 17 B test3
- `[1:] {
- t.Error("Unexpected result:", res, err)
- return
- }
- }
- func TestItemOpDelete(t *testing.T) {
- // Build up a tree from one branch
- cfg := map[string]interface{}{
- config.TreeSecret: "123",
- }
- tree, _ := NewTree(cfg, clientCert)
- branchRPC := fmt.Sprintf("%v:%v", branchConfigs["footest"][config.RPCHost], branchConfigs["footest"][config.RPCPort])
- tree.AddBranch("footest", branchRPC, "")
- branchRPC = fmt.Sprintf("%v:%v", branchConfigs["bartest"][config.RPCHost], branchConfigs["bartest"][config.RPCPort])
- tree.AddBranch("bartest", branchRPC, "")
- tree.AddMapping("/1", "footest", false)
- tree.AddMapping("/1/sub1", "footest", true)
- tree.AddMapping("///1///sub1", "bartest", false)
- conf := tree.Config()
- if conf != `{
- "branches": [
- {
- "branch": "footest",
- "fingerprint": "`+footest.SSLFingerprint()+`",
- "rpc": "localhost:9021"
- },
- {
- "branch": "bartest",
- "fingerprint": "`+bartest.SSLFingerprint()+`",
- "rpc": "localhost:9022"
- }
- ],
- "tree": [
- {
- "branch": "footest",
- "path": "/1",
- "writeable": false
- },
- {
- "branch": "footest",
- "path": "/1/sub1",
- "writeable": true
- },
- {
- "branch": "bartest",
- "path": "///1///sub1",
- "writeable": false
- }
- ]
- }` {
- t.Error("Unexpected config:", conf)
- return
- }
- if err := tree.SetMapping(conf); err != nil {
- t.Error(err)
- return
- }
- // We should now have the following structure:
- //
- // /1/test1
- // /1/test2
- // /1/sub1/test3
- // /1/sub1/test1
- // /1/sub1/test2
- // /1/sub1/sub1/test3
- if res := fmt.Sprint(tree); res != `
- /:
- 1/: footest(r)
- sub1/: footest(w), bartest(r)
- `[1:] {
- t.Error("Unexpected result:", res)
- return
- }
- // Create a new file
- ioutil.WriteFile("foo/test_to_delete", []byte("Test1 file"), 0770)
- paths, infos, err := tree.Dir("/", "", true, true)
- if res := DirResultToString(paths, infos); err != nil || res != `
- /
- drwxrwxrwx 0 B 1
- /1
- drwxrwxrwx 4.0 KiB sub1
- -rw-rw-rw- 10 B test1 [73b8af47]
- -rw-rw-rw- 10 B test2 [b0c1fadd]
- -rw-rw-rw- 10 B test_to_delete [73b8af47]
- /1/sub1
- drwxrwxrwx 4.0 KiB sub1
- -rw-rw-rw- 10 B test1 [73b8af47]
- -rw-rw-rw- 10 B test2 [b0c1fadd]
- -rw-rw-rw- 17 B test3 [f89782b1]
- -rw-rw-rw- 10 B test_to_delete [73b8af47]
- /1/sub1/sub1
- -rw-rw-rw- 17 B test3 [f89782b1]
- `[1:] {
- t.Error("Unexpected result:", res, err)
- return
- }
- // Delete file
- ok, err := tree.ItemOp("/1", map[string]string{
- ItemOpAction: ItemOpActDelete,
- ItemOpName: "test_to_delete",
- })
- if ok || err == nil || err.Error() != "All applicable branches for the requested path were mounted as not writable" {
- t.Error(ok, err)
- return
- }
- _, err = tree.WriteFile("/1/bla", []byte("test"), 0)
- if err == nil || err.Error() != "All applicable branches for the requested path were mounted as not writable" {
- t.Error(ok, err)
- return
- }
- ok, err = tree.ItemOp("/1/sub1", map[string]string{
- ItemOpAction: ItemOpActDelete,
- ItemOpName: "test_to_delete",
- })
- if !ok || err != nil {
- t.Error(ok, err)
- return
- }
- paths, infos, err = tree.Dir("/", "", true, true)
- if res := DirResultToString(paths, infos); err != nil || res != `
- /
- drwxrwxrwx 0 B 1
- /1
- drwxrwxrwx 4.0 KiB sub1
- -rw-rw-rw- 10 B test1 [73b8af47]
- -rw-rw-rw- 10 B test2 [b0c1fadd]
- /1/sub1
- drwxrwxrwx 4.0 KiB sub1
- -rw-rw-rw- 10 B test1 [73b8af47]
- -rw-rw-rw- 10 B test2 [b0c1fadd]
- -rw-rw-rw- 17 B test3 [f89782b1]
- /1/sub1/sub1
- -rw-rw-rw- 17 B test3 [f89782b1]
- `[1:] {
- t.Error("Unexpected result:", res, err)
- return
- }
- tree.Reset(false) // Just reset mappings not known branches
- tree.AddMapping("/1", "footest", false)
- tree.AddMapping("/1/sub1", "footest", true)
- tree.AddMapping("/1/sub1", "bartest", true)
- // Create new files to delete
- ioutil.WriteFile("foo/test_to_delete1", []byte("Test1 file"), 0770)
- ioutil.WriteFile("bar/test_to_delete2", []byte("Test1 file"), 0770)
- // Delete non existing file
- ok, err = tree.ItemOp("/1/sub1", map[string]string{
- ItemOpAction: ItemOpActDelete,
- ItemOpName: "test_to_delete3",
- })
- if err == nil || err.Error() != "RufsError: Remote error (file does not exist)" || ok {
- t.Error("Unexpected result:", ok, err)
- return
- }
- ok, err = tree.ItemOp("/1/sub1", map[string]string{
- ItemOpAction: ItemOpActDelete,
- ItemOpName: "",
- })
- if err == nil || err.Error() != "RufsError: Remote error (This operation requires a specific file or directory)" || ok {
- t.Error("Unexpected result:", ok, err)
- return
- }
- // Do a wildcard delete
- ok, err = tree.ItemOp("/1/sub1", map[string]string{
- ItemOpAction: ItemOpActDelete,
- ItemOpName: "test_to_delete*",
- })
- if !ok || err != nil {
- t.Error(ok, err)
- return
- }
- paths, infos, err = tree.Dir("/", "", true, true)
- if res := DirResultToString(paths, infos); err != nil || res != `
- /
- drwxrwxrwx 0 B 1
- /1
- drwxrwxrwx 4.0 KiB sub1
- -rw-rw-rw- 10 B test1 [73b8af47]
- -rw-rw-rw- 10 B test2 [b0c1fadd]
- /1/sub1
- drwxrwxrwx 4.0 KiB sub1
- -rw-rw-rw- 10 B test1 [73b8af47]
- -rw-rw-rw- 10 B test2 [b0c1fadd]
- -rw-rw-rw- 17 B test3 [f89782b1]
- /1/sub1/sub1
- -rw-rw-rw- 17 B test3 [f89782b1]
- `[1:] {
- t.Error("Unexpected result:", res, err)
- return
- }
- tree.Reset(true)
- paths, infos, err = tree.Dir("/", "", true, true)
- if res := DirResultToString(paths, infos); err != nil || res != `
- /
- `[1:] {
- t.Error("Unexpected result:", res, err)
- return
- }
- }
- func TestItemOpMkdir(t *testing.T) {
- // Build up a tree from one branch
- cfg := map[string]interface{}{
- config.TreeSecret: "123",
- }
- tree, _ := NewTree(cfg, clientCert)
- branchRPC := fmt.Sprintf("%v:%v", branchConfigs["footest"][config.RPCHost], branchConfigs["footest"][config.RPCPort])
- tree.AddBranch("footest", branchRPC, "")
- tree.AddMapping("/1", "footest", true)
- paths, infos, err := tree.Dir("/", "", true, true)
- if res := DirResultToString(paths, infos); err != nil || res != `
- /
- drwxrwxrwx 0 B 1
- /1
- drwxrwxrwx 4.0 KiB sub1
- -rw-rw-rw- 10 B test1 [73b8af47]
- -rw-rw-rw- 10 B test2 [b0c1fadd]
- /1/sub1
- -rw-rw-rw- 17 B test3 [f89782b1]
- `[1:] {
- t.Error("Unexpected result:", res, err)
- return
- }
- ok, err := tree.ItemOp("/1/sub1", map[string]string{
- ItemOpAction: ItemOpActMkDir,
- ItemOpName: "aaa/bbb",
- })
- if !ok || err != nil {
- t.Error(ok, err)
- return
- }
- paths, infos, err = tree.Dir("/", "", true, true)
- if res := DirResultToString(paths, infos); err != nil || res != `
- /
- drwxrwxrwx 0 B 1
- /1
- drwxrwxrwx 4.0 KiB sub1
- -rw-rw-rw- 10 B test1 [73b8af47]
- -rw-rw-rw- 10 B test2 [b0c1fadd]
- /1/sub1
- drwxrwxrwx 4.0 KiB bbb
- -rw-rw-rw- 17 B test3 [f89782b1]
- /1/sub1/bbb
- `[1:] {
- t.Error("Unexpected result:", res, err)
- return
- }
- errorutil.AssertOk(os.RemoveAll("./foo/sub1/bbb"))
- }
- func TestRelPath(t *testing.T) {
- if res := relPath("/1/", ""); res != "/1" {
- t.Error("Unexpected result:", res)
- return
- }
- if res := relPath("/1/", "/1/"); res != "/" {
- t.Error("Unexpected result:", res)
- return
- }
- if res := relPath("/1/test", "/1/"); res != "/test" {
- t.Error("Unexpected result:", res)
- return
- }
- if res := relPath("/1/test/", "/1"); res != "/test" {
- t.Error("Unexpected result:", res)
- return
- }
- }
- func TestSync(t *testing.T) {
- var buf bytes.Buffer
- // Build up a tree from one branch
- cfg := map[string]interface{}{
- config.TreeSecret: "123",
- }
- tree, _ := NewTree(cfg, clientCert)
- fooRPC := fmt.Sprintf("%v:%v", branchConfigs["footest"][config.RPCHost], branchConfigs["footest"][config.RPCPort])
- barRPC := fmt.Sprintf("%v:%v", branchConfigs["bartest"][config.RPCHost], branchConfigs["bartest"][config.RPCPort])
- errorutil.AssertOk(tree.AddBranch("footest", fooRPC, ""))
- errorutil.AssertOk(tree.AddBranch("bartest", barRPC, ""))
- errorutil.AssertOk(tree.AddMapping("/2", "footest", false))
- errorutil.AssertOk(tree.AddMapping("/3", "bartest", true))
- paths, infos, err := tree.Dir("/", "", true, true)
- if res := DirResultToString(paths, infos); err != nil || res != `
- /
- drwxrwxrwx 0 B 2
- drwxrwxrwx 0 B 3
- /2
- drwxrwxrwx 4.0 KiB sub1
- -rw-rw-rw- 10 B test1 [73b8af47]
- -rw-rw-rw- 10 B test2 [b0c1fadd]
- /2/sub1
- -rw-rw-rw- 17 B test3 [f89782b1]
- /3
- -rw-rw-rw- 10 B test1 [5b62da0f]
- `[1:] {
- t.Error("Unexpected result:", res, err)
- return
- }
- ioutil.WriteFile("./bar/test1", []byte("Test1 file"), 0770)
- ioutil.WriteFile("./bar/test2", []byte("Testx file"), 0770)
- ioutil.WriteFile("./bar/test5", []byte("Testx file"), 0770)
- os.Mkdir("./bar/sub2", 0755)
- ioutil.WriteFile("./bar/sub2/test2", []byte("Testx file"), 0770)
- ioutil.WriteFile("./foo/test3", []byte("Testx file"), 0770)
- ioutil.WriteFile("./foo/testempty", nil, 0770)
- paths, infos, err = tree.Dir("/", "", true, true)
- if res := DirResultToString(paths, infos); err != nil || res != `
- /
- drwxrwxrwx 0 B 2
- drwxrwxrwx 0 B 3
- /2
- drwxrwxrwx 4.0 KiB sub1
- -rw-rw-rw- 10 B test1 [73b8af47]
- -rw-rw-rw- 10 B test2 [b0c1fadd]
- -rw-rw-rw- 10 B test3 [91767c28]
- -rw-rw-rw- 0 B testempty
- /2/sub1
- -rw-rw-rw- 17 B test3 [f89782b1]
- /3
- drwxrwxrwx 4.0 KiB sub2
- -rw-rw-rw- 10 B test1 [73b8af47]
- -rw-rw-rw- 10 B test2 [91767c28]
- -rw-rw-rw- 10 B test5 [91767c28]
- /3/sub2
- -rw-rw-rw- 10 B test2 [91767c28]
- `[1:] {
- t.Error("Unexpected result:", res, err)
- return
- }
- updFunc := func(op, srcFile, dstFile string, writtenBytes, totalBytes, currentFile, totalFiles int64) {
- buf.WriteString(fmt.Sprintf("%v %v -> %v", op, srcFile, dstFile))
- if writtenBytes > 0 {
- buf.WriteString(" " + bitutil.ByteSizeString(writtenBytes, false))
- buf.WriteString("/" + bitutil.ByteSizeString(totalBytes, false))
- } else if writtenBytes == -1 {
- buf.WriteString(" finished")
- }
- buf.WriteString("\n")
- }
- if err := tree.Sync("/2", "/3", false, updFunc); err != nil {
- t.Error(buf.String(), err)
- return
- }
- // Check log
- if buf.String() != `
- Create directory -> /3/sub1
- Copy file /2/test2 -> /3/test2 10 B/10 B
- Copy file /2/test2 -> /3/test2 finished
- Copy file /2/test3 -> /3/test3 10 B/10 B
- Copy file /2/test3 -> /3/test3 finished
- Copy file /2/testempty -> /3/testempty
- Remove directory -> /3/sub2
- Remove file -> /3/test5
- `[1:] {
- t.Error("Unexpected log:", buf.String())
- return
- }
- // Check new directory structure - Almost but not quite - sub1 in /3
- // is empty as the call was not recursive
- paths, infos, err = tree.Dir("/", "", true, true)
- if res := DirResultToString(paths, infos); err != nil || res != `
- /
- drwxrwxrwx 0 B 2
- drwxrwxrwx 0 B 3
- /2
- drwxrwxrwx 4.0 KiB sub1
- -rw-rw-rw- 10 B test1 [73b8af47]
- -rw-rw-rw- 10 B test2 [b0c1fadd]
- -rw-rw-rw- 10 B test3 [91767c28]
- -rw-rw-rw- 0 B testempty
- /2/sub1
- -rw-rw-rw- 17 B test3 [f89782b1]
- /3
- drwxrwxrwx 4.0 KiB sub1
- -rw-rw-rw- 10 B test1 [73b8af47]
- -rw-rw-rw- 10 B test2 [b0c1fadd]
- -rw-rw-rw- 10 B test3 [91767c28]
- -rw-rw-rw- 0 B testempty
- /3/sub1
- `[1:] {
- t.Error("Unexpected result:", res, err)
- return
- }
- // Now reset the whole thing and do it recursive this time!
- os.RemoveAll("./bar/test3")
- os.RemoveAll("./bar/sub1")
- os.Mkdir("./bar/sub2", 0755)
- ioutil.WriteFile("./bar/sub2/test2", []byte("Testx file"), 0770)
- ioutil.WriteFile("./bar/test2", []byte("Testx file"), 0770)
- buf.Reset()
- paths, infos, err = tree.Dir("/", "", true, true)
- if res := DirResultToString(paths, infos); err != nil || res != `
- /
- drwxrwxrwx 0 B 2
- drwxrwxrwx 0 B 3
- /2
- drwxrwxrwx 4.0 KiB sub1
- -rw-rw-rw- 10 B test1 [73b8af47]
- -rw-rw-rw- 10 B test2 [b0c1fadd]
- -rw-rw-rw- 10 B test3 [91767c28]
- -rw-rw-rw- 0 B testempty
- /2/sub1
- -rw-rw-rw- 17 B test3 [f89782b1]
- /3
- drwxrwxrwx 4.0 KiB sub2
- -rw-rw-rw- 10 B test1 [73b8af47]
- -rw-rw-rw- 10 B test2 [91767c28]
- -rw-rw-rw- 0 B testempty
- /3/sub2
- -rw-rw-rw- 10 B test2 [91767c28]
- `[1:] {
- t.Error("Unexpected result:", res, err)
- return
- }
- if err := tree.Sync("/2", "/3", true, updFunc); err != nil {
- t.Error(buf.String(), err)
- return
- }
- if buf.String() != `
- Create directory -> /3/sub1
- Copy file /2/test2 -> /3/test2 10 B/10 B
- Copy file /2/test2 -> /3/test2 finished
- Copy file /2/test3 -> /3/test3 10 B/10 B
- Copy file /2/test3 -> /3/test3 finished
- Remove directory -> /3/sub2
- Copy file /2/sub1/test3 -> /3/sub1/test3 17 B/17 B
- Copy file /2/sub1/test3 -> /3/sub1/test3 finished
- `[1:] {
- t.Error("Unexpected log:", buf.String())
- return
- }
- // Check new directory structure - Now sub1 in /3
- // is not empty as the call was recursive
- paths, infos, err = tree.Dir("/", "", true, true)
- if res := DirResultToString(paths, infos); err != nil || res != `
- /
- drwxrwxrwx 0 B 2
- drwxrwxrwx 0 B 3
- /2
- drwxrwxrwx 4.0 KiB sub1
- -rw-rw-rw- 10 B test1 [73b8af47]
- -rw-rw-rw- 10 B test2 [b0c1fadd]
- -rw-rw-rw- 10 B test3 [91767c28]
- -rw-rw-rw- 0 B testempty
- /2/sub1
- -rw-rw-rw- 17 B test3 [f89782b1]
- /3
- drwxrwxrwx 4.0 KiB sub1
- -rw-rw-rw- 10 B test1 [73b8af47]
- -rw-rw-rw- 10 B test2 [b0c1fadd]
- -rw-rw-rw- 10 B test3 [91767c28]
- -rw-rw-rw- 0 B testempty
- /3/sub1
- -rw-rw-rw- 17 B test3 [f89782b1]
- `[1:] {
- t.Error("Unexpected result:", res, err)
- return
- }
- // Reset everything
- ioutil.WriteFile("bar/test1", []byte("Test3 file"), 0770)
- os.RemoveAll("./bar/test2")
- os.RemoveAll("./bar/test3")
- os.RemoveAll("./bar/sub1")
- os.RemoveAll("./bar/testempty")
- os.RemoveAll("./foo/test3")
- os.RemoveAll("./foo/testempty")
- // Test error reporting
- buf.Reset()
- if err := tree.Sync("/3", "/2", true, updFunc); err == nil || err.Error() != "All applicable branches for the requested path were mounted as not writable" {
- t.Error(buf.String(), err)
- return
- }
- // Make sure we "bomb out" after the first write attempt
- if buf.String() != `
- Copy file /3/test1 -> /2/test1
- `[1:] {
- t.Error("Unexpected log:", buf.String())
- return
- }
- // Make sure everything is in the state it should be
- if err := tree.CopyFile("/bla", "/3/xxx", nil); err == nil ||
- err.Error() != "RufsError: Remote error (file does not exist)" {
- t.Error("Unexpected result:", err)
- return
- }
- paths, infos, err = tree.Dir("/", "", true, true)
- if res := DirResultToString(paths, infos); err != nil || res != `
- /
- drwxrwxrwx 0 B 2
- drwxrwxrwx 0 B 3
- /2
- drwxrwxrwx 4.0 KiB sub1
- -rw-rw-rw- 10 B test1 [73b8af47]
- -rw-rw-rw- 10 B test2 [b0c1fadd]
- /2/sub1
- -rw-rw-rw- 17 B test3 [f89782b1]
- /3
- -rw-rw-rw- 10 B test1 [5b62da0f]
- `[1:] {
- t.Error("Unexpected result:", res, err)
- return
- }
- }
- func TestDirPattern(t *testing.T) {
- // Build up a tree from one branch
- cfg := map[string]interface{}{
- config.TreeSecret: "123",
- }
- tree, _ := NewTree(cfg, clientCert)
- fooRPC := fmt.Sprintf("%v:%v", branchConfigs["footest"][config.RPCHost], branchConfigs["footest"][config.RPCPort])
- barRPC := fmt.Sprintf("%v:%v", branchConfigs["bartest"][config.RPCHost], branchConfigs["bartest"][config.RPCPort])
- errorutil.AssertOk(tree.AddBranch("footest", fooRPC, ""))
- errorutil.AssertOk(tree.AddBranch("bartest", barRPC, ""))
- errorutil.AssertOk(tree.AddMapping("/2", "footest", false))
- errorutil.AssertOk(tree.AddMapping("/3", "bartest", true))
- paths, infos, err := tree.Dir("/", "", true, true)
- if res := DirResultToString(paths, infos); err != nil || res != `
- /
- drwxrwxrwx 0 B 2
- drwxrwxrwx 0 B 3
- /2
- drwxrwxrwx 4.0 KiB sub1
- -rw-rw-rw- 10 B test1 [73b8af47]
- -rw-rw-rw- 10 B test2 [b0c1fadd]
- /2/sub1
- -rw-rw-rw- 17 B test3 [f89782b1]
- /3
- -rw-rw-rw- 10 B test1 [5b62da0f]
- `[1:] {
- t.Error("Unexpected result:", res, err)
- return
- }
- paths, infos, err = tree.Dir("/", "2", true, true)
- if res := DirResultToString(paths, infos); err != nil || res != `
- /
- drwxrwxrwx 0 B 2
- /2
- -rw-rw-rw- 10 B test2 [b0c1fadd]
- /2/sub1
- /3
- `[1:] {
- t.Error("Unexpected result:", res, err)
- return
- }
- paths, infos, err = tree.Dir("/", "1", true, true)
- if res := DirResultToString(paths, infos); err != nil || res != `
- /
- /2
- drwxrwxrwx 4.0 KiB sub1
- -rw-rw-rw- 10 B test1 [73b8af47]
- /2/sub1
- /3
- -rw-rw-rw- 10 B test1 [5b62da0f]
- `[1:] {
- t.Error("Unexpected result:", res, err)
- return
- }
- // Test error case
- if _, _, err := tree.Dir("/", "(", true, true); err == nil || err.Error() != "error parsing regexp: missing closing ): `(`" {
- t.Error("Unexpected result:", err)
- return
- }
- }
- func TestCopy(t *testing.T) {
- var buf bytes.Buffer
- // Build up a tree from one branch
- cfg := map[string]interface{}{
- config.TreeSecret: "123",
- }
- tree, _ := NewTree(cfg, clientCert)
- fooRPC := fmt.Sprintf("%v:%v", branchConfigs["footest"][config.RPCHost], branchConfigs["footest"][config.RPCPort])
- barRPC := fmt.Sprintf("%v:%v", branchConfigs["bartest"][config.RPCHost], branchConfigs["bartest"][config.RPCPort])
- errorutil.AssertOk(tree.AddBranch("footest", fooRPC, ""))
- errorutil.AssertOk(tree.AddBranch("bartest", barRPC, ""))
- errorutil.AssertOk(tree.AddMapping("/2", "footest", false))
- errorutil.AssertOk(tree.AddMapping("/3", "bartest", true))
- paths, infos, err := tree.Dir("/", "", true, true)
- if res := DirResultToString(paths, infos); err != nil || res != `
- /
- drwxrwxrwx 0 B 2
- drwxrwxrwx 0 B 3
- /2
- drwxrwxrwx 4.0 KiB sub1
- -rw-rw-rw- 10 B test1 [73b8af47]
- -rw-rw-rw- 10 B test2 [b0c1fadd]
- /2/sub1
- -rw-rw-rw- 17 B test3 [f89782b1]
- /3
- -rw-rw-rw- 10 B test1 [5b62da0f]
- `[1:] {
- t.Error("Unexpected result:", res, err)
- return
- }
- // Test stat
- if fi, err := tree.Stat("/2/test1"); err != nil || fi.Name() != "test1" {
- t.Error("Unexpected result:", fi, err)
- return
- }
- if fi, err := tree.Stat("/2/test0"); err == nil || err.Error() != "RufsError: Remote error (file does not exist)" {
- t.Error("Unexpected result:", fi, err)
- return
- }
- updFunc := func(file string, writtenBytes, totalBytes, currentFile, totalFiles int64) {
- buf.WriteString(fmt.Sprintf("%v %v/%v (%v of %v)\n", file, writtenBytes, totalBytes, currentFile, totalFiles))
- }
- if err := tree.Copy([]string{"/2", "/2/sub1", "/2/test1"}, "/3/4/5/6", updFunc); err != nil {
- t.Error(buf.String(), err)
- return
- }
- // Check the result
- paths, infos, err = tree.Dir("/", "", true, true)
- if res := DirResultToString(paths, infos); err != nil || res != `
- /
- drwxrwxrwx 0 B 2
- drwxrwxrwx 0 B 3
- /2
- drwxrwxrwx 4.0 KiB sub1
- -rw-rw-rw- 10 B test1 [73b8af47]
- -rw-rw-rw- 10 B test2 [b0c1fadd]
- /2/sub1
- -rw-rw-rw- 17 B test3 [f89782b1]
- /3
- drwxrwxrwx 4.0 KiB 4
- -rw-rw-rw- 10 B test1 [5b62da0f]
- /3/4
- drwxrwxrwx 4.0 KiB 5
- /3/4/5
- drwxrwxrwx 4.0 KiB 6
- /3/4/5/6
- drwxrwxrwx 4.0 KiB 2
- drwxrwxrwx 4.0 KiB sub1
- -rw-rw-rw- 10 B test1 [73b8af47]
- /3/4/5/6/2
- drwxrwxrwx 4.0 KiB sub1
- -rw-rw-rw- 10 B test1 [73b8af47]
- -rw-rw-rw- 10 B test2 [b0c1fadd]
- /3/4/5/6/2/sub1
- -rw-rw-rw- 17 B test3 [f89782b1]
- /3/4/5/6/sub1
- -rw-rw-rw- 17 B test3 [f89782b1]
- `[1:] {
- t.Error("Unexpected result:", res, err)
- return
- }
- // Test error case
- if err := tree.Copy([]string{"/2", "/2/test5"}, "/", updFunc); err == nil ||
- err.Error() != "Cannot stat /2/test5: RufsError: Remote error (file does not exist)" {
- t.Error(buf.String(), err)
- return
- }
- if err := tree.Copy([]string{"/2", "/2/test1"}, "/", updFunc); err == nil ||
- err.Error() != "Cannot copy /2/test1 to /: All applicable branches for the requested path were mounted as not writable" {
- t.Error(buf.String(), err)
- return
- }
- }
|