mirror of
https://github.com/SecurityBrewery/catalyst.git
synced 2025-12-06 15:22:47 +01:00
@@ -36,6 +36,7 @@ func (i *Index) Index(incidents []*model.TicketSimpleResponse) {
|
||||
for _, incident := range incidents {
|
||||
if incident.ID == 0 {
|
||||
log.Println(errors.New("no ID"), incident)
|
||||
|
||||
continue
|
||||
}
|
||||
|
||||
@@ -44,8 +45,8 @@ func (i *Index) Index(incidents []*model.TicketSimpleResponse) {
|
||||
log.Println(err)
|
||||
}
|
||||
}
|
||||
err := i.internal.Batch(b)
|
||||
if err != nil {
|
||||
|
||||
if err := i.internal.Batch(b); err != nil {
|
||||
log.Println(err)
|
||||
}
|
||||
}
|
||||
@@ -59,6 +60,7 @@ func (i *Index) Search(term string) (ids []string, err error) {
|
||||
for _, match := range result.Hits {
|
||||
ids = append(ids, match.ID)
|
||||
}
|
||||
|
||||
return ids, nil
|
||||
}
|
||||
|
||||
@@ -76,6 +78,7 @@ func (i *Index) Truncate() error {
|
||||
return err
|
||||
}
|
||||
i.internal = index
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
|
||||
@@ -9,6 +9,8 @@ import (
|
||||
)
|
||||
|
||||
func TestIndex(t *testing.T) {
|
||||
t.Parallel()
|
||||
|
||||
type args struct {
|
||||
term string
|
||||
}
|
||||
@@ -22,7 +24,10 @@ func TestIndex(t *testing.T) {
|
||||
{name: "Not exists", args: args{"bar"}},
|
||||
}
|
||||
for _, tt := range tests {
|
||||
tt := tt
|
||||
t.Run(tt.name, func(t *testing.T) {
|
||||
t.Parallel()
|
||||
|
||||
i, cleanup, err := test.Index(t)
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
@@ -37,6 +42,7 @@ func TestIndex(t *testing.T) {
|
||||
gotIds, err := i.Search(tt.args.term)
|
||||
if (err != nil) != tt.wantErr {
|
||||
t.Errorf("Search() error = %v, wantErr %v", err, tt.wantErr)
|
||||
|
||||
return
|
||||
}
|
||||
if !reflect.DeepEqual(gotIds, tt.wantIds) {
|
||||
@@ -47,6 +53,8 @@ func TestIndex(t *testing.T) {
|
||||
}
|
||||
|
||||
func TestIndex_Truncate(t *testing.T) {
|
||||
t.Parallel()
|
||||
|
||||
tests := []struct {
|
||||
name string
|
||||
wantErr bool
|
||||
@@ -54,7 +62,10 @@ func TestIndex_Truncate(t *testing.T) {
|
||||
{name: "Truncate"},
|
||||
}
|
||||
for _, tt := range tests {
|
||||
tt := tt
|
||||
t.Run(tt.name, func(t *testing.T) {
|
||||
t.Parallel()
|
||||
|
||||
i, cleanup, err := test.Index(t)
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
|
||||
Reference in New Issue
Block a user