Kk db tests

models/test/board_model_test.go

@@ -0,0 +1,18 @@
+package test
+
+import (
+	"testing"
+
+	"github.com/merico-dev/lake/models"
+	"github.com/merico-dev/lake/models/test/factory"
+	"github.com/stretchr/testify/assert"
+)
+
+func TestInsertBoard(t *testing.T) {
+	board, err := factory.CreateBoard()
+	assert.Nil(t, err)
+	tx := models.Db.Create(&board)
+	assert.Nil(t, tx.Error)
+	expected := int64(1)
+	assert.Equal(t, tx.RowsAffected, expected)
+}

models/test/build_model_test.go

@@ -0,0 +1,20 @@
+package test
+
+import (
+	"testing"
+
+	"github.com/merico-dev/lake/models"
+	"github.com/merico-dev/lake/models/test/factory"
+	"github.com/stretchr/testify/assert"
+)
+
+func TestInsertBuild(t *testing.T) {
+	job, err := factory.CreateJob()
+	assert.Nil(t, err)
+	build, err := factory.CreateBuild(job.DomainEntity.Id)
+	assert.Nil(t, err)
+	tx := models.Db.Create(&build)
+	assert.Nil(t, tx.Error)
+	expected := int64(1)
+	assert.Equal(t, tx.RowsAffected, expected)
+}

models/test/changelog_model_test.go

@@ -0,0 +1,22 @@
+package test
+
+import (
+	"testing"
+
+	"github.com/merico-dev/lake/models"
+	"github.com/merico-dev/lake/models/test/factory"
+	"github.com/stretchr/testify/assert"
+)
+
+func TestInsertChangelog(t *testing.T) {
+	board, err := factory.CreateBoard()
+	assert.Nil(t, err)
+	issue, err := factory.CreateIssue(board.DomainEntity.Id)
+	assert.Nil(t, err)
+	changelog, err := factory.CreateChangelog(issue.DomainEntity.Id)
+	assert.Nil(t, err)
+	tx := models.Db.Create(&changelog)
+	assert.Nil(t, tx.Error)
+	expected := int64(1)
+	assert.Equal(t, tx.RowsAffected, expected)
+}

models/test/factory/board_factory.go

@@ -0,0 +1,17 @@
+package factory
+
+import (
+	"github.com/merico-dev/lake/models/domainlayer"
+	"github.com/merico-dev/lake/models/domainlayer/ticket"
+)
+
+func CreateBoard() (*ticket.Board, error) {
+	board := &ticket.Board{
+		DomainEntity: domainlayer.DomainEntity{
+			Id: RandIntString(),
+		},
+		Name: "",
+		Url:  "",
+	}
+	return board, nil
+}

models/test/factory/build_factory.go

@@ -0,0 +1,23 @@
+package factory
+
+import (
+	"time"
+
+	"github.com/merico-dev/lake/models/domainlayer"
+	"github.com/merico-dev/lake/models/domainlayer/devops"
+)
+
+func CreateBuild(jobId string) (*devops.Build, error) {
+	build := &devops.Build{
+		DomainEntity: domainlayer.DomainEntity{
+			Id: RandIntString(),
+		},
+		JobId:       jobId, // ref to job
+		Name:        "",
+		CommitSha:   "",
+		DurationSec: uint64(RandInt()),
+		Status:      "",
+		StartedDate: time.Now(),
+	}
+	return build, nil
+}

models/test/factory/changelog_factory.go

@@ -0,0 +1,25 @@
+package factory
+
+import (
+	"time"
+
+	"github.com/merico-dev/lake/models/domainlayer"
+	"github.com/merico-dev/lake/models/domainlayer/ticket"
+)
+
+func CreateChangelog(issueId string) (*ticket.Changelog, error) {
+	changelog := &ticket.Changelog{
+		DomainEntity: domainlayer.DomainEntity{
+			Id: RandIntString(),
+		},
+		IssueId:     issueId, // ref to issue
+		AuthorId:    "",
+		AuthorName:  "",
+		FieldId:     "",
+		FieldName:   "",
+		From:        "",
+		To:          "",
+		CreatedDate: time.Now(),
+	}
+	return changelog, nil
+}

