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("\n", ts.URL) defer ts.Close() req := httptest.NewRequest(http.MethodPost, "/api/import/site", bytes.NewBuffer([]byte(fmt.Sprintf(` { "url" : "%s", "include_resources" : false } `, ts.URL)))) req.Header.Set("Content-Type", "application/json") response := httptest.NewRecorder() ctx.apiServer.ImportSite(response, req) cs := cloneResponse{} err := json.NewDecoder(response.Body).Decode(&cs) if err != nil { t.Fatalf("error decoding response: %v", err) } if cs.HTML != expected { t.Fatalf("unexpected response received. expected %s got %s", expected, cs.HTML) } }