fixed quick diff test

Author: Hubtrick-Git

vulndb/quick_diff.go

@@ -68,6 +68,7 @@ type quickDiffCVE struct {
 	References            string
 	CISAExploitAdd        *time.Time
 	CISAActionDue         *time.Time
+	EUVDExploitAdd        *time.Time
 	CISARequiredAction    *string
 	CISAVulnerabilityName *string
 	EPSS                  *float64
@@ -142,7 +143,8 @@ func ComputeQuickDiff(ctx context.Context, tx pgx.Tx, fromVersion time.Time) (*Q
 		       c.description, c.cvss, c."references", c.cisa_required_action,
 		       c.cisa_vulnerability_name, c.epss, c.percentile, c.vector,
 		       c.cisa_exploit_add,
-		       c.cisa_action_due
+		       c.cisa_action_due,
+		       c.euvd_exploit_add
 		FROM cves c
 		WHERE NOT EXISTS (SELECT 1 FROM _snap_cves s WHERE s.id = c.id)
 	`)
@@ -159,7 +161,8 @@ func ComputeQuickDiff(ctx context.Context, tx pgx.Tx, fromVersion time.Time) (*Q
 		       c.description, c.cvss, c."references", c.cisa_required_action,
 		       c.cisa_vulnerability_name, c.epss, c.percentile, c.vector,
 		       c.cisa_exploit_add,
-		       c.cisa_action_due
+		       c.cisa_action_due,
+		       c.euvd_exploit_add
 		FROM cves c
 		JOIN _snap_cves s ON s.id = c.id
 		WHERE s.content_hash != c.content_hash
@@ -390,25 +393,25 @@ func computeDiffFromQuickDiff(ctx context.Context, tx pgx.Tx, diff *QuickDiff) e
 		return fmt.Errorf("computeDiffFromQuickDiff: copy _diff_del_cves: %w", err)
 	}
 
-	cvePlain := []string{"id", "content_hash", "cve", "date_published", "date_last_modified", "description", "cvss", "references", "cisa_exploit_add", "cisa_action_due", "cisa_required_action", "cisa_vulnerability_name", "epss", "percentile", "vector"}
-	if err := createLike("_diff_ins_cves", "cves", `id, content_hash, cve, date_published, date_last_modified, description, cvss, "references", cisa_exploit_add, cisa_action_due, cisa_required_action, cisa_vulnerability_name, epss, percentile, vector`); err != nil {
+	cvePlain := []string{"id", "content_hash", "cve", "date_published", "date_last_modified", "description", "cvss", "references", "cisa_exploit_add", "cisa_action_due", "cisa_required_action", "cisa_vulnerability_name", "epss", "percentile", "vector", "euvd_exploit_add"}
+	if err := createLike("_diff_ins_cves", "cves", `id, content_hash, cve, date_published, date_last_modified, description, cvss, "references", cisa_exploit_add, cisa_action_due, cisa_required_action, cisa_vulnerability_name, epss, percentile, vector, euvd_exploit_add`); err != nil {
 		return fmt.Errorf("computeDiffFromQuickDiff: create _diff_ins_cves: %w", err)
 	}
 	if len(diff.CVEsInserted) > 0 {
 		if _, err := tx.CopyFrom(ctx, pgx.Identifier{"_diff_ins_cves"}, cvePlain, pgx.CopyFromSlice(len(diff.CVEsInserted), func(i int) ([]any, error) {
 			c := diff.CVEsInserted[i]
-			return []any{c.ID, c.ContentHash, c.CVE, c.DatePublished, c.DateLastModified, c.Description, c.CVSS, c.References, c.CISAExploitAdd, c.CISAActionDue, c.CISARequiredAction, c.CISAVulnerabilityName, c.EPSS, c.Percentile, c.Vector}, nil
+			return []any{c.ID, c.ContentHash, c.CVE, c.DatePublished, c.DateLastModified, c.Description, c.CVSS, c.References, c.CISAExploitAdd, c.CISAActionDue, c.CISARequiredAction, c.CISAVulnerabilityName, c.EPSS, c.Percentile, c.Vector, c.EUVDExploitAdd}, nil
 		})); err != nil {
 			return fmt.Errorf("computeDiffFromQuickDiff: copy _diff_ins_cves: %w", err)
 		}
 	}
-	if err := createLike("_diff_upd_cves", "cves", `id, content_hash, cve, date_published, date_last_modified, description, cvss, "references", cisa_exploit_add, cisa_action_due, cisa_required_action, cisa_vulnerability_name, epss, percentile, vector`); err != nil {
+	if err := createLike("_diff_upd_cves", "cves", `id, content_hash, cve, date_published, date_last_modified, description, cvss, "references", cisa_exploit_add, cisa_action_due, cisa_required_action, cisa_vulnerability_name, epss, percentile, vector, euvd_exploit_add`); err != nil {
 		return fmt.Errorf("computeDiffFromQuickDiff: create _diff_upd_cves: %w", err)
 	}
 	if len(diff.CVEsUpdated) > 0 {
 		if _, err := tx.CopyFrom(ctx, pgx.Identifier{"_diff_upd_cves"}, cvePlain, pgx.CopyFromSlice(len(diff.CVEsUpdated), func(i int) ([]any, error) {
 			c := diff.CVEsUpdated[i]
-			return []any{c.ID, c.ContentHash, c.CVE, c.DatePublished, c.DateLastModified, c.Description, c.CVSS, c.References, c.CISAExploitAdd, c.CISAActionDue, c.CISARequiredAction, c.CISAVulnerabilityName, c.EPSS, c.Percentile, c.Vector}, nil
+			return []any{c.ID, c.ContentHash, c.CVE, c.DatePublished, c.DateLastModified, c.Description, c.CVSS, c.References, c.CISAExploitAdd, c.CISAActionDue, c.CISARequiredAction, c.CISAVulnerabilityName, c.EPSS, c.Percentile, c.Vector, c.EUVDExploitAdd}, nil
 		})); err != nil {
 			return fmt.Errorf("computeDiffFromQuickDiff: copy _diff_upd_cves: %w", err)
 		}
@@ -629,7 +632,7 @@ func collectCVERows(rows pgx.Rows) ([]quickDiffCVE, error) {
 			&c.ID, &c.ContentHash, &c.CVE, &c.DatePublished, &c.DateLastModified,
 			&c.Description, &c.CVSS, &c.References, &c.CISARequiredAction,
 			&c.CISAVulnerabilityName, &c.EPSS, &c.Percentile, &c.Vector,
-			&c.CISAExploitAdd, &c.CISAActionDue,
+			&c.CISAExploitAdd, &c.CISAActionDue, &c.EUVDExploitAdd,
 		); err != nil {
 			return nil, err
 		}