You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
80 lines
1.8 KiB
80 lines
1.8 KiB
package oidbased
|
|
|
|
import (
|
|
"encoding/json"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"go.mongodb.org/mongo-driver/bson/primitive"
|
|
|
|
"github.com/mainnika/mongox-go-driver/v2/mongox-testing/database"
|
|
)
|
|
|
|
func Test_GetID(t *testing.T) {
|
|
|
|
type DocWithObjectID struct {
|
|
Primary `bson:",inline" json:",inline" collection:"1"`
|
|
}
|
|
|
|
doc := &DocWithObjectID{Primary{[12]byte{1, 2, 3, 4, 5, 6, 7, 8, 9, 0, 1, 2}}}
|
|
|
|
assert.Equal(t, primitive.ObjectID([12]byte{1, 2, 3, 4, 5, 6, 7, 8, 9, 0, 1, 2}), doc.GetID())
|
|
}
|
|
|
|
func Test_SetID(t *testing.T) {
|
|
|
|
type DocWithObjectID struct {
|
|
Primary `bson:",inline" json:",inline" collection:"1"`
|
|
}
|
|
|
|
doc := &DocWithObjectID{}
|
|
|
|
doc.SetID([12]byte{1, 2, 3, 4, 5, 6, 7, 8, 9, 0, 1, 2})
|
|
|
|
assert.Equal(t, primitive.ObjectID([12]byte{1, 2, 3, 4, 5, 6, 7, 8, 9, 0, 1, 2}), doc.Primary.ID)
|
|
assert.Equal(t, primitive.ObjectID([12]byte{1, 2, 3, 4, 5, 6, 7, 8, 9, 0, 1, 2}), doc.GetID())
|
|
}
|
|
|
|
func Test_SaveLoad(t *testing.T) {
|
|
|
|
type DocWithObjectID struct {
|
|
Primary `bson:",inline" json:",inline" collection:"1"`
|
|
}
|
|
|
|
db, err := database.NewEphemeral("mongodb://localhost")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
defer db.Close()
|
|
|
|
doc1 := &DocWithObjectID{}
|
|
doc2 := &DocWithObjectID{}
|
|
|
|
err = db.SaveOne(doc1)
|
|
assert.NoError(t, err)
|
|
|
|
err = db.LoadOne(doc2)
|
|
assert.NoError(t, err)
|
|
|
|
assert.Equal(t, doc1, doc2)
|
|
|
|
bytes1, _ := json.Marshal(doc1)
|
|
bytes2, _ := json.Marshal(doc2)
|
|
|
|
assert.Equal(t, bytes1, bytes2)
|
|
}
|
|
|
|
func Test_Marshal(t *testing.T) {
|
|
|
|
type DocWithObjectID struct {
|
|
Primary `bson:",inline" json:",inline" collection:"1"`
|
|
}
|
|
|
|
id, _ := primitive.ObjectIDFromHex("feadbeeffeadbeeffeadbeef")
|
|
doc := &DocWithObjectID{Primary{id}}
|
|
|
|
bytes, err := json.Marshal(doc)
|
|
assert.NoError(t, err)
|
|
assert.Equal(t, `{"_id":"feadbeeffeadbeeffeadbeef"}`, string(bytes))
|
|
}
|
|
|