editor_test.go

package main

import (
	"fmt"
	"log"
	"testing"
)

type I = Issue

func createIssuesImpl(children, level, recurse, id int, status IssueStatus) *Issue {
	issue := &Issue{
		ID:     id,
		Title:  fmt.Sprintf("Child Issue id(%v) level(%v)", id, level),
		Status: status,
	}

	if children > 0 {
		issue.Children = make([]*Issue, children)
	}

	for i := 0; i < children; i++ {
		// yes, for tests we support not more than 10 subissues
		issueID := 10*id + i

		if recurse > 0 {
			issue.Children[i] = createIssuesImpl(children, level+1, recurse-1, issueID, status)
		} else {
			issue.Children[i] = &Issue{
				Title: fmt.Sprintf("Child Issue id(%v) level(%v)", issueID, level+1),
			}
		}

		issue.Children[i].ID = issueID
		issue.Children[i].Status = status

		log.Printf("Created issue: %v", issue.Children[i].FormatTitle(0))
	}
	return issue
}

func createIssues(children, level, recurse int, status IssueStatus) *Issue {
	return createIssuesImpl(children, level, recurse, 1 /*parent id*/, status)
}

func EditorSuite(t *testing.T, e *Editor, i *Issue, addMissing bool, initialBody, expectedBody string, changes int) {
	i.Body = initialBody
	body, changeLog, err := e.Update(i, addMissing)
	if err != nil {
		t.Fatal(err)
	}

	if body != expectedBody {
		t.Errorf("Body does not match. actual=%v expected=%v", body, expectedBody)
	}

	if len(changeLog) != changes {
		t.Errorf("Changes count does not match. actual=%v expected=%v", len(changeLog), changes)
	}
}

func EditSuite(t *testing.T, i *Issue, initialBody, expectedBody string, changes int) {
	e := &Editor{}
	EditorSuite(t, e, i, false /*addMissing*/, initialBody, expectedBody, changes)
}

func EditAppendSuite(t *testing.T, i *Issue, initialBody, expectedBody string, changes int) {
	e := &Editor{}
	EditorSuite(t, e, i, true /*addMissing*/, initialBody, expectedBody, changes)
}

func TestNoChildren(t *testing.T) {
	body := `
	abcd

	efgh
	`
	issue := createIssues(
		0 /*children*/, 0 /*level*/, 0 /*recurse*/, StatusOpened)
	EditSuite(t, issue, body, body, 0 /*changes*/)
}

func TestAddOneChildSingleLineBody(t *testing.T) {
	body := "abcd"
	expected := `abcd

### Child issues:

- [ ] Child Issue id(10) level(1) #10
`
	issue := createIssues(

		1 /*children*/, 0 /*level*/, 0 /*recurse*/, StatusOpened)
	EditSuite(t, issue, body, expected, 1 /*changes*/)
}

func TestAddOneChildEmptyBody(t *testing.T) {
	body := ""
	expected := `### Child issues:

- [ ] Child Issue id(10) level(1) #10
`
	issue := createIssues(
		1 /*children*/, 0 /*level*/, 0 /*recurse*/, StatusOpened)
	EditSuite(t, issue, body, expected, 1 /*changes*/)
}

func TestAddFewChildrenSingleLineBody(t *testing.T) {
	body := "abcd"
	expected := `abcd

### Child issues:

- [ ] Child Issue id(10) level(1) #10
- [ ] Child Issue id(11) level(1) #11
`
	issue := createIssues(
		2 /*children*/, 0 /*level*/, 0 /*recurse*/, StatusOpened)
	EditSuite(t, issue, body, expected, 1 /*changes*/)
}

func TestAddHierarchySingleLineBody(t *testing.T) {
	body := "abcd"
	expected := `abcd

### Child issues:

- [ ] Child Issue id(10) level(1) #10
  - [ ] Child Issue id(100) level(2) #100
  - [ ] Child Issue id(101) level(2) #101
- [ ] Child Issue id(11) level(1) #11
  - [ ] Child Issue id(110) level(2) #110
  - [ ] Child Issue id(111) level(2) #111
`
	issue := createIssues(
		2 /*children*/, 0 /*level*/, 1 /*recurse*/, StatusOpened)
	EditSuite(t, issue, body, expected, 1 /*changes*/)
}

