Browse Source

Fixed unit tests

Torkel Ödegaard 11 years ago
parent
commit
cd5843e977
2 changed files with 12 additions and 12 deletions
  1. 10 10
      pkg/services/sqlstore/accounts_test.go
  2. 2 2
      pkg/services/sqlstore/tokens.go

+ 10 - 10
pkg/services/sqlstore/accounts_test.go

@@ -30,6 +30,16 @@ func TestAccountDataAccess(t *testing.T) {
 
 
 				So(err, ShouldBeNil)
 				So(err, ShouldBeNil)
 				So(query.Result.Email, ShouldEqual, "ac1@test.com")
 				So(query.Result.Email, ShouldEqual, "ac1@test.com")
+				So(query.Result.Login, ShouldEqual, "ac1")
+			})
+
+			Convey("Can search accounts", func() {
+				query := m.SearchAccountsQuery{Query: ""}
+				err := SearchAccounts(&query)
+
+				So(err, ShouldBeNil)
+				So(query.Result[0].Email, ShouldEqual, "ac1@test.com")
+				So(query.Result[1].Email, ShouldEqual, "ac2@test.com")
 			})
 			})
 
 
 			Convey("Can add collaborator", func() {
 			Convey("Can add collaborator", func() {
@@ -44,16 +54,6 @@ func TestAccountDataAccess(t *testing.T) {
 					So(err, ShouldBeNil)
 					So(err, ShouldBeNil)
 				})
 				})
 
 
-				Convey("Collaborator should be included in account info projection", func() {
-					query := m.GetAccountInfoQuery{Id: ac1.Id}
-					err = GetAccountInfo(&query)
-
-					So(err, ShouldBeNil)
-					So(query.Result.Collaborators[0].CollaboratorId, ShouldEqual, ac2.Id)
-					So(query.Result.Collaborators[0].Role, ShouldEqual, m.ROLE_READ_WRITE)
-					So(query.Result.Collaborators[0].Email, ShouldEqual, "ac2@test.com")
-				})
-
 				Convey("Can get other accounts", func() {
 				Convey("Can get other accounts", func() {
 					query := m.GetOtherAccountsQuery{AccountId: ac2.Id}
 					query := m.GetOtherAccountsQuery{AccountId: ac2.Id}
 					err := GetOtherAccounts(&query)
 					err := GetOtherAccounts(&query)

+ 2 - 2
pkg/services/sqlstore/tokens.go

@@ -1,10 +1,11 @@
 package sqlstore
 package sqlstore
 
 
 import (
 import (
+	"time"
+
 	"github.com/go-xorm/xorm"
 	"github.com/go-xorm/xorm"
 	"github.com/torkelo/grafana-pro/pkg/bus"
 	"github.com/torkelo/grafana-pro/pkg/bus"
 	m "github.com/torkelo/grafana-pro/pkg/models"
 	m "github.com/torkelo/grafana-pro/pkg/models"
-	"time"
 )
 )
 
 
 func init() {
 func init() {
@@ -30,7 +31,6 @@ func DeleteToken(cmd *m.DeleteTokenCommand) error {
 }
 }
 
 
 func AddToken(cmd *m.AddTokenCommand) error {
 func AddToken(cmd *m.AddTokenCommand) error {
-
 	return inTransaction(func(sess *xorm.Session) error {
 	return inTransaction(func(sess *xorm.Session) error {
 		t := m.Token{
 		t := m.Token{
 			AccountId: cmd.AccountId,
 			AccountId: cmd.AccountId,