76 lines
2.1 KiB
Go
76 lines
2.1 KiB
Go
// Copyright (c) quickfixengine.org All rights reserved.
|
|
//
|
|
// This file may be distributed under the terms of the quickfixengine.org
|
|
// license as defined by quickfixengine.org and appearing in the file
|
|
// LICENSE included in the packaging of this file.
|
|
//
|
|
// This file is provided AS IS with NO WARRANTY OF ANY KIND, INCLUDING
|
|
// THE WARRANTY OF DESIGN, MERCHANTABILITY AND FITNESS FOR A
|
|
// PARTICULAR PURPOSE.
|
|
//
|
|
// See http://www.quickfixengine.org/LICENSE for licensing information.
|
|
//
|
|
// Contact ask@quickfixengine.org if any conditions of this licensing
|
|
// are not clear to you.
|
|
|
|
package mongo
|
|
|
|
import (
|
|
"fmt"
|
|
"log"
|
|
"os"
|
|
"strings"
|
|
"testing"
|
|
|
|
"quantex.com/qfixdpl/quickfix"
|
|
"quantex.com/qfixdpl/quickfix/internal/testsuite"
|
|
"github.com/stretchr/testify/require"
|
|
"github.com/stretchr/testify/suite"
|
|
)
|
|
|
|
// MongoStoreTestSuite runs all tests in the message.StoreTestSuite against the MongoStore implementation.
|
|
type MongoStoreTestSuite struct {
|
|
testsuite.StoreTestSuite
|
|
}
|
|
|
|
func (suite *MongoStoreTestSuite) SetupTest() {
|
|
mongoDbCxn := os.Getenv("MONGODB_TEST_CXN")
|
|
if len(mongoDbCxn) <= 0 {
|
|
log.Println("MONGODB_TEST_CXN environment arg is not provided, skipping...")
|
|
suite.T().SkipNow()
|
|
}
|
|
mongoDatabase := "automated_testing_database"
|
|
mongoReplicaSet := "replicaset"
|
|
|
|
// create settings
|
|
sessionID := quickfix.SessionID{BeginString: "FIX.4.4", SenderCompID: "SENDER", TargetCompID: "TARGET"}
|
|
settings, err := quickfix.ParseSettings(strings.NewReader(fmt.Sprintf(`
|
|
[DEFAULT]
|
|
MongoStoreConnection=%s
|
|
MongoStoreDatabase=%s
|
|
MongoStoreReplicaSet=%s
|
|
|
|
[SESSION]
|
|
BeginString=%s
|
|
SenderCompID=%s
|
|
TargetCompID=%s`, mongoDbCxn, mongoDatabase, mongoReplicaSet, sessionID.BeginString, sessionID.SenderCompID, sessionID.TargetCompID)))
|
|
require.Nil(suite.T(), err)
|
|
|
|
// create store
|
|
suite.MsgStore, err = NewStoreFactory(settings).Create(sessionID)
|
|
require.Nil(suite.T(), err)
|
|
err = suite.MsgStore.Reset()
|
|
require.Nil(suite.T(), err)
|
|
}
|
|
|
|
func (suite *MongoStoreTestSuite) TearDownTest() {
|
|
if suite.MsgStore != nil {
|
|
err := suite.MsgStore.Close()
|
|
require.Nil(suite.T(), err)
|
|
}
|
|
}
|
|
|
|
func TestMongoStoreTestSuite(t *testing.T) {
|
|
suite.Run(t, new(MongoStoreTestSuite))
|
|
}
|