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)