code.gitea.io/gitea@v1.21.7/models/migrations/v1_18/v229_test.go (about)

     1  // Copyright 2022 The Gitea Authors. All rights reserved.
     2  // SPDX-License-Identifier: MIT
     3  
     4  package v1_18 //nolint
     5  
     6  import (
     7  	"testing"
     8  
     9  	"code.gitea.io/gitea/models/issues"
    10  	"code.gitea.io/gitea/models/migrations/base"
    11  
    12  	"github.com/stretchr/testify/assert"
    13  )
    14  
    15  func Test_UpdateOpenMilestoneCounts(t *testing.T) {
    16  	type ExpectedMilestone issues.Milestone
    17  
    18  	// Prepare and load the testing database
    19  	x, deferable := base.PrepareTestEnv(t, 0, new(issues.Milestone), new(ExpectedMilestone), new(issues.Issue))
    20  	defer deferable()
    21  	if x == nil || t.Failed() {
    22  		return
    23  	}
    24  
    25  	if err := UpdateOpenMilestoneCounts(x); err != nil {
    26  		assert.NoError(t, err)
    27  		return
    28  	}
    29  
    30  	expected := []ExpectedMilestone{}
    31  	if err := x.Table("expected_milestone").Asc("id").Find(&expected); !assert.NoError(t, err) {
    32  		return
    33  	}
    34  
    35  	got := []issues.Milestone{}
    36  	if err := x.Table("milestone").Asc("id").Find(&got); !assert.NoError(t, err) {
    37  		return
    38  	}
    39  
    40  	for i, e := range expected {
    41  		got := got[i]
    42  		assert.Equal(t, e.ID, got.ID)
    43  		assert.Equal(t, e.NumIssues, got.NumIssues)
    44  		assert.Equal(t, e.NumClosedIssues, got.NumClosedIssues)
    45  	}
    46  }