Prechádzať zdrojové kódy

Make recently used auth_module test more robust by adding another 'log in'

Sean Lafferty 6 rokov pred
rodič
commit
f3c52710df
1 zmenil súbory, kde vykonal 18 pridanie a 1 odobranie
  1. 18 1
      pkg/services/sqlstore/user_auth_test.go

+ 18 - 1
pkg/services/sqlstore/user_auth_test.go

@@ -175,7 +175,7 @@ func TestUserAuth(t *testing.T) {
 			login := "loginuser0"
 
 			// Calling GetUserByAuthInfoQuery on an existing user will populate an entry in the user_auth table
-			query := &m.GetUserByAuthInfoQuery{Login: login, AuthModule: "test", AuthId: "test"}
+			query := &m.GetUserByAuthInfoQuery{Login: login, AuthModule: "test1", AuthId: "test1"}
 			err = GetUserByAuthInfo(query)
 
 			So(err, ShouldBeNil)
@@ -200,6 +200,23 @@ func TestUserAuth(t *testing.T) {
 			So(err, ShouldBeNil)
 			So(getAuthQuery.Result.AuthModule, ShouldEqual, "test2")
 
+			// "log in" again with the first auth module
+			// resolution of `Created` column is 1sec, so we need a delay
+			time.Sleep(time.Second)
+			updateAuthCmd := &m.UpdateAuthInfoCommand{UserId: query.Result.Id, AuthModule: "test1", AuthId: "test1"}
+			err = UpdateAuthInfo(updateAuthCmd)
+
+			So(err, ShouldBeNil)
+
+			// Get the latest entry by not supply an authmodule or authid
+			getAuthQuery = &m.GetAuthInfoQuery{
+				UserId: query.Result.Id,
+			}
+
+			err = GetAuthInfo(getAuthQuery)
+
+			So(err, ShouldBeNil)
+			So(getAuthQuery.Result.AuthModule, ShouldEqual, "test1")
 		})
 	})
 }