Close #13

yaravind · May 26, 2016 · a4ae410 · a4ae410
1 parent 015c5a1
commit a4ae410
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 1 deletion.
diff --git a/src/main/java/org/aravind/oss/kafka/connector/JenkinsSourceConnector.java b/src/main/java/org/aravind/oss/kafka/connector/JenkinsSourceConnector.java
@@ -56,7 +56,7 @@ public void start(Map<String, String> props) {
 
     @Override
     public Class<? extends Task> taskClass() {
-        return null;
+        return JenkinsSourceTask.class;
     }
 
     /**

diff --git a/src/test/groovy/org/aravind/oss/kafka/connector/JenkinsSourceConnectorTest.groovy b/src/test/groovy/org/aravind/oss/kafka/connector/JenkinsSourceConnectorTest.groovy
@@ -28,6 +28,17 @@ class JenkinsSourceConnectorTest extends Specification {
     def cleanup() {
         if (connector != null) connector.stop()
     }
+
+    //API Contracts
+
+    def "taskClass()"() {
+        when:
+        Class taskClass = connector.taskClass()
+
+        then:
+        taskClass == JenkinsSourceTask.class
+    }
+
     //Error scenarios
 
     def "Config - Wrong jenkins url should throw exception"() {