package/storage
1
0
Fork 0
mirror of https://github.com/gofiber/storage.git synced 2025-10-05 16:48:25 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

♻️ improve test

Browse Source
This commit is contained in:
Wei Lun
2020-11-01 20:50:47 +08:00
parent fa1e1591bb
commit d6a5f1431e
2 changed files with 35 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
mongodb/mongodb.go
View File

@@ -106,10 +106,10 @@ func (s *Storage) Get(key string) ([]byte, error) {
result := MongoStorage{} result := MongoStorage{}
if err := res.Err(); err != nil { if err := res.Err(); err != nil {
return []byte{}, err return nil, err
} }
if err := res.Decode(&result); err != nil { if err := res.Decode(&result); err != nil {
return []byte{}, err return nil, err
} }
return result.Value, nil return result.Value, nil

41
mongodb/mongodb_test.go
View File

@@ -46,13 +46,32 @@ func TestMongoStore_Set_Get(t *testing.T) {
key := "example_key" key := "example_key"
value := []byte("123") value := []byte("123")
_ = store.Set(key, value, 0) err := store.Set(key, value, 0)
utils.AssertEqual(t, nil, err)
getVal, _ := store.Get(key) getVal, getErr := store.Get(key)
utils.AssertEqual(t, nil, getErr)
utils.AssertEqual(t, value, getVal, "correctly set and get value") utils.AssertEqual(t, value, getVal, "correctly set and get value")
} }
func TestMongoStore_Get_Invalid(t *testing.T) {
if uri == "" {
t.Skip()
}
store := New(getConfig())
defer func() {
_ = store.db.Client().Disconnect(context.TODO())
}()
key := "random_invalid_key"
getVal, getErr := store.Get(key)
utils.AssertEqual(t, true, getErr != nil)
utils.AssertEqual(t, true, getVal == nil, "get nil if key not found")
}
func TestMongoStore_Delete(t *testing.T) { func TestMongoStore_Delete(t *testing.T) {
if uri == "" { if uri == "" {
t.Skip() t.Skip()
@@ -65,12 +84,15 @@ func TestMongoStore_Delete(t *testing.T) {
key := "example_key_2" key := "example_key_2"
value := []byte("123") value := []byte("123")
_ = store.Set(key, value, 10) err := store.Set(key, value, 0)
_ = store.Delete(key) utils.AssertEqual(t, nil, err)
getVal, _ := store.Get(key) err = store.Delete(key)
utils.AssertEqual(t, nil, err)
utils.AssertEqual(t, []byte{}, getVal, "correctly delete value") getVal, getErr := store.Get(key)
utils.AssertEqual(t, true, getErr != nil)
utils.AssertEqual(t, true, getVal == nil, "correctly delete value")
} }
func TestMongoStore_Clear(t *testing.T) { func TestMongoStore_Clear(t *testing.T) {
@@ -85,12 +107,15 @@ func TestMongoStore_Clear(t *testing.T) {
key := "example_key_2" key := "example_key_2"
value := []byte("123") value := []byte("123")
_ = store.Set(key, value, 10) err := store.Set(key, value, 10)
names, _ := store.db.ListCollectionNames(context.TODO(), bson.D{}) names, _ := store.db.ListCollectionNames(context.TODO(), bson.D{})
utils.AssertEqual(t, nil, err)
utils.AssertEqual(t, true, contains(names, colName), "has collection") utils.AssertEqual(t, true, contains(names, colName), "has collection")
_ = store.Clear()
cErr := store.Clear()
names2, _ := store.db.ListCollectionNames(context.TODO(), bson.D{}) names2, _ := store.db.ListCollectionNames(context.TODO(), bson.D{})
utils.AssertEqual(t, nil, cErr)
utils.AssertEqual(t, false, contains(names2, colName), "do not have collection") utils.AssertEqual(t, false, contains(names2, colName), "do not have collection")
} }