func TestAddHierarchyMaxLevelsBody(t *testing.T) {
	body := "abcd"
	expected := `abcd

### Child issues:

- [ ] Child Issue id(10) level(1) #10
- [ ] Child Issue id(11) level(1) #11
`
	issue := createIssues(
		2 /*children*/, 0 /*level*/, 1 /*recurse*/, StatusOpened)
	EditorSuite(t,
		&Editor{MaxLevels: 1},
		issue, false, body, expected, 1 /*changes*/)
}

func TestAddFewChildrenNewlines(t *testing.T) {
	body := `abcd  

`
	expected := `abcd

### Child issues:

- [ ] Child Issue id(10) level(1) #10
- [ ] Child Issue id(11) level(1) #11
`
	issue := createIssues(
		2 /*children*/, 0 /*level*/, 0 /*recurse*/, StatusOpened)
	EditSuite(t, issue, body, expected, 1 /*changes*/)
}

func TestFewChildrenEmptyBody(t *testing.T) {
	body := ""
	expected := `### Child issues:

- [ ] Child Issue id(10) level(1) #10
- [ ] Child Issue id(11) level(1) #11
- [ ] Child Issue id(12) level(1) #12
`
	issue := createIssues(
		3 /*children*/, 0 /*level*/, 0 /*recurse*/, StatusOpened)
	EditSuite(t, issue, body, expected, 1 /*changes*/)
}

func TestUpdateCheckOneChildEmptyBody(t *testing.T) {
	body := `### Child issues:

- [ ] Child Issue id(10) level(1) #10
`

	expected := `### Child issues:

- [x] Child Issue id(10) level(1) #10
`

	issue := createIssues(
		1 /*children*/, 0 /*level*/, 0 /*recurse*/, StatusClosed)
	EditSuite(t, issue, body, expected, 1 /*changes*/)
}

func TestUpdateUncheckOneChildEmptyBody(t *testing.T) {
	body := `### Child issues:

- [x] Child Issue id(10) level(1) #10
`

	expected := `### Child issues:

- [ ] Child Issue id(10) level(1) #10
`

	issue := createIssues(
		1 /*children*/, 0 /*level*/, 0 /*recurse*/, StatusOpened)
	EditSuite(t, issue, body, expected, 1 /*changes*/)
}

func TestUpdateCheckFewChildrenNonEmptyBody(t *testing.T) {
	body := `abcd
efgh

### Child issues:

- [ ] Child Issue id(11) level(1) #11
- [ ] Child Issue id(10) level(1) #10
`

	expected := `abcd
efgh

### Child issues:

- [x] Child Issue id(11) level(1) #11
- [x] Child Issue id(10) level(1) #10
`

	issue := createIssues(
		2 /*children*/, 0 /*level*/, 0 /*recurse*/, StatusClosed)
	EditSuite(t, issue, body, expected, 2 /*changes*/)
}

func TestUpdateUncheckFewChildrenNonEmptyBody(t *testing.T) {
	body := `abcd
efgh

### Child issues:

- [x] Child Issue id(11) level(1) #11
- [x] Child Issue id(10) level(1) #10
`

	expected := `abcd
efgh

### Child issues:

- [ ] Child Issue id(11) level(1) #11
- [ ] Child Issue id(10) level(1) #10
`

	issue := createIssues(
		2 /*children*/, 0 /*level*/, 0 /*recurse*/, StatusOpened)
	EditSuite(t, issue, body, expected, 2 /*changes*/)
}

func TestMultiLevelHangingChildrenUpdate(t *testing.T) {
	body := `abcd
efgh

### Child issues:

- [x] Child Issue id(10) level(1) #10
  - [x] Child Issue id(100) level(2) #100
    - [x] Child Issue id(1000) level(3) #1000
      - [x] Child Issue id(10000) level(4) #10000
`

	expected := `abcd
efgh

### Child issues:

- [ ] Child Issue id(10) level(1) #10
  - [ ] Child Issue id(100) level(2) #100
    - [ ] Child Issue id(1000) level(3) #1000
      - [ ] Child Issue id(10000) level(4) #10000
`

	issue := createIssues(
		1 /*children*/, 0 /*level*/, 3 /*recurse*/, StatusOpened)
	EditSuite(t, issue, body, expected, 4 /*changes*/)
}

