1 files changed, 7 insertions, 1 deletions
diff --git a/policy-endpoints/src/test/java/org/onap/policy/common/endpoints/event/comm/bus/internal/SingleThreadedKafkaTopicSourceTest.java b/policy-endpoints/src/test/java/org/onap/policy/common/endpoints/event/comm/bus/internal/SingleThreadedKafkaTopicSourceTest.java
index cc096585..6b63c9f4 100644
--- a/policy-endpoints/src/test/java/org/onap/policy/common/endpoints/event/comm/bus/internal/SingleThreadedKafkaTopicSourceTest.java
+++ b/policy-endpoints/src/test/java/org/onap/policy/common/endpoints/event/comm/bus/internal/SingleThreadedKafkaTopicSourceTest.java
@@ -42,7 +42,7 @@ public class SingleThreadedKafkaTopicSourceTest extends TopicTestBase {
     public void setUp() {
         super.setUp();
 
-        source = new SingleThreadedKafkaTopicSource(makeBuilder().build());
+        source = new SingleThreadedKafkaTopicSource(makeKafkaBuilder().build());
     }
 
     @After
@@ -50,9 +50,15 @@ public class SingleThreadedKafkaTopicSourceTest extends TopicTestBase {
         source.shutdown();
     }
 
+    public void testSerialize() {
+        assertThatCode(() -> new GsonTestUtils().compareGson(source, SingleThreadedKafkaTopicSourceTest.class))
+                        .doesNotThrowAnyException();
+    }
+
     @Test
     public void testToString() {
         assertTrue(source.toString().startsWith("SingleThreadedKafkaTopicSource ["));
+        source.shutdown();
     }
 
     @Test