331 lines
9.3 KiB
Go
331 lines
9.3 KiB
Go
// Copyright (c) 2014 Hewlett-Packard Development Company, L.P.
|
|
//
|
|
// 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 objectstorage_test
|
|
|
|
import (
|
|
"errors"
|
|
"git.openstack.org/openstack/golang-client.git/objectstorage/v1"
|
|
"io/ioutil"
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"os"
|
|
"strconv"
|
|
"testing"
|
|
)
|
|
|
|
var znHome = "./"
|
|
var objFile = "objectstorage_test.go"
|
|
var srcFile = znHome + objFile
|
|
var tokn = "eaaafd18-0fed-4b3a-81b4-663c99ec1cbb"
|
|
var containerName = "John's container"
|
|
var containerPrefix = "/" + containerName
|
|
var objPrefix = containerPrefix + "/" + objFile
|
|
|
|
func TestGetAccountMeta(t *testing.T) {
|
|
var apiServer = httptest.NewServer(http.HandlerFunc(
|
|
func(w http.ResponseWriter, r *http.Request) {
|
|
if r.Method == "HEAD" {
|
|
w.Header().Set("X-Account-Container-Count", "7")
|
|
w.Header().Set("X-Account-Object-Count", "413")
|
|
w.Header().Set("X-Account-Bytes-Used", "987654321000")
|
|
w.WriteHeader(204)
|
|
return
|
|
}
|
|
t.Error(errors.New("Failed: r.Method == HEAD"))
|
|
}))
|
|
defer apiServer.Close()
|
|
meta, err := objectstorage.GetAccountMeta(apiServer.URL, tokn)
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
if meta.Get("X-Account-Container-Count") != "7" ||
|
|
meta.Get("X-Account-Object-Count") != "413" ||
|
|
meta.Get("X-Account-Bytes-Used") != "987654321000" {
|
|
t.Error("Failed: meta not matching")
|
|
}
|
|
}
|
|
|
|
func TestListContainers(t *testing.T) {
|
|
var containerList = `[
|
|
{"name":"container 1",
|
|
"count":2, "bytes":78},
|
|
{"name":"container 2",
|
|
"count":1,
|
|
"bytes":17}]`
|
|
var apiServer = httptest.NewServer(http.HandlerFunc(
|
|
func(w http.ResponseWriter, r *http.Request) {
|
|
if r.Method == "GET" {
|
|
w.Header().Set("Content-Type", "application/json; charset=utf-8")
|
|
w.WriteHeader(200)
|
|
w.Write([]byte(containerList))
|
|
return
|
|
}
|
|
t.Error(errors.New("Failed: r.Method == GET"))
|
|
}))
|
|
defer apiServer.Close()
|
|
myList, err := objectstorage.ListContainers(0, "", apiServer.URL, tokn)
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
if string(myList) != containerList {
|
|
t.Error(errors.New("Failed: input != output"))
|
|
}
|
|
}
|
|
|
|
func TestListObjects(t *testing.T) {
|
|
var objList = `[
|
|
{"name":"test obj 1",
|
|
"hash":"4281c348eaf83e70ddce0e07221c3d28",
|
|
"bytes":14,
|
|
"content_type":"application\/octet-stream",
|
|
"last_modified":"2009-02-03T05:26:32.612278"},
|
|
{"name":"test obj 2",
|
|
"hash":"b039efe731ad111bc1b0ef221c3849d0",
|
|
"bytes":64,
|
|
"content_type":"application\/octet-stream",
|
|
"last_modified":"2009-02-03T05:26:32.612278"}
|
|
]`
|
|
var apiServer = httptest.NewServer(http.HandlerFunc(
|
|
func(w http.ResponseWriter, r *http.Request) {
|
|
if r.Method == "GET" {
|
|
w.Header().Set("Content-Type", "application/json; charset=utf-8")
|
|
w.WriteHeader(200)
|
|
w.Write([]byte(objList))
|
|
return
|
|
}
|
|
t.Error(errors.New("Failed: r.Method == GET"))
|
|
}))
|
|
defer apiServer.Close()
|
|
myList, err := objectstorage.ListObjects(
|
|
0, "", "", "", "", apiServer.URL+containerPrefix, tokn)
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
if string(myList) != objList {
|
|
t.Error(errors.New("Failed: input != output"))
|
|
}
|
|
}
|
|
|
|
func TestDeleteContainer(t *testing.T) {
|
|
var apiServer = httptest.NewServer(http.HandlerFunc(
|
|
func(w http.ResponseWriter, r *http.Request) {
|
|
if r.Method == "DELETE" {
|
|
w.WriteHeader(204)
|
|
return
|
|
}
|
|
t.Error(errors.New("Failed: r.Method == DELETE"))
|
|
}))
|
|
defer apiServer.Close()
|
|
if err := objectstorage.DeleteContainer(apiServer.URL+containerPrefix,
|
|
tokn); err != nil {
|
|
t.Error(err)
|
|
}
|
|
}
|
|
|
|
func TestGetContainerMeta(t *testing.T) {
|
|
var apiServer = httptest.NewServer(http.HandlerFunc(
|
|
func(w http.ResponseWriter, r *http.Request) {
|
|
if r.Method == "HEAD" {
|
|
w.Header().Set("X-Container-Object-Count", "7")
|
|
w.Header().Set("X-Container-Bytes-Used", "413")
|
|
w.Header().Set("X-Container-Meta-InspectedBy", "Jack Wolf")
|
|
w.WriteHeader(204)
|
|
return
|
|
}
|
|
t.Error(errors.New("Failed: r.Method == HEAD"))
|
|
}))
|
|
defer apiServer.Close()
|
|
meta, err := objectstorage.GetContainerMeta(apiServer.URL+containerPrefix, tokn)
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
if meta.Get("X-Container-Object-Count") != "7" ||
|
|
meta.Get("X-Container-Bytes-Used") != "413" ||
|
|
meta.Get("X-Container-Meta-InspectedBy") != "Jack Wolf" {
|
|
t.Error("Failed: meta not matching")
|
|
}
|
|
}
|
|
|
|
func TestSetContainerMeta(t *testing.T) {
|
|
var apiServer = httptest.NewServer(http.HandlerFunc(
|
|
func(w http.ResponseWriter, r *http.Request) {
|
|
if r.Method == "POST" && r.Header.Get("X-Container-Meta-Fruit") == "Apple" {
|
|
w.WriteHeader(204)
|
|
return
|
|
}
|
|
t.Error(errors.New(
|
|
"Failed: r.Method == POST && X-Container-Meta-Fruit == Apple"))
|
|
}))
|
|
defer apiServer.Close()
|
|
if err := objectstorage.SetContainerMeta(
|
|
apiServer.URL+containerPrefix, tokn,
|
|
"X-Container-Meta-Fruit", "Apple"); err != nil {
|
|
t.Error(err)
|
|
}
|
|
}
|
|
|
|
func TestPutContainer(t *testing.T) {
|
|
var apiServer = httptest.NewServer(http.HandlerFunc(
|
|
func(w http.ResponseWriter, r *http.Request) {
|
|
if r.Method == "PUT" {
|
|
w.WriteHeader(201)
|
|
return
|
|
}
|
|
t.Error(errors.New("Failed: r.Method == PUT"))
|
|
}))
|
|
defer apiServer.Close()
|
|
if err := objectstorage.PutContainer(apiServer.URL+containerPrefix,
|
|
tokn, "X-TTL", "259200", "X-Log-Retention", "true"); err != nil {
|
|
t.Error(err)
|
|
}
|
|
}
|
|
|
|
func TestPutObject(t *testing.T) {
|
|
var fContent []byte
|
|
f, err := os.Open(srcFile)
|
|
defer f.Close()
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
fContent, err = ioutil.ReadAll(f)
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
f.Close()
|
|
var apiServer = httptest.NewServer(http.HandlerFunc(
|
|
func(w http.ResponseWriter, r *http.Request) {
|
|
rBody, err := ioutil.ReadAll(r.Body)
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
if r.Method == "PUT" && len(fContent) == len(rBody) {
|
|
w.WriteHeader(201)
|
|
return
|
|
}
|
|
t.Error(errors.New("Failed: Not 201"))
|
|
}))
|
|
defer apiServer.Close()
|
|
if err = objectstorage.PutObject(&fContent, apiServer.URL+objPrefix,
|
|
tokn); err != nil {
|
|
t.Error(err)
|
|
}
|
|
}
|
|
|
|
func TestCopyObject(t *testing.T) {
|
|
destURL := "/destContainer/dest/Obj"
|
|
var apiServer = httptest.NewServer(http.HandlerFunc(
|
|
func(w http.ResponseWriter, r *http.Request) {
|
|
if r.Method == "COPY" && r.Header.Get("Destination") == destURL {
|
|
w.WriteHeader(200)
|
|
return
|
|
}
|
|
t.Error(errors.New(
|
|
"Failed: r.Method == COPY && r.Header.Get(Destination) == destURL"))
|
|
}))
|
|
defer apiServer.Close()
|
|
if err := objectstorage.CopyObject(apiServer.URL+objPrefix, destURL,
|
|
tokn); err != nil {
|
|
t.Error(err)
|
|
}
|
|
}
|
|
|
|
func TestGetObjectMeta(t *testing.T) {
|
|
var apiServer = httptest.NewServer(http.HandlerFunc(
|
|
func(w http.ResponseWriter, r *http.Request) {
|
|
if r.Method == "HEAD" {
|
|
w.Header().Set("X-Object-Meta-Fruit", "Apple")
|
|
w.Header().Set("X-Object-Meta-Veggie", "Carrot")
|
|
w.WriteHeader(200)
|
|
return
|
|
}
|
|
t.Error(errors.New(
|
|
"Failed: r.Method == HEAD && r.Header.Get(X-Auth-Token) == tokn"))
|
|
}))
|
|
defer apiServer.Close()
|
|
meta, err := objectstorage.GetObjectMeta(apiServer.URL+objPrefix, tokn)
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
if meta.Get("X-Object-Meta-Fruit") != "Apple" ||
|
|
meta.Get("X-Object-Meta-Veggie") != "Carrot" {
|
|
t.Error("Failed: meta not matching")
|
|
}
|
|
}
|
|
|
|
func TestSetObjectMeta(t *testing.T) {
|
|
var apiServer = httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter,
|
|
r *http.Request) {
|
|
if r.Method == "POST" && r.Header.Get("X-Object-Meta-Fruit") == "Apple" {
|
|
w.WriteHeader(202)
|
|
return
|
|
}
|
|
t.Error(errors.New("Failed: r.Method == POST && X-Object-Meta-Fruit == Apple"))
|
|
}))
|
|
defer apiServer.Close()
|
|
if err := objectstorage.SetObjectMeta(apiServer.URL+objPrefix,
|
|
tokn, "X-Object-Meta-Fruit", "Apple"); err != nil {
|
|
t.Error(err)
|
|
}
|
|
}
|
|
|
|
func TestGetObject(t *testing.T) {
|
|
var unCompressedLen int
|
|
var apiServer = httptest.NewServer(http.HandlerFunc(
|
|
func(w http.ResponseWriter, r *http.Request) {
|
|
if r.Method == "GET" {
|
|
fContent, err := ioutil.ReadFile(srcFile)
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
unCompressedLen = len(fContent)
|
|
w.Header().Set("Content-Length", strconv.Itoa(unCompressedLen))
|
|
w.Header().Set("X-Object-ModTime", "93000299")
|
|
w.Header().Set("X-Object-Mode", "rwxrwxrwx")
|
|
w.Write(fContent)
|
|
return
|
|
}
|
|
t.Error(errors.New("Failed: r.Method == GET"))
|
|
}))
|
|
defer apiServer.Close()
|
|
hdr, body, err := objectstorage.GetObject(apiServer.URL+objPrefix, tokn)
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
if unCompressedLen != len(body) {
|
|
t.Error(errors.New("GET: incorrect uncompressed len"))
|
|
}
|
|
if hdr.Get("X-Object-ModTime") != "93000299" ||
|
|
hdr.Get("Content-Length") != strconv.Itoa(len(body)) ||
|
|
hdr.Get("X-Object-Mode") != "rwxrwxrwx" {
|
|
//
|
|
t.Error(errors.New("GET: incorrect hdr"))
|
|
}
|
|
}
|
|
|
|
func TestDeleteObject(t *testing.T) {
|
|
var apiServer = httptest.NewServer(http.HandlerFunc(
|
|
func(w http.ResponseWriter, r *http.Request) {
|
|
if r.Method == "DELETE" {
|
|
w.WriteHeader(204)
|
|
return
|
|
}
|
|
t.Error(errors.New("Failed: r.Method == DELETE"))
|
|
}))
|
|
defer apiServer.Close()
|
|
if err := objectstorage.DeleteObject(apiServer.URL+objPrefix, tokn); err != nil {
|
|
t.Error(err)
|
|
}
|
|
}
|