diff options
Diffstat (limited to 'store')
-rw-r--r-- | store/sqlstore/emoji_store.go | 19 | ||||
-rw-r--r-- | store/store.go | 2 | ||||
-rw-r--r-- | store/storetest/emoji_store.go | 26 | ||||
-rw-r--r-- | store/storetest/mocks/EmojiStore.go | 10 |
4 files changed, 38 insertions, 19 deletions
diff --git a/store/sqlstore/emoji_store.go b/store/sqlstore/emoji_store.go index 9ef071f02..734190dbb 100644 --- a/store/sqlstore/emoji_store.go +++ b/store/sqlstore/emoji_store.go @@ -123,18 +123,19 @@ func (es SqlEmojiStore) GetByName(name string) store.StoreChannel { }) } -func (es SqlEmojiStore) GetList(offset, limit int) store.StoreChannel { +func (es SqlEmojiStore) GetList(offset, limit int, sort string) store.StoreChannel { return store.Do(func(result *store.StoreResult) { var emoji []*model.Emoji - if _, err := es.GetReplica().Select(&emoji, - `SELECT - * - FROM - Emoji - WHERE - DeleteAt = 0 - LIMIT :Limit OFFSET :Offset`, map[string]interface{}{"Offset": offset, "Limit": limit}); err != nil { + query := "SELECT * FROM Emoji WHERE DeleteAt = 0" + + if sort == model.EMOJI_SORT_BY_NAME { + query += " ORDER BY Name" + } + + query += " LIMIT :Limit OFFSET :Offset" + + if _, err := es.GetReplica().Select(&emoji, query, map[string]interface{}{"Offset": offset, "Limit": limit}); err != nil { result.Err = model.NewAppError("SqlEmojiStore.GetList", "store.sql_emoji.get_all.app_error", nil, err.Error(), http.StatusInternalServerError) } else { result.Data = emoji diff --git a/store/store.go b/store/store.go index c66daec7f..8cb5093ea 100644 --- a/store/store.go +++ b/store/store.go @@ -391,7 +391,7 @@ type EmojiStore interface { Save(emoji *model.Emoji) StoreChannel Get(id string, allowFromCache bool) StoreChannel GetByName(name string) StoreChannel - GetList(offset, limit int) StoreChannel + GetList(offset, limit int, sort string) StoreChannel Delete(id string, time int64) StoreChannel } diff --git a/store/storetest/emoji_store.go b/store/storetest/emoji_store.go index fc5a31a43..a862440e5 100644 --- a/store/storetest/emoji_store.go +++ b/store/storetest/emoji_store.go @@ -9,6 +9,8 @@ import ( "github.com/mattermost/mattermost-server/model" "github.com/mattermost/mattermost-server/store" + + "github.com/stretchr/testify/assert" ) func TestEmojiStore(t *testing.T, ss store.Store) { @@ -133,15 +135,15 @@ func testEmojiGetList(t *testing.T, ss store.Store) { emojis := []model.Emoji{ { CreatorId: model.NewId(), - Name: model.NewId(), + Name: "00000000000000000000000000a" + model.NewId(), }, { CreatorId: model.NewId(), - Name: model.NewId(), + Name: "00000000000000000000000000b" + model.NewId(), }, { CreatorId: model.NewId(), - Name: model.NewId(), + Name: "00000000000000000000000000c" + model.NewId(), }, } @@ -154,7 +156,7 @@ func testEmojiGetList(t *testing.T, ss store.Store) { } }() - if result := <-ss.Emoji().GetList(0, 100); result.Err != nil { + if result := <-ss.Emoji().GetList(0, 100, ""); result.Err != nil { t.Fatal(result.Err) } else { for _, emoji := range emojis { @@ -172,4 +174,20 @@ func testEmojiGetList(t *testing.T, ss store.Store) { } } } + + result := <-ss.Emoji().GetList(0, 3, model.EMOJI_SORT_BY_NAME) + assert.Nil(t, result.Err) + remojis := result.Data.([]*model.Emoji) + assert.Equal(t, 3, len(remojis)) + assert.Equal(t, emojis[0].Name, remojis[0].Name) + assert.Equal(t, emojis[1].Name, remojis[1].Name) + assert.Equal(t, emojis[2].Name, remojis[2].Name) + + result = <-ss.Emoji().GetList(1, 2, model.EMOJI_SORT_BY_NAME) + assert.Nil(t, result.Err) + remojis = result.Data.([]*model.Emoji) + assert.Equal(t, 2, len(remojis)) + assert.Equal(t, emojis[1].Name, remojis[0].Name) + assert.Equal(t, emojis[2].Name, remojis[1].Name) + } diff --git a/store/storetest/mocks/EmojiStore.go b/store/storetest/mocks/EmojiStore.go index c01e64578..d1bfe7f00 100644 --- a/store/storetest/mocks/EmojiStore.go +++ b/store/storetest/mocks/EmojiStore.go @@ -61,13 +61,13 @@ func (_m *EmojiStore) GetByName(name string) store.StoreChannel { return r0 } -// GetList provides a mock function with given fields: offset, limit -func (_m *EmojiStore) GetList(offset int, limit int) store.StoreChannel { - ret := _m.Called(offset, limit) +// GetList provides a mock function with given fields: offset, limit, sort +func (_m *EmojiStore) GetList(offset int, limit int, sort string) store.StoreChannel { + ret := _m.Called(offset, limit, sort) var r0 store.StoreChannel - if rf, ok := ret.Get(0).(func(int, int) store.StoreChannel); ok { - r0 = rf(offset, limit) + if rf, ok := ret.Get(0).(func(int, int, string) store.StoreChannel); ok { + r0 = rf(offset, limit, sort) } else { if ret.Get(0) != nil { r0 = ret.Get(0).(store.StoreChannel) |