Skip to content

Commit 4558a74

Browse files
committed
[fix] Fixed tests
1 parent 29857b0 commit 4558a74

1 file changed

Lines changed: 19 additions & 3 deletions

File tree

openwisp_radius/tests/test_users_integration.py

Lines changed: 19 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -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

Comments
 (0)