+
+func TestAddMemberSuccessfully(t *testing.T) {
+ m, i := setup()
+
+ groupExpected := string("{namespace},group")
+ membersExpected := []interface{}{"member1", "member2"}
+
+ m.On("SAdd", groupExpected, membersExpected).Return(nil)
+
+ err := i.AddMember("group", "member1", "member2")
+ assert.Nil(t, err)
+ m.AssertExpectations(t)
+}
+
+func TestAddMemberFail(t *testing.T) {
+ m, i := setup()
+
+ groupExpected := string("{namespace},group")
+ membersExpected := []interface{}{"member1", "member2"}
+
+ m.On("SAdd", groupExpected, membersExpected).Return(errors.New("Some error"))
+
+ err := i.AddMember("group", "member1", "member2")
+ assert.NotNil(t, err)
+ m.AssertExpectations(t)
+}
+func TestRemoveMemberSuccessfully(t *testing.T) {
+ m, i := setup()
+
+ groupExpected := string("{namespace},group")
+ membersExpected := []interface{}{"member1", "member2"}
+
+ m.On("SRem", groupExpected, membersExpected).Return(nil)
+
+ err := i.RemoveMember("group", "member1", "member2")
+ assert.Nil(t, err)
+ m.AssertExpectations(t)
+}
+
+func TestRemoveMemberFail(t *testing.T) {
+ m, i := setup()
+
+ groupExpected := string("{namespace},group")
+ membersExpected := []interface{}{"member1", "member2"}
+
+ m.On("SRem", groupExpected, membersExpected).Return(errors.New("Some error"))
+
+ err := i.RemoveMember("group", "member1", "member2")
+ assert.NotNil(t, err)
+ m.AssertExpectations(t)
+}
+
+func TestRemoveGroupSuccessfully(t *testing.T) {
+ m, i := setup()
+
+ groupExpected := []string{"{namespace},group"}
+
+ m.On("Del", groupExpected).Return(nil)
+
+ err := i.RemoveGroup("group")
+ assert.Nil(t, err)
+ m.AssertExpectations(t)
+}
+func TestRemoveGroupFail(t *testing.T) {
+ m, i := setup()
+
+ groupExpected := []string{"{namespace},group"}
+
+ m.On("Del", groupExpected).Return(errors.New("Some error"))
+
+ err := i.RemoveGroup("group")
+ assert.NotNil(t, err)
+ m.AssertExpectations(t)
+}
+
+func TestGetMembersSuccessfully(t *testing.T) {
+ m, i := setup()
+
+ groupExpected := "{namespace},group"
+ returnExpected := []string{"member1", "member2"}
+
+ m.On("SMembers", groupExpected).Return(returnExpected, nil)
+
+ result, err := i.GetMembers("group")
+ assert.Nil(t, err)
+ assert.Equal(t, result, returnExpected)
+ m.AssertExpectations(t)
+}
+func TestGetMembersFail(t *testing.T) {
+ m, i := setup()
+
+ groupExpected := "{namespace},group"
+ returnExpected := []string{"member1", "member2"}
+
+ m.On("SMembers", groupExpected).Return(returnExpected, errors.New("Some error"))
+
+ result, err := i.GetMembers("group")
+ assert.NotNil(t, err)
+ assert.Equal(t, []string{}, result)
+ m.AssertExpectations(t)
+}
+
+func TestIsMemberSuccessfullyIsMember(t *testing.T) {
+ m, i := setup()
+
+ groupExpected := "{namespace},group"
+ memberExpected := "member"
+
+ m.On("SIsMember", groupExpected, memberExpected).Return(true, nil)
+
+ result, err := i.IsMember("group", "member")
+ assert.Nil(t, err)
+ assert.True(t, result)
+ m.AssertExpectations(t)
+}
+func TestIsMemberSuccessfullyIsNotMember(t *testing.T) {
+ m, i := setup()
+
+ groupExpected := "{namespace},group"
+ memberExpected := "member"
+
+ m.On("SIsMember", groupExpected, memberExpected).Return(false, nil)
+
+ result, err := i.IsMember("group", "member")
+ assert.Nil(t, err)
+ assert.False(t, result)
+ m.AssertExpectations(t)
+}
+func TestIsMemberFailure(t *testing.T) {
+ m, i := setup()
+
+ groupExpected := "{namespace},group"
+ memberExpected := "member"
+
+ m.On("SIsMember", groupExpected, memberExpected).Return(true, errors.New("Some error"))
+
+ result, err := i.IsMember("group", "member")
+ assert.NotNil(t, err)
+ assert.False(t, result)
+ m.AssertExpectations(t)
+}
+
+func TestGroupSizeSuccessfully(t *testing.T) {
+ m, i := setup()
+
+ var expectedSize int64
+ expectedSize = 2
+ groupExpected := "{namespace},group"
+
+ m.On("SCard", groupExpected).Return(expectedSize, nil)
+
+ result, err := i.GroupSize("group")
+ assert.Nil(t, err)
+ assert.Equal(t, expectedSize, result)
+ m.AssertExpectations(t)
+}
+func TestGroupSizeFail(t *testing.T) {
+ m, i := setup()
+
+ var expectedSize int64
+ expectedSize = 2
+ groupExpected := "{namespace},group"
+
+ m.On("SCard", groupExpected).Return(expectedSize, errors.New("Some error"))
+
+ result, err := i.GroupSize("group")
+ assert.NotNil(t, err)
+ assert.Equal(t, int64(0), result)
+ m.AssertExpectations(t)
+}
+
+func TestLockResourceSuccessfully(t *testing.T) {
+ m, i := setup()
+
+ resourceExpected := "{namespace},resource"
+ m.On("SetNX", resourceExpected, mock.Anything, time.Duration(1)).Return(true, nil)
+
+ lock, err := i.LockResource("resource", time.Duration(1), &sdlgo.Options{})
+ assert.Nil(t, err)
+ assert.NotNil(t, lock)
+ m.AssertExpectations(t)
+}
+
+func TestLockResourceFailure(t *testing.T) {
+ m, i := setup()
+
+ resourceExpected := "{namespace},resource"
+ m.On("SetNX", resourceExpected, mock.Anything, time.Duration(1)).Return(true, errors.New("Some error"))
+
+ lock, err := i.LockResource("resource", time.Duration(1), &sdlgo.Options{})
+ assert.NotNil(t, err)
+ assert.Nil(t, lock)
+ m.AssertExpectations(t)
+}
+
+func TestLockResourceTrySeveralTimesSuccessfully(t *testing.T) {
+ m, i := setup()
+
+ resourceExpected := "{namespace},resource"
+ m.On("SetNX", resourceExpected, mock.Anything, time.Duration(1)).Return(false, nil).Once()
+ m.On("SetNX", resourceExpected, mock.Anything, time.Duration(1)).Return(true, nil).Once()
+
+ lock, err := i.LockResource("resource", time.Duration(1), &sdlgo.Options{
+ RetryCount: 2,
+ })
+ assert.Nil(t, err)
+ assert.NotNil(t, lock)
+ m.AssertExpectations(t)
+}
+
+func TestLockResourceTrySeveralTimesFailure(t *testing.T) {
+ m, i := setup()
+
+ resourceExpected := "{namespace},resource"
+ m.On("SetNX", resourceExpected, mock.Anything, time.Duration(1)).Return(false, nil).Once()
+ m.On("SetNX", resourceExpected, mock.Anything, time.Duration(1)).Return(true, errors.New("Some error")).Once()
+
+ lock, err := i.LockResource("resource", time.Duration(1), &sdlgo.Options{
+ RetryCount: 2,
+ })
+ assert.NotNil(t, err)
+ assert.Nil(t, lock)
+ m.AssertExpectations(t)
+}
+
+func TestLockResourceTrySeveralTimesUnableToGetResource(t *testing.T) {
+ m, i := setup()
+
+ resourceExpected := "{namespace},resource"
+ m.On("SetNX", resourceExpected, mock.Anything, time.Duration(1)).Return(false, nil).Once()
+ m.On("SetNX", resourceExpected, mock.Anything, time.Duration(1)).Return(false, nil).Once()
+
+ lock, err := i.LockResource("resource", time.Duration(1), &sdlgo.Options{
+ RetryCount: 1,
+ })
+ assert.NotNil(t, err)
+ assert.EqualError(t, err, "Lock not obtained")
+ assert.Nil(t, lock)
+ m.AssertExpectations(t)
+}
+
+func TestReleaseResourceSuccessfully(t *testing.T) {
+ m, i := setup()
+
+ resourceExpected := "{namespace},resource"
+ m.On("SetNX", resourceExpected, mock.Anything, time.Duration(1)).Return(true, nil).Once()
+ m.On("DelIE", resourceExpected, mock.Anything).Return(true, nil).Once()
+
+ lock, err := i.LockResource("resource", time.Duration(1), &sdlgo.Options{
+ RetryCount: 1,
+ })
+ err2 := lock.ReleaseResource()
+ assert.Nil(t, err)
+ assert.NotNil(t, lock)
+ assert.Nil(t, err2)
+ m.AssertExpectations(t)
+}
+
+func TestReleaseResourceFailure(t *testing.T) {
+ m, i := setup()
+
+ resourceExpected := "{namespace},resource"
+ m.On("SetNX", resourceExpected, mock.Anything, time.Duration(1)).Return(true, nil).Once()
+ m.On("DelIE", resourceExpected, mock.Anything).Return(true, errors.New("Some error")).Once()
+
+ lock, err := i.LockResource("resource", time.Duration(1), &sdlgo.Options{
+ RetryCount: 1,
+ })
+ err2 := lock.ReleaseResource()
+ assert.Nil(t, err)
+ assert.NotNil(t, lock)
+ assert.NotNil(t, err2)
+ m.AssertExpectations(t)
+}
+
+func TestReleaseResourceLockNotHeld(t *testing.T) {
+ m, i := setup()
+
+ resourceExpected := "{namespace},resource"
+ m.On("SetNX", resourceExpected, mock.Anything, time.Duration(1)).Return(true, nil).Once()
+ m.On("DelIE", resourceExpected, mock.Anything).Return(false, nil).Once()
+
+ lock, err := i.LockResource("resource", time.Duration(1), &sdlgo.Options{
+ RetryCount: 1,
+ })
+ err2 := lock.ReleaseResource()
+ assert.Nil(t, err)
+ assert.NotNil(t, lock)
+ assert.NotNil(t, err2)
+ assert.EqualError(t, err2, "Lock not held")
+ m.AssertExpectations(t)
+}
+
+func TestRefreshResourceSuccessfully(t *testing.T) {
+ m, i := setup()
+
+ resourceExpected := "{namespace},resource"
+ m.On("SetNX", resourceExpected, mock.Anything, time.Duration(1)).Return(true, nil).Once()
+ m.On("PExpireIE", resourceExpected, mock.Anything, time.Duration(1)).Return(nil).Once()
+
+ lock, err := i.LockResource("resource", time.Duration(1), &sdlgo.Options{
+ RetryCount: 1,
+ })
+ err2 := lock.RefreshResource(time.Duration(1))
+ assert.Nil(t, err)
+ assert.NotNil(t, lock)
+ assert.Nil(t, err2)
+ m.AssertExpectations(t)
+}
+
+func TestRefreshResourceFailure(t *testing.T) {
+ m, i := setup()
+
+ resourceExpected := "{namespace},resource"
+ m.On("SetNX", resourceExpected, mock.Anything, time.Duration(1)).Return(true, nil).Once()
+ m.On("PExpireIE", resourceExpected, mock.Anything, time.Duration(1)).Return(errors.New("Some error")).Once()
+
+ lock, err := i.LockResource("resource", time.Duration(1), &sdlgo.Options{
+ RetryCount: 1,
+ })
+ err2 := lock.RefreshResource(time.Duration(1))
+ assert.Nil(t, err)
+ assert.NotNil(t, lock)
+ assert.NotNil(t, err2)
+ m.AssertExpectations(t)
+}
+
+func TestCheckResourceSuccessfully(t *testing.T) {
+ m, i := setup()
+
+ resourceExpected := "{namespace},resource"
+ m.On("PTTL", resourceExpected).Return(time.Duration(1), nil)
+ result, err := i.CheckResource("resource")
+ assert.Nil(t, err)
+ assert.Equal(t, result, time.Duration(1))
+ m.AssertExpectations(t)
+}
+
+func TestCheckResourceFailure(t *testing.T) {
+ m, i := setup()
+
+ resourceExpected := "{namespace},resource"
+ m.On("PTTL", resourceExpected).Return(time.Duration(1), errors.New("Some error"))
+ result, err := i.CheckResource("resource")
+ assert.NotNil(t, err)
+ assert.EqualError(t, err, "Some error")
+ assert.Equal(t, result, time.Duration(0))
+ m.AssertExpectations(t)
+}
+
+func TestCheckResourceInvalidResource(t *testing.T) {
+ m, i := setup()
+
+ resourceExpected := "{namespace},resource"
+ m.On("PTTL", resourceExpected).Return(time.Duration(-1), nil)
+ result, err := i.CheckResource("resource")
+ assert.NotNil(t, err)
+ assert.EqualError(t, err, "invalid resource given, no expiration time attached")
+ assert.Equal(t, result, time.Duration(0))
+ m.AssertExpectations(t)
+}