harness-drone/pkg/store/builtin/blob_test.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()
		})

	})
}