66 lines
1.5 KiB
Go
66 lines
1.5 KiB
Go
package builtin
|
|
|
|
import (
|
|
"bytes"
|
|
"io/ioutil"
|
|
"testing"
|
|
|
|
"github.com/drone/drone/Godeps/_workspace/src/github.com/franela/goblin"
|
|
)
|
|
|
|
func TestBlobstore(t *testing.T) {
|
|
db := mustConnectTest()
|
|
bs := NewBlobstore(db)
|
|
defer db.Close()
|
|
|
|
g := goblin.Goblin(t)
|
|
g.Describe("Blobstore", func() {
|
|
|
|
// before each test be sure to purge the package
|
|
// table data from the database.
|
|
g.BeforeEach(func() {
|
|
db.Exec("DELETE FROM blobs")
|
|
})
|
|
|
|
g.It("Should Set a Blob", func() {
|
|
err := bs.SetBlob("foo", []byte("bar"))
|
|
g.Assert(err == nil).IsTrue()
|
|
})
|
|
|
|
g.It("Should Set a Blob reader", func() {
|
|
var buf bytes.Buffer
|
|
buf.Write([]byte("bar"))
|
|
err := bs.SetBlobReader("foo", &buf)
|
|
g.Assert(err == nil).IsTrue()
|
|
})
|
|
|
|
g.It("Should Overwrite a Blob", func() {
|
|
bs.SetBlob("foo", []byte("bar"))
|
|
bs.SetBlob("foo", []byte("baz"))
|
|
blob, err := bs.GetBlob("foo")
|
|
g.Assert(err == nil).IsTrue()
|
|
g.Assert(string(blob)).Equal("baz")
|
|
})
|
|
|
|
g.It("Should Get a Blob", func() {
|
|
bs.SetBlob("foo", []byte("bar"))
|
|
blob, err := bs.GetBlob("foo")
|
|
g.Assert(err == nil).IsTrue()
|
|
g.Assert(string(blob)).Equal("bar")
|
|
})
|
|
|
|
g.It("Should Get a Blob reader", func() {
|
|
bs.SetBlob("foo", []byte("bar"))
|
|
r, _ := bs.GetBlobReader("foo")
|
|
blob, _ := ioutil.ReadAll(r)
|
|
g.Assert(string(blob)).Equal("bar")
|
|
})
|
|
|
|
g.It("Should Del a Blob", func() {
|
|
bs.SetBlob("foo", []byte("bar"))
|
|
err := bs.DelBlob("foo")
|
|
g.Assert(err == nil).IsTrue()
|
|
})
|
|
|
|
})
|
|
}
|