@@ -98,12 +98,20 @@ def test_export_users_command(self):
9898 temp_file = NamedTemporaryFile (delete = False )
9999 org_user = self ._create_org_user ()
100100 user = org_user .user
101- reg_user = RegisteredUser .objects .create (
101+ org2 = self ._create_org (name = "Test Organization 2" )
102+ self ._create_org_user (organization = org2 , user = user )
103+ org1_reg_user = RegisteredUser .objects .create (
102104 user = user ,
103105 organization = org_user .organization ,
104106 method = "mobile_phone" ,
105107 is_verified = False ,
106108 )
109+ org2_reg_user = RegisteredUser .objects .create (
110+ user = user ,
111+ organization = org2 ,
112+ method = "mobile_phone" ,
113+ is_verified = True ,
114+ )
107115 with self .assertNumQueries (3 ):
108116 call_command ("export_users" , filename = temp_file .name )
109117
@@ -112,10 +120,18 @@ def test_export_users_command(self):
112120 csv_data = list (csv_reader )
113121
114122 self .assertEqual (len (csv_data ), 2 )
115- self .assertIn ("registered_users" , csv_data [0 ])
123+ self .assertIn (
124+ "registered_users (organization_id, method, is_verified)" , csv_data [0 ]
125+ )
116126 self .assertEqual (
117127 csv_data [1 ][- 1 ],
118- f"(({ reg_user .organization_id } ,{ reg_user .method } ,{ reg_user .is_verified } ))" ,
128+ (
129+ f"({ org1_reg_user .organization_id } ,{ org1_reg_user .method } ,"
130+ f"{ org1_reg_user .is_verified } )"
131+ "\n "
132+ f"({ org2_reg_user .organization_id } ,{ org2_reg_user .method } ,"
133+ f"{ org2_reg_user .is_verified } )"
134+ ),
119135 )
120136
121137 def test_radiususergroup_inline (self ):
0 commit comments