package api import ( "bytes" "encoding/json" "fmt" "net/http" "net/http/httptest" "testing" "github.com/gophish/gophish/config" "github.com/gophish/gophish/models" ) type testContext struct { apiKey string config *config.Config apiServer *Server admin models.User } func setupTest(t *testing.T) *testContext { conf := &config.Config{ DBName: "sqlite3", DBPath: ":memory:", MigrationsPath: "../../db/db_sqlite3/migrations/", } err := models.Setup(conf) if err != nil { t.Fatalf("Failed creating database: %v", err) } ctx := &testContext{} ctx.config = conf // Get the API key to use for these tests u, err := models.GetUser(1) if err != nil { t.Fatalf("error getting admin user: %v", err) } ctx.apiKey = u.ApiKey ctx.admin = u ctx.apiServer = NewServer() return ctx } func tearDown(t *testing.T, ctx *testContext) { // Cleanup all users except the original admin // users, _ := models.GetUsers() // for _, user := range users { // if user.Id == 1 { // continue // } // err := models.DeleteUser(user.Id) // s.Nil(err) // } } func createTestData(t *testing.T) { // Add a group group := models.Group{Name: "Test Group"} group.Targets = []models.Target{ models.Target{BaseRecipient: models.BaseRecipient{Email: "test1@example.com", FirstName: "First", LastName: "Example"}}, models.Target{BaseRecipient: models.BaseRecipient{Email: "test2@example.com", FirstName: "Second", LastName: "Example"}}, } group.UserId = 1 models.PostGroup(&group) // Add a template template := models.Template{Name: "Test Template"} template.Subject = "Test subject" template.Text = "Text text" template.HTML = "Test" template.UserId = 1 models.PostTemplate(&template) // Add a landing page p := models.Page{Name: "Test Page"} p.HTML = "Test" p.UserId = 1 models.PostPage(&p) // Add a sending profile smtp := models.SMTP{Name: "Test Page"} smtp.UserId = 1 smtp.Host = "example.com" smtp.FromAddress = "test@test.com" models.PostSMTP(&smtp) // Setup and "launch" our campaign // Set the status such that no emails are attempted c := models.Campaign{Name: "Test campaign"} c.UserId = 1 c.Template = template c.Page = p c.SMTP = smtp c.Groups = []models.Group{group} models.PostCampaign(&c, c.UserId) c.UpdateStatus(models.CampaignEmailsSent) } func TestSiteImportBaseHref(t *testing.T) { ctx := setupTest(t) h := "
" ts := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) { fmt.Fprintln(w, h) })) expected := fmt.Sprintf("