func TestMultiLevelOneChildUpdate(t *testing.T) {
	body := `abcd
efgh

### Child issues:

- [x] Child Issue id(10) level(1) #10
  - [x] Child Issue id(100) level(2) #100
- [x] Child Issue id(11) level(1) #11
`

	expected := `abcd
efgh

### Child issues:

- [ ] Child Issue id(10) level(1) #10
  - [ ] Child Issue id(100) level(2) #100
- [ ] Child Issue id(11) level(1) #11
`

	issue := createIssues(
		2 /*children*/, 0 /*level*/, 1 /*recurse*/, StatusOpened)
	EditSuite(t, issue, body, expected, 3 /*changes*/)
}

func TestMultiLevelFewChildrenUpdate(t *testing.T) {
	body := `abcd
efgh

### Child issues:

- [ ] Child Issue id(10) level(1) #10
  - [ ] Child Issue id(100) level(2) #100
  - [x] Child Issue id(101) level(2) #101
    - [x] Child Issue id(1010) level(3) #1010
    - [ ] Child Issue id(1011) level(3) #1011
- [x] Child Issue id(11) level(1) #11
  - [ ] Child Issue id(110) level(2) #110
`

	expected := `abcd
efgh

### Child issues:

- [ ] Child Issue id(10) level(1) #10
  - [x] Child Issue id(100) level(2) #100
  - [ ] Child Issue id(101) level(2) #101
    - [x] Child Issue id(1010) level(3) #1010
    - [x] Child Issue id(1011) level(3) #1011
- [x] Child Issue id(11) level(1) #11
  - [ ] Child Issue id(110) level(2) #110
`

	issue := createIssues(
		2 /*children*/, 0 /*level*/, 2 /*recurse*/, StatusOpened)
	issue.Children[0].Children[0].Status = StatusClosed
	issue.Children[0].Children[1].Children[0].Status = StatusClosed
	issue.Children[0].Children[1].Children[1].Status = StatusClosed
	issue.Children[1].Status = StatusClosed

	EditSuite(t, issue, body, expected, 3 /*changes*/)
}

func TestMultiLevelUpdateMaxLevel(t *testing.T) {
	body := `abcd
efgh

### Child issues:

- [ ] Child Issue id(10) level(1) #10
  - [ ] Child Issue id(100) level(2) #100
  - [x] Child Issue id(101) level(2) #101
    - [x] Child Issue id(1010) level(3) #1010
    - [ ] Child Issue id(1011) level(3) #1011
- [x] Child Issue id(11) level(1) #11
  - [ ] Child Issue id(110) level(2) #110
`

	expected := `abcd
efgh

### Child issues:

- [ ] Child Issue id(10) level(1) #10
  - [x] Child Issue id(100) level(2) #100
  - [ ] Child Issue id(101) level(2) #101
    - [x] Child Issue id(1010) level(3) #1010
    - [ ] Child Issue id(1011) level(3) #1011
- [x] Child Issue id(11) level(1) #11
  - [ ] Child Issue id(110) level(2) #110
`

	issue := createIssues(
		2 /*children*/, 0 /*level*/, 2 /*recurse*/, StatusOpened)
	issue.Children[0].Children[0].Status = StatusClosed
	issue.Children[0].Children[1].Children[0].Status = StatusClosed
	issue.Children[0].Children[1].Children[1].Status = StatusClosed
	issue.Children[1].Status = StatusClosed

	EditorSuite(t, &Editor{MaxLevels: 2},
		issue, false /*add missing*/, body, expected, 2 /*changes*/)
}

func TestAppendNewChild(t *testing.T) {
	body := `abcd
efgh

### Child issues:

- [ ] Child Issue id(10) level(1) #10
`

	expected := `abcd
efgh

### Child issues:

- [ ] Child Issue id(10) level(1) #10
- [ ] Child Issue id(11) level(1) #11
`

	issue := createIssues(
		2 /*children*/, 0 /*level*/, 0 /*recurse*/, StatusOpened)
	EditAppendSuite(t, issue, body, expected, 1 /*changes*/)
}

