mirror of
https://github.com/aclindsa/moneygo.git
synced 2024-10-31 16:00:05 -04:00
92 lines
2.6 KiB
Go
92 lines
2.6 KiB
Go
package handlers_test
|
|
|
|
import (
|
|
"fmt"
|
|
"github.com/aclindsa/moneygo/internal/handlers"
|
|
"sort"
|
|
"strconv"
|
|
"testing"
|
|
)
|
|
|
|
type LuaTest struct {
|
|
Name string
|
|
Lua string
|
|
Expected string
|
|
}
|
|
|
|
// Int64Slice attaches the methods of int64 to []int64, sorting in increasing order.
|
|
type Int64Slice []int64
|
|
|
|
func (p Int64Slice) Len() int { return len(p) }
|
|
func (p Int64Slice) Less(i, j int) bool { return p[i] < p[j] }
|
|
func (p Int64Slice) Swap(i, j int) { p[i], p[j] = p[j], p[i] }
|
|
|
|
// Sort is a convenience method.
|
|
func (p Int64Slice) Sort() { sort.Sort(p) }
|
|
|
|
func TestLuaSecurities(t *testing.T) {
|
|
RunWith(t, &data[0], func(t *testing.T, d *TestData) {
|
|
defaultSecurity, err := getSecurity(d.clients[0], d.users[0].DefaultCurrency)
|
|
if err != nil {
|
|
t.Fatalf("Error getting default security: %s", err)
|
|
}
|
|
securities, err := getSecurities(d.clients[0])
|
|
if err != nil {
|
|
t.Fatalf("Error getting securities: %s", err)
|
|
}
|
|
securityids := make(Int64Slice, len(*securities.Securities))
|
|
for i, s := range *securities.Securities {
|
|
securityids[i] = s.SecurityId
|
|
}
|
|
securityids.Sort()
|
|
|
|
for _, lt := range []LuaTest{
|
|
{"SecurityId", `return get_default_currency().SecurityId`, strconv.FormatInt(defaultSecurity.SecurityId, 10)},
|
|
{"Name", `return get_default_currency().Name`, defaultSecurity.Name},
|
|
{"Description", `return get_default_currency().Description`, defaultSecurity.Description},
|
|
{"Symbol", `return get_default_currency().Symbol`, defaultSecurity.Symbol},
|
|
{"Precision", `return get_default_currency().Precision`, strconv.FormatInt(int64(defaultSecurity.Precision), 10)},
|
|
{"Type", `return get_default_currency().Type`, strconv.FormatInt(int64(defaultSecurity.Type), 10)},
|
|
{"AlternateId", `return get_default_currency().AlternateId`, defaultSecurity.AlternateId},
|
|
{"get_securities()", `
|
|
sorted = {}
|
|
for id in pairs(get_securities()) do
|
|
table.insert(sorted, id)
|
|
end
|
|
table.sort(sorted)
|
|
str = "["
|
|
for i,id in ipairs(sorted) do
|
|
str = str .. id .. " "
|
|
end
|
|
return string.sub(str, 1, -2) .. "]"`, fmt.Sprint(securityids)},
|
|
} {
|
|
lua := fmt.Sprintf(`function test()
|
|
%s
|
|
end
|
|
|
|
function generate()
|
|
t = tabulation.new(0)
|
|
t:title(tostring(test()))
|
|
return t
|
|
end`, lt.Lua)
|
|
r := handlers.Report{
|
|
Name: lt.Name,
|
|
Lua: lua,
|
|
}
|
|
report, err := createReport(d.clients[0], &r)
|
|
if err != nil {
|
|
t.Fatalf("Error creating report: %s", err)
|
|
}
|
|
|
|
tab, err := tabulateReport(d.clients[0], report.ReportId)
|
|
if err != nil {
|
|
t.Fatalf("Error tabulating report: %s", err)
|
|
}
|
|
|
|
if tab.Title != lt.Expected {
|
|
t.Errorf("%s: Returned '%s', expected '%s'", lt.Name, tab.Title, lt.Expected)
|
|
}
|
|
}
|
|
})
|
|
}
|