[java] Update toJson() method creation for integer type in CdpClientGenerator

pujagani · web-flow · commit ad48d31a7d88 · 2021-12-31T10:46:57.000+05:30
Fixes #10155
diff --git a/java/src/org/openqa/selenium/devtools/CdpClientGenerator.java b/java/src/org/openqa/selenium/devtools/CdpClientGenerator.java
@@ -769,6 +769,9 @@ public TypeDeclaration<?> toTypeDeclaration() {
       if (type.equals("object")) {
         toJson.setType("java.util.Map<String, Object>");
         toJson.getBody().get().addStatement(String.format("return %s;", propertyName));
+      } else if (type.equals("number")) {
+        toJson.setType(Number.class);
+        toJson.getBody().get().addStatement(String.format("return %s;", propertyName));
       } else if (type.equals("integer")) {
         toJson.setType(Integer.class);
         toJson.getBody().get().addStatement(String.format("return %s;", propertyName));
diff --git a/java/src/org/openqa/selenium/devtools/v85/V85Events.java b/java/src/org/openqa/selenium/devtools/v85/V85Events.java
@@ -64,7 +64,7 @@ protected Event<ExceptionThrown> exceptionThrownEvent() {
 
   @Override
   protected ConsoleEvent toConsoleEvent(ConsoleAPICalled event) {
-    long ts = new BigDecimal(event.getTimestamp().toJson()).longValue();
+    long ts = event.getTimestamp().toJson().longValue();
 
     List<Object> modifiedArgs = event.getArgs().stream()
       .map(obj -> new RemoteObject(
diff --git a/java/src/org/openqa/selenium/devtools/v94/V94Events.java b/java/src/org/openqa/selenium/devtools/v94/V94Events.java
@@ -64,7 +64,7 @@ protected Event<ExceptionThrown> exceptionThrownEvent() {
 
   @Override
   protected ConsoleEvent toConsoleEvent(ConsoleAPICalled event) {
-    long ts = new BigDecimal(event.getTimestamp().toJson()).longValue();
+    long ts = event.getTimestamp().toJson().longValue();
 
     List<Object> modifiedArgs = event.getArgs().stream()
       .map(obj -> new RemoteObject(
diff --git a/java/src/org/openqa/selenium/devtools/v95/V95Events.java b/java/src/org/openqa/selenium/devtools/v95/V95Events.java
@@ -64,7 +64,7 @@ protected Event<ExceptionThrown> exceptionThrownEvent() {
 
   @Override
   protected ConsoleEvent toConsoleEvent(ConsoleAPICalled event) {
-    long ts = new BigDecimal(event.getTimestamp().toJson()).longValue();
+    long ts = event.getTimestamp().toJson().longValue();
 
     List<Object> modifiedArgs = event.getArgs().stream()
       .map(obj -> new RemoteObject(
diff --git a/java/src/org/openqa/selenium/devtools/v96/V96Events.java b/java/src/org/openqa/selenium/devtools/v96/V96Events.java
@@ -64,7 +64,7 @@ protected Event<ExceptionThrown> exceptionThrownEvent() {
 
   @Override
   protected ConsoleEvent toConsoleEvent(ConsoleAPICalled event) {
-    long ts = new BigDecimal(event.getTimestamp().toJson()).longValue();
+    long ts = event.getTimestamp().toJson().longValue();
 
     List<Object> modifiedArgs = event.getArgs().stream()
       .map(obj -> new RemoteObject(
