158 lines
4.1 KiB
Go
158 lines
4.1 KiB
Go
package gateway
|
|
|
|
import (
|
|
"io/ioutil"
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"os"
|
|
"strings"
|
|
|
|
"github.com/gorilla/mux"
|
|
"github.com/minio-io/minio/pkg/utils"
|
|
. "gopkg.in/check.v1"
|
|
)
|
|
|
|
type GatewaySuite struct{}
|
|
|
|
var _ = Suite(&GatewaySuite{})
|
|
|
|
func (s *GatewaySuite) TestPrintsGateway(c *C) {
|
|
// set up router with in memory storage driver
|
|
router := mux.NewRouter()
|
|
config := GatewayConfig{
|
|
StorageDriver: InMemoryStorageDriver,
|
|
BucketDriver: SynchronizedBucketDriver,
|
|
}
|
|
RegisterGatewayHandlers(router, config)
|
|
server := httptest.NewServer(router)
|
|
defer server.Close()
|
|
|
|
// GET request, empty
|
|
getReq1, _ := http.NewRequest("GET", server.URL+"/one/two/three", nil)
|
|
client := &http.Client{}
|
|
resp, err := client.Do(getReq1)
|
|
c.Assert(resp.StatusCode, Equals, 404)
|
|
c.Assert(err, IsNil)
|
|
|
|
// assert object not found response
|
|
body, _ := ioutil.ReadAll(resp.Body)
|
|
c.Assert(string(body), Equals, "Object not found\n")
|
|
c.Assert(err, IsNil)
|
|
|
|
// add new object
|
|
putReq, _ := http.NewRequest("PUT", server.URL+"/one/two/three", strings.NewReader("hello"))
|
|
resp, err = client.Do(putReq)
|
|
c.Assert(resp.StatusCode, Equals, 200)
|
|
c.Assert(err, IsNil)
|
|
|
|
// verify object exists
|
|
getReq2, _ := http.NewRequest("GET", server.URL+"/one/two/three", strings.NewReader("hello"))
|
|
resp, err = client.Do(getReq2)
|
|
c.Assert(resp.StatusCode, Equals, 200)
|
|
c.Assert(err, IsNil)
|
|
|
|
// verify object's contents
|
|
body2, _ := ioutil.ReadAll(resp.Body)
|
|
c.Assert(string(body2), Equals, "hello")
|
|
c.Assert(err, IsNil)
|
|
}
|
|
|
|
type TestContext struct{}
|
|
|
|
func (s *GatewaySuite) TestBucketCreation(c *C) {
|
|
config := GatewayConfig{
|
|
StorageDriver: InMemoryStorageDriver,
|
|
requestBucketChan: make(chan BucketRequest),
|
|
}
|
|
defer close(config.requestBucketChan)
|
|
go SynchronizedBucketDriver(config)
|
|
context := TestContext{}
|
|
|
|
// get new bucket A
|
|
var bucketA1 Bucket
|
|
callback := make(chan Bucket)
|
|
config.requestBucketChan <- BucketRequest{
|
|
name: "bucketA",
|
|
context: context,
|
|
callback: callback,
|
|
}
|
|
bucketA1 = <-callback
|
|
c.Assert(bucketA1.GetName(context), Equals, "bucketA")
|
|
|
|
// get bucket A again
|
|
var bucketA2 Bucket
|
|
callback = make(chan Bucket)
|
|
config.requestBucketChan <- BucketRequest{
|
|
name: "bucketA",
|
|
context: context,
|
|
callback: callback,
|
|
}
|
|
bucketA2 = <-callback
|
|
c.Assert(bucketA2.GetName(context), Equals, "bucketA")
|
|
c.Assert(bucketA1, DeepEquals, bucketA2)
|
|
|
|
// get new bucket B
|
|
var bucketB Bucket
|
|
callback = make(chan Bucket)
|
|
config.requestBucketChan <- BucketRequest{
|
|
name: "bucketB",
|
|
context: context,
|
|
callback: callback,
|
|
}
|
|
bucketB = <-callback
|
|
c.Assert(bucketB.GetName(context), Equals, "bucketB")
|
|
}
|
|
|
|
func (s *GatewaySuite) TestInMemoryBucketOperations(c *C) {
|
|
simpleFileStorageRootDir, err1 := utils.MakeTempTestDir()
|
|
c.Assert(err1, IsNil)
|
|
simpleEncodedStorageRootDir, err2 := utils.MakeTempTestDir()
|
|
c.Assert(err2, IsNil)
|
|
defer os.RemoveAll(simpleFileStorageRootDir)
|
|
defer os.RemoveAll(simpleEncodedStorageRootDir)
|
|
|
|
configs := []GatewayConfig{
|
|
GatewayConfig{
|
|
StorageDriver: InMemoryStorageDriver,
|
|
requestBucketChan: make(chan BucketRequest),
|
|
},
|
|
GatewayConfig{
|
|
StorageDriver: SimpleEncodedStorageDriver,
|
|
requestBucketChan: make(chan BucketRequest),
|
|
DataDir: simpleEncodedStorageRootDir,
|
|
K: 10,
|
|
M: 6,
|
|
BlockSize: 1024 * 1024,
|
|
},
|
|
}
|
|
for _, config := range configs {
|
|
defer close(config.requestBucketChan)
|
|
go SynchronizedBucketDriver(config)
|
|
context := TestContext{}
|
|
|
|
// get bucket
|
|
callback := make(chan Bucket)
|
|
config.requestBucketChan <- BucketRequest{
|
|
name: "bucket",
|
|
context: context,
|
|
callback: callback,
|
|
}
|
|
bucket := <-callback
|
|
c.Assert(bucket.GetName(context), Equals, "bucket")
|
|
|
|
// get missing value
|
|
nilResult, err := bucket.Get(context, "foo")
|
|
c.Assert(nilResult, IsNil)
|
|
c.Assert(err, Not(IsNil))
|
|
c.Assert(err.Error(), Equals, "Object not found")
|
|
|
|
// add new value
|
|
err = bucket.Put(context, "foo", []byte("bar"))
|
|
c.Assert(err, IsNil)
|
|
|
|
// retrieve value
|
|
barResult, err := bucket.Get(context, "foo")
|
|
c.Assert(err, IsNil)
|
|
c.Assert(string(barResult), Equals, "bar")
|
|
}
|
|
}
|