testing: Test get_securities() Lua call

This commit is contained in:
Aaron Lindsay 2017-11-05 21:25:20 -05:00
parent 0dec4001eb
commit c4ed7d3c50
1 changed files with 31 additions and 0 deletions

View File

@ -3,6 +3,7 @@ package handlers_test
import (
"fmt"
"github.com/aclindsa/moneygo/internal/handlers"
"sort"
"strconv"
"testing"
)
@ -13,12 +14,31 @@ type LuaTest struct {
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)},
@ -28,6 +48,17 @@ func TestLuaSecurities(t *testing.T) {
{"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