Skip to content
This repository was archived by the owner on Dec 9, 2024. It is now read-only.

Commit 0bf2d7f

Browse files
authored
fix listing all subscriptions (#366)
1 parent d70a637 commit 0bf2d7f

File tree

3 files changed

+11
-14
lines changed

3 files changed

+11
-14
lines changed

libkv/space.go

Lines changed: 0 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -9,9 +9,6 @@ import (
99
const defaultSpace = "default"
1010

1111
func spacePath(space string) string {
12-
if space == "" {
13-
return ""
14-
}
1512
return space + "/"
1613
}
1714

libkv/subscription.go

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -96,7 +96,7 @@ func (service Service) DeleteSubscription(space string, id subscription.ID) erro
9696
func (service Service) GetSubscriptions(space string) (subscription.Subscriptions, error) {
9797
subs := []*subscription.Subscription{}
9898

99-
kvs, err := service.SubscriptionStore.List("", &store.ReadOptions{Consistent: true})
99+
kvs, err := service.SubscriptionStore.List(spacePath(space), &store.ReadOptions{Consistent: true})
100100
if err != nil && err.Error() != errKeyNotFound {
101101
return nil, err
102102
}

libkv/subscription_test.go

Lines changed: 10 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -245,35 +245,35 @@ func TestDeleteSubscription_DeleteEndpointError(t *testing.T) {
245245
assert.EqualError(t, err, "KV Delete err")
246246
}
247247

248-
func TestGetAllSubscriptions_OK(t *testing.T) {
248+
func TestGetSubscriptions_OK(t *testing.T) {
249249
ctrl := gomock.NewController(t)
250250
defer ctrl.Finish()
251251

252252
kvs := []*store.KVPair{
253-
{Value: []byte(`{"subscriptionId":"s1","event":"test","functionId":"f1"}`)},
254-
{Value: []byte(`{"subscriptionId":"s2","event":"test","functionId":"f2"}`)},
253+
{Value: []byte(`{"subscriptionId":"s1","space":"default","event":"test","functionId":"f1"}`)},
254+
{Value: []byte(`{"subscriptionId":"s2","space":"default","event":"test","functionId":"f2"}`)},
255255
}
256256
subscriptionsDB := mock.NewMockStore(ctrl)
257-
subscriptionsDB.EXPECT().List("", &store.ReadOptions{Consistent: true}).Return(kvs, nil)
257+
subscriptionsDB.EXPECT().List("default/", &store.ReadOptions{Consistent: true}).Return(kvs, nil)
258258
subs := &Service{SubscriptionStore: subscriptionsDB, Log: zap.NewNop()}
259259

260-
list, _ := subs.GetSubscriptions("")
260+
list, _ := subs.GetSubscriptions("default")
261261

262262
assert.Equal(t, subscription.Subscriptions{
263-
{ID: subscription.ID("s1"), Event: "test", FunctionID: function.ID("f1")},
264-
{ID: subscription.ID("s2"), Event: "test", FunctionID: function.ID("f2")},
263+
{ID: subscription.ID("s1"), Space: "default", Event: "test", FunctionID: function.ID("f1")},
264+
{ID: subscription.ID("s2"), Space: "default", Event: "test", FunctionID: function.ID("f2")},
265265
}, list)
266266
}
267267

268-
func TestGetAllSubscriptions_ListError(t *testing.T) {
268+
func TestGetSubscriptions_ListError(t *testing.T) {
269269
ctrl := gomock.NewController(t)
270270
defer ctrl.Finish()
271271

272272
subscriptionsDB := mock.NewMockStore(ctrl)
273-
subscriptionsDB.EXPECT().List("", gomock.Any()).Return(nil, errors.New("KV error"))
273+
subscriptionsDB.EXPECT().List("default/", gomock.Any()).Return(nil, errors.New("KV error"))
274274
subs := &Service{SubscriptionStore: subscriptionsDB, Log: zap.NewNop()}
275275

276-
_, err := subs.GetSubscriptions("")
276+
_, err := subs.GetSubscriptions("default")
277277
assert.EqualError(t, err, "KV error")
278278
}
279279

0 commit comments

Comments
 (0)