@@ -59,12 +59,10 @@ async def _assert_connection_org_manager(self, client, view_perm=True, conn=True
5959 )
6060 t = await database_sync_to_async (self ._create_topology )(organization = org )
6161 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'
6764 )
65+ await test_user .user_permissions .aadd (topology_view_permission )
6866 await database_sync_to_async (client .force_login )(test_user )
6967 communicator = await self ._get_communicator (client , t .pk )
7068 connected , _ = await communicator .connect ()
@@ -134,7 +132,7 @@ async def test_node_topology_update(self, admin_user, admin_client):
134132 response = await communicator .receive_json_from ()
135133 assert response ['topology' ] is not None
136134 assert response ['topology' ] == expected_response
137- await database_sync_to_async ( node .delete ) ()
135+ await node .adelete ()
138136 expected_response = await database_sync_to_async (topo .json )()
139137 response = await communicator .receive_json_from ()
140138 assert response ['topology' ] is not None
@@ -164,12 +162,12 @@ async def test_link_topology_update(self, admin_user, admin_client):
164162 assert response ['topology' ] == expected_response
165163 link .status = 'down'
166164 await database_sync_to_async (link .full_clean )()
167- await database_sync_to_async ( link .save ) ()
165+ await link .asave ()
168166 expected_response = await database_sync_to_async (topo .json )()
169167 response = await communicator .receive_json_from ()
170168 assert response ['topology' ] is not None
171169 assert response ['topology' ] == expected_response
172- await database_sync_to_async ( link .delete ) ()
170+ await link .adelete ()
173171 expected_response = await database_sync_to_async (topo .json )()
174172 response = await communicator .receive_json_from ()
175173 assert response ['topology' ] is not None
@@ -184,7 +182,7 @@ async def test_topology_properties_update(self, admin_user, admin_client):
184182 assert connected is True
185183 topo .name = 'test'
186184 await database_sync_to_async (topo .full_clean )()
187- await database_sync_to_async ( topo .save ) ()
185+ await topo .asave ()
188186 expected_response = await database_sync_to_async (topo .json )()
189187 response = await communicator .receive_json_from ()
190188 assert response ['topology' ] is not None
0 commit comments