mirror of
https://github.com/ceph/go-ceph
synced 2025-03-03 18:09:00 +00:00
rados/striper: add an initial set of tests for the striper package
Signed-off-by: John Mulligan <jmulligan@redhat.com>
This commit is contained in:
John Mulligan
mergify[bot]
parent
72e2ac4d8f
commit
6143ff682d
109
rados/striper/striper_test.go
Normal file
109
rados/striper/striper_test.go
Normal file
@ -0,0 +1,109 @@
|
||||
//go:build ceph_preview
|
||||
|
||||
package striper
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"testing"
|
||||
"time"
|
||||
|
||||
"github.com/gofrs/uuid"
|
||||
"github.com/stretchr/testify/assert"
|
||||
"github.com/stretchr/testify/require"
|
||||
tsuite "github.com/stretchr/testify/suite"
|
||||
|
||||
"github.com/ceph/go-ceph/rados"
|
||||
)
|
||||
|
||||
type StriperTestSuite struct {
|
||||
tsuite.Suite
|
||||
|
||||
pool string
|
||||
conn *rados.Conn
|
||||
}
|
||||
|
||||
func (suite *StriperTestSuite) connect() *rados.Conn {
|
||||
conn, err := rados.NewConn()
|
||||
require.NoError(suite.T(), err)
|
||||
|
||||
err = conn.ReadDefaultConfigFile()
|
||||
require.NoError(suite.T(), err)
|
||||
|
||||
timeout := time.After(time.Second * 5)
|
||||
ch := make(chan error)
|
||||
go func(conn *rados.Conn) {
|
||||
ch <- conn.Connect()
|
||||
}(conn)
|
||||
select {
|
||||
case err = <-ch:
|
||||
case <-timeout:
|
||||
err = fmt.Errorf("timed out waiting for connect")
|
||||
}
|
||||
require.NoError(suite.T(), err)
|
||||
return conn
|
||||
}
|
||||
|
||||
func (suite *StriperTestSuite) defaultContext() *rados.IOContext {
|
||||
ioctx, err := suite.conn.OpenIOContext(suite.pool)
|
||||
require.NoError(suite.T(), err)
|
||||
return ioctx
|
||||
}
|
||||
|
||||
func (suite *StriperTestSuite) SetupSuite() {
|
||||
suite.pool = fmt.Sprintf("striper%s", uuid.Must(uuid.NewV4()).String())
|
||||
conn := suite.connect()
|
||||
defer conn.Shutdown()
|
||||
require.NoError(suite.T(), conn.MakePool(suite.pool))
|
||||
}
|
||||
|
||||
func (suite *StriperTestSuite) TearDownSuite() {
|
||||
conn := suite.connect()
|
||||
defer conn.Shutdown()
|
||||
assert.NoError(suite.T(), conn.DeletePool(suite.pool))
|
||||
}
|
||||
|
||||
func (suite *StriperTestSuite) SetupTest() {
|
||||
suite.conn = suite.connect()
|
||||
}
|
||||
|
||||
func (suite *StriperTestSuite) TearDownTest() {
|
||||
suite.conn.Shutdown()
|
||||
suite.conn = nil
|
||||
}
|
||||
|
||||
func (suite *StriperTestSuite) TestNewStriper() {
|
||||
ioctx := suite.defaultContext()
|
||||
defer ioctx.Destroy()
|
||||
|
||||
striper, err := New(ioctx)
|
||||
assert.NoError(suite.T(), err)
|
||||
striper.Destroy()
|
||||
|
||||
}
|
||||
|
||||
func (suite *StriperTestSuite) TestStriperSetObjectLayout() {
|
||||
ioctx := suite.defaultContext()
|
||||
defer ioctx.Destroy()
|
||||
|
||||
striper, err := New(ioctx)
|
||||
assert.NoError(suite.T(), err)
|
||||
defer striper.Destroy()
|
||||
|
||||
assert.NoError(suite.T(), striper.SetObjectLayoutStripeUnit(65536))
|
||||
assert.NoError(suite.T(), striper.SetObjectLayoutStripeCount(16))
|
||||
assert.NoError(suite.T(), striper.SetObjectLayoutObjectSize(8388608))
|
||||
}
|
||||
|
||||
func (suite *StriperTestSuite) TestNewStriperWithLayout() {
|
||||
ioctx := suite.defaultContext()
|
||||
defer ioctx.Destroy()
|
||||
|
||||
p := StriperLayout{65536, 16, 8388608}
|
||||
striper, err := NewWithLayout(ioctx, p)
|
||||
assert.NoError(suite.T(), err)
|
||||
striper.Destroy()
|
||||
}
|
||||
|
||||
func TestStriperTestSuite(t *testing.T) {
|
||||
tsuite.Run(t, new(StriperTestSuite))
|
||||
}
|
||||
163
rados/striper/write_test.go
Normal file
163
rados/striper/write_test.go
Normal file
@ -0,0 +1,163 @@
|
||||
//go:build ceph_preview
|
||||
|
||||
package striper
|
||||
|
||||
import (
|
||||
"time"
|
||||
|
||||
"github.com/stretchr/testify/assert"
|
||||
"github.com/stretchr/testify/require"
|
||||
)
|
||||
|
||||
func (suite *StriperTestSuite) TestReadWrite() {
|
||||
ioctx := suite.defaultContext()
|
||||
defer ioctx.Destroy()
|
||||
|
||||
striper, err := New(ioctx)
|
||||
require.NoError(suite.T(), err)
|
||||
defer striper.Destroy()
|
||||
|
||||
name := "TestReadWrite"
|
||||
err = striper.Write(name, []byte("hello world"), 0)
|
||||
require.NoError(suite.T(), err)
|
||||
|
||||
err = striper.Write(name, []byte("earthlings"), 6)
|
||||
require.NoError(suite.T(), err)
|
||||
|
||||
buf := make([]byte, 32)
|
||||
size, err := striper.Read(name, buf, 0)
|
||||
require.NoError(suite.T(), err)
|
||||
assert.Equal(suite.T(), 16, size)
|
||||
assert.Equal(suite.T(), "hello earthlings", string(buf[:size]))
|
||||
}
|
||||
|
||||
func (suite *StriperTestSuite) TestReadWriteFull() {
|
||||
ioctx := suite.defaultContext()
|
||||
defer ioctx.Destroy()
|
||||
|
||||
striper, err := New(ioctx)
|
||||
require.NoError(suite.T(), err)
|
||||
defer striper.Destroy()
|
||||
|
||||
name := "TestReadWriteFull"
|
||||
err = striper.WriteFull(name, []byte("hello world"))
|
||||
require.NoError(suite.T(), err)
|
||||
|
||||
err = striper.WriteFull(name, []byte("earthlings"))
|
||||
require.NoError(suite.T(), err)
|
||||
|
||||
buf := make([]byte, 32)
|
||||
size, err := striper.Read(name, buf, 0)
|
||||
require.NoError(suite.T(), err)
|
||||
assert.Equal(suite.T(), 10, size)
|
||||
assert.Equal(suite.T(), "earthlings", string(buf[:size]))
|
||||
}
|
||||
|
||||
func (suite *StriperTestSuite) TestReadAppend() {
|
||||
ioctx := suite.defaultContext()
|
||||
defer ioctx.Destroy()
|
||||
|
||||
striper, err := New(ioctx)
|
||||
require.NoError(suite.T(), err)
|
||||
defer striper.Destroy()
|
||||
|
||||
name := "TestReadAppend"
|
||||
err = striper.Append(name, []byte("hello world"))
|
||||
require.NoError(suite.T(), err)
|
||||
|
||||
err = striper.Append(name, []byte(" and all you earthlings"))
|
||||
require.NoError(suite.T(), err)
|
||||
|
||||
buf := make([]byte, 64)
|
||||
size, err := striper.Read(name, buf, 0)
|
||||
require.NoError(suite.T(), err)
|
||||
assert.Equal(suite.T(), 34, size)
|
||||
assert.Equal(suite.T(),
|
||||
"hello world and all you earthlings",
|
||||
string(buf[:size]))
|
||||
}
|
||||
|
||||
func (suite *StriperTestSuite) TestRemove() {
|
||||
ioctx := suite.defaultContext()
|
||||
defer ioctx.Destroy()
|
||||
|
||||
striper, err := New(ioctx)
|
||||
require.NoError(suite.T(), err)
|
||||
defer striper.Destroy()
|
||||
|
||||
name := "TestRemove"
|
||||
err = striper.Append(name, []byte("goodbye world"))
|
||||
require.NoError(suite.T(), err)
|
||||
|
||||
err = striper.Remove(name)
|
||||
require.NoError(suite.T(), err)
|
||||
|
||||
buf := make([]byte, 32)
|
||||
_, err = striper.Read(name, buf, 0)
|
||||
require.Error(suite.T(), err)
|
||||
require.Equal(suite.T(), -2, err.(radosStriperError).ErrorCode())
|
||||
}
|
||||
|
||||
func (suite *StriperTestSuite) TestTruncate() {
|
||||
ioctx := suite.defaultContext()
|
||||
defer ioctx.Destroy()
|
||||
|
||||
striper, err := New(ioctx)
|
||||
require.NoError(suite.T(), err)
|
||||
defer striper.Destroy()
|
||||
|
||||
name := "TestTruncate"
|
||||
err = striper.Write(name, []byte("that's all folks"), 0)
|
||||
require.NoError(suite.T(), err)
|
||||
|
||||
err = striper.Truncate(name, 10)
|
||||
require.NoError(suite.T(), err)
|
||||
|
||||
buf := make([]byte, 32)
|
||||
size, err := striper.Read(name, buf, 0)
|
||||
require.NoError(suite.T(), err)
|
||||
assert.Equal(suite.T(), 10, size)
|
||||
assert.Equal(suite.T(), "that's all", string(buf[:size]))
|
||||
}
|
||||
|
||||
func (suite *StriperTestSuite) TestStat() {
|
||||
ut := time.Now().Unix()
|
||||
|
||||
ioctx := suite.defaultContext()
|
||||
defer ioctx.Destroy()
|
||||
|
||||
striper, err := New(ioctx)
|
||||
require.NoError(suite.T(), err)
|
||||
defer striper.Destroy()
|
||||
|
||||
name := "TestStat"
|
||||
err = striper.Write(name, []byte("that's all..."), 0)
|
||||
require.NoError(suite.T(), err)
|
||||
err = striper.Write(name, []byte("...folks"), 4096)
|
||||
require.NoError(suite.T(), err)
|
||||
|
||||
ss, err := striper.Stat(name)
|
||||
require.NoError(suite.T(), err)
|
||||
|
||||
assert.Equal(suite.T(), uint64(4104), ss.Size)
|
||||
assert.Equal(suite.T(), uint64(4104), ss.Size)
|
||||
// the Seconds field of the timespec should be between the time we started
|
||||
// the test and two seconds after the time we started the test (as that
|
||||
// ought to be a pretty large window for this operation)
|
||||
assert.GreaterOrEqual(suite.T(), ss.ModTime.Sec, ut)
|
||||
assert.GreaterOrEqual(suite.T(), ut+2, ss.ModTime.Sec)
|
||||
}
|
||||
|
||||
func (suite *StriperTestSuite) TestStatMissing() {
|
||||
ioctx := suite.defaultContext()
|
||||
defer ioctx.Destroy()
|
||||
|
||||
striper, err := New(ioctx)
|
||||
require.NoError(suite.T(), err)
|
||||
defer striper.Destroy()
|
||||
|
||||
name := "TestStatMissing"
|
||||
_, err = striper.Stat(name)
|
||||
require.Error(suite.T(), err)
|
||||
require.Equal(suite.T(), -2, err.(radosStriperError).ErrorCode())
|
||||
}
|
||||
Loading…
Reference in New Issue
Block a user