Query for instruments

pkg/models/instrument.go

@@ -1,5 +1,7 @@
 package models
 
+import "open-outcry/pkg/db"
+
 // `instrument.pub_id` db reference
 type InstrumentId string
 
@@ -13,9 +15,17 @@ type InstrumentBaseCurrency CurrencyName
 type InstrumentQuoteCurrency CurrencyName
 
 type Instrument struct {
-	Id            InstrumentId
+	Id            InstrumentId `db:"pub_id"`
 	Name          InstrumentName
-	BaseCurrency  InstrumentBaseCurrency
-	QuoteCurrency InstrumentQuoteCurrency
+	BaseCurrency  InstrumentBaseCurrency  `db:"base_currency"`
+	QuoteCurrency InstrumentQuoteCurrency `db:"quote_currency"`
 	Active        bool
 }
+
+func GetInstruments() []Instrument {
+	return db.QueryList[Instrument](`SELECT pub_id, name, quote_currency FROM instrument WHERE fx_instrument = FALSE`)
+}
+
+func GetFxInstruments() []Instrument {
+	return db.QueryList[Instrument](`SELECT pub_id, name, base_currency, quote_currency FROM instrument WHERE fx_instrument = TRUE`)
+}

pkg/models/instrument_test.go

@@ -0,0 +1,9 @@
+package models
+
+func (assert *ModelsTestSuite) TestGetInstruments() {
+	assert.GreaterOrEqual(1, len(GetInstruments()))
+}
+
+func (assert *ModelsTestSuite) TestGetFxInstruments() {
+	assert.GreaterOrEqual(1, len(GetFxInstruments()))
+}

pkg/models/models_suite_test.go

@@ -17,15 +17,15 @@ func TestModelSuite(t *testing.T) {
 	suite.Run(t, &ModelsTestSuite{})
 }
 
-func (suite *ModelsTestSuite) SetupTest() {
+func (assert *ModelsTestSuite) SetupTest() {
 	err := db.SetupInstance()
 	if err != nil {
 		panic(err)
 	}
 	db.MigrateUp()
 }
 
-func (suite *ModelsTestSuite) TearDownTest() {
+func (assert *ModelsTestSuite) TearDownTest() {
 	db.MigrateDown()
 	db.Instance().Close()
 }