mirror of
https://github.com/aclindsa/ofxgo.git
synced 2024-11-14 09:10:05 -05:00
Aaron Lindsay
4521bb377a
This adds a python script to generate constants.go when `go generate` is called, and updates the structs, tests, and command-line client to all use the new named constants.
902 lines
23 KiB
Go
902 lines
23 KiB
Go
package ofxgo_test
|
|
|
|
import (
|
|
"github.com/aclindsa/ofxgo"
|
|
"strings"
|
|
"testing"
|
|
"time"
|
|
)
|
|
|
|
func TestMarshalInvStatementRequest(t *testing.T) {
|
|
var expectedString string = `<?xml version="1.0" encoding="UTF-8" standalone="no"?>
|
|
<?OFX OFXHEADER="200" VERSION="203" SECURITY="NONE" OLDFILEUID="NONE" NEWFILEUID="NONE"?>
|
|
<OFX>
|
|
<SIGNONMSGSRQV1>
|
|
<SONRQ>
|
|
<DTCLIENT>20160224131905.000[-5:EST]</DTCLIENT>
|
|
<USERID>1998124</USERID>
|
|
<USERPASS>Sup3eSekrit</USERPASS>
|
|
<LANGUAGE>ENG</LANGUAGE>
|
|
<FI>
|
|
<ORG>First Bank</ORG>
|
|
<FID>01</FID>
|
|
</FI>
|
|
<APPID>MYAPP</APPID>
|
|
<APPVER>1234</APPVER>
|
|
</SONRQ>
|
|
</SIGNONMSGSRQV1>
|
|
<INVSTMTMSGSRQV1>
|
|
<INVSTMTTRNRQ>
|
|
<TRNUID>382827d6-e2d0-4396-bf3b-665979285420</TRNUID>
|
|
<INVSTMTRQ>
|
|
<INVACCTFROM>
|
|
<BROKERID>fi.example.com</BROKERID>
|
|
<ACCTID>82736664</ACCTID>
|
|
</INVACCTFROM>
|
|
<INCTRAN>
|
|
<DTSTART>20160101000000.000[-5:EST]</DTSTART>
|
|
<INCLUDE>Y</INCLUDE>
|
|
</INCTRAN>
|
|
<INCOO>Y</INCOO>
|
|
<INCPOS>
|
|
<INCLUDE>Y</INCLUDE>
|
|
</INCPOS>
|
|
<INCBAL>Y</INCBAL>
|
|
</INVSTMTRQ>
|
|
</INVSTMTTRNRQ>
|
|
</INVSTMTMSGSRQV1>
|
|
</OFX>`
|
|
|
|
var client = ofxgo.Client{
|
|
AppId: "MYAPP",
|
|
AppVer: "1234",
|
|
SpecVersion: "203",
|
|
}
|
|
|
|
var request ofxgo.Request
|
|
request.Signon.UserId = "1998124"
|
|
request.Signon.UserPass = "Sup3eSekrit"
|
|
request.Signon.Org = "First Bank"
|
|
request.Signon.Fid = "01"
|
|
|
|
EST := time.FixedZone("EST", -5*60*60)
|
|
|
|
statementRequest := ofxgo.InvStatementRequest{
|
|
TrnUID: "382827d6-e2d0-4396-bf3b-665979285420",
|
|
InvAcctFrom: ofxgo.InvAcct{
|
|
BrokerId: "fi.example.com",
|
|
AcctId: "82736664",
|
|
},
|
|
DtStart: ofxgo.NewDate(2016, 1, 1, 0, 0, 0, 0, EST),
|
|
Include: true,
|
|
IncludeOO: true,
|
|
IncludePos: true,
|
|
IncludeBalance: true,
|
|
}
|
|
request.InvStmt = append(request.InvStmt, &statementRequest)
|
|
|
|
request.SetClientFields(&client)
|
|
// Overwrite the DtClient value set by SetClientFields to time.Now()
|
|
request.Signon.DtClient = *ofxgo.NewDate(2016, 2, 24, 13, 19, 5, 0, EST)
|
|
|
|
marshalCheckRequest(t, &request, expectedString)
|
|
}
|
|
|
|
func TestUnmarshalInvStatementResponse(t *testing.T) {
|
|
responseReader := strings.NewReader(`<?xml version="1.0" encoding="UTF-8" standalone="no"?>
|
|
<?OFX OFXHEADER="200" VERSION="203" SECURITY="NONE" OLDFILEUID="NONE" NEWFILEUID="NONE"?>
|
|
<OFX>
|
|
<SIGNONMSGSRSV1>
|
|
<SONRS>
|
|
<STATUS>
|
|
<CODE>0</CODE>
|
|
<SEVERITY>INFO</SEVERITY>
|
|
</STATUS>
|
|
<DTSERVER>20170401201244</DTSERVER>
|
|
<LANGUAGE>ENG</LANGUAGE>
|
|
<FI>
|
|
<ORG>INVSTRUS</ORG>
|
|
<FID>9999</FID>
|
|
</FI>
|
|
</SONRS>
|
|
</SIGNONMSGSRSV1>
|
|
<INVSTMTMSGSRSV1>
|
|
<INVSTMTTRNRS>
|
|
<TRNUID>1a0117ad-692b-4c6a-a21b-020d37d34d49</TRNUID>
|
|
<STATUS>
|
|
<CODE>0</CODE>
|
|
<SEVERITY>INFO</SEVERITY>
|
|
</STATUS>
|
|
<INVSTMTRS>
|
|
<DTASOF>20170331000000</DTASOF>
|
|
<CURDEF>USD</CURDEF>
|
|
<INVACCTFROM>
|
|
<BROKERID>invstrus.com</BROKERID>
|
|
<ACCTID>91827364</ACCTID>
|
|
</INVACCTFROM>
|
|
<INVTRANLIST>
|
|
<DTSTART>20170101000000</DTSTART>
|
|
<DTEND>20170331000000</DTEND>
|
|
<BUYSTOCK>
|
|
<INVBUY>
|
|
<INVTRAN>
|
|
<FITID>729483191</FITID>
|
|
<DTTRADE>20170203</DTTRADE>
|
|
<DTSETTLE>20170207</DTSETTLE>
|
|
</INVTRAN>
|
|
<SECID>
|
|
<UNIQUEID>78462F103</UNIQUEID>
|
|
<UNIQUEIDTYPE>CUSIP</UNIQUEIDTYPE>
|
|
</SECID>
|
|
<UNITS>100</UNITS>
|
|
<UNITPRICE>229.00</UNITPRICE>
|
|
<COMMISSION>9.00</COMMISSION>
|
|
<TOTAL>-22909.00</TOTAL>
|
|
<SUBACCTSEC>CASH</SUBACCTSEC>
|
|
<SUBACCTFUND>CASH</SUBACCTFUND>
|
|
</INVBUY>
|
|
<BUYTYPE>BUY</BUYTYPE>
|
|
</BUYSTOCK>
|
|
<INVBANKTRAN>
|
|
<STMTTRN>
|
|
<TRNTYPE>CREDIT</TRNTYPE>
|
|
<DTPOSTED>20170120</DTPOSTED>
|
|
<DTUSER>20170118</DTUSER>
|
|
<DTAVAIL>20170123</DTAVAIL>
|
|
<TRNAMT>22000.00</TRNAMT>
|
|
<FITID>993838</FITID>
|
|
<NAME>DEPOSIT</NAME>
|
|
<MEMO>CHECK 19980</MEMO>
|
|
</STMTTRN>
|
|
<SUBACCTFUND>CASH</SUBACCTFUND>
|
|
</INVBANKTRAN>
|
|
</INVTRANLIST>
|
|
<INVPOSLIST>
|
|
<POSSTOCK>
|
|
<INVPOS>
|
|
<SECID>
|
|
<UNIQUEID>78462F103</UNIQUEID>
|
|
<UNIQUEIDTYPE>CUSIP</UNIQUEIDTYPE>
|
|
</SECID>
|
|
<HELDINACCT>CASH</HELDINACCT>
|
|
<POSTYPE>LONG</POSTYPE>
|
|
<UNITS>200</UNITS>
|
|
<UNITPRICE>235.74</UNITPRICE>
|
|
<MKTVAL>47148.00</MKTVAL>
|
|
<DTPRICEASOF>20170331160000</DTPRICEASOF>
|
|
<MEMO>Price as of previous close</MEMO>
|
|
</INVPOS>
|
|
</POSSTOCK>
|
|
<POSOPT>
|
|
<INVPOS>
|
|
<SECID>
|
|
<UNIQUEID>129887339</UNIQUEID>
|
|
<UNIQUEIDTYPE>CUSIP</UNIQUEIDTYPE>
|
|
</SECID>
|
|
<HELDINACCT>CASH</HELDINACCT>
|
|
<POSTYPE>LONG</POSTYPE>
|
|
<UNITS>1</UNITS>
|
|
<UNITPRICE>3</UNITPRICE>
|
|
<MKTVAL>300</MKTVAL>
|
|
<DTPRICEASOF>20170331160000</DTPRICEASOF>
|
|
</INVPOS>
|
|
</POSOPT>
|
|
</INVPOSLIST>
|
|
<INVBAL>
|
|
<AVAILCASH>16.73</AVAILCASH>
|
|
<MARGINBALANCE>-819.20</MARGINBALANCE>
|
|
<SHORTBALANCE>0</SHORTBALANCE>
|
|
<BALLIST>
|
|
<BAL>
|
|
<NAME>Sweep Int Rate</NAME>
|
|
<DESC>Current interest rate for sweep account balances</DESC>
|
|
<BALTYPE>PERCENT</BALTYPE>
|
|
<VALUE>0.25</VALUE>
|
|
<DTASOF>20170401</DTASOF>
|
|
</BAL>
|
|
</BALLIST>
|
|
</INVBAL>
|
|
<INVOOLIST>
|
|
<OOBUYMF>
|
|
<OO>
|
|
<FITID>76464632</FITID>
|
|
<SECID>
|
|
<UNIQUEID>922908645</UNIQUEID>
|
|
<UNIQUEIDTYPE>CUSIP</UNIQUEIDTYPE>
|
|
</SECID>
|
|
<DTPLACED>20170310124445</DTPLACED>
|
|
<UNITS>10</UNITS>
|
|
<SUBACCT>CASH</SUBACCT>
|
|
<DURATION>GOODTILCANCEL</DURATION>
|
|
<RESTRICTION>NONE</RESTRICTION>
|
|
<LIMITPRICE>168.50</LIMITPRICE>
|
|
</OO>
|
|
<BUYTYPE>BUY</BUYTYPE>
|
|
<UNITTYPE>SHARES</UNITTYPE>
|
|
</OOBUYMF>
|
|
<OOBUYSTOCK>
|
|
<OO>
|
|
<FITID>999387423</FITID>
|
|
<SECID>
|
|
<UNIQUEID>899422348</UNIQUEID>
|
|
<UNIQUEIDTYPE>CUSIP</UNIQUEIDTYPE>
|
|
</SECID>
|
|
<DTPLACED>20170324031900</DTPLACED>
|
|
<UNITS>25</UNITS>
|
|
<SUBACCT>CASH</SUBACCT>
|
|
<DURATION>GOODTILCANCEL</DURATION>
|
|
<RESTRICTION>ALLORNONE</RESTRICTION>
|
|
<LIMITPRICE>19.75</LIMITPRICE>
|
|
</OO>
|
|
<BUYTYPE>BUY</BUYTYPE>
|
|
</OOBUYSTOCK>
|
|
</INVOOLIST>
|
|
</INVSTMTRS>
|
|
</INVSTMTTRNRS>
|
|
</INVSTMTMSGSRSV1>
|
|
<SECLISTMSGSRSV1>
|
|
<SECLIST>
|
|
<STOCKINFO>
|
|
<SECINFO>
|
|
<SECID>
|
|
<UNIQUEID>78462F103</UNIQUEID>
|
|
<UNIQUEIDTYPE>CUSIP</UNIQUEIDTYPE>
|
|
</SECID>
|
|
<SECNAME>S&P 500 ETF</SECNAME>
|
|
<TICKER>SPY</TICKER>
|
|
<FIID>99184</FIID>
|
|
</SECINFO>
|
|
<YIELD>1.92</YIELD>
|
|
<ASSETCLASS>OTHER</ASSETCLASS>
|
|
</STOCKINFO>
|
|
<OPTINFO>
|
|
<SECINFO>
|
|
<SECID>
|
|
<UNIQUEID>129887339</UNIQUEID>
|
|
<UNIQUEIDTYPE>CUSIP</UNIQUEIDTYPE>
|
|
</SECID>
|
|
<SECNAME>John's Fertilizer Puts</SECNAME>
|
|
<TICKER>FERTP</TICKER>
|
|
<FIID>882919</FIID>
|
|
</SECINFO>
|
|
<OPTTYPE>PUT</OPTTYPE>
|
|
<STRIKEPRICE>79.00</STRIKEPRICE>
|
|
<DTEXPIRE>20170901</DTEXPIRE>
|
|
<SHPERCTRCT>100</SHPERCTRCT>
|
|
<SECID>
|
|
<UNIQUEID>983322180</UNIQUEID>
|
|
<UNIQUEIDTYPE>CUSIP</UNIQUEIDTYPE>
|
|
</SECID>
|
|
<ASSETCLASS>LARGESTOCK</ASSETCLASS>
|
|
</OPTINFO>
|
|
<STOCKINFO>
|
|
<SECINFO>
|
|
<SECID>
|
|
<UNIQUEID>899422348</UNIQUEID>
|
|
<UNIQUEIDTYPE>CUSIP</UNIQUEIDTYPE>
|
|
</SECID>
|
|
<SECNAME>Whatchamacallit, Inc.</SECNAME>
|
|
<TICKER>WHAT</TICKER>
|
|
<FIID>883897</FIID>
|
|
</SECINFO>
|
|
<YIELD>17</YIELD>
|
|
<ASSETCLASS>SMALLSTOCK</ASSETCLASS>
|
|
</STOCKINFO>
|
|
<MFINFO>
|
|
<SECINFO>
|
|
<SECID>
|
|
<UNIQUEID>922908645</UNIQUEID>
|
|
<UNIQUEIDTYPE>CUSIP</UNIQUEIDTYPE>
|
|
</SECID>
|
|
<SECNAME>Mid-Cap Index Fund Admiral Shares</SECNAME>
|
|
<TICKER>VIMAX</TICKER>
|
|
</SECINFO>
|
|
</MFINFO>
|
|
</SECLIST>
|
|
</SECLISTMSGSRSV1>
|
|
</OFX>`)
|
|
var expected ofxgo.Response
|
|
|
|
expected.Version = "203"
|
|
expected.Signon.Status.Code = 0
|
|
expected.Signon.Status.Severity = "INFO"
|
|
expected.Signon.DtServer = *ofxgo.NewDateGMT(2017, 4, 1, 20, 12, 44, 0)
|
|
expected.Signon.Language = "ENG"
|
|
expected.Signon.Org = "INVSTRUS"
|
|
expected.Signon.Fid = "9999"
|
|
|
|
var units1, unitprice1, commission1, total1, amount2 ofxgo.Amount
|
|
units1.SetFrac64(100, 1)
|
|
unitprice1.SetFrac64(229, 1)
|
|
commission1.SetFrac64(9, 1)
|
|
total1.SetFrac64(-22909, 1)
|
|
amount2.SetFrac64(22000, 1)
|
|
|
|
invtranlist := ofxgo.InvTranList{
|
|
DtStart: *ofxgo.NewDateGMT(2017, 1, 1, 0, 0, 0, 0),
|
|
DtEnd: *ofxgo.NewDateGMT(2017, 3, 31, 0, 0, 0, 0),
|
|
InvTransactions: []ofxgo.InvTransaction{
|
|
ofxgo.BuyStock{
|
|
InvBuy: ofxgo.InvBuy{
|
|
InvTran: ofxgo.InvTran{
|
|
FiTId: "729483191",
|
|
DtTrade: *ofxgo.NewDateGMT(2017, 2, 3, 0, 0, 0, 0),
|
|
DtSettle: ofxgo.NewDateGMT(2017, 2, 7, 0, 0, 0, 0),
|
|
},
|
|
SecId: ofxgo.SecurityId{
|
|
UniqueId: "78462F103",
|
|
UniqueIdType: "CUSIP",
|
|
},
|
|
Units: units1,
|
|
UnitPrice: unitprice1,
|
|
Commission: commission1,
|
|
Total: total1,
|
|
SubAcctSec: ofxgo.SubAcctTypeCash,
|
|
SubAcctFund: ofxgo.SubAcctTypeCash,
|
|
},
|
|
BuyType: ofxgo.BuyTypeBuy,
|
|
},
|
|
},
|
|
BankTransactions: []ofxgo.InvBankTransaction{
|
|
ofxgo.InvBankTransaction{
|
|
Transactions: []ofxgo.Transaction{
|
|
ofxgo.Transaction{
|
|
TrnType: ofxgo.TrnTypeCredit,
|
|
DtPosted: *ofxgo.NewDateGMT(2017, 1, 20, 0, 0, 0, 0),
|
|
DtUser: ofxgo.NewDateGMT(2017, 1, 18, 0, 0, 0, 0),
|
|
DtAvail: ofxgo.NewDateGMT(2017, 1, 23, 0, 0, 0, 0),
|
|
|
|
TrnAmt: amount2,
|
|
FiTId: "993838",
|
|
Name: "DEPOSIT",
|
|
Memo: "CHECK 19980",
|
|
},
|
|
},
|
|
SubAcctFund: ofxgo.SubAcctTypeCash,
|
|
},
|
|
},
|
|
}
|
|
|
|
var availcash, marginbalance, shortbalance, balvalue ofxgo.Amount
|
|
availcash.SetFrac64(1673, 100)
|
|
marginbalance.SetFrac64(-8192, 10)
|
|
shortbalance.SetFrac64(0, 1)
|
|
balvalue.SetFrac64(25, 100)
|
|
|
|
invbalance := ofxgo.InvBalance{
|
|
AvailCash: availcash,
|
|
MarginBalance: marginbalance,
|
|
ShortBalance: shortbalance,
|
|
BalList: []ofxgo.Balance{
|
|
ofxgo.Balance{
|
|
Name: "Sweep Int Rate",
|
|
Desc: "Current interest rate for sweep account balances",
|
|
BalType: ofxgo.BalTypePercent,
|
|
Value: balvalue,
|
|
DtAsOf: ofxgo.NewDateGMT(2017, 4, 1, 0, 0, 0, 0),
|
|
},
|
|
},
|
|
}
|
|
|
|
var balamt, availbalamt, posunits1, posunitprice1, posmktval1, posunits2, posunitprice2, posmktval2, oounits1, oolimitprice1, oounits2, oolimitprice2 ofxgo.Amount
|
|
balamt.SetFrac64(20029, 100)
|
|
availbalamt.SetFrac64(20029, 100)
|
|
posunits1.SetFrac64(200, 1)
|
|
posunitprice1.SetFrac64(23574, 100)
|
|
posmktval1.SetFrac64(47148, 1)
|
|
posunits2.SetFrac64(1, 1)
|
|
posunitprice2.SetFrac64(3, 1)
|
|
posmktval2.SetFrac64(300, 1)
|
|
oounits1.SetFrac64(10, 1)
|
|
oolimitprice1.SetFrac64(16850, 100)
|
|
oounits2.SetFrac64(25, 1)
|
|
oolimitprice2.SetFrac64(1975, 100)
|
|
|
|
statementResponse := ofxgo.InvStatementResponse{
|
|
TrnUID: "1a0117ad-692b-4c6a-a21b-020d37d34d49",
|
|
Status: ofxgo.Status{
|
|
Code: 0,
|
|
Severity: "INFO",
|
|
},
|
|
DtAsOf: *ofxgo.NewDateGMT(2017, 3, 31, 0, 0, 0, 0),
|
|
CurDef: "USD",
|
|
InvAcctFrom: ofxgo.InvAcct{
|
|
BrokerId: "invstrus.com",
|
|
AcctId: "91827364",
|
|
},
|
|
InvTranList: &invtranlist,
|
|
InvPosList: ofxgo.PositionList{
|
|
ofxgo.StockPosition{
|
|
InvPos: ofxgo.InvPosition{
|
|
SecId: ofxgo.SecurityId{
|
|
UniqueId: "78462F103",
|
|
UniqueIdType: "CUSIP",
|
|
},
|
|
HeldInAcct: ofxgo.SubAcctTypeCash,
|
|
PosType: ofxgo.PosTypeLong,
|
|
Units: posunits1,
|
|
UnitPrice: posunitprice1,
|
|
MktVal: posmktval1,
|
|
DtPriceAsOf: *ofxgo.NewDateGMT(2017, 3, 31, 16, 0, 0, 0),
|
|
Memo: "Price as of previous close",
|
|
},
|
|
},
|
|
ofxgo.OptPosition{
|
|
InvPos: ofxgo.InvPosition{
|
|
SecId: ofxgo.SecurityId{
|
|
UniqueId: "129887339",
|
|
UniqueIdType: "CUSIP",
|
|
},
|
|
HeldInAcct: ofxgo.SubAcctTypeCash,
|
|
PosType: ofxgo.PosTypeLong,
|
|
Units: posunits2,
|
|
UnitPrice: posunitprice2,
|
|
MktVal: posmktval2,
|
|
DtPriceAsOf: *ofxgo.NewDateGMT(2017, 3, 31, 16, 0, 0, 0),
|
|
},
|
|
},
|
|
},
|
|
InvBal: &invbalance,
|
|
InvOOList: ofxgo.OOList{
|
|
ofxgo.OOBuyMF{
|
|
OO: ofxgo.OO{
|
|
FiTId: "76464632",
|
|
SecId: ofxgo.SecurityId{
|
|
UniqueId: "922908645",
|
|
UniqueIdType: "CUSIP",
|
|
},
|
|
DtPlaced: *ofxgo.NewDateGMT(2017, 3, 10, 12, 44, 45, 0),
|
|
Units: oounits1,
|
|
SubAcct: ofxgo.SubAcctTypeCash,
|
|
Duration: ofxgo.DurationGoodTilCancel,
|
|
Restriction: ofxgo.RestrictionNone,
|
|
LimitPrice: oolimitprice1,
|
|
},
|
|
BuyType: ofxgo.BuyTypeBuy,
|
|
UnitType: ofxgo.UnitTypeShares,
|
|
},
|
|
ofxgo.OOBuyStock{
|
|
OO: ofxgo.OO{
|
|
FiTId: "999387423",
|
|
SecId: ofxgo.SecurityId{
|
|
UniqueId: "899422348",
|
|
UniqueIdType: "CUSIP",
|
|
},
|
|
DtPlaced: *ofxgo.NewDateGMT(2017, 3, 24, 3, 19, 0, 0),
|
|
Units: oounits2,
|
|
SubAcct: ofxgo.SubAcctTypeCash,
|
|
Duration: ofxgo.DurationGoodTilCancel,
|
|
Restriction: ofxgo.RestrictionAllOrNone,
|
|
LimitPrice: oolimitprice2,
|
|
},
|
|
BuyType: ofxgo.BuyTypeBuy,
|
|
},
|
|
},
|
|
}
|
|
expected.InvStmt = append(expected.InvStmt, &statementResponse)
|
|
|
|
var yield1, yield2, strikeprice ofxgo.Amount
|
|
yield1.SetFrac64(192, 100)
|
|
yield2.SetFrac64(17, 1)
|
|
strikeprice.SetFrac64(79, 1)
|
|
|
|
seclist := ofxgo.SecurityList{
|
|
Securities: []ofxgo.Security{
|
|
ofxgo.StockInfo{
|
|
SecInfo: ofxgo.SecInfo{
|
|
SecId: ofxgo.SecurityId{
|
|
UniqueId: "78462F103",
|
|
UniqueIdType: "CUSIP",
|
|
},
|
|
SecName: "S&P 500 ETF",
|
|
Ticker: "SPY",
|
|
FiId: "99184",
|
|
},
|
|
Yield: yield1,
|
|
AssetClass: ofxgo.AssetClassOther,
|
|
},
|
|
ofxgo.OptInfo{
|
|
SecInfo: ofxgo.SecInfo{
|
|
SecId: ofxgo.SecurityId{
|
|
UniqueId: "129887339",
|
|
UniqueIdType: "CUSIP",
|
|
},
|
|
SecName: "John's Fertilizer Puts",
|
|
Ticker: "FERTP",
|
|
FiId: "882919",
|
|
},
|
|
OptType: ofxgo.OptTypePut,
|
|
StrikePrice: strikeprice,
|
|
DtExpire: *ofxgo.NewDateGMT(2017, 9, 1, 0, 0, 0, 0),
|
|
ShPerCtrct: 100,
|
|
SecId: &ofxgo.SecurityId{
|
|
UniqueId: "983322180",
|
|
UniqueIdType: "CUSIP",
|
|
},
|
|
AssetClass: ofxgo.AssetClassLargeStock,
|
|
},
|
|
ofxgo.StockInfo{
|
|
SecInfo: ofxgo.SecInfo{
|
|
SecId: ofxgo.SecurityId{
|
|
UniqueId: "899422348",
|
|
UniqueIdType: "CUSIP",
|
|
},
|
|
SecName: "Whatchamacallit, Inc.",
|
|
Ticker: "WHAT",
|
|
FiId: "883897",
|
|
},
|
|
Yield: yield2,
|
|
AssetClass: ofxgo.AssetClassSmallStock,
|
|
},
|
|
ofxgo.MFInfo{
|
|
SecInfo: ofxgo.SecInfo{
|
|
SecId: ofxgo.SecurityId{
|
|
UniqueId: "922908645",
|
|
UniqueIdType: "CUSIP",
|
|
},
|
|
SecName: "Mid-Cap Index Fund Admiral Shares",
|
|
Ticker: "VIMAX",
|
|
},
|
|
},
|
|
},
|
|
}
|
|
expected.SecList = append(expected.SecList, &seclist)
|
|
|
|
response, err := ofxgo.ParseResponse(responseReader)
|
|
if err != nil {
|
|
t.Fatalf("Unexpected error unmarshalling response: %s\n", err)
|
|
}
|
|
|
|
checkResponsesEqual(t, &expected, response)
|
|
}
|
|
|
|
func TestUnmarshalInvStatementResponse102(t *testing.T) {
|
|
responseReader := strings.NewReader(`OFXHEADER: 100
|
|
DATA: OFXSGML
|
|
VERSION: 102
|
|
SECURITY: NONE
|
|
ENCODING: USASCII
|
|
CHARSET: 1252
|
|
COMPRESSION: NONE
|
|
OLDFILEUID: NONE
|
|
NEWFILEUID: NONE
|
|
|
|
<OFX>
|
|
<SIGNONMSGSRSV1>
|
|
<SONRS>
|
|
<STATUS>
|
|
<CODE>0
|
|
<SEVERITY>INFO
|
|
</STATUS>
|
|
<DTSERVER>20170403120000
|
|
<LANGUAGE>ENG
|
|
<FI>
|
|
<ORG>VV
|
|
<FID>1000
|
|
</FI>
|
|
<INTU.BID>1000
|
|
</SONRS>
|
|
</SIGNONMSGSRSV1>
|
|
<INVSTMTMSGSRSV1>
|
|
<INVSTMTTRNRS>
|
|
<TRNUID>1283719872
|
|
<STATUS>
|
|
<CODE>0
|
|
<SEVERITY>INFO
|
|
</STATUS>
|
|
<INVSTMTRS>
|
|
<DTASOF>20170403120000
|
|
<CURDEF>USD
|
|
<INVACCTFROM>
|
|
<BROKERID>www.exampletrader.com
|
|
<ACCTID>12341234
|
|
</INVACCTFROM>
|
|
<INVTRANLIST>
|
|
<DTSTART>20161206120000
|
|
<DTEND>20170403120000
|
|
<SELLOPT>
|
|
<INVSELL>
|
|
<INVTRAN>
|
|
<FITID>12341234-20161207-1
|
|
<DTTRADE>20161207120000
|
|
<DTSETTLE>20161208120000
|
|
</INVTRAN>
|
|
<SECID>
|
|
<UNIQUEID>SPY161216C00226000
|
|
<UNIQUEIDTYPE>CUSIP
|
|
</SECID>
|
|
<UNITS>-1.0000
|
|
<UNITPRICE>0.3500
|
|
<COMMISSION>8.8500
|
|
<FEES>0.2600
|
|
<TOTAL>200.8900
|
|
<SUBACCTSEC>CASH
|
|
<SUBACCTFUND>CASH
|
|
</INVSELL>
|
|
<OPTSELLTYPE>SELLTOOPEN
|
|
<SHPERCTRCT>100
|
|
</SELLOPT>
|
|
<CLOSUREOPT>
|
|
<INVTRAN>
|
|
<FITID>12341234-20161215-1
|
|
<DTTRADE>20161215120000
|
|
<DTSETTLE>20161220120000
|
|
</INVTRAN>
|
|
<SECID>
|
|
<UNIQUEID>78462F10
|
|
<UNIQUEIDTYPE>CUSIP
|
|
</SECID>
|
|
<OPTACTION>ASSIGN
|
|
<UNITS>-100.0000
|
|
<SHPERCTRCT>100
|
|
<SUBACCTSEC>CASH
|
|
</CLOSUREOPT>
|
|
<CLOSUREOPT>
|
|
<INVTRAN>
|
|
<FITID>12341234-20161215-2
|
|
<DTTRADE>20161215120000
|
|
<DTSETTLE>20161215120000
|
|
</INVTRAN>
|
|
<SECID>
|
|
<UNIQUEID>SPY161216C00226000
|
|
<UNIQUEIDTYPE>CUSIP
|
|
</SECID>
|
|
<OPTACTION>ASSIGN
|
|
<UNITS>1.0000
|
|
<SHPERCTRCT>100
|
|
<SUBACCTSEC>CASH
|
|
</CLOSUREOPT>
|
|
</INVTRANLIST>
|
|
<INVPOSLIST>
|
|
<POSSTOCK>
|
|
<INVPOS>
|
|
<SECID>
|
|
<UNIQUEID>04956010
|
|
<UNIQUEIDTYPE>CUSIP
|
|
</SECID>
|
|
<HELDINACCT>CASH
|
|
<POSTYPE>LONG
|
|
<UNITS>100
|
|
<UNITPRICE>79.0000
|
|
<MKTVAL>79000
|
|
<DTPRICEASOF>20170403120000
|
|
</INVPOS>
|
|
</POSSTOCK>
|
|
<POSSTOCK>
|
|
<INVPOS>
|
|
<SECID>
|
|
<UNIQUEID>36960410
|
|
<UNIQUEIDTYPE>CUSIP
|
|
</SECID>
|
|
<HELDINACCT>CASH
|
|
<POSTYPE>LONG
|
|
<UNITS>100.00
|
|
<UNITPRICE>29.8700
|
|
<MKTVAL>2987.00
|
|
<DTPRICEASOF>20170403120000
|
|
</INVPOS>
|
|
</POSSTOCK>
|
|
</INVPOSLIST>
|
|
<INVBAL>
|
|
<AVAILCASH>0.0
|
|
<MARGINBALANCE>-0.00
|
|
<SHORTBALANCE>0.00
|
|
</INVBAL>
|
|
</INVSTMTRS>
|
|
</INVSTMTTRNRS>
|
|
</INVSTMTMSGSRSV1>
|
|
<SECLISTMSGSRSV1>
|
|
<SECLIST>
|
|
<STOCKINFO>
|
|
<SECINFO>
|
|
<SECID>
|
|
<UNIQUEID>78462F10
|
|
<UNIQUEIDTYPE>CUSIP
|
|
</SECID>
|
|
<SECNAME>SPDR S&P 500 ETF TRUST
|
|
<TICKER>SPY
|
|
</SECINFO>
|
|
</STOCKINFO>
|
|
<OPTINFO>
|
|
<SECINFO>
|
|
<SECID>
|
|
<UNIQUEID>SPY161216C00226000
|
|
<UNIQUEIDTYPE>CUSIP
|
|
</SECID>
|
|
<SECNAME>SPY Dec 16 2016 226.00 Call
|
|
<TICKER>SPY 161216C00226000
|
|
</SECINFO>
|
|
<OPTTYPE>CALL
|
|
<STRIKEPRICE>226.00
|
|
<DTEXPIRE>20161216120000
|
|
<SHPERCTRCT>100
|
|
</OPTINFO>
|
|
</SECLIST>
|
|
</SECLISTMSGSRSV1>
|
|
</OFX>`)
|
|
var expected ofxgo.Response
|
|
|
|
expected.Version = "102"
|
|
expected.Signon.Status.Code = 0
|
|
expected.Signon.Status.Severity = "INFO"
|
|
expected.Signon.DtServer = *ofxgo.NewDateGMT(2017, 4, 3, 12, 0, 0, 0)
|
|
expected.Signon.Language = "ENG"
|
|
expected.Signon.Org = "VV"
|
|
expected.Signon.Fid = "1000"
|
|
// Ignored <INTU.BID>1000
|
|
|
|
var units1, unitprice1, commission1, fees1, total1, units2, units3 ofxgo.Amount
|
|
units1.SetFrac64(-1, 1)
|
|
unitprice1.SetFrac64(35, 100)
|
|
commission1.SetFrac64(885, 100)
|
|
fees1.SetFrac64(26, 100)
|
|
total1.SetFrac64(20089, 100)
|
|
units2.SetFrac64(-100, 1)
|
|
units3.SetFrac64(1, 1)
|
|
|
|
invtranlist := ofxgo.InvTranList{
|
|
DtStart: *ofxgo.NewDateGMT(2016, 12, 6, 12, 0, 0, 0),
|
|
DtEnd: *ofxgo.NewDateGMT(2017, 4, 3, 12, 0, 0, 0),
|
|
InvTransactions: []ofxgo.InvTransaction{
|
|
ofxgo.SellOpt{
|
|
InvSell: ofxgo.InvSell{
|
|
InvTran: ofxgo.InvTran{
|
|
FiTId: "12341234-20161207-1",
|
|
DtTrade: *ofxgo.NewDateGMT(2016, 12, 7, 12, 0, 0, 0),
|
|
DtSettle: ofxgo.NewDateGMT(2016, 12, 8, 12, 0, 0, 0),
|
|
},
|
|
SecId: ofxgo.SecurityId{
|
|
UniqueId: "SPY161216C00226000",
|
|
UniqueIdType: "CUSIP",
|
|
},
|
|
Units: units1,
|
|
UnitPrice: unitprice1,
|
|
Commission: commission1,
|
|
Fees: fees1,
|
|
Total: total1,
|
|
SubAcctSec: ofxgo.SubAcctTypeCash,
|
|
SubAcctFund: ofxgo.SubAcctTypeCash,
|
|
},
|
|
OptSellType: ofxgo.OptSellTypeSellToOpen,
|
|
ShPerCtrct: 100,
|
|
},
|
|
ofxgo.ClosureOpt{
|
|
InvTran: ofxgo.InvTran{
|
|
FiTId: "12341234-20161215-1",
|
|
DtTrade: *ofxgo.NewDateGMT(2016, 12, 15, 12, 0, 0, 0),
|
|
DtSettle: ofxgo.NewDateGMT(2016, 12, 20, 12, 0, 0, 0),
|
|
},
|
|
SecId: ofxgo.SecurityId{
|
|
UniqueId: "78462F10",
|
|
UniqueIdType: "CUSIP",
|
|
},
|
|
OptAction: ofxgo.OptActionAssign,
|
|
Units: units2,
|
|
ShPerCtrct: 100,
|
|
SubAcctSec: ofxgo.SubAcctTypeCash,
|
|
},
|
|
ofxgo.ClosureOpt{
|
|
InvTran: ofxgo.InvTran{
|
|
FiTId: "12341234-20161215-2",
|
|
DtTrade: *ofxgo.NewDateGMT(2016, 12, 15, 12, 0, 0, 0),
|
|
DtSettle: ofxgo.NewDateGMT(2016, 12, 15, 12, 0, 0, 0),
|
|
},
|
|
SecId: ofxgo.SecurityId{
|
|
UniqueId: "SPY161216C00226000",
|
|
UniqueIdType: "CUSIP",
|
|
},
|
|
OptAction: ofxgo.OptActionAssign,
|
|
Units: units3,
|
|
ShPerCtrct: 100,
|
|
SubAcctSec: ofxgo.SubAcctTypeCash,
|
|
},
|
|
},
|
|
}
|
|
|
|
var availcash, marginbalance, shortbalance ofxgo.Amount
|
|
availcash.SetFrac64(0, 1)
|
|
marginbalance.SetFrac64(-0, 1)
|
|
shortbalance.SetFrac64(0, 1)
|
|
|
|
invbalance := ofxgo.InvBalance{
|
|
AvailCash: availcash,
|
|
MarginBalance: marginbalance,
|
|
ShortBalance: shortbalance,
|
|
}
|
|
|
|
var posunits1, posunitprice1, posmktval1, posunits2, posunitprice2, posmktval2 ofxgo.Amount
|
|
posunits1.SetFrac64(100, 1)
|
|
posunitprice1.SetFrac64(79, 1)
|
|
posmktval1.SetFrac64(79000, 1)
|
|
posunits2.SetFrac64(100, 1)
|
|
posunitprice2.SetFrac64(2987, 100)
|
|
posmktval2.SetFrac64(2987, 1)
|
|
|
|
statementResponse := ofxgo.InvStatementResponse{
|
|
TrnUID: "1283719872",
|
|
Status: ofxgo.Status{
|
|
Code: 0,
|
|
Severity: "INFO",
|
|
},
|
|
DtAsOf: *ofxgo.NewDateGMT(2017, 4, 3, 12, 0, 0, 0),
|
|
CurDef: "USD",
|
|
InvAcctFrom: ofxgo.InvAcct{
|
|
BrokerId: "www.exampletrader.com",
|
|
AcctId: "12341234",
|
|
},
|
|
InvTranList: &invtranlist,
|
|
InvPosList: ofxgo.PositionList{
|
|
ofxgo.StockPosition{
|
|
InvPos: ofxgo.InvPosition{
|
|
SecId: ofxgo.SecurityId{
|
|
UniqueId: "04956010",
|
|
UniqueIdType: "CUSIP",
|
|
},
|
|
HeldInAcct: ofxgo.SubAcctTypeCash,
|
|
PosType: ofxgo.PosTypeLong,
|
|
Units: posunits1,
|
|
UnitPrice: posunitprice1,
|
|
MktVal: posmktval1,
|
|
DtPriceAsOf: *ofxgo.NewDateGMT(2017, 4, 3, 12, 0, 0, 0),
|
|
},
|
|
},
|
|
ofxgo.StockPosition{
|
|
InvPos: ofxgo.InvPosition{
|
|
SecId: ofxgo.SecurityId{
|
|
UniqueId: "36960410",
|
|
UniqueIdType: "CUSIP",
|
|
},
|
|
HeldInAcct: ofxgo.SubAcctTypeCash,
|
|
PosType: ofxgo.PosTypeLong,
|
|
Units: posunits2,
|
|
UnitPrice: posunitprice2,
|
|
MktVal: posmktval2,
|
|
DtPriceAsOf: *ofxgo.NewDateGMT(2017, 4, 3, 12, 0, 0, 0),
|
|
},
|
|
},
|
|
},
|
|
InvBal: &invbalance,
|
|
}
|
|
expected.InvStmt = append(expected.InvStmt, &statementResponse)
|
|
|
|
var strikeprice ofxgo.Amount
|
|
strikeprice.SetFrac64(226, 1)
|
|
|
|
seclist := ofxgo.SecurityList{
|
|
Securities: []ofxgo.Security{
|
|
ofxgo.StockInfo{
|
|
SecInfo: ofxgo.SecInfo{
|
|
SecId: ofxgo.SecurityId{
|
|
UniqueId: "78462F10",
|
|
UniqueIdType: "CUSIP",
|
|
},
|
|
SecName: "SPDR S&P 500 ETF TRUST",
|
|
Ticker: "SPY",
|
|
},
|
|
},
|
|
ofxgo.OptInfo{
|
|
SecInfo: ofxgo.SecInfo{
|
|
SecId: ofxgo.SecurityId{
|
|
UniqueId: "SPY161216C00226000",
|
|
UniqueIdType: "CUSIP",
|
|
},
|
|
SecName: "SPY Dec 16 2016 226.00 Call",
|
|
Ticker: "SPY 161216C00226000",
|
|
},
|
|
OptType: ofxgo.OptTypeCall,
|
|
StrikePrice: strikeprice,
|
|
DtExpire: *ofxgo.NewDateGMT(2016, 12, 16, 12, 0, 0, 0),
|
|
ShPerCtrct: 100,
|
|
},
|
|
},
|
|
}
|
|
expected.SecList = append(expected.SecList, &seclist)
|
|
|
|
response, err := ofxgo.ParseResponse(responseReader)
|
|
if err != nil {
|
|
t.Fatalf("Unexpected error unmarshalling response: %s\n", err)
|
|
}
|
|
|
|
checkResponsesEqual(t, &expected, response)
|
|
}
|