+++ /dev/null
-package shell
-
-import (
- "bytes"
- "context"
- "crypto/md5"
- "fmt"
- "io"
- "math/rand"
- "sort"
- "strings"
- "testing"
- "time"
-
- "github.com/cheekybits/is"
- "github.com/ipfs/go-ipfs-api/options"
-)
-
-const (
- examplesHash = "QmS4ustL54uo8FzR9455qaxZwuMiUhyvMcX9Ba8nUH4uVv"
- shellUrl = "localhost:5001"
-)
-
-func TestAdd(t *testing.T) {
- is := is.New(t)
- s := NewShell(shellUrl)
-
- mhash, err := s.Add(bytes.NewBufferString("Hello IPFS Shell tests"))
- is.Nil(err)
- is.Equal(mhash, "QmUfZ9rAdhV5ioBzXKdUTh2ZNsz9bzbkaLVyQ8uc8pj21F")
-}
-
-func TestRedirect(t *testing.T) {
- is := is.New(t)
- s := NewShell(shellUrl)
-
- err := s.
- Request("/version").
- Exec(context.Background(), nil)
- is.NotNil(err)
- is.True(strings.Contains(err.Error(), "unexpected redirect"))
-}
-
-func TestAddWithCat(t *testing.T) {
- is := is.New(t)
- s := NewShell(shellUrl)
- s.SetTimeout(1 * time.Second)
-
- rand := randString(32)
-
- mhash, err := s.Add(bytes.NewBufferString(rand))
- is.Nil(err)
-
- reader, err := s.Cat(mhash)
- is.Nil(err)
-
- buf := new(bytes.Buffer)
- buf.ReadFrom(reader)
- catRand := buf.String()
-
- is.Equal(rand, catRand)
-}
-
-func TestAddOnlyHash(t *testing.T) {
- is := is.New(t)
- s := NewShell(shellUrl)
- s.SetTimeout(1 * time.Second)
-
- rand := randString(32)
-
- mhash, err := s.Add(bytes.NewBufferString(rand), OnlyHash(true))
- is.Nil(err)
-
- _, err = s.Cat(mhash)
- is.Err(err) // we expect an http timeout error because `cat` won't find the `rand` string
-}
-
-func TestAddNoPin(t *testing.T) {
- is := is.New(t)
- s := NewShell(shellUrl)
-
- h, err := s.Add(bytes.NewBufferString(randString(32)), Pin(false))
- is.Nil(err)
-
- pins, err := s.Pins()
- is.Nil(err)
-
- _, ok := pins[h]
- is.False(ok)
-}
-
-func TestAddNoPinDeprecated(t *testing.T) {
- is := is.New(t)
- s := NewShell(shellUrl)
-
- h, err := s.AddNoPin(bytes.NewBufferString(randString(32)))
- is.Nil(err)
-
- pins, err := s.Pins()
- is.Nil(err)
-
- _, ok := pins[h]
- is.False(ok)
-}
-
-func TestAddDir(t *testing.T) {
- is := is.New(t)
- s := NewShell(shellUrl)
-
- cid, err := s.AddDir("./testdata")
- is.Nil(err)
- is.Equal(cid, "QmS4ustL54uo8FzR9455qaxZwuMiUhyvMcX9Ba8nUH4uVv")
-}
-
-func TestLocalShell(t *testing.T) {
- is := is.New(t)
- s := NewLocalShell()
- is.NotNil(s)
-
- mhash, err := s.Add(bytes.NewBufferString("Hello IPFS Shell tests"))
- is.Nil(err)
- is.Equal(mhash, "QmUfZ9rAdhV5ioBzXKdUTh2ZNsz9bzbkaLVyQ8uc8pj21F")
-}
-
-func TestCat(t *testing.T) {
- is := is.New(t)
- s := NewShell(shellUrl)
-
- rc, err := s.Cat(fmt.Sprintf("/ipfs/%s/readme", examplesHash))
- is.Nil(err)
-
- md5 := md5.New()
- _, err = io.Copy(md5, rc)
- is.Nil(err)
- is.Equal(fmt.Sprintf("%x", md5.Sum(nil)), "3fdcaad186e79983a6920b4c7eeda949")
-}
-
-func TestList(t *testing.T) {
- is := is.New(t)
- s := NewShell(shellUrl)
-
- list, err := s.List(fmt.Sprintf("/ipfs/%s", examplesHash))
- is.Nil(err)
-
- is.Equal(len(list), 7)
-
- // TODO: document difference in size between 'ipfs ls' and 'ipfs file ls -v'. additional object encoding in data block?
- expected := map[string]LsLink{
- "about": {Type: TFile, Hash: "QmZTR5bcpQD7cFgTorqxZDYaew1Wqgfbd2ud9QqGPAkK2V", Name: "about", Size: 1677},
- "contact": {Type: TFile, Hash: "QmYCvbfNbCwFR45HiNP45rwJgvatpiW38D961L5qAhUM5Y", Name: "contact", Size: 189},
- "help": {Type: TFile, Hash: "QmY5heUM5qgRubMDD1og9fhCPA6QdkMp3QCwd4s7gJsyE7", Name: "help", Size: 311},
- "ping": {Type: TFile, Hash: "QmejvEPop4D7YUadeGqYWmZxHhLc4JBUCzJJHWMzdcMe2y", Name: "ping", Size: 4},
- "quick-start": {Type: TFile, Hash: "QmXgqKTbzdh83pQtKFb19SpMCpDDcKR2ujqk3pKph9aCNF", Name: "quick-start", Size: 1681},
- "readme": {Type: TFile, Hash: "QmPZ9gcCEpqKTo6aq61g2nXGUhM4iCL3ewB6LDXZCtioEB", Name: "readme", Size: 1091},
- "security-notes": {Type: TFile, Hash: "QmQ5vhrL7uv6tuoN9KeVBwd4PwfQkXdVVmDLUZuTNxqgvm", Name: "security-notes", Size: 1162},
- }
- for _, l := range list {
- el, ok := expected[l.Name]
- is.True(ok)
- is.NotNil(el)
- is.Equal(*l, el)
- }
-}
-
-func TestFileList(t *testing.T) {
- is := is.New(t)
- s := NewShell(shellUrl)
-
- list, err := s.FileList(fmt.Sprintf("/ipfs/%s", examplesHash))
- is.Nil(err)
-
- is.Equal(list.Type, "Directory")
- is.Equal(list.Size, 0)
- is.Equal(len(list.Links), 7)
-
- // TODO: document difference in sice betwen 'ipfs ls' and 'ipfs file ls -v'. additional object encoding in data block?
- expected := map[string]UnixLsLink{
- "about": {Type: "File", Hash: "QmZTR5bcpQD7cFgTorqxZDYaew1Wqgfbd2ud9QqGPAkK2V", Name: "about", Size: 1677},
- "contact": {Type: "File", Hash: "QmYCvbfNbCwFR45HiNP45rwJgvatpiW38D961L5qAhUM5Y", Name: "contact", Size: 189},
- "help": {Type: "File", Hash: "QmY5heUM5qgRubMDD1og9fhCPA6QdkMp3QCwd4s7gJsyE7", Name: "help", Size: 311},
- "ping": {Type: "File", Hash: "QmejvEPop4D7YUadeGqYWmZxHhLc4JBUCzJJHWMzdcMe2y", Name: "ping", Size: 4},
- "quick-start": {Type: "File", Hash: "QmXgqKTbzdh83pQtKFb19SpMCpDDcKR2ujqk3pKph9aCNF", Name: "quick-start", Size: 1681},
- "readme": {Type: "File", Hash: "QmPZ9gcCEpqKTo6aq61g2nXGUhM4iCL3ewB6LDXZCtioEB", Name: "readme", Size: 1091},
- "security-notes": {Type: "File", Hash: "QmQ5vhrL7uv6tuoN9KeVBwd4PwfQkXdVVmDLUZuTNxqgvm", Name: "security-notes", Size: 1162},
- }
- for _, l := range list.Links {
- el, ok := expected[l.Name]
- is.True(ok)
- is.NotNil(el)
- is.Equal(*l, el)
- }
-}
-
-func TestPins(t *testing.T) {
- is := is.New(t)
- s := NewShell(shellUrl)
-
- // Add a thing, which pins it by default
- h, err := s.Add(bytes.NewBufferString("go-ipfs-api pins test 9F3D1F30-D12A-4024-9477-8F0C8E4B3A63"))
- is.Nil(err)
-
- pins, err := s.Pins()
- is.Nil(err)
-
- _, ok := pins[h]
- is.True(ok)
-
- err = s.Unpin(h)
- is.Nil(err)
-
- pins, err = s.Pins()
- is.Nil(err)
-
- _, ok = pins[h]
- is.False(ok)
-
- err = s.Pin(h)
- is.Nil(err)
-
- pins, err = s.Pins()
- is.Nil(err)
-
- info, ok := pins[h]
- is.True(ok)
- is.Equal(info.Type, RecursivePin)
-}
-
-func TestPatch_rmLink(t *testing.T) {
- is := is.New(t)
- s := NewShell(shellUrl)
- newRoot, err := s.Patch(examplesHash, "rm-link", "about")
- is.Nil(err)
- is.Equal(newRoot, "QmPmCJpciopaZnKcwymfQyRAEjXReR6UL2rdSfEscZfzcp")
-}
-
-func TestPatchLink(t *testing.T) {
- is := is.New(t)
- s := NewShell(shellUrl)
- newRoot, err := s.PatchLink(examplesHash, "about", "QmUXTtySmd7LD4p6RG6rZW6RuUuPZXTtNMmRQ6DSQo3aMw", true)
- is.Nil(err)
- is.Equal(newRoot, "QmVfe7gesXf4t9JzWePqqib8QSifC1ypRBGeJHitSnF7fA")
- newRoot, err = s.PatchLink(examplesHash, "about", "QmUXTtySmd7LD4p6RG6rZW6RuUuPZXTtNMmRQ6DSQo3aMw", false)
- is.Nil(err)
- is.Equal(newRoot, "QmVfe7gesXf4t9JzWePqqib8QSifC1ypRBGeJHitSnF7fA")
- newHash, err := s.NewObject("unixfs-dir")
- is.Nil(err)
- _, err = s.PatchLink(newHash, "a/b/c", newHash, false)
- is.NotNil(err)
- newHash, err = s.PatchLink(newHash, "a/b/c", newHash, true)
- is.Nil(err)
- is.Equal(newHash, "QmQ5D3xbMWFQRC9BKqbvnSnHri31GqvtWG1G6rE8xAZf1J")
-}
-
-func TestResolvePath(t *testing.T) {
- is := is.New(t)
- s := NewShell(shellUrl)
-
- childHash, err := s.ResolvePath(fmt.Sprintf("/ipfs/%s/about", examplesHash))
- is.Nil(err)
- is.Equal(childHash, "QmZTR5bcpQD7cFgTorqxZDYaew1Wqgfbd2ud9QqGPAkK2V")
-}
-
-func TestPubSub(t *testing.T) {
- is := is.New(t)
- s := NewShell(shellUrl)
-
- var (
- topic = "test"
-
- sub *PubSubSubscription
- err error
- )
-
- t.Log("subscribing...")
- sub, err = s.PubSubSubscribe(topic)
- is.Nil(err)
- is.NotNil(sub)
- t.Log("sub: done")
-
- time.Sleep(10 * time.Millisecond)
-
- t.Log("publishing...")
- is.Nil(s.PubSubPublish(topic, "Hello World!"))
- t.Log("pub: done")
-
- t.Log("next()...")
- r, err := sub.Next()
- t.Log("next: done. ")
-
- is.Nil(err)
- is.NotNil(r)
- is.Equal(r.Data, "Hello World!")
-
- sub2, err := s.PubSubSubscribe(topic)
- is.Nil(err)
- is.NotNil(sub2)
-
- is.Nil(s.PubSubPublish(topic, "Hallo Welt!"))
-
- r, err = sub2.Next()
- is.Nil(err)
- is.NotNil(r)
- is.Equal(r.Data, "Hallo Welt!")
-
- r, err = sub.Next()
- is.NotNil(r)
- is.Nil(err)
- is.Equal(r.Data, "Hallo Welt!")
-
- is.Nil(sub.Cancel())
-}
-
-func TestObjectStat(t *testing.T) {
- obj := "QmZTR5bcpQD7cFgTorqxZDYaew1Wqgfbd2ud9QqGPAkK2V"
- is := is.New(t)
- s := NewShell(shellUrl)
- stat, err := s.ObjectStat("QmZTR5bcpQD7cFgTorqxZDYaew1Wqgfbd2ud9QqGPAkK2V")
- is.Nil(err)
- is.Equal(stat.Hash, obj)
- is.Equal(stat.LinksSize, 3)
- is.Equal(stat.CumulativeSize, 1688)
-}
-
-func TestDagPut(t *testing.T) {
- is := is.New(t)
- s := NewShell(shellUrl)
-
- c, err := s.DagPut(`{"x": "abc","y":"def"}`, "json", "cbor")
- is.Nil(err)
- is.Equal(c, "zdpuAt47YjE9XTgSxUBkiYCbmnktKajQNheQBGASHj3FfYf8M")
-}
-
-func TestDagPutWithOpts(t *testing.T) {
- is := is.New(t)
- s := NewShell(shellUrl)
-
- c, err := s.DagPutWithOpts(`{"x": "abc","y":"def"}`, options.Dag.Pin("true"))
- is.Nil(err)
- is.Equal(c, "zdpuAt47YjE9XTgSxUBkiYCbmnktKajQNheQBGASHj3FfYf8M")
-}
-
-func TestStatsBW(t *testing.T) {
- is := is.New(t)
- s := NewShell(shellUrl)
- _, err := s.StatsBW(context.Background())
- is.Nil(err)
-}
-
-func TestSwarmPeers(t *testing.T) {
- is := is.New(t)
- s := NewShell(shellUrl)
- _, err := s.SwarmPeers(context.Background())
- is.Nil(err)
-}
-
-const letterBytes = "abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ"
-const (
- letterIdxBits = 6 // 6 bits to represent a letter index
- letterIdxMask = 1<<letterIdxBits - 1 // All 1-bits, as many as letterIdxBits
- letterIdxMax = 63 / letterIdxBits // # of letter indices fitting in 63 bits
-)
-
-var src = rand.NewSource(time.Now().UnixNano())
-
-func randString(n int) string {
- b := make([]byte, n)
- // A src.Int63() generates 63 random bits, enough for letterIdxMax characters!
- for i, cache, remain := n-1, src.Int63(), letterIdxMax; i >= 0; {
- if remain == 0 {
- cache, remain = src.Int63(), letterIdxMax
- }
- if idx := int(cache & letterIdxMask); idx < len(letterBytes) {
- b[i] = letterBytes[idx]
- i--
- }
- cache >>= letterIdxBits
- remain--
- }
-
- return string(b)
-}
-
-func TestRefs(t *testing.T) {
- is := is.New(t)
- s := NewShell(shellUrl)
-
- cid, err := s.AddDir("./testdata")
- is.Nil(err)
- is.Equal(cid, "QmS4ustL54uo8FzR9455qaxZwuMiUhyvMcX9Ba8nUH4uVv")
- refs, err := s.Refs(cid, false)
- is.Nil(err)
- expected := []string{
- "QmZTR5bcpQD7cFgTorqxZDYaew1Wqgfbd2ud9QqGPAkK2V",
- "QmYCvbfNbCwFR45HiNP45rwJgvatpiW38D961L5qAhUM5Y",
- "QmY5heUM5qgRubMDD1og9fhCPA6QdkMp3QCwd4s7gJsyE7",
- "QmejvEPop4D7YUadeGqYWmZxHhLc4JBUCzJJHWMzdcMe2y",
- "QmXgqKTbzdh83pQtKFb19SpMCpDDcKR2ujqk3pKph9aCNF",
- "QmPZ9gcCEpqKTo6aq61g2nXGUhM4iCL3ewB6LDXZCtioEB",
- "QmQ5vhrL7uv6tuoN9KeVBwd4PwfQkXdVVmDLUZuTNxqgvm",
- }
- var actual []string
- for r := range refs {
- actual = append(actual, r)
- }
-
- sort.Strings(expected)
- sort.Strings(actual)
- is.Equal(expected, actual)
-}
OSDN Git Service
