@@ -182,8 +182,17 @@ func TestBuildIngestHeaders(t *testing.T) {
182182 if _ , ok := headers .Properties ["delivery_id" ]; ok {
183183 t .Fatal ("expected delivery_id to be omitted when empty" )
184184 }
185- if len (headers .Tags ) != 2 {
186- t .Fatalf ("expected 2 tags, got %d" , len (headers .Tags ))
185+ if len (headers .Tags ) != 3 {
186+ t .Fatalf ("expected 3 tags, got %d: %v" , len (headers .Tags ), headers .Tags )
187+ }
188+ if headers .Tags [0 ] != "def-001" {
189+ t .Fatalf ("expected first tag %q, got %q" , "def-001" , headers .Tags [0 ])
190+ }
191+ if headers .Tags [1 ] != "schema_finding" {
192+ t .Fatalf ("expected second tag %q, got %q" , "schema_finding" , headers .Tags [1 ])
193+ }
194+ if headers .Tags [2 ] != string (integrationgenerated .IntegrationIngestSourceWebhook ) {
195+ t .Fatalf ("expected third tag %q, got %q" , string (integrationgenerated .IntegrationIngestSourceWebhook ), headers .Tags [2 ])
187196 }
188197 })
189198
@@ -195,11 +204,14 @@ func TestBuildIngestHeaders(t *testing.T) {
195204
196205 headers := buildIngestHeaders (record , metadata )
197206
198- if len (headers .Tags ) != 1 {
199- t .Fatalf ("expected 1 tag, got %d: %v" , len (headers .Tags ), headers .Tags )
207+ if len (headers .Tags ) != 2 {
208+ t .Fatalf ("expected 2 tags, got %d: %v" , len (headers .Tags ), headers .Tags )
209+ }
210+ if headers .Tags [0 ] != "" {
211+ t .Fatalf ("expected first tag to be empty definition id, got %q" , headers .Tags [0 ])
200212 }
201- if headers .Tags [0 ] != "asset " {
202- t .Fatalf ("expected tag %q, got %q" , "asset " , headers .Tags [0 ])
213+ if headers .Tags [1 ] != "schema_asset " {
214+ t .Fatalf ("expected second tag %q, got %q" , "schema_asset " , headers .Tags [1 ])
203215 }
204216 })
205217}
0 commit comments