[feat][python] Add basic authentication (#17482)

Author: nodece

pulsar/__init__.py

@@ -344,6 +344,23 @@ def __init__(self, auth_params_string):
         _check_type(str, auth_params_string, 'auth_params_string')
         self.auth = _pulsar.AuthenticationOauth2(auth_params_string)
 
+class AuthenticationBasic(Authentication):
+    """
+    Basic Authentication implementation
+    """
+    def __init__(self, username, password):
+        """
+        Create the Basic authentication provider instance.
+
+        **Args**
+
+        * `username`: Used to authentication as username
+        * `password`: Used to authentication as password
+        """
+        _check_type(str, username, 'username')
+        _check_type(str, password, 'password')
+        self.auth = _pulsar.AuthenticationBasic(username, password)
+
 class Client:
     """
     The Pulsar client. A single client instance can be used to create producers

pulsar_test.py

@@ -31,6 +31,7 @@
     CompressionType,
     ConsumerType,
     PartitionsRoutingMode,
+    AuthenticationBasic,
     AuthenticationTLS,
     Authentication,
     AuthenticationToken,
@@ -1282,6 +1283,28 @@ def _check_type_error(self, fun):
         with self.assertRaises(TypeError):
             fun()
 
+    def test_basic_auth(self):
+        username = "admin"
+        password = "123456"
+        client = Client(self.adminUrl, authentication=AuthenticationBasic(username, password))
+
+        topic = "persistent://private/auth/my-python-topic-basic-auth"
+        consumer = client.subscribe(topic, "my-sub", consumer_type=ConsumerType.Shared)
+        producer = client.create_producer(topic)
+        producer.send(b"hello")
+
+        msg = consumer.receive(TM)
+        self.assertTrue(msg)
+        self.assertEqual(msg.data(), b"hello")
+        client.close()
+
+    def test_invalid_basic_auth(self):
+        username = "invalid"
+        password = "123456"
+        client = Client(self.adminUrl, authentication=AuthenticationBasic(username, password))
+        topic = "persistent://private/auth/my-python-topic-invalid-basic-auth"
+        with self.assertRaises(pulsar.ConnectError):
+            client.subscribe(topic, "my-sub", consumer_type=ConsumerType.Shared)
 
 if __name__ == "__main__":
     main()

src/authentication.cc

@@ -90,6 +90,13 @@ struct AuthenticationOauth2Wrapper : public AuthenticationWrapper {
     }
 };
 
+struct AuthenticationBasicWrapper : public AuthenticationWrapper {
+    AuthenticationBasicWrapper(const std::string& username, const std::string& password)
+        : AuthenticationWrapper() {
+        this->auth = AuthBasic::create(username, password);
+    }
+};
+
 void export_authentication() {
     using namespace boost::python;
 
@@ -106,4 +113,7 @@ void export_authentication() {
 
     class_<AuthenticationOauth2Wrapper, bases<AuthenticationWrapper> >("AuthenticationOauth2",
                                                                        init<const std::string&>());
+
+    class_<AuthenticationBasicWrapper, bases<AuthenticationWrapper> >(
+        "AuthenticationBasic", init<const std::string&, const std::string&>());
 }