mirror of
https://github.com/minio/minio.git
synced 2024-12-26 23:25:54 -05:00
370 lines
11 KiB
Go
370 lines
11 KiB
Go
/*
|
|
* Mini Object Storage, (C) 2014 Minio, Inc.
|
|
*
|
|
* Licensed under the Apache License, Version 2.0 (the "License");
|
|
* you may not use this file except in compliance with the License.
|
|
* You may obtain a copy of the License at
|
|
*
|
|
* http://www.apache.org/licenses/LICENSE-2.0
|
|
*
|
|
* Unless required by applicable law or agreed to in writing, software
|
|
* distributed under the License is distributed on an "AS IS" BASIS,
|
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
* See the License for the specific language governing permissions and
|
|
* limitations under the License.
|
|
*/
|
|
|
|
package minioapi
|
|
|
|
import (
|
|
"bytes"
|
|
"encoding/xml"
|
|
"io"
|
|
"io/ioutil"
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"sort"
|
|
"strconv"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/minio-io/minio/pkg/storage/inmemory"
|
|
. "gopkg.in/check.v1"
|
|
)
|
|
|
|
func Test(t *testing.T) { TestingT(t) }
|
|
|
|
type MySuite struct{}
|
|
|
|
var _ = Suite(&MySuite{})
|
|
|
|
func (s *MySuite) TestNonExistantObject(c *C) {
|
|
_, _, storage := inmemory.Start()
|
|
httpHandler := HttpHandler(storage)
|
|
testServer := httptest.NewServer(httpHandler)
|
|
defer testServer.Close()
|
|
|
|
response, err := http.Get(testServer.URL + "/bucket/object")
|
|
c.Assert(err, IsNil)
|
|
c.Assert(response.StatusCode, Equals, http.StatusNotFound)
|
|
}
|
|
|
|
func (s *MySuite) TestEmptyObject(c *C) {
|
|
_, _, storage := inmemory.Start()
|
|
httpHandler := HttpHandler(storage)
|
|
testServer := httptest.NewServer(httpHandler)
|
|
defer testServer.Close()
|
|
|
|
buffer := bytes.NewBufferString("")
|
|
storage.StoreBucket("bucket")
|
|
storage.StoreObject("bucket", "object", buffer)
|
|
|
|
response, err := http.Get(testServer.URL + "/bucket/object")
|
|
c.Assert(err, IsNil)
|
|
c.Assert(response.StatusCode, Equals, http.StatusOK)
|
|
|
|
responseBody, err := ioutil.ReadAll(response.Body)
|
|
c.Assert(err, IsNil)
|
|
c.Assert(true, Equals, bytes.Equal(responseBody, buffer.Bytes()))
|
|
|
|
metadata, err := storage.GetObjectMetadata("bucket", "object")
|
|
c.Assert(err, IsNil)
|
|
verifyHeaders(c, response.Header, metadata.Created, 0, "text/plain", metadata.ETag)
|
|
|
|
// TODO Test Headers
|
|
}
|
|
|
|
func (s *MySuite) TestObject(c *C) {
|
|
_, _, storage := inmemory.Start()
|
|
httpHandler := HttpHandler(storage)
|
|
testServer := httptest.NewServer(httpHandler)
|
|
defer testServer.Close()
|
|
|
|
buffer := bytes.NewBufferString("hello world")
|
|
storage.StoreBucket("bucket")
|
|
storage.StoreObject("bucket", "object", buffer)
|
|
|
|
response, err := http.Get(testServer.URL + "/bucket/object")
|
|
c.Assert(err, IsNil)
|
|
c.Assert(response.StatusCode, Equals, http.StatusOK)
|
|
|
|
responseBody, err := ioutil.ReadAll(response.Body)
|
|
c.Assert(err, IsNil)
|
|
c.Assert(true, Equals, bytes.Equal(responseBody, []byte("hello world")))
|
|
|
|
metadata, err := storage.GetObjectMetadata("bucket", "object")
|
|
c.Assert(err, IsNil)
|
|
verifyHeaders(c, response.Header, metadata.Created, len("hello world"), "text/plain", metadata.ETag)
|
|
}
|
|
|
|
func (s *MySuite) TestMultipleObjects(c *C) {
|
|
_, _, storage := inmemory.Start()
|
|
httpHandler := HttpHandler(storage)
|
|
testServer := httptest.NewServer(httpHandler)
|
|
defer testServer.Close()
|
|
|
|
buffer1 := bytes.NewBufferString("hello one")
|
|
buffer2 := bytes.NewBufferString("hello two")
|
|
buffer3 := bytes.NewBufferString("hello three")
|
|
|
|
storage.StoreBucket("bucket")
|
|
storage.StoreObject("bucket", "object1", buffer1)
|
|
storage.StoreObject("bucket", "object2", buffer2)
|
|
storage.StoreObject("bucket", "object3", buffer3)
|
|
|
|
// test non-existant object
|
|
response, err := http.Get(testServer.URL + "/bucket/object")
|
|
c.Assert(err, IsNil)
|
|
c.Assert(response.StatusCode, Equals, http.StatusNotFound)
|
|
// TODO Test Headers
|
|
|
|
//// test object 1
|
|
|
|
// get object
|
|
response, err = http.Get(testServer.URL + "/bucket/object1")
|
|
c.Assert(err, IsNil)
|
|
|
|
// get metadata
|
|
metadata, err := storage.GetObjectMetadata("bucket", "object1")
|
|
c.Assert(err, IsNil)
|
|
c.Assert(response.StatusCode, Equals, http.StatusOK)
|
|
|
|
// verify headers
|
|
verifyHeaders(c, response.Header, metadata.Created, len("hello one"), "text/plain", metadata.ETag)
|
|
c.Assert(err, IsNil)
|
|
|
|
// verify response data
|
|
responseBody, err := ioutil.ReadAll(response.Body)
|
|
c.Assert(err, IsNil)
|
|
c.Assert(true, Equals, bytes.Equal(responseBody, []byte("hello one")))
|
|
|
|
// test object 2
|
|
// get object
|
|
response, err = http.Get(testServer.URL + "/bucket/object2")
|
|
c.Assert(err, IsNil)
|
|
|
|
// get metadata
|
|
metadata, err = storage.GetObjectMetadata("bucket", "object2")
|
|
c.Assert(err, IsNil)
|
|
c.Assert(response.StatusCode, Equals, http.StatusOK)
|
|
|
|
// verify headers
|
|
verifyHeaders(c, response.Header, metadata.Created, len("hello two"), "text/plain", metadata.ETag)
|
|
c.Assert(err, IsNil)
|
|
|
|
// verify response data
|
|
responseBody, err = ioutil.ReadAll(response.Body)
|
|
c.Assert(err, IsNil)
|
|
c.Assert(true, Equals, bytes.Equal(responseBody, []byte("hello two")))
|
|
|
|
// test object 3
|
|
// get object
|
|
response, err = http.Get(testServer.URL + "/bucket/object3")
|
|
c.Assert(err, IsNil)
|
|
|
|
// get metadata
|
|
metadata, err = storage.GetObjectMetadata("bucket", "object3")
|
|
c.Assert(err, IsNil)
|
|
c.Assert(response.StatusCode, Equals, http.StatusOK)
|
|
|
|
// verify headers
|
|
verifyHeaders(c, response.Header, metadata.Created, len("hello three"), "text/plain", metadata.ETag)
|
|
c.Assert(err, IsNil)
|
|
|
|
// verify object
|
|
responseBody, err = ioutil.ReadAll(response.Body)
|
|
c.Assert(err, IsNil)
|
|
c.Assert(true, Equals, bytes.Equal(responseBody, []byte("hello three")))
|
|
}
|
|
|
|
func (s *MySuite) TestNotImplemented(c *C) {
|
|
_, _, storage := inmemory.Start()
|
|
httpHandler := HttpHandler(storage)
|
|
testServer := httptest.NewServer(httpHandler)
|
|
defer testServer.Close()
|
|
|
|
response, err := http.Get(testServer.URL + "/bucket/object?acl")
|
|
c.Assert(err, IsNil)
|
|
c.Assert(response.StatusCode, Equals, http.StatusNotImplemented)
|
|
}
|
|
|
|
func (s *MySuite) TestHeader(c *C) {
|
|
_, _, storage := inmemory.Start()
|
|
httpHandler := HttpHandler(storage)
|
|
testServer := httptest.NewServer(httpHandler)
|
|
defer testServer.Close()
|
|
|
|
response, err := http.Get(testServer.URL + "/bucket/object")
|
|
c.Assert(err, IsNil)
|
|
c.Assert(response.StatusCode, Equals, http.StatusNotFound)
|
|
|
|
buffer := bytes.NewBufferString("hello world")
|
|
storage.StoreBucket("bucket")
|
|
storage.StoreObject("bucket", "object", buffer)
|
|
|
|
response, err = http.Get(testServer.URL + "/bucket/object")
|
|
c.Assert(err, IsNil)
|
|
c.Assert(response.StatusCode, Equals, http.StatusOK)
|
|
|
|
metadata, err := storage.GetObjectMetadata("bucket", "object")
|
|
c.Assert(err, IsNil)
|
|
verifyHeaders(c, response.Header, metadata.Created, len("hello world"), "text/plain", metadata.ETag)
|
|
}
|
|
|
|
func (s *MySuite) TestPutBucket(c *C) {
|
|
_, _, storage := inmemory.Start()
|
|
httpHandler := HttpHandler(storage)
|
|
testServer := httptest.NewServer(httpHandler)
|
|
defer testServer.Close()
|
|
|
|
buckets, err := storage.ListBuckets("bucket")
|
|
c.Assert(len(buckets), Equals, 0)
|
|
c.Assert(err, IsNil)
|
|
|
|
request, err := http.NewRequest("PUT", testServer.URL+"/bucket/", bytes.NewBufferString(""))
|
|
c.Assert(err, IsNil)
|
|
|
|
client := http.Client{}
|
|
response, err := client.Do(request)
|
|
c.Assert(err, IsNil)
|
|
c.Assert(response.StatusCode, Equals, http.StatusOK)
|
|
|
|
// check bucket exists
|
|
buckets, err = storage.ListBuckets("bucket")
|
|
c.Assert(len(buckets), Equals, 1)
|
|
c.Assert(err, IsNil)
|
|
c.Assert(buckets[0].Name, Equals, "bucket")
|
|
}
|
|
|
|
func (s *MySuite) TestPutObject(c *C) {
|
|
_, _, storage := inmemory.Start()
|
|
httpHandler := HttpHandler(storage)
|
|
testServer := httptest.NewServer(httpHandler)
|
|
defer testServer.Close()
|
|
|
|
objects, isTruncated, err := storage.ListObjects("bucket", "", 1000)
|
|
c.Assert(len(objects), Equals, 0)
|
|
c.Assert(isTruncated, Equals, false)
|
|
c.Assert(err, IsNil)
|
|
|
|
date1 := time.Now()
|
|
|
|
// Put Bucket before - Put Object into a bucket
|
|
request, err := http.NewRequest("PUT", testServer.URL+"/bucket/", bytes.NewBufferString(""))
|
|
c.Assert(err, IsNil)
|
|
|
|
client := http.Client{}
|
|
response, err := client.Do(request)
|
|
c.Assert(err, IsNil)
|
|
c.Assert(response.StatusCode, Equals, http.StatusOK)
|
|
|
|
request, err = http.NewRequest("PUT", testServer.URL+"/bucket/two", bytes.NewBufferString("hello world"))
|
|
c.Assert(err, IsNil)
|
|
|
|
response, err = client.Do(request)
|
|
c.Assert(err, IsNil)
|
|
c.Assert(response.StatusCode, Equals, http.StatusOK)
|
|
|
|
date2 := time.Now()
|
|
|
|
objects, isTruncated, err = storage.ListObjects("bucket", "", 1000)
|
|
c.Assert(len(objects), Equals, 1)
|
|
c.Assert(isTruncated, Equals, false)
|
|
c.Assert(err, IsNil)
|
|
|
|
var writer bytes.Buffer
|
|
|
|
storage.CopyObjectToWriter(&writer, "bucket", "two")
|
|
|
|
c.Assert(bytes.Equal(writer.Bytes(), []byte("hello world")), Equals, true)
|
|
|
|
metadata, err := storage.GetObjectMetadata("bucket", "two")
|
|
c.Assert(err, IsNil)
|
|
lastModified := metadata.Created
|
|
|
|
c.Assert(date1.Before(lastModified), Equals, true)
|
|
c.Assert(lastModified.Before(date2), Equals, true)
|
|
}
|
|
|
|
func (s *MySuite) TestListBuckets(c *C) {
|
|
_, _, storage := inmemory.Start()
|
|
httpHandler := HttpHandler(storage)
|
|
testServer := httptest.NewServer(httpHandler)
|
|
defer testServer.Close()
|
|
|
|
response, err := http.Get(testServer.URL + "/")
|
|
defer response.Body.Close()
|
|
c.Assert(err, IsNil)
|
|
c.Assert(response.StatusCode, Equals, http.StatusOK)
|
|
|
|
listResponse, err := readListBucket(response.Body)
|
|
c.Assert(err, IsNil)
|
|
c.Assert(len(listResponse.Buckets.Bucket), Equals, 0)
|
|
|
|
storage.StoreBucket("foo")
|
|
|
|
response, err = http.Get(testServer.URL + "/")
|
|
defer response.Body.Close()
|
|
c.Assert(err, IsNil)
|
|
c.Assert(response.StatusCode, Equals, http.StatusOK)
|
|
|
|
listResponse, err = readListBucket(response.Body)
|
|
c.Assert(err, IsNil)
|
|
c.Assert(len(listResponse.Buckets.Bucket), Equals, 1)
|
|
c.Assert(listResponse.Buckets.Bucket[0].Name, Equals, "foo")
|
|
|
|
storage.StoreBucket("bar")
|
|
|
|
response, err = http.Get(testServer.URL + "/")
|
|
defer response.Body.Close()
|
|
c.Assert(err, IsNil)
|
|
c.Assert(response.StatusCode, Equals, http.StatusOK)
|
|
|
|
listResponse, err = readListBucket(response.Body)
|
|
c.Assert(err, IsNil)
|
|
c.Assert(len(listResponse.Buckets.Bucket), Equals, 2)
|
|
var buckets []string
|
|
buckets = append(buckets, listResponse.Buckets.Bucket[0].Name)
|
|
buckets = append(buckets, listResponse.Buckets.Bucket[1].Name)
|
|
sort.Strings(buckets)
|
|
c.Assert(buckets[0], Equals, "bar")
|
|
c.Assert(buckets[1], Equals, "foo")
|
|
}
|
|
|
|
func readListBucket(reader io.Reader) (BucketListResponse, error) {
|
|
var results BucketListResponse
|
|
decoder := xml.NewDecoder(reader)
|
|
err := decoder.Decode(&results)
|
|
return results, err
|
|
}
|
|
|
|
func (s *MySuite) TestListObjects(c *C) {
|
|
// TODO Implement
|
|
}
|
|
|
|
func (s *MySuite) TestShouldNotBeAbleToCreateObjectInNonexistantBucket(c *C) {
|
|
// TODO Implement
|
|
}
|
|
|
|
func (s *MySuite) TestHeadOnObject(c *C) {
|
|
// TODO
|
|
}
|
|
|
|
func (s *MySuite) TestDateFormat(c *C) {
|
|
// TODO
|
|
}
|
|
|
|
func verifyHeaders(c *C, header http.Header, date time.Time, size int, contentType string, etag string) {
|
|
// Verify date
|
|
c.Assert(header["Last-Modified"][0], Equals, date.Format(time.RFC1123))
|
|
|
|
// verify size
|
|
c.Assert(header["Content-Length"][0], Equals, strconv.Itoa(size))
|
|
|
|
// verify content type
|
|
c.Assert(header["Content-Type"][0], Equals, contentType)
|
|
|
|
// verify etag
|
|
c.Assert(header["Etag"][0], Equals, etag)
|
|
}
|