mirror of
https://github.com/aclindsa/ofxgo.git
synced 2024-11-22 03:30:04 -05:00
Add additional investment statement response test
This commit is contained in:
parent
cd39e4f8af
commit
9b03829645
355
invstmt_test.go
355
invstmt_test.go
@ -557,3 +557,358 @@ func TestUnmarshalInvStatementResponse(t *testing.T) {
|
|||||||
|
|
||||||
checkResponsesEqual(t, &expected, response)
|
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
|
||||||
|
GMT := time.FixedZone("GMT", 0)
|
||||||
|
|
||||||
|
expected.Version = "102"
|
||||||
|
expected.Signon.Status.Code = 0
|
||||||
|
expected.Signon.Status.Severity = "INFO"
|
||||||
|
expected.Signon.DtServer = ofxgo.Date(time.Date(2017, 4, 3, 12, 0, 0, 0, GMT))
|
||||||
|
expected.Signon.Language = "ENG"
|
||||||
|
expected.Signon.Org = "VV"
|
||||||
|
expected.Signon.Fid = "1000"
|
||||||
|
// Ignored <INTU.BID>1000
|
||||||
|
|
||||||
|
var units1, unitprice1, commission1, fees1, total1, units2, units3 big.Rat
|
||||||
|
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)
|
||||||
|
|
||||||
|
dtsettle1 := ofxgo.Date(time.Date(2016, 12, 8, 12, 0, 0, 0, GMT))
|
||||||
|
dtsettle2 := ofxgo.Date(time.Date(2016, 12, 20, 12, 0, 0, 0, GMT))
|
||||||
|
dtsettle3 := ofxgo.Date(time.Date(2016, 12, 15, 12, 0, 0, 0, GMT))
|
||||||
|
|
||||||
|
invtranlist := ofxgo.InvTranList{
|
||||||
|
DtStart: ofxgo.Date(time.Date(2016, 12, 6, 12, 0, 0, 0, GMT)),
|
||||||
|
DtEnd: ofxgo.Date(time.Date(2017, 4, 3, 12, 0, 0, 0, GMT)),
|
||||||
|
InvTransactions: []ofxgo.InvTransaction{
|
||||||
|
ofxgo.SellOpt{
|
||||||
|
InvSell: ofxgo.InvSell{
|
||||||
|
InvTran: ofxgo.InvTran{
|
||||||
|
FiTId: "12341234-20161207-1",
|
||||||
|
DtTrade: ofxgo.Date(time.Date(2016, 12, 7, 12, 0, 0, 0, GMT)),
|
||||||
|
DtSettle: &dtsettle1,
|
||||||
|
},
|
||||||
|
SecId: ofxgo.SecurityId{
|
||||||
|
UniqueId: "SPY161216C00226000",
|
||||||
|
UniqueIdType: "CUSIP",
|
||||||
|
},
|
||||||
|
Units: ofxgo.Amount(units1),
|
||||||
|
UnitPrice: ofxgo.Amount(unitprice1),
|
||||||
|
Commission: ofxgo.Amount(commission1),
|
||||||
|
Fees: ofxgo.Amount(fees1),
|
||||||
|
Total: ofxgo.Amount(total1),
|
||||||
|
SubAcctSec: "CASH",
|
||||||
|
SubAcctFund: "CASH",
|
||||||
|
},
|
||||||
|
OptSellType: "SELLTOOPEN",
|
||||||
|
ShPerCtrct: 100,
|
||||||
|
},
|
||||||
|
ofxgo.ClosureOpt{
|
||||||
|
InvTran: ofxgo.InvTran{
|
||||||
|
FiTId: "12341234-20161215-1",
|
||||||
|
DtTrade: ofxgo.Date(time.Date(2016, 12, 15, 12, 0, 0, 0, GMT)),
|
||||||
|
DtSettle: &dtsettle2,
|
||||||
|
},
|
||||||
|
SecId: ofxgo.SecurityId{
|
||||||
|
UniqueId: "78462F10",
|
||||||
|
UniqueIdType: "CUSIP",
|
||||||
|
},
|
||||||
|
OptAction: "ASSIGN",
|
||||||
|
Units: ofxgo.Amount(units2),
|
||||||
|
ShPerCtrct: 100,
|
||||||
|
SubAcctSec: "CASH",
|
||||||
|
},
|
||||||
|
ofxgo.ClosureOpt{
|
||||||
|
InvTran: ofxgo.InvTran{
|
||||||
|
FiTId: "12341234-20161215-2",
|
||||||
|
DtTrade: ofxgo.Date(time.Date(2016, 12, 15, 12, 0, 0, 0, GMT)),
|
||||||
|
DtSettle: &dtsettle3,
|
||||||
|
},
|
||||||
|
SecId: ofxgo.SecurityId{
|
||||||
|
UniqueId: "SPY161216C00226000",
|
||||||
|
UniqueIdType: "CUSIP",
|
||||||
|
},
|
||||||
|
OptAction: "ASSIGN",
|
||||||
|
Units: ofxgo.Amount(units3),
|
||||||
|
ShPerCtrct: 100,
|
||||||
|
SubAcctSec: "CASH",
|
||||||
|
},
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
|
var availcash, marginbalance, shortbalance big.Rat
|
||||||
|
availcash.SetFrac64(0, 1)
|
||||||
|
marginbalance.SetFrac64(-0, 1)
|
||||||
|
shortbalance.SetFrac64(0, 1)
|
||||||
|
|
||||||
|
invbalance := ofxgo.InvBalance{
|
||||||
|
AvailCash: ofxgo.Amount(availcash),
|
||||||
|
MarginBalance: ofxgo.Amount(marginbalance),
|
||||||
|
ShortBalance: ofxgo.Amount(shortbalance),
|
||||||
|
}
|
||||||
|
|
||||||
|
var posunits1, posunitprice1, posmktval1, posunits2, posunitprice2, posmktval2 big.Rat
|
||||||
|
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.Date(time.Date(2017, 4, 3, 12, 0, 0, 0, GMT)),
|
||||||
|
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: "CASH",
|
||||||
|
PosType: "LONG",
|
||||||
|
Units: ofxgo.Amount(posunits1),
|
||||||
|
UnitPrice: ofxgo.Amount(posunitprice1),
|
||||||
|
MktVal: ofxgo.Amount(posmktval1),
|
||||||
|
DtPriceAsOf: ofxgo.Date(time.Date(2017, 4, 3, 12, 0, 0, 0, GMT)),
|
||||||
|
},
|
||||||
|
},
|
||||||
|
ofxgo.StockPosition{
|
||||||
|
InvPos: ofxgo.InvPosition{
|
||||||
|
SecId: ofxgo.SecurityId{
|
||||||
|
UniqueId: "36960410",
|
||||||
|
UniqueIdType: "CUSIP",
|
||||||
|
},
|
||||||
|
HeldInAcct: "CASH",
|
||||||
|
PosType: "LONG",
|
||||||
|
Units: ofxgo.Amount(posunits2),
|
||||||
|
UnitPrice: ofxgo.Amount(posunitprice2),
|
||||||
|
MktVal: ofxgo.Amount(posmktval2),
|
||||||
|
DtPriceAsOf: ofxgo.Date(time.Date(2017, 4, 3, 12, 0, 0, 0, GMT)),
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
InvBal: &invbalance,
|
||||||
|
}
|
||||||
|
expected.InvStmt = append(expected.InvStmt, &statementResponse)
|
||||||
|
|
||||||
|
var strikeprice big.Rat
|
||||||
|
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: "CALL",
|
||||||
|
StrikePrice: ofxgo.Amount(strikeprice),
|
||||||
|
DtExpire: ofxgo.Date(time.Date(2016, 12, 16, 12, 0, 0, 0, GMT)),
|
||||||
|
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)
|
||||||
|
}
|
||||||
|
|||||||
Loading…
Reference in New Issue
Block a user