summaryrefslogtreecommitdiffstats
path: root/store
diff options
context:
space:
mode:
Diffstat (limited to 'store')
-rw-r--r--store/sql_channel_store.go23
-rw-r--r--store/sql_channel_store_test.go27
-rw-r--r--store/store.go1
3 files changed, 51 insertions, 0 deletions
diff --git a/store/sql_channel_store.go b/store/sql_channel_store.go
index 07c037075..eb150a63c 100644
--- a/store/sql_channel_store.go
+++ b/store/sql_channel_store.go
@@ -83,6 +83,29 @@ func (s SqlChannelStore) Save(channel *model.Channel) StoreChannel {
return storeChannel
}
+func (s SqlChannelStore) CreateDirectChannel(userId string, otherUserId string) StoreChannel {
+ channel := new(model.Channel)
+
+ channel.DisplayName = ""
+ channel.Name = model.GetDMNameFromIds(otherUserId, userId)
+
+ channel.Header = ""
+ channel.Type = model.CHANNEL_DIRECT
+
+ cm1 := &model.ChannelMember{
+ UserId: userId,
+ NotifyProps: model.GetDefaultChannelNotifyProps(),
+ Roles: model.ROLE_CHANNEL_USER.Id,
+ }
+ cm2 := &model.ChannelMember{
+ UserId: otherUserId,
+ NotifyProps: model.GetDefaultChannelNotifyProps(),
+ Roles: model.ROLE_CHANNEL_USER.Id,
+ }
+
+ return s.SaveDirectChannel(channel, cm1, cm2)
+}
+
func (s SqlChannelStore) SaveDirectChannel(directchannel *model.Channel, member1 *model.ChannelMember, member2 *model.ChannelMember) StoreChannel {
storeChannel := make(StoreChannel, 1)
diff --git a/store/sql_channel_store_test.go b/store/sql_channel_store_test.go
index 0bd059e5f..19db3d003 100644
--- a/store/sql_channel_store_test.go
+++ b/store/sql_channel_store_test.go
@@ -94,7 +94,34 @@ func TestChannelStoreSaveDirectChannel(t *testing.T) {
if err := (<-store.Channel().SaveDirectChannel(&o1, &m1, &m2)).Err; err == nil {
t.Fatal("Should not be able to save non-direct channel")
}
+}
+
+func TestChannelStoreCreateDirectChannel(t *testing.T) {
+ Setup()
+ u1 := &model.User{}
+ u1.Email = model.NewId()
+ u1.Nickname = model.NewId()
+ Must(store.User().Save(u1))
+ Must(store.Team().SaveMember(&model.TeamMember{TeamId: model.NewId(), UserId: u1.Id}))
+
+ u2 := &model.User{}
+ u2.Email = model.NewId()
+ u2.Nickname = model.NewId()
+ Must(store.User().Save(u2))
+ Must(store.Team().SaveMember(&model.TeamMember{TeamId: model.NewId(), UserId: u2.Id}))
+
+ res := <-store.Channel().CreateDirectChannel(u1.Id, u2.Id)
+ if res.Err != nil {
+ t.Fatal("couldn't create direct channel", res.Err)
+ }
+
+ c1 := res.Data.(*model.Channel)
+
+ members := (<-store.Channel().GetMembers(c1.Id)).Data.([]model.ChannelMember)
+ if len(members) != 2 {
+ t.Fatal("should have saved 2 members")
+ }
}
func TestChannelStoreUpdate(t *testing.T) {
diff --git a/store/store.go b/store/store.go
index c01b6d8bc..7474d3afb 100644
--- a/store/store.go
+++ b/store/store.go
@@ -75,6 +75,7 @@ type TeamStore interface {
type ChannelStore interface {
Save(channel *model.Channel) StoreChannel
+ CreateDirectChannel(userId string, otherUserId string) StoreChannel
SaveDirectChannel(channel *model.Channel, member1 *model.ChannelMember, member2 *model.ChannelMember) StoreChannel
Update(channel *model.Channel) StoreChannel
Get(id string) StoreChannel