3
0

Fix repository tests

This commit is contained in:
Denis Arh 2018-07-30 08:00:12 +02:00
parent 49bfdc26f8
commit a6e86066f4
6 changed files with 51 additions and 60 deletions

View File

@ -14,7 +14,7 @@ func TestChannel(t *testing.T) {
return
}
rpo := Channel()
rpo := New()
ctx := context.Background()
chn := &types.Channel{}
@ -24,7 +24,7 @@ func TestChannel(t *testing.T) {
chn.Name = name1
chn, err = rpo.CreateChannel(ctx, chn)
chn, err = rpo.CreateChannel(chn)
must(t, err)
if chn.Name != name1 {
t.Fatal("Changes were not stored")
@ -32,27 +32,27 @@ func TestChannel(t *testing.T) {
chn.Name = name2
chn, err = rpo.UpdateChannel(ctx, chn)
chn, err = rpo.UpdateChannel(chn)
must(t, err)
if chn.Name != name2 {
t.Fatal("Changes were not stored")
}
chn, err = rpo.FindChannelByID(ctx, chn.ID)
chn, err = rpo.FindChannelByID(chn.ID)
must(t, err)
if chn.Name != name2 {
t.Fatal("Changes were not stored")
}
cc, err = rpo.FindChannels(ctx, &types.ChannelFilter{Query: name2})
cc, err = rpo.FindChannels(&types.ChannelFilter{Query: name2})
must(t, err)
if len(cc) == 0 {
t.Fatal("No results found")
}
must(t, rpo.ArchiveChannel(ctx, chn.ID))
must(t, rpo.UnarchiveChannel(ctx, chn.ID))
must(t, rpo.DeleteChannelByID(ctx, chn.ID))
must(t, rpo.ArchiveChannelByID(chn.ID))
must(t, rpo.UnarchiveChannelByID(chn.ID))
must(t, rpo.DeleteChannelByID(chn.ID))
}
func TestChannelMembers(t *testing.T) {
@ -63,17 +63,16 @@ func TestChannelMembers(t *testing.T) {
return
}
rpo := Channel()
ctx := context.Background()
rpo := New()
chn := &types.Channel{}
chn, err = rpo.CreateChannel(ctx, chn)
chn, err = rpo.CreateChannel(chn)
must(t, err)
usr := &types.User{}
usr, err = User().Create(ctx, usr)
usr, err = rpo.CreateUser(usr)
must(t, err)
must(t, rpo.AddChannelMember(ctx, chn.ID, usr.ID))
must(t, rpo.RemoveChannelMember(ctx, chn.ID, usr.ID))
must(t, rpo.AddChannelMember(chn.ID, usr.ID))
must(t, rpo.RemoveChannelMember(chn.ID, usr.ID))
}

View File

@ -1,7 +1,6 @@
package repository
import (
"context"
"github.com/crusttech/crust/sam/types"
"testing"
)
@ -14,8 +13,7 @@ func TestMessage(t *testing.T) {
return
}
rpo := Message()
ctx := context.Background()
rpo := New()
msg := &types.Message{}
var msg1, msg2 = "Test message v1", "Test message v2"
@ -24,7 +22,7 @@ func TestMessage(t *testing.T) {
msg.Message = msg1
msg, err = rpo.CreateMessage(ctx, msg)
msg, err = rpo.CreateMessage(msg)
must(t, err)
if msg.Message != msg1 {
t.Fatal("Changes were not stored")
@ -32,23 +30,23 @@ func TestMessage(t *testing.T) {
msg.Message = msg2
msg, err = rpo.Update(ctx, msg)
msg, err = rpo.UpdateMessage(msg)
must(t, err)
if msg.Message != msg2 {
t.Fatal("Changes were not stored")
}
msg, err = rpo.FindMessageByID(ctx, msg.ID)
msg, err = rpo.FindMessageByID(msg.ID)
must(t, err)
if msg.Message != msg2 {
t.Fatal("Changes were not stored")
}
mm, err = rpo.FindMessages(ctx, &types.MessageFilter{Query: msg2})
mm, err = rpo.FindMessages(&types.MessageFilter{Query: msg2})
must(t, err)
if len(mm) == 0 {
t.Fatal("No results found")
}
must(t, rpo.Delete(ctx, msg.ID))
must(t, rpo.DeleteMessageByID(msg.ID))
}

View File

@ -24,7 +24,7 @@ func TestOrganisation(t *testing.T) {
org.Name = name1
org, err = rpo.Create(ctx, org)
org, err = rpo.CreateOrganisation(org)
must(t, err)
if org.Name != name1 {
t.Fatal("Changes were not stored")
@ -32,25 +32,25 @@ func TestOrganisation(t *testing.T) {
org.Name = name2
org, err = rpo.Update(ctx, org)
org, err = rpo.UpdateOrganisation(org)
must(t, err)
if org.Name != name2 {
t.Fatal("Changes were not stored")
}
org, err = rpo.FindByID(ctx, org.ID)
org, err = rpo.FindOrganisationByID(org.ID)
must(t, err)
if org.Name != name2 {
t.Fatal("Changes were not stored")
}
oo, err = rpo.Find(ctx, &types.OrganisationFilter{Query: name2})
oo, err = rpo.FindOrganisations(&types.OrganisationFilter{Query: name2})
must(t, err)
if len(oo) == 0 {
t.Fatal("No results found")
}
must(t, rpo.Archive(ctx, org.ID))
must(t, rpo.Unarchive(ctx, org.ID))
must(t, rpo.Delete(ctx, org.ID))
must(t, rpo.ArchiveOrganisationByID(org.ID))
must(t, rpo.UnarchiveOrganisationByID(org.ID))
must(t, rpo.DeleteOrganisationByID(org.ID))
}

View File

@ -1,7 +1,6 @@
package repository
import (
"context"
"github.com/crusttech/crust/sam/types"
"testing"
)
@ -14,25 +13,24 @@ func TestReaction(t *testing.T) {
return
}
rpo := Reaction()
ctx := context.Background()
rpo := New()
react := &types.Reaction{}
var reaction = ":laugh:"
react.Reaction = reaction
react, err = rpo.Create(ctx, react)
react, err = rpo.CreateReaction(react)
must(t, err)
if react.Reaction != reaction {
t.Fatal("Changes were not stored")
}
react, err = rpo.FindByID(ctx, react.ID)
react, err = rpo.FindReactionByID(react.ID)
must(t, err)
if react.Reaction != reaction {
t.Fatal("Changes were not stored")
}
must(t, rpo.Delete(ctx, react.ID))
must(t, rpo.DeleteReactionByID(react.ID))
}

View File

@ -1,7 +1,6 @@
package repository
import (
"context"
"github.com/crusttech/crust/sam/types"
"testing"
)
@ -14,8 +13,7 @@ func TestTeam(t *testing.T) {
return
}
rpo := Team()
ctx := context.Background()
rpo := New()
team := &types.Team{}
var name1, name2 = "Test team v1", "Test team v2"
@ -24,7 +22,7 @@ func TestTeam(t *testing.T) {
team.Name = name1
team, err = rpo.Create(ctx, team)
team, err = rpo.CreateTeam(team)
must(t, err)
if team.Name != name1 {
t.Fatal("Changes were not stored")
@ -32,25 +30,25 @@ func TestTeam(t *testing.T) {
team.Name = name2
team, err = rpo.Update(ctx, team)
team, err = rpo.UpdateTeam(team)
must(t, err)
if team.Name != name2 {
t.Fatal("Changes were not stored")
}
team, err = rpo.FindByID(ctx, team.ID)
team, err = rpo.FindTeamByID(team.ID)
must(t, err)
if team.Name != name2 {
t.Fatal("Changes were not stored")
}
aa, err = rpo.Find(ctx, &types.TeamFilter{Query: name2})
aa, err = rpo.FindTeams(&types.TeamFilter{Query: name2})
must(t, err)
if len(aa) == 0 {
t.Fatal("No results found")
}
must(t, rpo.Archive(ctx, team.ID))
must(t, rpo.Unarchive(ctx, team.ID))
must(t, rpo.Delete(ctx, team.ID))
must(t, rpo.ArchiveTeamByID(team.ID))
must(t, rpo.UnarchiveTeamByID(team.ID))
must(t, rpo.DeleteTeamByID(team.ID))
}

View File

@ -1,7 +1,6 @@
package repository
import (
"context"
"github.com/crusttech/crust/sam/types"
"testing"
)
@ -14,41 +13,40 @@ func TestUser(t *testing.T) {
return
}
rpo := User()
ctx := context.Background()
usr := &types.User{}
rpo := New()
team := &types.Team{}
var name1, name2 = "Test user v1", "Test user v2"
var aa []*types.User
var aa []*types.Team
usr.Username = name1
team.Name = name1
usr, err = rpo.Create(ctx, usr)
team, err = rpo.CreateTeam(team)
must(t, err)
if usr.Username != name1 {
if team.Name != name1 {
t.Fatal("Changes were not stored")
}
usr.Username = name2
team.Name = name2
usr, err = rpo.Update(ctx, usr)
team, err = rpo.UpdateTeam(team)
must(t, err)
if usr.Username != name2 {
if team.Name != name2 {
t.Fatal("Changes were not stored")
}
usr, err = rpo.FindByID(ctx, usr.ID)
team, err = rpo.FindTeamByID(team.ID)
must(t, err)
if usr.Username != name2 {
if team.Name != name2 {
t.Fatal("Changes were not stored")
}
aa, err = rpo.Find(ctx, &types.UserFilter{Query: name2})
aa, err = rpo.FindTeams(&types.TeamFilter{Query: name2})
must(t, err)
if len(aa) == 0 {
t.Fatal("No results found")
}
must(t, rpo.Delete(ctx, usr.ID))
must(t, rpo.DeleteTeamByID(team.ID))
}