Fix unit test

This commit is contained in:
Tyrone Yeh 2025-07-24 13:49:13 +08:00
parent 1e51590711
commit 420b8520d2
No known key found for this signature in database
GPG Key ID: AC8B5AA00375E170
2 changed files with 9 additions and 13 deletions

View File

@ -30,7 +30,7 @@ type IndexerData struct {
LabelIDs []int64 `json:"label_ids"`
NoLabel bool `json:"no_label"` // True if LabelIDs is empty
MilestoneID int64 `json:"milestone_id"`
ProjectIDs []int64 `json:"project_ids"`
ProjectIDs []int64 `json:"project_id"`
ProjectColumnID int64 `json:"project_board_id"` // the key should be kept as project_board_id to keep compatible
PosterID int64 `json:"poster_id"`
AssigneeID int64 `json:"assignee_id"`

View File

@ -312,9 +312,7 @@ var cases = []*testIndexerCase{
Expected: func(t *testing.T, data map[int64]*internal.IndexerData, result *internal.SearchResult) {
assert.Len(t, result.Hits, 5)
for _, v := range result.Hits {
if len(data[v.ID].ProjectIDs) > 0 {
assert.Equal(t, int64(1), data[v.ID].ProjectIDs[0])
}
assert.Contains(t, data[v.ID].ProjectIDs, int64(1))
}
assert.Equal(t, countIndexerData(data, func(v *internal.IndexerData) bool {
if len(data[v.ID].ProjectIDs) > 0 {
@ -333,18 +331,12 @@ var cases = []*testIndexerCase{
ProjectIDs: []int64{0},
},
Expected: func(t *testing.T, data map[int64]*internal.IndexerData, result *internal.SearchResult) {
assert.Len(t, result.Hits, 5)
assert.Len(t, result.Hits, 0)
for _, v := range result.Hits {
if len(data[v.ID].ProjectIDs) > 0 {
assert.Equal(t, int64(0), data[v.ID].ProjectIDs[0])
assert.Equal(t, int64(1), data[v.ID].ProjectIDs[0])
}
}
assert.Equal(t, countIndexerData(data, func(v *internal.IndexerData) bool {
if len(data[v.ID].ProjectIDs) > 0 {
return v.ProjectIDs[0] == 1
}
return false
}), result.Total)
},
},
{
@ -717,6 +709,10 @@ func generateDefaultIndexerData() []*internal.IndexerData {
for i := range subscriberIDs {
subscriberIDs[i] = int64(i) + 1 // SubscriberID should not be 0
}
projectIDs := make([]int64, id%5)
for i := range projectIDs {
projectIDs[i] = int64(i) + 1 // projectID should not be 0
}
data = append(data, &internal.IndexerData{
ID: id,
@ -730,7 +726,7 @@ func generateDefaultIndexerData() []*internal.IndexerData {
LabelIDs: labelIDs,
NoLabel: len(labelIDs) == 0,
MilestoneID: issueIndex % 4,
ProjectIDs: []int64{issueIndex % 5},
ProjectIDs: projectIDs,
ProjectColumnID: issueIndex % 6,
PosterID: id%10 + 1, // PosterID should not be 0
AssigneeID: issueIndex % 10,