Add test case for upsert insertId 0 fix

Signed-off-by: Leo Xuzhang Lin <[email protected]>
before30 · Aug 15, 2018 · b195f30 · b195f30
1 parent 125c9fe
commit b195f30
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/java/jdbc/src/test/java/io/vitess/jdbc/VitessStatementTest.java b/java/jdbc/src/test/java/io/vitess/jdbc/VitessStatementTest.java
@@ -742,5 +742,15 @@ private void testExecute(int fetchSize, boolean simpleExecute, boolean shouldRun
             Assert.assertEquals(i, 0); // we should only have one
             i++;
         }
+
+        VitessStatement noUpdate = new VitessStatement(mockConn);
+        PowerMockito.when(mockCursor.getInsertId()).thenReturn(0L);
+        PowerMockito.when(mockCursor.getRowsAffected()).thenReturn(1L);
+
+        noUpdate.addBatch(sqlUpsert);
+        noUpdate.executeBatch();
+
+        ResultSet empty = noUpdate.getGeneratedKeys();
+        Assert.assertFalse(empty.next());
     }
 }