mage-ai 0.9.45__py3-none-any.whl → 0.9.46__py3-none-any.whl

mage_ai/tests/streaming/sources/test_source_factory.py

@@ -1,13 +1,27 @@
+from unittest.mock import patch
+
+from mage_ai.streaming.sources.activemq import ActiveMQSource
 from mage_ai.streaming.sources.azure_event_hub import AzureEventHubSource
 from mage_ai.streaming.sources.google_cloud_pubsub import GoogleCloudPubSubSource
 from mage_ai.streaming.sources.kafka import KafkaSource
+from mage_ai.streaming.sources.nats_js import NATSSource
 from mage_ai.streaming.sources.rabbitmq import RabbitMQSource
 from mage_ai.streaming.sources.source_factory import SourceFactory
 from mage_ai.tests.base_test import TestCase
-from unittest.mock import patch
 
 
 class SourceFactoryTests(TestCase):
+    def test_get_source_nats(self):
+        with patch.object(NATSSource,
+                          '__init__',
+                          return_value=None) as mock_init:
+            config = dict(
+                connector_type='nats',
+            )
+            source = SourceFactory.get_source(config)
+            self.assertIsInstance(source, NATSSource)
+            mock_init.assert_called_once_with(config)
+
     def test_get_source_kafka(self):
         with patch.object(KafkaSource,
                           '__init__',
@@ -52,6 +66,17 @@ class SourceFactoryTests(TestCase):
             self.assertIsInstance(source, RabbitMQSource)
             mock_init.assert_called_once_with(config)
 
+    def test_get_source_activemq(self):
+        with patch.object(ActiveMQSource,
+                          '__init__',
+                          return_value=None) as mock_init:
+            config = dict(
+                connector_type='activemq',
+            )
+            source = SourceFactory.get_source(config)
+            self.assertIsInstance(source, ActiveMQSource)
+            mock_init.assert_called_once_with(config)
+
     def test_get_source_other(self):
         s = "Consuming data from random is not supported " \
             "in streaming pipelines yet."

{mage_ai-0.9.45.dist-info → mage_ai-0.9.46.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: mage-ai
-Version: 0.9.45
+Version: 0.9.46
 Summary: Mage is a tool for building and deploying data pipelines.
 Home-page: https://github.com/mage-ai/mage-ai
 Author: Mage
@@ -41,6 +41,7 @@ Requires-Dist: pandas >=1.3.0
 Requires-Dist: polars <=0.19.2
 Requires-Dist: protobuf
 Requires-Dist: pyarrow ==10.0.1
+Requires-Dist: pyarrow-hotfix ==0.5
 Requires-Dist: python-dateutil ==2.8.2
 Requires-Dist: pytz ==2022.2.1
 Requires-Dist: pyyaml ~=6.0
@@ -102,6 +103,8 @@ Requires-Dist: kafka-python ==2.0.2 ; extra == 'all'
 Requires-Dist: kubernetes ==25.3.0 ; extra == 'all'
 Requires-Dist: langchain >=0.0.222 ; extra == 'all'
 Requires-Dist: ldap3 ==2.9.1 ; extra == 'all'
+Requires-Dist: nats-py ==2.6.0 ; extra == 'all'
+Requires-Dist: nkeys ; extra == 'all'
 Requires-Dist: openai <1.0.0,>=0.27.8 ; extra == 'all'
 Requires-Dist: opensearch-py ==2.0.0 ; extra == 'all'
 Requires-Dist: opentelemetry-exporter-prometheus >=1.12.0rc1 ; extra == 'all'
@@ -119,6 +122,7 @@ Requires-Dist: redshift-connector ==2.0.909 ; extra == 'all'
 Requires-Dist: requests-aws4auth ==1.1.2 ; extra == 'all'
 Requires-Dist: snowflake-connector-python ==3.2.1 ; extra == 'all'
 Requires-Dist: sshtunnel ==0.4.0 ; extra == 'all'
+Requires-Dist: stomp.py ==8.1.0 ; extra == 'all'
 Requires-Dist: thefuzz[speedup] ==0.19.0 ; extra == 'all'
 Provides-Extra: azure
 Requires-Dist: azure-eventhub ==5.11.2 ; extra == 'azure'
@@ -173,9 +177,12 @@ Requires-Dist: confluent-avro ; extra == 'streaming'
 Requires-Dist: influxdb-client ==1.36.1 ; extra == 'streaming'
 Requires-Dist: kafka-python ==2.0.2 ; extra == 'streaming'
 Requires-Dist: opensearch-py ==2.0.0 ; extra == 'streaming'
+Requires-Dist: nats-py ==2.6.0 ; extra == 'streaming'
+Requires-Dist: nkeys ; extra == 'streaming'
 Requires-Dist: pika ==1.3.1 ; extra == 'streaming'
 Requires-Dist: pymongo ==4.3.3 ; extra == 'streaming'
 Requires-Dist: requests-aws4auth ==1.1.2 ; extra == 'streaming'
+Requires-Dist: stomp.py ==8.1.0 ; extra == 'streaming'
 Requires-Dist: elasticsearch ==8.9.0 ; extra == 'streaming'
 
 <h1 align="center">