package badger import ( "sync" "testing" "github.com/appleboy/gorush/config" "github.com/appleboy/gorush/core" "github.com/stretchr/testify/assert" ) func TestBadgerEngine(t *testing.T) { var val int64 cfg, _ := config.LoadConf() badger := New(cfg) err := badger.Init() assert.Nil(t, err) badger.Add(core.HuaweiSuccessKey, 10) val = badger.Get(core.HuaweiSuccessKey) assert.Equal(t, int64(10), val) badger.Add(core.HuaweiSuccessKey, 10) val = badger.Get(core.HuaweiSuccessKey) assert.Equal(t, int64(20), val) badger.Set(core.HuaweiSuccessKey, 0) val = badger.Get(core.HuaweiSuccessKey) assert.Equal(t, int64(0), val) // test concurrency issues var wg sync.WaitGroup for i := 0; i < 10; i++ { wg.Add(1) go func() { badger.Add(core.HuaweiSuccessKey, 1) wg.Done() }() } wg.Wait() val = badger.Get(core.HuaweiSuccessKey) assert.Equal(t, int64(10), val) assert.NoError(t, badger.Close()) }