@@ -90,7 +90,6 @@ func newGroupTest(t *testing.T) *GroupTest {
90
90
t .Fatal (err )
91
91
}
92
92
93
-
94
93
time .Sleep (time .Second )
95
94
96
95
checker := & defaultAuthChecker {}
@@ -612,3 +611,101 @@ func Test_server_RefreshGroupToken(t *testing.T) {
612
611
assert .True (t , batchResp .Code .Value == v1 .NotAllowedAccess , batchResp .Info .GetValue ())
613
612
})
614
613
}
614
+
615
+ func Test_AuthServer_NormalOperateUserGroup (t * testing.T ) {
616
+ suit := & AuthTestSuit {}
617
+ if err := suit .initialize (); err != nil {
618
+ t .Fatal (err )
619
+ }
620
+ defer suit .Destroy ()
621
+
622
+ users := createApiMockUser (10 , "test" )
623
+ for i := range users {
624
+ users [i ].Id = utils .NewStringValue (utils .NewUUID ())
625
+ }
626
+
627
+ groups := createMockApiUserGroup ([]* api.User {users [0 ]})
628
+
629
+ t .Run ("正常创建用户组" , func (t * testing.T ) {
630
+ bresp := suit .server .CreateUsers (suit .defaultCtx , users )
631
+ if ! respSuccess (bresp ) {
632
+ t .Fatal (bresp .GetInfo ().GetValue ())
633
+ }
634
+
635
+ time .Sleep (suit .updateCacheInterval )
636
+
637
+ resp := suit .server .CreateGroup (suit .defaultCtx , groups [0 ])
638
+
639
+ if ! respSuccess (resp ) {
640
+ t .Fatal (resp .GetInfo ().GetValue ())
641
+ }
642
+
643
+ groups [0 ].Id = utils .NewStringValue (resp .GetUserGroup ().Id .Value )
644
+ })
645
+
646
+ t .Run ("正常更新用户组" , func (t * testing.T ) {
647
+
648
+ time .Sleep (time .Second )
649
+
650
+ req := []* api.ModifyUserGroup {
651
+ & v1.ModifyUserGroup {
652
+ Id : utils .NewStringValue (groups [0 ].GetId ().GetValue ()),
653
+ Name : utils .NewStringValue (groups [0 ].GetName ().GetValue ()),
654
+ Comment : & wrapperspb.StringValue {
655
+ Value : "update user group" ,
656
+ },
657
+ AddRelations : & v1.UserGroupRelation {
658
+ Users : users [3 :],
659
+ },
660
+ },
661
+ }
662
+
663
+ resp := suit .server .UpdateGroups (suit .defaultCtx , req )
664
+ if ! respSuccess (resp ) {
665
+ t .Fatal (resp .GetInfo ().GetValue ())
666
+ }
667
+
668
+ time .Sleep (suit .updateCacheInterval )
669
+
670
+ qresp := suit .server .GetGroup (suit .defaultCtx , groups [0 ])
671
+
672
+ if ! respSuccess (resp ) {
673
+ t .Fatal (resp .GetInfo ().GetValue ())
674
+ }
675
+
676
+ assert .Equal (t , req [0 ].GetComment ().GetValue (), qresp .GetUserGroup ().GetComment ().GetValue ())
677
+ assert .Equal (t , len (users [3 :])+ 1 , len (qresp .GetUserGroup ().GetRelation ().GetUsers ()))
678
+ })
679
+
680
+ t .Run ("正常更新用户组Token" , func (t * testing.T ) {
681
+ resp := suit .server .ResetGroupToken (suit .defaultCtx , groups [0 ])
682
+
683
+ if ! respSuccess (resp ) {
684
+ t .Fatal (resp .GetInfo ().GetValue ())
685
+ }
686
+
687
+ time .Sleep (suit .updateCacheInterval )
688
+
689
+ qresp := suit .server .GetGroupToken (suit .defaultCtx , groups [0 ])
690
+ if ! respSuccess (qresp ) {
691
+ t .Fatal (resp .GetInfo ().GetValue ())
692
+ }
693
+ assert .Equal (t , resp .GetUserGroup ().GetAuthToken ().GetValue (), qresp .GetUserGroup ().GetAuthToken ().GetValue ())
694
+ })
695
+
696
+ t .Run ("正常删除用户组" , func (t * testing.T ) {
697
+ resp := suit .server .DeleteGroups (suit .defaultCtx , groups )
698
+
699
+ if ! respSuccess (resp ) {
700
+ t .Fatal (resp .GetInfo ().GetValue ())
701
+ }
702
+
703
+ qresp := suit .server .GetGroup (suit .defaultCtx , groups [0 ])
704
+
705
+ if respSuccess (qresp ) {
706
+ t .Fatal (resp .GetInfo ().GetValue ())
707
+ }
708
+
709
+ assert .Equal (t , v1 .NotFoundUserGroup , qresp .GetCode ().GetValue ())
710
+ })
711
+ }
0 commit comments