(GH-1612) Updated classes to override methods

- To allow for correct generation of hash codes - This will be used to validate changes to chocolatey.config file
chocolatey · Oct 9, 2018 · 4c85719 · 4c85719
1 parent 134596e
commit 4c85719
Show file tree

Hide file tree

Showing 5 changed files with 137 additions and 0 deletions.
diff --git a/src/chocolatey/infrastructure.app/configuration/ConfigFileApiKeySetting.cs b/src/chocolatey/infrastructure.app/configuration/ConfigFileApiKeySetting.cs
@@ -31,5 +31,26 @@ public sealed class ConfigFileApiKeySetting
 
         [XmlAttribute(AttributeName = "key")]
         public string Key { get; set; }
+
+        public override bool Equals(object obj)
+        {
+            // Check for null values and compare run-time types.
+            if (obj == null || GetType() != obj.GetType())
+            {
+                return false;
+            }
+
+            var item = (ConfigFileApiKeySetting) obj;
+
+            return (Source == item.Source)
+                   && (Key == item.Source);
+        }
+
+        public override int GetHashCode()
+        {
+            return HashCode
+                .Of(Source)
+                .And(Key);
+        }
     }
 }
diff --git a/src/chocolatey/infrastructure.app/configuration/ConfigFileConfigSetting.cs b/src/chocolatey/infrastructure.app/configuration/ConfigFileConfigSetting.cs
@@ -34,5 +34,28 @@ public sealed class ConfigFileConfigSetting
 
         [XmlAttribute(AttributeName = "description")]
         public string Description { get; set; }
+
+        public override bool Equals(object obj)
+        {
+            // Check for null values and compare run-time types.
+            if (obj == null || GetType() != obj.GetType())
+            {
+                return false;
+            }
+
+            var item = (ConfigFileConfigSetting) obj;
+
+            return (Key == item.Key)
+                   && (Value == item.Value)
+                   && (Description == item.Description);
+        }
+
+        public override int GetHashCode()
+        {
+            return HashCode
+                .Of(Key)
+                .And(Value)
+                .And(Description);
+        }
     }
 }
diff --git a/src/chocolatey/infrastructure.app/configuration/ConfigFileFeatureSetting.cs b/src/chocolatey/infrastructure.app/configuration/ConfigFileFeatureSetting.cs
@@ -37,5 +37,30 @@ public sealed class ConfigFileFeatureSetting
 
         [XmlAttribute(AttributeName = "description")]
         public string Description { get; set; }
+
+        public override bool Equals(object obj)
+        {
+            // Check for null values and compare run-time types.
+            if (obj == null || GetType() != obj.GetType())
+            {
+                return false;
+            }
+
+            var item = (ConfigFileFeatureSetting) obj;
+
+            return (Name == item.Name)
+                   && (Enabled == item.Enabled)
+                   && (SetExplicitly == item.SetExplicitly)
+                   && (Description == item.Description);
+        }
+
+        public override int GetHashCode()
+        {
+            return HashCode
+                .Of(Name)
+                .And(Enabled)
+                .And(SetExplicitly)
+                .And(Description);
+        }
     }
 }
diff --git a/src/chocolatey/infrastructure.app/configuration/ConfigFileSettings.cs b/src/chocolatey/infrastructure.app/configuration/ConfigFileSettings.cs
@@ -46,5 +46,34 @@ public class ConfigFileSettings
 
         [XmlArray("apiKeys")]
         public HashSet<ConfigFileApiKeySetting> ApiKeys { get; set; }
+
+        public override bool Equals(object obj)
+        {
+            // Check for null values and compare run-time types.
+            if (obj == null || GetType() != obj.GetType())
+            {
+                return false;
+            }
+
+            var item = (ConfigFileSettings) obj;
+
+            return (CacheLocation == item.CacheLocation)
+                && (CommandExecutionTimeoutSeconds == item.CommandExecutionTimeoutSeconds)
+                && (ConfigSettings == item.ConfigSettings)
+                && (Sources == item.Sources)
+                && (Features == item.Features)
+                && (ApiKeys == item.ApiKeys);
+        }
+
+        public override int GetHashCode()
+        {
+            return HashCode
+                .Of(CacheLocation)
+                .And(CommandExecutionTimeoutSeconds)
+                .AndEach(ConfigSettings)
+                .AndEach(Sources)
+                .AndEach(Features)
+                .AndEach(ApiKeys);
+        }
     }
 }
diff --git a/src/chocolatey/infrastructure.app/configuration/ConfigFileSourceSetting.cs b/src/chocolatey/infrastructure.app/configuration/ConfigFileSourceSetting.cs
@@ -58,5 +58,44 @@ public sealed class ConfigFileSourceSetting
 
         [XmlAttribute(AttributeName = "certificatePassword")]
         public string CertificatePassword { get; set; }
+
+        public override bool Equals(object obj)
+        {
+            // Check for null values and compare run-time types.
+            if (obj == null || GetType() != obj.GetType())
+            {
+                return false;
+            }
+
+            var item = (ConfigFileSourceSetting) obj;
+
+            return (Id == item.Id)
+                && (Value == item.Value)
+                && (Disabled == item.Disabled)
+                && (BypassProxy == item.BypassProxy)
+                && (AllowSelfService == item.AllowSelfService)
+                && (VisibleToAdminsOnly == item.VisibleToAdminsOnly)
+                && (UserName == item.UserName)
+                && (Password == item.Password)
+                && (Priority == item.Priority)
+                && (Certificate == item.Certificate)
+                && (CertificatePassword == item.CertificatePassword);
+        }
+
+        public override int GetHashCode()
+        {
+            return HashCode
+                .Of(Id)
+                .And(Value)
+                .And(Disabled)
+                .And(BypassProxy)
+                .And(AllowSelfService)
+                .And(VisibleToAdminsOnly)
+                .And(UserName)
+                .And(Password)
+                .And(Priority)
+                .And(Certificate)
+                .And(CertificatePassword);
+        }
     }
 }