dlp: add a utility conversion function (#3028)

jeanbza · pongad · commit 4fbe33177071 · 2018-03-15T10:19:37.000+11:00
Fixes #2321
diff --git a/google-cloud-dlp/src/main/java/com/google/cloud/dlp/v2beta1/Colors.java b/google-cloud-dlp/src/main/java/com/google/cloud/dlp/v2beta1/Colors.java
@@ -0,0 +1,35 @@
+/*
+ * Copyright 2018 Google LLC
+ *
+ * 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
+ *
+ *     https://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.
+ */
+package com.google.cloud.dlp.v2beta1;
+
+import com.google.privacy.dlp.v2beta1.Color;
+
+public class Colors {
+
+  private Colors() {}
+
+  public static final Color asDlpColor(java.awt.Color c) {
+    return Color.newBuilder()
+        .setBlue(c.getBlue())
+        .setGreen(c.getGreen())
+        .setRed(c.getRed())
+        .build();
+  }
+
+  public static final java.awt.Color asAwtColor(Color c) {
+    return new java.awt.Color(c.getRed() / 255, c.getGreen() / 255, c.getBlue() / 255);
+  }
+}
diff --git a/google-cloud-dlp/src/test/java/com/google/cloud/dlp/v2beta1/ColorTest.java b/google-cloud-dlp/src/test/java/com/google/cloud/dlp/v2beta1/ColorTest.java
@@ -0,0 +1,26 @@
+package com.google.cloud.dlp.v2beta1;
+
+import com.google.privacy.dlp.v2beta1.Color;
+import org.junit.Assert;
+import org.junit.Test;
+
+public class ColorTest {
+
+  @Test
+  public void asDlpColorTest() {
+    Color got = Colors.asDlpColor(new java.awt.Color(1, 2, 3));
+    Assert.assertEquals(got.getBlue(), 3, 0);
+    Assert.assertEquals(got.getGreen(), 2, 0);
+    Assert.assertEquals(got.getRed(), 1, 0);
+  }
+
+  @Test
+  public void asAwtColorTest() {
+    java.awt.Color got =
+        Colors.asAwtColor(Color.newBuilder().setRed(1).setGreen(2).setBlue(3).build());
+
+    Assert.assertEquals(got.getBlue(), 3, 0);
+    Assert.assertEquals(got.getGreen(), 2, 0);
+    Assert.assertEquals(got.getRed(), 1, 0);
+  }
+}