models/test/factory/commit_factory.go

@@ -10,13 +10,13 @@ import (
 func CreateCommit(repoId uint64) (*code.Commit, error) {
 	commit := &code.Commit{
 		DomainEntity: domainlayer.DomainEntity{
-			Id: "something",
+			Id: RandIntString(),
 		},
 		RepoId:         repoId,
 		Sha:            "dosifj9302hf80h23f",
-		Additions:      33,
-		Deletions:      33,
-		DevEq:          33,
+		Additions:      RandInt(),
+		Deletions:      RandInt(),
+		DevEq:          RandInt(),
 		Message:        "",
 		AuthorName:     "",
 		AuthorEmail:    "",

models/test/factory/issue_factory.go

@@ -0,0 +1,39 @@
+package factory
+
+import (
+	"time"
+
+	"github.com/merico-dev/lake/models/domainlayer"
+	"github.com/merico-dev/lake/models/domainlayer/ticket"
+)
+
+func CreateIssue(boardId string) (*ticket.Issue, error) {
+	issue := &ticket.Issue{
+		DomainEntity: domainlayer.DomainEntity{
+			Id: RandIntString(),
+		},
+		BoardId:                  boardId, // ref to board
+		Url:                      "",
+		Key:                      "",
+		Title:                    "",
+		Summary:                  "",
+		EpicKey:                  "",
+		Type:                     "",
+		Status:                   "",
+		StoryPoint:               1,
+		OriginalEstimateMinutes:  1, // user input?
+		AggregateEstimateMinutes: 1, // sum up of all subtasks?
+		RemainingEstimateMinutes: 1, // could it be negative value?
+		CreatorId:                "",
+		AssigneeId:               "",
+		ResolutionDate:           nil,
+		Priority:                 "", // not sure how to deal with it yet, copy the name for now
+		ParentId:                 "",
+		SprintId:                 "",
+		CreatedDate:              time.Now(),
+		UpdatedDate:              time.Now(),
+		SpentMinutes:             1,
+		LeadTimeMinutes:          1,
+	}
+	return issue, nil
+}

models/test/factory/job_factory.go

@@ -0,0 +1,16 @@
+package factory
+
+import (
+	"github.com/merico-dev/lake/models/domainlayer"
+	"github.com/merico-dev/lake/models/domainlayer/devops"
+)
+
+func CreateJob() (*devops.Job, error) {
+	job := &devops.Job{
+		DomainEntity: domainlayer.DomainEntity{
+			Id: RandIntString(),
+		},
+		Name: "",
+	}
+	return job, nil
+}

models/test/factory/note_factory.go

@@ -10,7 +10,7 @@ import (
 func CreateNote(prId uint64) (*code.Note, error) {
 	note := &code.Note{
 		DomainEntity: domainlayer.DomainEntity{
-			Id: "something",
+			Id: RandIntString(),
 		},
 		PrId:        prId,
 		Author:      "note.AuthorUsername",

models/test/factory/pipeline_factory.go

@@ -7,10 +7,10 @@ import (
 func CreatePipeline() (*models.Pipeline, error) {
 	pipeline := &models.Pipeline{
 		Name:          "My Pipeline",
-		FinishedTasks: 0,
+		FinishedTasks: RandInt(),
 		Status:        "MY_STATUS",
 		Message:       "",
-		SpentSeconds:  0,
+		SpentSeconds:  RandInt(),
 	}
 	return pipeline, nil
 }

models/test/factory/pr_factory.go

@@ -10,7 +10,7 @@ import (
 func CreatePr(repoId uint64) (*code.Pr, error) {
 	pr := &code.Pr{
 		DomainEntity: domainlayer.DomainEntity{
-			Id: "something",
+			Id: RandIntString(),
 		},
 		RepoId:      repoId,
 		State:       "",

models/test/factory/repo_factory.go

@@ -8,7 +8,7 @@ import (
 func CreateRepo() (*code.Repo, error) {
 	repo := &code.Repo{
 		DomainEntity: domainlayer.DomainEntity{
-			Id: "something",
+			Id: RandIntString(),
 		},
 	}
 	return repo, nil

models/test/factory/sprint_factory.go

@@ -0,0 +1,22 @@
+package factory
+
+import (
+	"github.com/merico-dev/lake/models/domainlayer"
+	"github.com/merico-dev/lake/models/domainlayer/ticket"
+)
+
+func CreateSprint(boardId string) (*ticket.Sprint, error) {
+	sprint := &ticket.Sprint{
+		DomainEntity: domainlayer.DomainEntity{
+			Id: RandIntString(),
+		},
+		BoardId:      boardId, // ref to board
+		Url:          "",
+		State:        "",
+		Name:         "",
+		StartDate:    nil,
+		EndDate:      nil,
+		CompleteDate: nil,
+	}
+	return sprint, nil
+}

models/test/factory/task_factory.go

@@ -10,7 +10,7 @@ func CreateTask(pipelineId uint64) (*models.Task, error) {
 		Options:    nil,
 		Status:     "MY_STATUS",
 		Message:    "my message",
-		PipelineId: pipelineId,
+		PipelineId: pipelineId, // ref to pipeline
 	}
 	return task, nil
 }

models/test/factory/user_factory.go

@@ -0,0 +1,19 @@
+package factory
+
+import (
+	"github.com/merico-dev/lake/models/domainlayer"
+	"github.com/merico-dev/lake/models/domainlayer/user"
+)
+
+func CreateUser() (*user.User, error) {
+	user := &user.User{
+		DomainEntity: domainlayer.DomainEntity{
+			Id: RandIntString(),
+		},
+		Name:      "",
+		Email:     "",
+		AvatarUrl: "",
+		Timezone:  "",
+	}
+	return user, nil
+}

models/test/factory/util.go

@@ -0,0 +1,17 @@
+package factory
+
+import (
+	"fmt"
+	"math"
+	"math/rand"
+)
+
+func RandInt() int {
+	seed := 1000000
+	r := int(math.Floor(rand.Float64() * float64(seed)))
+	return r
+}
+
+func RandIntString() string {
+	return fmt.Sprintf("something%v", RandInt())
+}

models/test/factory/worklog_factory.go

@@ -0,0 +1,25 @@
+package factory
+
+import (
+	"time"
+
+	"github.com/merico-dev/lake/models/domainlayer"
+	"github.com/merico-dev/lake/models/domainlayer/ticket"
+)
+
+func CreateWorklog(boardId string, issueId string) (*ticket.Worklog, error) {
+	worklog := &ticket.Worklog{
+		DomainEntity: domainlayer.DomainEntity{
+			Id: RandIntString(),
+		},
+		IssueId:          issueId, // ref to issue
+		BoardId:          boardId, // ref to board
+		AuthorId:         "",
+		UpdateAuthorId:   "",
+		TimeSpent:        "",
+		TimeSpentSeconds: RandInt(),
+		Updated:          time.Now(),
+		Started:          time.Now(),
+	}
+	return worklog, nil
+}

models/test/issue_model_test.go

@@ -0,0 +1,20 @@
+package test
+
+import (
+	"testing"
+
+	"github.com/merico-dev/lake/models"
+	"github.com/merico-dev/lake/models/test/factory"
+	"github.com/stretchr/testify/assert"
+)
+
+func TestInsertIssue(t *testing.T) {
+	board, err := factory.CreateBoard()
+	assert.Nil(t, err)
+	issue, err := factory.CreateIssue(board.DomainEntity.Id)
+	assert.Nil(t, err)
+	tx := models.Db.Create(&issue)
+	assert.Nil(t, tx.Error)
+	expected := int64(1)
+	assert.Equal(t, tx.RowsAffected, expected)
+}

models/test/job_model_test.go

@@ -0,0 +1,18 @@
+package test
+
+import (
+	"testing"
+
+	"github.com/merico-dev/lake/models"
+	"github.com/merico-dev/lake/models/test/factory"
+	"github.com/stretchr/testify/assert"
+)
+
+func TestInsertJob(t *testing.T) {
+	job, err := factory.CreateJob()
+	assert.Nil(t, err)
+	tx := models.Db.Create(&job)
+	assert.Nil(t, tx.Error)
+	expected := int64(1)
+	assert.Equal(t, tx.RowsAffected, expected)
+}