From c4ed7d3c50d4cb346369c1c69e1771c7aa233393 Mon Sep 17 00:00:00 2001
From: Aaron Lindsay <aaron@aclindsay.com>
Date: Sun, 5 Nov 2017 21:25:20 -0500
Subject: [PATCH] testing: Test get_securities() Lua call

---
 internal/handlers/securities_lua_test.go | 31 ++++++++++++++++++++++++
 1 file changed, 31 insertions(+)

diff --git a/internal/handlers/securities_lua_test.go b/internal/handlers/securities_lua_test.go
index 11d2d5f..51dcfdb 100644
--- a/internal/handlers/securities_lua_test.go
+++ b/internal/handlers/securities_lua_test.go
@@ -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