Fix unit test for MySQLSinkTest

Author: sgarg-CS

mysql-plugin/src/test/java/io/cdap/plugin/mysql/MysqlSinkTest.java

@@ -20,14 +20,23 @@
 import org.junit.Assert;
 import org.junit.Test;
 
+import static org.mockito.Mockito.mock;
+import static org.mockito.Mockito.when;
+
 public class MysqlSinkTest {
   @Test
   public void testSetColumnsInfo() {
     Schema outputSchema = Schema.recordOf("output",
       Schema.Field.of("id", Schema.of(Schema.Type.INT)),
       Schema.Field.of("name", Schema.of(Schema.Type.STRING)),
       Schema.Field.of("insert", Schema.of(Schema.Type.STRING)));
-    MysqlSink mySQLSink = new MysqlSink(new MysqlSink.MysqlSinkConfig());
+
+    MysqlSink.MysqlSinkConfig mockConfig = mock(MysqlSink.MysqlSinkConfig.class);
+    when(mockConfig.getInitialRetryDuration()).thenReturn(5); // or appropriate value
+    when(mockConfig.getMaxRetryDuration()).thenReturn(80); // or appropriate value
+    when(mockConfig.getMaxRetryCount()).thenReturn(5); // or appropriate value
+
+    MysqlSink mySQLSink = new MysqlSink(mockConfig);
     Assert.assertNotNull(outputSchema.getFields());
     mySQLSink.setColumnsInfo(outputSchema.getFields());
     Assert.assertEquals("`id`,`name`,`insert`", mySQLSink.getDbColumns());