Fix YAML serialization for IntOrString values. (#310)

src/KubernetesClient/IntstrIntOrString.cs

@@ -1,8 +1,43 @@
 using System;
 using Newtonsoft.Json;
+using YamlDotNet.Core;
+using YamlDotNet.Core.Events;
+using YamlDotNet.Core.Tokens;
+using YamlDotNet.Serialization;
 
 namespace k8s.Models
 {
+    public class IntOrStringYamlConverter: IYamlTypeConverter
+    {
+        public bool Accepts(Type type)
+        {
+            return type == typeof(IntstrIntOrString);
+        }
+
+        public object ReadYaml(IParser parser, Type type)
+        {
+            if (parser.Current is YamlDotNet.Core.Events.Scalar scalar)
+            {
+                try {
+                    if (string.IsNullOrEmpty(scalar.Value))
+                    {
+                        return null;
+                    }
+                    return new IntstrIntOrString(scalar.Value);
+                } finally {
+                    parser.MoveNext();
+                }
+            }
+            throw new InvalidOperationException(parser.Current?.ToString());
+        }
+
+        public void WriteYaml(IEmitter emitter, object value, Type type)
+        {
+            var obj = (IntstrIntOrString) value;
+            emitter.Emit(new YamlDotNet.Core.Events.Scalar(obj.Value));
+        }
+    }
+
     internal class IntOrStringConverter : JsonConverter
     {
         public override void WriteJson(JsonWriter writer, object value, JsonSerializer serializer)

src/KubernetesClient/Yaml.cs

@@ -9,6 +9,7 @@ using YamlDotNet.Core;
 using YamlDotNet.Core.Events;
 using YamlDotNet.Serialization;
 using YamlDotNet.Serialization.NamingConventions;
+using k8s.Models;
 
 namespace k8s
 {
@@ -34,6 +35,7 @@ namespace k8s
                 new DeserializerBuilder()
                 .WithNamingConvention(new CamelCaseNamingConvention())
                 .WithTypeInspector(ti => new AutoRestTypeInspector(ti))
+                .WithTypeConverter(new IntOrStringYamlConverter())
                 .Build();
             var obj = deserializer.Deserialize<T>(content);
             return obj;
@@ -47,8 +49,10 @@ namespace k8s
 
             var serializer =
                 new SerializerBuilder()
+                .DisableAliases()
                 .WithNamingConvention(new CamelCaseNamingConvention())
                 .WithTypeInspector(ti => new AutoRestTypeInspector(ti))
+                .WithTypeConverter(new IntOrStringYamlConverter())
                 .BuildValueSerializer();
             emitter.Emit(new StreamStart());
             emitter.Emit(new DocumentStart());

tests/KubernetesClient.Tests/YamlTests.cs

@@ -7,7 +7,8 @@ using Xunit;
 
 namespace k8s.Tests
 {
-    public class YamlTests {
+    public class YamlTests
+    {
         [Fact]
         public void LoadFromString()
         {
@@ -236,5 +237,62 @@ spec:
             Assert.Equal("cpu", cpuRequest.Key);
             Assert.Equal("500m", cpuRequest.Value.ToString());
         }
+
+        [Fact]
+        public void LoadIntOrString()
+        {
+            var content = @"apiVersion: v1
+kind: Service
+spec:
+  ports:
+  - port: 3000
+    targetPort: 3000
+";
+
+            var obj = Yaml.LoadFromString<V1Service>(content);
+
+            Assert.Equal(3000, obj.Spec.Ports[0].Port);
+            Assert.Equal(3000, (int)obj.Spec.Ports[0].TargetPort);
+        }
+
+        [Fact]
+        public void SerializeIntOrString()
+        {
+            var content = @"apiVersion: v1
+kind: Service
+metadata:
+  labels:
+    app: test
+  name: test-svc
+spec:
+  ports:
+  - port: 3000
+    targetPort: 3000";
+
+            Dictionary<string, string> labels = new Dictionary<string, string>
+                                                {
+                                                    {"app", "test"}
+                                                };
+            var obj = new V1Service
+            {
+                Kind = "Service",
+                Metadata = new V1ObjectMeta(labels: labels, name: "test-svc"),
+                ApiVersion = "v1",
+                Spec = new V1ServiceSpec
+                {
+                    Ports = new List<V1ServicePort>
+                                                       {
+                                                           new V1ServicePort
+                                                           {
+                                                               Port = 3000,
+                                                               TargetPort = 3000
+                                                           }
+                                                       }
+                }
+            };
+
+            var output = Yaml.SaveToString<V1Service>(obj);
+            Assert.Equal(output, content);
+        }
     }
 }