Albirew/nyaa-pantsu
Archivé
1
0
Bifurcation 0
Ce dépôt a été archivé le 2022-05-07. Vous pouvez voir ses fichiers ou le cloner, mais pas ouvrir de ticket ou de demandes d'ajout, ni soumettre de changements.
nyaa-pantsu/templates/template_test.go
2017-08-03 03:38:07 +02:00

288 lignes
11 Kio
Go

package templates
import (
"bytes"
"fmt"
"html/template"
"io/ioutil"
"net/url"
"path"
"testing"
"github.com/NyaaPantsu/nyaa/utils/validator/announcement"
"strings"
"time"
"github.com/CloudyKit/jet"
"github.com/NyaaPantsu/nyaa/config"
"github.com/NyaaPantsu/nyaa/models"
"github.com/NyaaPantsu/nyaa/utils/filelist"
"github.com/NyaaPantsu/nyaa/utils/oauth2/client"
"github.com/NyaaPantsu/nyaa/utils/publicSettings"
"github.com/NyaaPantsu/nyaa/utils/validator/api"
"github.com/NyaaPantsu/nyaa/utils/validator/torrent"
"github.com/NyaaPantsu/nyaa/utils/validator/user"
"github.com/gin-gonic/gin"
)
// run before router/init.go:init()
var _ = func() (_ struct{}) {
gin.SetMode(gin.TestMode)
config.Configpaths[1] = path.Join("..", config.Configpaths[1])
config.Configpaths[0] = path.Join("..", config.Configpaths[0])
config.Reload()
return
}()
func TestTemplates(t *testing.T) {
//var View = jet.NewHTMLSet(TemplateDir)
fmt.Print("JetTest Template started\n")
walkDirTest("/", t)
}
type ContextTest map[string]func(jet.VarMap) jet.VarMap
func walkDirTest(dir string, t *testing.T) {
fu := "http://nyaa.cat"
em := "cop@cat.fe"
fakeTag := &models.Tag{1, 1, "12345", "anidbid", 1, false, 0}
fakeUser := &models.User{1, "test", "test", "test", 1, time.Now(), time.Now(), "test", time.Now(), "en", "test", "test", "test", "test", 0, []models.User{}, []models.User{}, "test", []models.Torrent{}, []models.Notification{}, 1, models.UserSettings{}, []models.Tag{*fakeTag}}
fakeComment := &models.Comment{1, 1, 1, "test", time.Now(), time.Now(), nil, &models.Torrent{}, fakeUser}
fakeScrapeData := &models.Scrape{1, 0, 0, 10, time.Now()}
fakeFile := &models.File{1, 1, "l12:somefile.mp4e", 3}
fakeLanguages := []string{"fr", "en"}
fakeTorrent := &models.Torrent{1, "test", "test", 3, 12, 1, false, time.Now(), 1, 0, 3, "test", "test", "test", "test", "test", nil, fakeUser, "test", []models.OldComment{}, []models.Comment{*fakeComment, *fakeComment}, []models.Tag{*fakeTag, *fakeTag}, fakeScrapeData, []models.File{*fakeFile}, fakeLanguages}
fakeActivity := &models.Activity{1, "t", "e", "s", 1, fakeUser}
fakeDB := &models.DatabaseDump{time.Now(), 3, "test", "test"}
fakeLanguage := &publicSettings.Language{"English", "en", "en-us"}
fakeTorrentRequest := &torrentValidator.TorrentRequest{Name: "test", Magnet: "", Category: "", Remake: false, Description: "", Status: 1, Hidden: false, CaptchaID: "", WebsiteLink: "", SubCategory: 0, Languages: nil, Infohash: "", SubCategoryID: 0, CategoryID: 0, Filesize: 0, Filepath: "", FileList: nil, Trackers: nil, Tags: ""}
fakeLogin := &userValidator.LoginForm{"test", "test", "/"}
fakeRegistration := &userValidator.RegistrationForm{"test", "", "test", "test", "xxxx", "1"}
fakeReport := &models.TorrentReport{1, "test", 1, 1, time.Now(), fakeTorrent, fakeUser}
fakeOauthForm := apiValidator.CreateForm{"", "f", []string{fu}, []string{}, []string{}, "", "fedr", fu, fu, fu, fu, []string{em}, ""}
fakeOauthModel := fakeOauthForm.Bind(&models.OauthClient{})
fakeClient := client.Client{"", "", "", []string{""}, []string{""}, []string{""}, "", "", "", "", "", "", []string{""}, false}
fakeAnnouncement := announcementValidator.CreateForm{1, "", 2}
fakeNotification := &models.Notification{1, "test", true, "test", "test", time.Now(), 1}
contextvariables := ContextTest{
"dumps.jet.html": func(variables jet.VarMap) jet.VarMap {
variables.Set("GPGLink", "test")
variables.Set("ListDumps", []models.DatabaseDumpJSON{fakeDB.ToJSON(), fakeDB.ToJSON()})
return variables
},
"activities.jet.html": func(variables jet.VarMap) jet.VarMap {
variables.Set("Models", []models.Activity{*fakeActivity})
return variables
},
"listing.jet.html": func(variables jet.VarMap) jet.VarMap {
variables.Set("Models", []models.TorrentJSON{fakeTorrent.ToJSON(), fakeTorrent.ToJSON()})
return variables
},
"edit.jet.html": func(variables jet.VarMap) jet.VarMap {
variables.Set("NbTorrents", 0)
variables.Set("Form", fakeTorrentRequest)
variables.Set("Languages", publicSettings.Languages{*fakeLanguage, *fakeLanguage})
return variables
},
"torrents.jet.html": func(variables jet.VarMap) jet.VarMap {
variables.Set("NbTorrents", 0)
return variables
},
"profile.jet.html": func(variables jet.VarMap) jet.VarMap {
variables.Set("NbTorrents", 0)
return variables
},
"upload.jet.html": func(variables jet.VarMap) jet.VarMap {
variables.Set("Form", fakeTorrentRequest)
return variables
},
"view.jet.html": func(variables jet.VarMap) jet.VarMap {
variables.Set("NbTorrents", 0)
variables.Set("Torrent", fakeTorrent.ToJSON())
variables.Set("CaptchaID", "xxxxxx")
variables.Set("RootFolder", filelist.FileListToFolder(fakeTorrent.FileList, "root"))
return variables
},
"settings.jet.html": func(variables jet.VarMap) jet.VarMap {
variables.Set("Form", &LanguagesJSONResponse{"test", publicSettings.Languages{*fakeLanguage, *fakeLanguage}})
return variables
},
"login.jet.html": func(variables jet.VarMap) jet.VarMap {
variables.Set("Form", fakeLogin)
return variables
},
"register.jet.html": func(variables jet.VarMap) jet.VarMap {
variables.Set("Form", fakeRegistration)
return variables
},
"index.jet.html": func(variables jet.VarMap) jet.VarMap {
variables.Set("Torrents", []models.Torrent{*fakeTorrent, *fakeTorrent})
variables.Set("Users", []models.User{*fakeUser, *fakeUser})
variables.Set("Comments", []models.Comment{*fakeComment, *fakeComment})
variables.Set("TorrentReports", []models.TorrentReportJSON{fakeReport.ToJSON(), fakeReport.ToJSON()})
return variables
},
"paneltorrentedit.jet.html": func(variables jet.VarMap) jet.VarMap {
variables.Set("Form", *fakeTorrent)
return variables
},
"reassign.jet.html": func(variables jet.VarMap) jet.VarMap {
variables.Set("Form", torrentValidator.ReassignForm{1, "", "", []uint{1, 1}})
return variables
},
"torrentlist.jet.html": func(variables jet.VarMap) jet.VarMap {
variables.Set("Models", []models.Torrent{*fakeTorrent, *fakeTorrent})
return variables
},
"userlist.jet.html": func(variables jet.VarMap) jet.VarMap {
variables.Set("Models", []models.User{*fakeUser, *fakeUser})
return variables
},
"commentlist.jet.html": func(variables jet.VarMap) jet.VarMap {
variables.Set("Models", []models.Comment{*fakeComment, *fakeComment})
return variables
},
"torrent_report.jet.html": func(variables jet.VarMap) jet.VarMap {
variables.Set("Models", []models.TorrentReportJSON{fakeReport.ToJSON(), fakeReport.ToJSON()})
return variables
},
"notifications.jet.html": func(variables jet.VarMap) jet.VarMap {
variables.Set("NbTorrents", 0)
return variables
},
"report.jet.html": func(variables jet.VarMap) jet.VarMap {
type form struct {
ID int
CaptchaID string
}
variables.Set("Form", form{1, "test"})
return variables
},
"callback.jet.html": func(variables jet.VarMap) jet.VarMap {
variables.Set("Callback", true)
variables.Set("AccessToken", "")
variables.Set("RefreshToken", "")
variables.Set("Code", "")
return variables
},
"grant.jet.html": func(variables jet.VarMap) jet.VarMap {
variables.Set("Client", fakeClient)
variables.Set("Scopes", []string{})
return variables
},
"refresh.jet.html": func(variables jet.VarMap) jet.VarMap {
variables.Set("Refresh", true)
variables.Set("Response", "")
return variables
},
"revoke.jet.html": func(variables jet.VarMap) jet.VarMap {
variables.Set("Revoke", true)
variables.Set("ResponseCode", "")
variables.Set("Response", "")
return variables
},
"clientlist.jet.html": func(variables jet.VarMap) jet.VarMap {
variables.Set("Models", []models.OauthClient{*fakeOauthModel, *fakeOauthModel, *fakeOauthModel})
return variables
},
"oauth_client_form.jet.html": func(variables jet.VarMap) jet.VarMap {
variables.Set("Form", fakeOauthForm)
return variables
},
"announcement_form.jet.html": func(variables jet.VarMap) jet.VarMap {
variables.Set("Form", fakeAnnouncement)
return variables
},
"announcements.jet.html": func(variables jet.VarMap) jet.VarMap {
variables.Set("Models", []models.Notification{*fakeNotification, *fakeNotification})
return variables
},
"tag.jet.html": func(variables jet.VarMap) jet.VarMap {
variables.Set("Form", fakeTag)
return variables
},
}
fmt.Printf("\nTesting Folder: %s\n", dir)
view := jet.NewHTMLSet(path.Join("..", TemplateDir))
files, err := ioutil.ReadDir(path.Join("..", TemplateDir) + dir)
if err != nil {
t.Errorf("Couldn't find the folder %s", path.Join("..", TemplateDir)+dir)
}
if len(files) == 0 {
t.Errorf("Couldn't find any files in folder %s", path.Join("..", TemplateDir)+dir)
}
for _, f := range files {
variables := mockupCommonvariables(t)
if f.IsDir() {
walkDirTest(dir+f.Name()+"/", t)
continue
}
if strings.Contains(f.Name(), ".jet.html") {
template, err := view.GetTemplate(dir + f.Name())
fmt.Printf("\tJetTest Template of: %s", dir+f.Name())
if err != nil {
t.Errorf("\nParsing error: %s %s", err.Error(), dir+f.Name())
}
buff := bytes.NewBuffer(nil)
if contextvariables[f.Name()] != nil {
variables = contextvariables[f.Name()](variables)
}
if err = template.Execute(buff, variables, nil); err != nil {
t.Errorf("\nEval error: %q executing %s", err.Error(), template.Name)
}
fmt.Print("\tOK\n")
}
}
}
func mockupCommonvariables(t *testing.T) jet.VarMap {
variables := jet.VarMap{}
variables.Set("Navigation", NewNavigation())
variables.Set("Search", SearchForm{
Category: "_",
ShowItemsPerPage: true,
SizeType: "b",
DateType: "test",
MinSize: "test",
MaxSize: "test",
FromDate: "test",
ToDate: "test",
})
conf := config.Get().I18n
conf.Directory = path.Join("..", conf.Directory)
var retriever publicSettings.UserRetriever // not required during initialization
err := publicSettings.InitI18n(conf, retriever)
if err != nil {
t.Errorf("failed to initialize language translations: %v", err)
}
Ts, _, err := publicSettings.TfuncAndLanguageWithFallback("en-us", "test", "test")
if err != nil {
t.Error("Couldn't load language files!")
}
var T publicSettings.TemplateTfunc
T = func(id string, args ...interface{}) template.HTML {
return template.HTML(fmt.Sprintf(Ts(id), args...))
}
variables.Set("T", T)
variables.Set("Theme", "test")
variables.Set("Mascot", "test")
variables.Set("MascotURL", "test")
variables.Set("User", &models.User{})
variables.Set("URL", &url.URL{})
variables.Set("CsrfToken", "xxxxxx")
variables.Set("Config", config.Get())
variables.Set("Infos", make(map[string][]string))
variables.Set("Errors", make(map[string][]string))
variables.Set("UserProfile", &models.User{})
variables = templateFunctions(variables)
return variables
}