chore: import upstream snapshot with attribution

2026-07-13 13:02:50 +08:00
commit 0fc60fdcb1
5008 changed files with 910633 additions and 0 deletions
@@ -0,0 +1,93 @@
+/*
+ * QUANTCONNECT.COM - Democratizing Finance, Empowering Individuals.
+ * Lean Algorithmic Trading Engine v2.0. Copyright 2014 QuantConnect Corporation.
+ * 
+ * Licensed under the Apache License, Version 2.0 (the "License"); 
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at http://www.apache.org/licenses/LICENSE-2.0
+ * 
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+*/
+
+using NUnit.Framework;
+using Python.Runtime;
+using QuantConnect.Data;
+using System;
+using System.Collections.Generic;
+
+namespace QuantConnect.Tests.Common.Data
+{
+    [TestFixture]
+    public class SubscriptionDataSourceTests
+    {
+        [Test]
+        public void ComparesEqualWithIdenticalSourceAndTransportMedium()
+        {
+            var one = new SubscriptionDataSource("source", SubscriptionTransportMedium.LocalFile);
+            var two = new SubscriptionDataSource("source", SubscriptionTransportMedium.LocalFile);
+            Assert.IsTrue(one == two);
+            Assert.IsTrue(one.Equals(two));
+        }
+
+        [Test]
+        public void ComparesNotEqualWithDifferentSource()
+        {
+            var one = new SubscriptionDataSource("source1", SubscriptionTransportMedium.LocalFile);
+            var two = new SubscriptionDataSource("source2", SubscriptionTransportMedium.LocalFile);
+            Assert.IsTrue(one != two);
+            Assert.IsTrue(!one.Equals(two));
+        }
+
+        [Test]
+        public void ComparesNotEqualWithDifferentTransportMedium()
+        {
+            var one = new SubscriptionDataSource("source", SubscriptionTransportMedium.LocalFile);
+            var two = new SubscriptionDataSource("source", SubscriptionTransportMedium.RemoteFile);
+            Assert.IsTrue(one != two);
+            Assert.IsTrue(!one.Equals(two));
+        }
+
+        [Test]
+        public void SupportsPythonDictionaryHeaders()
+        {
+            using (Py.GIL())
+            {
+                using var headers = new PyDict();
+                headers.SetItem("Authorization".ToPython(), "Basic test-token".ToPython());
+                headers.SetItem("X-Api-Key".ToPython(), "abc123".ToPython());
+
+                var dataSource = new SubscriptionDataSource("https://example.com", SubscriptionTransportMedium.RemoteFile, FileFormat.Csv, headers);
+                CollectionAssert.AreEquivalent(new[]
+                {
+                    new KeyValuePair<string, string>("Authorization", "Basic test-token"),
+                    new KeyValuePair<string, string>("X-Api-Key", "abc123")
+                }, dataSource.Headers);
+            }
+        }
+
+        [Test]
+        public void SupportsNullPythonDictionaryHeaders()
+        {
+            var dataSource = new SubscriptionDataSource("https://example.com", SubscriptionTransportMedium.RemoteFile, FileFormat.Csv, (PyObject)null);
+            Assert.IsEmpty(dataSource.Headers);
+        }
+
+        [Test]
+        public void ThrowsForInvalidPythonHeadersType()
+        {
+            using (Py.GIL())
+            {
+                using var invalidHeaders = "invalid-headers".ToPython();
+
+                var exception = Assert.Throws<ArgumentException>(() =>
+                    new SubscriptionDataSource("https://example.com", SubscriptionTransportMedium.RemoteFile, FileFormat.Csv, invalidHeaders));
+
+                StringAssert.Contains("ConvertToDictionary cannot be used", exception.Message);
+            }
+        }
+    }
+}