Skip to content

Commit

Permalink
add more tests
Browse files Browse the repository at this point in the history
  • Loading branch information
blind-oracle committed May 2, 2019
1 parent b63ae91 commit 8f2a71f
Show file tree
Hide file tree
Showing 4 changed files with 39 additions and 3 deletions.
3 changes: 2 additions & 1 deletion .travis.yml
Original file line number Diff line number Diff line change
@@ -1,10 +1,11 @@
language: go

sudo: false
go:
- 1.x

before_install:
- go get github.com/mattn/goveralls

script:
- go test ./... -coverprofile=psql-streamer.coverprofile
- $GOPATH/bin/goveralls -service=travis-ci -coverprofile psql-streamer.coverprofile
1 change: 0 additions & 1 deletion sink/kafka/kafka.go
Original file line number Diff line number Diff line change
Expand Up @@ -71,7 +71,6 @@ func New(name string, v *viper.Viper) (s *Kfk, err error) {
Balancer: &kafka.Hash{},
Dialer: &kafka.Dialer{Timeout: v.GetDuration("timeout")},
Brokers: v.GetStringSlice("hosts"),
Topic: v.GetString("topic"),
BatchTimeout: v.GetDuration("batchTimeout"),
BatchSize: v.GetInt("batchSize"),
RequiredAcks: v.GetInt("requiredAcks"),
Expand Down
36 changes: 36 additions & 0 deletions sink/kafka/kafka_test.go
Original file line number Diff line number Diff line change
@@ -1 +1,37 @@
package kafkasink

import (
"github.com/blind-oracle/psql-streamer/event"
"github.com/spf13/viper"
"github.com/stretchr/testify/assert"
"testing"
"time"
)

func Test_Kafka(t *testing.T) {
ev := event.Event{
Host: "foo",
Database: "bar",
Table: "baz",
Action: "insert",
Columns: map[string]interface{}{
"a": "B",
},
Timestamp: time.Unix(123456789, 0),
}

v := viper.New()
v.Set("hosts", []string{"127.0.0.1:1234"})
v.Set("handlers", []string{"passthrough"})
v.Set("topicFallback", "tfb")
v.Set("async", true)

kfk, err := New("test", v)
assert.Nil(t, err)

err = kfk.ProcessEventsBatch([]event.Event{ev})
assert.Nil(t, err)

err = kfk.Close()
assert.Nil(t, err)
}
2 changes: 1 addition & 1 deletion source/postgres/postgres_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -7,13 +7,13 @@ import (
"time"

"github.com/blind-oracle/psql-streamer/mux"
"github.com/stretchr/testify/assert"

"github.com/blind-oracle/pgoutput"
"github.com/blind-oracle/psql-streamer/common"
"github.com/blind-oracle/psql-streamer/event"
"github.com/blind-oracle/psql-streamer/sink"
"github.com/blind-oracle/psql-streamer/sink/stub"
"github.com/stretchr/testify/assert"
)

var (
Expand Down

0 comments on commit 8f2a71f

Please sign in to comment.