Implement ID bases in packages

mongox/base/stringbased/id.go

@@ -0,0 +1,22 @@
+package stringbased
+
+import (
+	"github.com/mainnika/mongox-go-driver/v2/mongox"
+)
+
+var _ mongox.StringBased = (*Primary)(nil)
+
+// Primary is a structure with string as an _id field
+type Primary struct {
+	ID string `bson:"_id" json:"_id"`
+}
+
+// GetID returns an _id
+func (p *Primary) GetID() string {
+	return p.ID
+}
+
+// SetID sets an _id
+func (p *Primary) SetID(id string) {
+	p.ID = id
+}

mongox/base/stringbased/id_test.go

@@ -0,0 +1,78 @@
+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))
+}