所以我想在我的本地机器上测试一个与Cassandra交互的API。在我的func TestMain(m *testing.M)
函数中,我想在运行测试之前清除表。 TestMain
函数看起来像这样......
func TestMain(m *testing.M) {
keyspace = "staging"
cassandra.SetKeyspace(keyspace)
client = http.DefaultClient
// Ensure all tables are empty before tests run
err := ClearAllTables()
if err != nil {
logrus.Errorf("Failed to clear all tables: %v.", err)
os.Exit(1)
}
// Run tests
statusCode := m.Run()
os.Exit(statusCode)
}
ClearAllTables
函数看起来像这样......
func ClearAllTables() (err error) {
// Create a DB session
session := cassandra.CreateSession()
defer session.Close()
// Get names of all existing tables
tableNameList := []string{"cliques", "users"}
// Remove all rows from each table
var count int
for _, tableName := range tableNameList {
if err := session.Query(`TRUNCATE TABLE ` + tableName).Exec(); err != nil {
return err
}
}
return nil
}
出于某种原因,当我尝试TRUNCATE表时,Cassandra超时,我得到了错误......
level=error msg="Failed to clear all tables: gocql: no response received from cassandra within timeout period."
这似乎只在我测试程序时发生。我在main函数中编写了一段代码,工作正常。
func main () {
session := cassandra.CreateSession()
defer session.Close()
if err := session.Query(`TRUNCATE TABLE cliques`).Exec(); err != nil {
return
}
fmt.Println("Table truncated") //works
}
我还写了一段代码片段,它从数据库中返回一些行,并且在main函数中也可以正常工作。
以下是我如何创建我的cassandra会话......
// CreateSession connect to a cassandra instance
func CreateSession() *gocql.Session {
// Connect to the cluster
cluster := gocql.NewCluster("127.0.0.1")
cluster.Keyspace = "staging"
cluster.ProtoVersion = 4
cluster.CQLVersion = "3.0.0"
cluster.Consistency = gocql.One
session, err := cluster.CreateSession()
if err != nil {
logrus.Errorf("Failed to connect to Cassandra: %v.", err)
}
return session
}
我错过了什么,为什么Cassandra可以与go run main.go
一起使用,但不能与go test
一起使用?