package stringbased import ( "encoding/json" "testing" "github.com/stretchr/testify/assert" "github.com/mainnika/mongox-go-driver/v2/mongox/tempdb" ) func Test_GetID(t *testing.T) { type DocWithString struct { Primary `bson:",inline" json:",inline" collection:"1"` } doc := &DocWithString{Primary{"foobar"}} assert.Equal(t, "foobar", doc.GetID()) } func Test_SetID(t *testing.T) { type DocWithString struct { Primary `bson:",inline" json:",inline" collection:"1"` } doc := &DocWithString{Primary{"foobar"}} doc.SetID("rockrockrock") assert.Equal(t, "rockrockrock", doc.Primary.ID) assert.Equal(t, "rockrockrock", doc.GetID()) } func Test_SaveLoad(t *testing.T) { type DocWithObjectID struct { Primary `bson:",inline" json:",inline" collection:"1"` } db, err := tempdb.NewTempDB("mongodb://localhost") if err != nil { t.Fatal(err) } defer db.Close() doc1 := &DocWithObjectID{Primary{"foobar"}} 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"` } doc := &DocWithObjectID{Primary{"foobar"}} bytes, err := json.Marshal(doc) assert.NoError(t, err) assert.Equal(t, `{"_id":"foobar"}`, string(bytes)) }