mirror of https://github.com/gophish/gophish
82 lines
1.7 KiB
Go
82 lines
1.7 KiB
Go
|
package controllers
|
||
|
|
|
||
|
|
import (
|
||
|
|
"bytes"
|
||
|
|
"fmt"
|
||
|
|
"io/ioutil"
|
||
|
|
"net/http"
|
||
|
|
"os"
|
||
|
|
"path/filepath"
|
||
|
|
)
|
||
|
|
|
||
|
|
var fileContent = []byte("Hello world")
|
||
|
|
|
||
|
|
func mustRemoveAll(dir string) {
|
||
|
|
err := os.RemoveAll(dir)
|
||
|
|
if err != nil {
|
||
|
|
panic(err)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func createTestFile(dir, filename string) error {
|
||
|
|
return ioutil.WriteFile(filepath.Join(dir, filename), fileContent, 0644)
|
||
|
|
}
|
||
|
|
|
||
|
|
func (s *ControllersSuite) TestGetStaticFile() {
|
||
|
|
dir, err := ioutil.TempDir("static/endpoint", "test-")
|
||
|
|
tempFolder := filepath.Base(dir)
|
||
|
|
|
||
|
|
s.Nil(err)
|
||
|
|
defer mustRemoveAll(dir)
|
||
|
|
|
||
|
|
err = createTestFile(dir, "foo.txt")
|
||
|
|
s.Nil(nil, err)
|
||
|
|
|
||
|
|
resp, err := http.Get(fmt.Sprintf("%s/static/%s/foo.txt", ps.URL, tempFolder))
|
||
|
|
s.Nil(err)
|
||
|
|
|
||
|
|
defer resp.Body.Close()
|
||
|
|
got, err := ioutil.ReadAll(resp.Body)
|
||
|
|
s.Nil(err)
|
||
|
|
|
||
|
|
s.Equal(bytes.Compare(fileContent, got), 0, fmt.Sprintf("Got %s", got))
|
||
|
|
}
|
||
|
|
|
||
|
|
func (s *ControllersSuite) TestStaticFileListing() {
|
||
|
|
dir, err := ioutil.TempDir("static/endpoint", "test-")
|
||
|
|
tempFolder := filepath.Base(dir)
|
||
|
|
|
||
|
|
s.Nil(err)
|
||
|
|
defer mustRemoveAll(dir)
|
||
|
|
|
||
|
|
err = createTestFile(dir, "foo.txt")
|
||
|
|
s.Nil(nil, err)
|
||
|
|
|
||
|
|
resp, err := http.Get(fmt.Sprintf("%s/static/%s/", ps.URL, tempFolder))
|
||
|
|
s.Nil(err)
|
||
|
|
|
||
|
|
defer resp.Body.Close()
|
||
|
|
s.Nil(err)
|
||
|
|
s.Equal(resp.StatusCode, http.StatusNotFound)
|
||
|
|
}
|
||
|
|
|
||
|
|
func (s *ControllersSuite) TestStaticIndex() {
|
||
|
|
dir, err := ioutil.TempDir("static/endpoint", "test-")
|
||
|
|
tempFolder := filepath.Base(dir)
|
||
|
|
|
||
|
|
s.Nil(err)
|
||
|
|
defer mustRemoveAll(dir)
|
||
|
|
|
||
|
|
err = createTestFile(dir, "index.html")
|
||
|
|
s.Nil(nil, err)
|
||
|
|
|
||
|
|
resp, err := http.Get(fmt.Sprintf("%s/static/%s/", ps.URL, tempFolder))
|
||
|
|
s.Nil(err)
|
||
|
|
|
||
|
|
defer resp.Body.Close()
|
||
|
|
got, err := ioutil.ReadAll(resp.Body)
|
||
|
|
s.Nil(err)
|
||
|
|
|
||
|
|
s.Equal(bytes.Compare(fileContent, got), 0, fmt.Sprintf("Got %s", got))
|
||
|
|
}
|