def test_04_get_bootstrap_servers_tls_and_plaintext(self, mock_client):
mock_client.return_value = self._kafka
from lambda_function import _get_bootstrap_brokers
bootstrap_servers = _get_bootstrap_brokers('my-cluster-arn')
self.assertEqual(self.BOOTSTRAP_SERVER_TLS, bootstrap_servers)
def test_05_get_bootstrap_servers_iam(self, mock_client):
mock_client.return_value = self._kafka
from lambda_function import _get_bootstrap_brokers
bootstrap_servers = _get_bootstrap_brokers('my-cluster-arn')
self.assertEqual(self.BOOTSTRAP_SERVER_IAM, bootstrap_servers)
def test_06_get_bootstrap_servers_iam_and_unauthenticated(self, mock_client):
mock_client.return_value = self._kafka
from lambda_function import _get_bootstrap_brokers
bootstrap_servers = _get_bootstrap_brokers('my-cluster-arn')
self.assertEqual(self.BOOTSTRAP_SERVER_IAM, bootstrap_servers)
self.assertNotEqual(self.BOOTSTRAP_SERVER_TLS, bootstrap_servers)
self.assertNotEqual(self.BOOTSTRAP_SERVER_PLAINTEXT, bootstrap_servers)