mirror of https://github.com/certusone/dc4bc.git
Fixed tests
This commit is contained in:
parent
a99273adaf
commit
e5cfb2a337
|
@ -6,12 +6,13 @@ import (
|
||||||
"encoding/json"
|
"encoding/json"
|
||||||
"errors"
|
"errors"
|
||||||
"fmt"
|
"fmt"
|
||||||
sipf "github.com/depools/dc4bc/fsm/state_machines/signing_proposal_fsm"
|
|
||||||
"log"
|
"log"
|
||||||
"path/filepath"
|
"path/filepath"
|
||||||
"sync"
|
"sync"
|
||||||
"time"
|
"time"
|
||||||
|
|
||||||
|
sipf "github.com/depools/dc4bc/fsm/state_machines/signing_proposal_fsm"
|
||||||
|
|
||||||
"github.com/depools/dc4bc/client/types"
|
"github.com/depools/dc4bc/client/types"
|
||||||
"github.com/depools/dc4bc/fsm/types/requests"
|
"github.com/depools/dc4bc/fsm/types/requests"
|
||||||
"github.com/google/uuid"
|
"github.com/google/uuid"
|
||||||
|
@ -115,7 +116,7 @@ func (c *Client) Poll() error {
|
||||||
if message.RecipientAddr == "" || message.RecipientAddr == c.GetAddr() {
|
if message.RecipientAddr == "" || message.RecipientAddr == c.GetAddr() {
|
||||||
c.Logger.Log("Handling message with offset %d, type %s", message.Offset, message.Event)
|
c.Logger.Log("Handling message with offset %d, type %s", message.Offset, message.Event)
|
||||||
if err := c.ProcessMessage(message); err != nil {
|
if err := c.ProcessMessage(message); err != nil {
|
||||||
c.Logger.Log("Failed to process message: %v", err)
|
c.Logger.Log("Failed to process message with offset %d: %v", message.Offset, err)
|
||||||
} else {
|
} else {
|
||||||
c.Logger.Log("Successfully processed message with offset %d, type %s",
|
c.Logger.Log("Successfully processed message with offset %d, type %s",
|
||||||
message.Offset, message.Event)
|
message.Offset, message.Event)
|
||||||
|
|
|
@ -52,7 +52,7 @@ func (s *KafkaStorage) Send(m Message) (Message, error) {
|
||||||
return Message{}, fmt.Errorf("failed to WriteMessages: %w", err)
|
return Message{}, fmt.Errorf("failed to WriteMessages: %w", err)
|
||||||
}
|
}
|
||||||
|
|
||||||
return Message{}, nil
|
return m, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func (s *KafkaStorage) GetMessages(offset uint64) ([]Message, error) {
|
func (s *KafkaStorage) GetMessages(offset uint64) ([]Message, error) {
|
||||||
|
|
|
@ -8,33 +8,63 @@ import (
|
||||||
)
|
)
|
||||||
|
|
||||||
// kafkacat -C -b localhost -t messages
|
// kafkacat -C -b localhost -t messages
|
||||||
func TestKafkaStorage_Send(t *testing.T) {
|
//func TestKafkaStorage_Send(t *testing.T) {
|
||||||
|
// if testing.Short() {
|
||||||
|
// t.Skip("skipping long test")
|
||||||
|
// }
|
||||||
|
//
|
||||||
|
// req := require.New(t)
|
||||||
|
// stg, err := NewKafkaStorage(context.Background(), "localhost:9092")
|
||||||
|
// req.NoError(err)
|
||||||
|
//
|
||||||
|
// t.Run("test_kafka_storage_send", func(t *testing.T) {
|
||||||
|
// _, err = stg.Send(Message{
|
||||||
|
// ID: "test_message_id",
|
||||||
|
// DkgRoundID: "test_dkg_round_id",
|
||||||
|
// Event: "test_event",
|
||||||
|
// Data: []byte("test_data"),
|
||||||
|
// Signature: []byte("test_signature"),
|
||||||
|
// SenderAddr: "test_sender_addr",
|
||||||
|
// RecipientAddr: "test_recipient_addr",
|
||||||
|
// })
|
||||||
|
//
|
||||||
|
// req.NoError(err)
|
||||||
|
// })
|
||||||
|
//}
|
||||||
|
|
||||||
|
func TestKafkaStorage_GetMessages(t *testing.T) {
|
||||||
if testing.Short() {
|
if testing.Short() {
|
||||||
t.Skip("skipping long test")
|
t.Skip("skipping long test")
|
||||||
}
|
}
|
||||||
|
|
||||||
|
N := 10
|
||||||
|
var offset uint64 = 5
|
||||||
|
|
||||||
req := require.New(t)
|
req := require.New(t)
|
||||||
stg, err := NewKafkaStorage(context.Background(), "localhost:9092")
|
stg, err := NewKafkaStorage(context.Background(), "localhost:9092")
|
||||||
req.NoError(err)
|
req.NoError(err)
|
||||||
|
|
||||||
t.Run("test_kafka_storage_send", func(t *testing.T) {
|
msgs := make([]Message, 0, N)
|
||||||
_, err = stg.Send(Message{
|
for i := 0; i < N; i++ {
|
||||||
ID: "test_message_id",
|
msg := Message{
|
||||||
DkgRoundID: "test_dkg_round_id",
|
Data: randomBytes(10),
|
||||||
Event: "test_event",
|
Signature: randomBytes(10),
|
||||||
Data: []byte("test_data"),
|
}
|
||||||
Signature: []byte("test_signature"),
|
msg, err = stg.Send(msg)
|
||||||
SenderAddr: "test_sender_addr",
|
if err != nil {
|
||||||
RecipientAddr: "test_recipient_addr",
|
t.Error(err)
|
||||||
})
|
}
|
||||||
|
msgs = append(msgs, msg)
|
||||||
|
}
|
||||||
|
|
||||||
req.NoError(err)
|
offsetMsgs, err := stg.GetMessages(offset)
|
||||||
})
|
if err != nil {
|
||||||
|
t.Error(err)
|
||||||
|
}
|
||||||
|
|
||||||
t.Run("test_kafka_storage_get_messages", func(t *testing.T) {
|
expectedOffsetMsgs := msgs[offset:]
|
||||||
msgs, err := stg.GetMessages(0)
|
|
||||||
|
|
||||||
req.NoError(err)
|
for idx, msg := range expectedOffsetMsgs {
|
||||||
req.Len(msgs, 1)
|
req.Equal(msg.Signature, offsetMsgs[idx].Signature)
|
||||||
})
|
}
|
||||||
}
|
}
|
||||||
|
|
|
@ -2,8 +2,6 @@
|
||||||
|
|
||||||
echo "-=Stop old docker-compose=-"
|
echo "-=Stop old docker-compose=-"
|
||||||
docker-compose down -v
|
docker-compose down -v
|
||||||
#docker stop dc4bc-kafka && docker rm dc4bc-kafka
|
|
||||||
#docker stop dc4bc-zookeeper && docker rm dc4bc-zookeeper
|
|
||||||
|
|
||||||
echo "-=Start new docker-compose=-"
|
echo "-=Start new docker-compose=-"
|
||||||
docker-compose up -d --build dc4bc-kafka dc4bc-zookeeper
|
docker-compose up -d --build dc4bc-kafka dc4bc-zookeeper
|
||||||
|
@ -15,8 +13,10 @@ echo "-=Start tests=-"
|
||||||
cd ../
|
cd ../
|
||||||
|
|
||||||
# shellcheck disable=SC2046
|
# shellcheck disable=SC2046
|
||||||
go test $(go list ./...)
|
go test $(go list ./... | grep storage)
|
||||||
|
|
||||||
echo "-=Stop docker-compose=-"
|
echo "-=Stop docker-compose=-"
|
||||||
cd tests
|
cd tests
|
||||||
docker-compose down -v
|
#docker-compose down -v
|
||||||
|
#docker rm dc4bc-kafka
|
||||||
|
#docker rm dc4bc-zookeeper
|
||||||
|
|
Loading…
Reference in New Issue