22
33namespace Jobcloud \Kafka \Tests \Unit \Kafka \Consumer ;
44
5+ use Jobcloud \Kafka \Consumer \AbstractKafkaConsumer ;
56use Jobcloud \Kafka \Consumer \KafkaHighLevelConsumer ;
67use Jobcloud \Kafka \Consumer \TopicSubscriptionInterface ;
78use Jobcloud \Kafka \Exception \KafkaConsumerConsumeException ;
@@ -465,7 +466,7 @@ public function testKafkaConsumeWithDecode(): void
465466 $ message ->partition = '9 ' ;
466467 $ message ->offset = '501 ' ;
467468 $ message ->timestamp = '500 ' ;
468- $ message ->headers = ' header ' ;
469+ $ message ->headers = [ ' key ' => ' value ' ] ;
469470 $ message ->err = RD_KAFKA_RESP_ERR_NO_ERROR ;
470471
471472 $ topics = [new TopicSubscription ('testTopic ' )];
@@ -491,7 +492,7 @@ function (KafkaConsumerMessageInterface $message) {
491492 self ::assertEquals (9 , $ message ->getPartition ());
492493 self ::assertEquals (501 , $ message ->getOffset ());
493494 self ::assertEquals (500 , $ message ->getTimestamp ());
494- self ::assertEquals (['header ' ], $ message ->getHeaders ());
495+ self ::assertEquals (['key ' => ' value ' ], $ message ->getHeaders ());
495496
496497 return true ;
497498 }
@@ -512,6 +513,7 @@ public function testKafkaConsumeWithoutDecode(): void
512513 $ message ->partition = 9 ;
513514 $ message ->offset = 501 ;
514515 $ message ->timestamp = 500 ;
516+ $ message ->headers = ['key ' => 'value ' ];
515517 $ message ->err = RD_KAFKA_RESP_ERR_NO_ERROR ;
516518
517519 $ topics = [new TopicSubscription ('testTopic ' )];
@@ -663,6 +665,16 @@ public function testGetTopicSubscriptionsReturnsTopicSubscriptions(): void
663665 self ::assertSame ($ topicSubscriptionsMock , $ kafkaConsumer ->getTopicSubscriptions ());
664666 }
665667
668+ public function testMethodVisibility (): void
669+ {
670+ $ reflectionClass = new \ReflectionClass (AbstractKafkaConsumer::class);
671+
672+ $ methodGetConsumerMessage = $ reflectionClass ->getMethod ('getConsumerMessage ' );
673+ $ methodGetConsumerMessage ->setAccessible (true );
674+
675+ $ this ->assertTrue ($ methodGetConsumerMessage ->isProtected ());
676+ }
677+
666678 /**
667679 * @param int $partitionId
668680 * @return RdKafkaMetadataPartition|MockObject
0 commit comments