Merge pull request #1364 from amzn/rdma-tests

Fix failing tests due to query_qp attribute mask
linux-rdma · Jul 11, 2023 · d56f3c4 · d56f3c4
2 parents 3dab4fc + b725a8a
commit d56f3c4
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/tests/base.py b/tests/base.py
@@ -349,8 +349,6 @@ def create_players(self, resource, sync_attrs=True, **resource_arg):
  self.pre_run()
  if sync_attrs:
  self.sync_remote_attr()
- self.server_qp_attr, _ = self.server.qp.query(0x1ffffff)
- self.client_qp_attr, _ = self.client.qp.query(0x1ffffff)
  self.traffic_args = {'client': self.client, 'server': self.server,
  'iters': self.iters, 'gid_idx': self.gid_index,
  'port': self.ib_port}

diff --git a/tests/test_mr.py b/tests/test_mr.py
@@ -114,6 +114,8 @@ def test_mr_rereg_pd(self):
  succeeds.
  """
  self.create_players(MRRes)
+ self.server_qp_attr, _ = self.server.qp.query(0x1ffffff)
+ self.client_qp_attr, _ = self.client.qp.query(0x1ffffff)
  u.traffic(**self.traffic_args)
  server_new_pd = PD(self.server.ctx)
  self.server.rereg_mr(flags=e.IBV_REREG_MR_CHANGE_PD, pd=server_new_pd)
@@ -128,6 +130,8 @@ def test_mr_rereg_pd(self):
 
  def test_mr_rereg_addr(self):
  self.create_players(MRRes)
+ self.server_qp_attr, _ = self.server.qp.query(0x1ffffff)
+ self.client_qp_attr, _ = self.client.qp.query(0x1ffffff)
  s_recv_wr = u.get_recv_wr(self.server)
  self.server.qp.post_recv(s_recv_wr)
  server_addr = posix_memalign(self.server.msg_size)