Updated validation function

models/attachment.go

@@ -22,23 +22,26 @@ type Attachment struct {
 }
 
 // ValidateAttachment ensures that the provided attachment uses the supported template variables correctly.
-func (a Attachment) ValidateAttachment() error {
+func (a Attachment) Validate() error {
 
-	ptx := PhishingTemplateContext{
+	vc := ValidationContext{
+		FromAddress: "foo@bar.com",
+		BaseURL:     "http://example.com",
+	}
+	td := Result{
 		BaseRecipient: BaseRecipient{
+			Email:     "foo@bar.com",
 			FirstName: "Foo",
 			LastName:  "Bar",
-			Email:     "foo@bar.com",
+			Position:  "Test",
 		},
-		BaseURL:     "http://testurl.com",
+		RId: "123456",
-		URL:         "http://testurl.com/?rid=1234567",
-		TrackingURL: "http://testurl.local/track?rid=1234567",
-		Tracker:     "<img alt='' style='display: none' src='http://testurl.local/track?rid=1234567'/>",
-		From:        "Foobus Barbus",
-		RId:         "1234567",
 	}
-
+	ptx, err := NewPhishingTemplateContext(vc, td.BaseRecipient, td.RId)
-	_, err := a.ApplyTemplate(ptx)
+	if err != nil {
+		return err
+	}
+	_, err = a.ApplyTemplate(ptx)
 	return err
 }
 

models/template.go

@@ -41,7 +41,7 @@ func (t *Template) Validate() error {
 		return err
 	}
 	for _, a := range t.Attachments {
-		if err := a.ValidateAttachment(); err != nil {
+		if err := a.Validate(); err != nil {
 			return err
 		}
 	}