func TestAppendTwoLevels(t *testing.T) {
	body := `abcd
efgh

### Child issues:

- [ ] Child Issue id(11) level(1) #11
- [ ] Child Issue id(10) level(1) #10
`

	expected := `abcd
efgh

### Child issues:

- [ ] Child Issue id(11) level(1) #11
  - [ ] Child Issue id(110) level(2) #110
  - [ ] Child Issue id(111) level(2) #111
- [ ] Child Issue id(10) level(1) #10
  - [ ] Child Issue id(100) level(2) #100
  - [ ] Child Issue id(101) level(2) #101
`
	issue := createIssues(
		2 /*children*/, 0 /*level*/, 1 /*recurse*/, StatusOpened)
	EditAppendSuite(t, issue, body, expected, 2 /*changes*/)
}

func TestAppendSkipChild(t *testing.T) {
	body := `abcd
efgh

### Child issues:

- [x] Child Issue id(11) level(1) #11
  - [x] Child Issue id(111) level(2) #111
`

	expected := `abcd
efgh

### Child issues:

- [ ] Child Issue id(11) level(1) #11
  - [ ] Child Issue id(111) level(2) #111
  - [ ] Child Issue id(110) level(2) #110
- [ ] Child Issue id(10) level(1) #10
  - [ ] Child Issue id(100) level(2) #100
  - [ ] Child Issue id(101) level(2) #101
`

	issue := createIssues(
		2 /*children*/, 0 /*level*/, 1 /*recurse*/, StatusOpened)
	EditAppendSuite(t, issue, body, expected, 4 /*changes*/)
}

func TestAppendMultilevel(t *testing.T) {
	body := `abcd
efgh

### Child issues:

- [x] Child Issue id(11) level(1) #11
  - [x] Child Issue id(110) level(2) #110
    - [x] Child Issue id(1100) level(3) #1100
`

	expected := `abcd
efgh

### Child issues:

- [ ] Child Issue id(11) level(1) #11
  - [ ] Child Issue id(110) level(2) #110
    - [ ] Child Issue id(1100) level(3) #1100
    - [ ] Child Issue id(1101) level(3) #1101
  - [ ] Child Issue id(111) level(2) #111
    - [ ] Child Issue id(1110) level(3) #1110
    - [ ] Child Issue id(1111) level(3) #1111
- [ ] Child Issue id(10) level(1) #10
  - [ ] Child Issue id(100) level(2) #100
    - [ ] Child Issue id(1000) level(3) #1000
    - [ ] Child Issue id(1001) level(3) #1001
  - [ ] Child Issue id(101) level(2) #101
    - [ ] Child Issue id(1010) level(3) #1010
    - [ ] Child Issue id(1011) level(3) #1011
`

	issue := createIssues(
		2 /*children*/, 0 /*level*/, 2 /*recurse*/, StatusOpened)
	EditAppendSuite(t, issue, body, expected, 6 /*changes*/)
}

func TestAppendSkipChildCustom(t *testing.T) {
	body := `abcd
efgh

### Child issues:

- [ ] Child Issue id(11) level(1) #11
  - [x] Child Issue id(111) level(2) #111
- [ ] Child Issue id(10) level(1) #10
`

	expected := `abcd
efgh

### Child issues:

- [ ] Child Issue id(11) level(1) #11
  - [x] Child Issue id(111) level(2) #111
  - [ ] Child Issue id(110) level(2) #110
- [ ] Child Issue id(10) level(1) #10
`

	issue := &Issue{
		ID:     1,
		Title:  "Child Issue id(1) level(0)",
		Status: StatusOpened,
		Children: []*Issue{
			&Issue{
				ID:     11,
				Title:  "Child Issue id(11) level(1)",
				Status: StatusOpened,
				Children: []*Issue{
					&Issue{
						ID:     111,
						Title:  "Child Issue id(111) level(2)",
						Status: StatusClosed,
					},
					&Issue{
						ID:     110,
						Title:  "Child Issue id(110) level(2)",
						Status: StatusOpened,
					},
				},
			},
			&Issue{
				ID:     10,
				Title:  "Child Issue id(10) level(1)",
				Status: StatusOpened,
			},
		},
	}

	EditAppendSuite(t, issue, body, expected, 1 /*changes*/)
}