@@ -59,12 +59,10 @@ async def _assert_connection_org_manager(self, client, view_perm=True, conn=True
59
59
)
60
60
t = await database_sync_to_async (self ._create_topology )(organization = org )
61
61
if view_perm :
62
- topology_view_permission = await database_sync_to_async (
63
- Permission .objects .get
64
- )(codename = 'view_topology' )
65
- await database_sync_to_async (test_user .user_permissions .add )(
66
- topology_view_permission
62
+ topology_view_permission = await Permission .objects .aget (
63
+ codename = 'view_topology'
67
64
)
65
+ await test_user .user_permissions .aadd (topology_view_permission )
68
66
await database_sync_to_async (client .force_login )(test_user )
69
67
communicator = await self ._get_communicator (client , t .pk )
70
68
connected , _ = await communicator .connect ()
@@ -134,7 +132,7 @@ async def test_node_topology_update(self, admin_user, admin_client):
134
132
response = await communicator .receive_json_from ()
135
133
assert response ['topology' ] is not None
136
134
assert response ['topology' ] == expected_response
137
- await database_sync_to_async ( node .delete ) ()
135
+ await node .adelete ()
138
136
expected_response = await database_sync_to_async (topo .json )()
139
137
response = await communicator .receive_json_from ()
140
138
assert response ['topology' ] is not None
@@ -164,12 +162,12 @@ async def test_link_topology_update(self, admin_user, admin_client):
164
162
assert response ['topology' ] == expected_response
165
163
link .status = 'down'
166
164
await database_sync_to_async (link .full_clean )()
167
- await database_sync_to_async ( link .save ) ()
165
+ await link .asave ()
168
166
expected_response = await database_sync_to_async (topo .json )()
169
167
response = await communicator .receive_json_from ()
170
168
assert response ['topology' ] is not None
171
169
assert response ['topology' ] == expected_response
172
- await database_sync_to_async ( link .delete ) ()
170
+ await link .adelete ()
173
171
expected_response = await database_sync_to_async (topo .json )()
174
172
response = await communicator .receive_json_from ()
175
173
assert response ['topology' ] is not None
@@ -184,7 +182,7 @@ async def test_topology_properties_update(self, admin_user, admin_client):
184
182
assert connected is True
185
183
topo .name = 'test'
186
184
await database_sync_to_async (topo .full_clean )()
187
- await database_sync_to_async ( topo .save ) ()
185
+ await topo .asave ()
188
186
expected_response = await database_sync_to_async (topo .json )()
189
187
response = await communicator .receive_json_from ()
190
188
assert response ['topology' ] is not None
0 commit comments