minor changes

sajjad-moradi · sajjad-moradi · commit 85db9b64b19a · 2022-05-16T17:36:27.000-07:00
diff --git a/pinot-core/src/main/java/org/apache/pinot/core/data/manager/realtime/RealtimeTableDataManager.java b/pinot-core/src/main/java/org/apache/pinot/core/data/manager/realtime/RealtimeTableDataManager.java
@@ -371,7 +371,7 @@ private void handleUpsert(ImmutableSegmentImpl immutableSegment) {
     String segmentName = immutableSegment.getSegmentName();
     Integer partitionGroupId = SegmentUtils
         .getRealtimeSegmentPartitionId(segmentName, _tableNameWithType, _helixManager, _primaryKeyColumns.get(0));
-    Preconditions.checkNotNull(String
+    Preconditions.checkNotNull(partitionGroupId, String
         .format("PartitionGroupId is not available for segment: '%s' (upsert-enabled table: %s)", segmentName,
             _tableNameWithType));
     PartitionUpsertMetadataManager partitionUpsertMetadataManager =
diff --git a/pinot-integration-tests/src/test/java/org/apache/pinot/integration/tests/RealtimeClusterIntegrationTest.java b/pinot-integration-tests/src/test/java/org/apache/pinot/integration/tests/RealtimeClusterIntegrationTest.java
@@ -65,7 +65,7 @@ public void setUp()
     // Push data into Kafka
     pushAvroIntoKafka(avroFiles);
 
-    // create segment and upload it to controller
+    // create segments and upload them to controller
     createSegmentsAndUpload(avroFiles, schema, tableConfig);
 
     // Set up the H2 connection
