Browse Source

PortfolioStub

- expose PortfolioStub
Alexey Kim 10 months ago
parent
commit
209c0ff38a
1 changed files with 11 additions and 11 deletions
  1. 11 11
      strategy/portfolio_test.go

+ 11 - 11
strategy/portfolio_test.go

@@ -5,41 +5,41 @@ import (
 	"github.com/samber/lo"
 )
 
-type portfolio_stub []position_stub
+type PortfolioStub []PositionStub
 
-func newStubPortfolio(elem ...position_stub) sentio.Portfolio {
-	var p portfolio_stub
+func newStubPortfolio(elem ...PositionStub) sentio.Portfolio {
+	var p PortfolioStub
 	p = append(p, elem...)
 
 	return p
 }
 
-func (p portfolio_stub) Get(symbol string) (sentio.Position, bool) {
-	return lo.Find(p, func(item position_stub) bool {
+func (p PortfolioStub) Get(symbol string) (sentio.Position, bool) {
+	return lo.Find(p, func(item PositionStub) bool {
 		return item.symbol == symbol
 	})
 }
 
-func (p portfolio_stub) Positions() []sentio.Position {
-	return lo.Map(p, func(item position_stub, _ int) sentio.Position {
+func (p PortfolioStub) Positions() []sentio.Position {
+	return lo.Map(p, func(item PositionStub, _ int) sentio.Position {
 		return item
 	})
 }
 
-type position_stub struct {
+type PositionStub struct {
 	symbol string
 	size   float64
 	price  float64
 }
 
-func (p position_stub) GetSize() float64 {
+func (p PositionStub) GetSize() float64 {
 	return p.size
 }
 
-func (p position_stub) GetAvgPrice() float64 {
+func (p PositionStub) GetAvgPrice() float64 {
 	return p.price
 }
 
-func (p position_stub) GetSymbol() string {
+func (p PositionStub) GetSymbol() string {
 	return p.symbol
 }