diamond-orm/testing.go

package orm

import (
	"context"
	"fmt"
	"time"

	"github.com/go-loremipsum/loremipsum"
	"go.mongodb.org/mongo-driver/bson"
	"go.mongodb.org/mongo-driver/bson/primitive"
	"go.mongodb.org/mongo-driver/mongo"
	"go.mongodb.org/mongo-driver/mongo/options"
)

type chapter struct {
	ID            primitive.ObjectID `bson:"_id" json:"_id"`
	Title         string             `bson:"chapterTitle" json:"chapterTitle" form:"chapterTitle"`
	ChapterID     int                `bson:"id" json:"chapterID"`
	Index         int                `bson:"index" json:"index" form:"index"`
	Words         int                `bson:"words" json:"words"`
	Notes         string             `bson:"notes" json:"notes" form:"notes"`
	Genre         []string           `bson:"genre" json:"genre" form:"genre"`
	Bands         []band             `json:"bands" bson:"bands" ref:"band,bands"`
	Characters    []string           `bson:"characters" json:"characters" form:"characters"`
	Relationships [][]string         `bson:"relationships" json:"relationships" form:"relationships"`
	Adult         bool               `bson:"adult" json:"adult" form:"adult"`
	Summary       string             `bson:"summary" json:"summary" form:"summary"`
	Hidden        bool               `bson:"hidden" json:"hidden" form:"hidden"`
	LoggedInOnly  bool               `bson:"loggedInOnly" json:"loggedInOnly" form:"loggedInOnly"`
	Posted        time.Time          `bson:"datePosted,omitempty" json:"datePosted"`
	FileName      string             `json:"fileName" bson:"-"`
	Text          string             `json:"text" bson:"-"`
}

type band struct {
	ID         int64 `bson:"_id" json:"_id"`
	Model      `bson:",inline" json:",inline" coll:"bands"`
	Name       string   `bson:"name" json:"name" form:"name"`
	Locked     bool     `bson:"locked" json:"locked" form:"locked"`
	Characters []string `bson:"characters" json:"characters" form:"characters"`
}
type user struct {
	ID       int64 `bson:"_id" json:"_id"`
	Model    `bson:",inline" json:",inline" coll:"users"`
	Username string `bson:"username" json:"username"`
}
type story struct {
	ID        int64 `bson:"_id" json:"_id"`
	Model     `bson:",inline" json:",inline" coll:"stories"`
	Title     string    `bson:"title" json:"title" form:"title"`
	Author    *user     `bson:"author" json:"author" ref:"user"`
	CoAuthor  *user     `bson:"coAuthor" json:"coAuthor" ref:"user"`
	Chapters  []chapter `bson:"chapters" json:"chapters"`
	Recs      int       `bson:"recs" json:"recs"`
	Favs      int       `bson:"favs" json:"favs"`
	Views     int       `bson:"views" json:"views"`
	Completed bool      `bson:"completed" json:"completed" form:"completed"`
	Downloads int       `bson:"downloads" json:"downloads"`
}

func (s *story) Id() any {
	return s.ID
}

func (s *band) Id() any {
	return s.ID
}
func (s *user) Id() any {
	return s.ID
}

func (s *story) SetId(id any) {
	s.ID = id.(int64)
	//var t IModel = s
}

func (s *band) SetId(id any) {
	s.ID = id.(int64)
}

func (s *user) SetId(id any) {
	s.ID = id.(int64)
}

var author = user{
	Username: "tablet.exe",
}

func genChaps(single bool) []chapter {
	var ret []chapter
	var ceil int
	if single {
		ceil = 1
	} else {
		ceil = 5
	}
	emptyRel := make([][]string, 0)
	emptyRel = append(emptyRel, make([]string, 0))
	relMap := [][][]string{
		{
			{"Sean Harris", "Brian Tatler"},
		},
		{
			{"Sean Harris", "Brian Tatler"},
			{"Duncan Scott", "Colin Kimberley"},
		},
		{
			{"Duncan Scott", "Colin Kimberley"},
		},
		emptyRel,
		{
			{"Sean Harris", "Colin Kimberley", "Brian Tatler"},
		},
	}
	var b band = band{
		Name:       "Diamond Head",
		Locked:     false,
		Characters: []string{"Brian Tatler", "Sean Harris", "Duncan Scott", "Colin Kimberley"},
	}
	b.ID = 503
	for i := 0; i < ceil; i++ {
		spf := fmt.Sprintf("%d.md", i+1)
		ret = append(ret, chapter{
			Title:         fmt.Sprintf("-%d-", i+1),
			Index:         int(i + 1),
			Words:         50,
			Notes:         "notenotenote !!!",
			Genre:         []string{"Slash"},
			Bands:         []band{b},
			Characters:    []string{"Sean Harris", "Brian Tatler", "Duncan Scott", "Colin Kimberley"},
			Relationships: relMap[i],
			Adult:         true,
			Summary:       loremipsum.New().Paragraph(),
			Hidden:        false,
			LoggedInOnly:  true,
			FileName:      spf,
		})
	}

	return ret
}

var iti_single story = story{
	Title:     "title",
	Completed: true,
	Chapters:  genChaps(true),
}

var iti_multi story = story{
	Title:     "Brian Tatler Fucked and Abused Sean Harris",
	Completed: false,
	Chapters:  genChaps(false),
}

func initTest() {
	uri := "mongodb://127.0.0.1:27017"
	db := "rockfic_ormTest"
	ic, _ := mongo.Connect(context.TODO(), options.Client().ApplyURI(uri))
	colls, _ := ic.Database(db).ListCollectionNames(context.TODO(), bson.M{})
	if len(colls) < 1 {
		mdb := ic.Database(db)
		mdb.CreateCollection(context.TODO(), "bands")
		mdb.CreateCollection(context.TODO(), "stories")
		mdb.CreateCollection(context.TODO(), "users")
	}
	defer ic.Disconnect(context.TODO())
	Connect(uri, db)
	author.ID = 696969
	ModelRegistry.Model(band{}, user{}, story{})
}
func after() {
	err := DBClient.Disconnect(context.TODO())
	panik(err)
}