Adjust tests

tests/providers/hashicorp/_internal_client/test_vault_client.py

@@ -128,6 +128,7 @@ def test_aws_iam(self, mock_hvac):
  access_key="user",
  secret_key="pass",
  role="role",
+ mount_point=None,
  )
  client.is_authenticated.assert_called_with()
  assert 2 == vault_client.kv_engine_version
@@ -148,14 +149,7 @@ def test_aws_iam_different_auth_mount_point(self, mock_hvac):
  client = vault_client.client
  mock_hvac.Client.assert_called_with(url="http://localhost:8180", session=None)
  client.auth.aws.iam_login.assert_called_with(
- access_key="user",
- secret_key="pass",
- session_token=None,
- header_value=None,
- role="role",
- use_token=True,
- region="us-east-1",
- mount_point="other",
+ access_key="user", secret_key="pass", role="role", mount_point="other"
  )
  client.is_authenticated.assert_called_with()
  assert 2 == vault_client.kv_engine_version

tests/providers/hashicorp/hooks/test_vault.py

@@ -315,6 +315,7 @@ def test_aws_iam_init_params(self, mock_hvac, mock_get_connection):
  access_key="user",
  secret_key="pass",
  role="role",
+ mount_point=None,
  )
  test_client.is_authenticated.assert_called_with()
  assert 2 == test_hook.vault_client.kv_engine_version
@@ -343,6 +344,7 @@ def test_aws_iam_dejson(self, mock_hvac, mock_get_connection):
  access_key="user",
  secret_key="pass",
  role="role",
+ mount_point=None,
  )
 
  @mock.patch("airflow.providers.hashicorp._internal_client.vault_client.hvac")
@@ -358,6 +360,7 @@ def test_aws_uri(self, mock_hvac):
  access_key="login",
  secret_key="pass",
  role="role",
+ mount_point=None,
  )
  test_client.is_authenticated.assert_called_with()
  assert 2 == test_hook.vault_client.kv_engine_version