Browse Source

fix kafka tests

pull/7231/head
chrislu 2 months ago
parent
commit
30b21abab9
  1. 1
      test/kafka/api_sequence_test.go
  2. 3
      test/kafka/client_integration_test.go
  3. 2
      test/kafka/comprehensive_e2e_test.go
  4. 1
      test/kafka/consumer_test.go
  5. 3
      test/kafka/debug_connection_test.go
  6. 9
      test/kafka/persistent_offset_integration_test.go
  7. 1
      test/kafka/produce_consume_test.go
  8. 1
      test/kafka/raw_protocol_test.go
  9. 13
      test/kafka/seaweedmq_integration_test.go

1
test/kafka/api_sequence_test.go

@ -15,7 +15,6 @@ func TestKafkaGateway_APISequence(t *testing.T) {
// Start the gateway server
srv := gateway.NewServer(gateway.Options{
Listen: ":0",
UseSeaweedMQ: false,
})
if err := srv.Start(); err != nil {

3
test/kafka/client_integration_test.go

@ -17,7 +17,6 @@ func TestKafkaGoClient_BasicProduceConsume(t *testing.T) {
// Start the gateway server
srv := gateway.NewServer(gateway.Options{
Listen: ":0", // Use random port
UseSeaweedMQ: false, // Use in-memory mode for testing
})
if err := srv.Start(); err != nil {
@ -84,7 +83,6 @@ func TestKafkaGoClient_ConsumerGroups(t *testing.T) {
// Start the gateway server
srv := gateway.NewServer(gateway.Options{
Listen: ":0",
UseSeaweedMQ: false,
})
if err := srv.Start(); err != nil {
@ -138,7 +136,6 @@ func TestKafkaGoClient_OffsetManagement(t *testing.T) {
// Start the gateway server
srv := gateway.NewServer(gateway.Options{
Listen: ":0",
UseSeaweedMQ: false,
})
if err := srv.Start(); err != nil {

2
test/kafka/comprehensive_e2e_test.go

@ -17,7 +17,6 @@ func TestComprehensiveE2E(t *testing.T) {
// Start gateway
gatewayServer := gateway.NewServer(gateway.Options{
Listen: "127.0.0.1:0",
UseSeaweedMQ: false, // Use in-memory mode for testing
})
go func() {
@ -360,7 +359,6 @@ func TestOffsetManagement(t *testing.T) {
// Start gateway
gatewayServer := gateway.NewServer(gateway.Options{
Listen: "127.0.0.1:0",
UseSeaweedMQ: false,
})
go func() {

1
test/kafka/consumer_test.go

@ -15,7 +15,6 @@ func TestKafkaGoReader(t *testing.T) {
// Start the gateway server
srv := gateway.NewServer(gateway.Options{
Listen: ":0",
UseSeaweedMQ: false,
})
if err := srv.Start(); err != nil {

3
test/kafka/debug_connection_test.go

@ -14,7 +14,6 @@ func TestGateway_BasicConnection(t *testing.T) {
// Start the gateway server
srv := gateway.NewServer(gateway.Options{
Listen: ":0",
UseSeaweedMQ: false,
})
if err := srv.Start(); err != nil {
@ -40,7 +39,6 @@ func TestGateway_ApiVersionsRequest(t *testing.T) {
// Start the gateway server
srv := gateway.NewServer(gateway.Options{
Listen: ":0",
UseSeaweedMQ: false,
})
if err := srv.Start(); err != nil {
@ -142,7 +140,6 @@ func TestGateway_CreateTopicsRequest(t *testing.T) {
// Start the gateway server
srv := gateway.NewServer(gateway.Options{
Listen: ":0",
UseSeaweedMQ: false,
})
if err := srv.Start(); err != nil {

9
test/kafka/persistent_offset_integration_test.go

@ -39,15 +39,13 @@ func TestPersistentOffsetIntegration(t *testing.T) {
func testOffsetPersistenceAndRecovery(t *testing.T, brokers []string) {
// Create offset storage
storage, err := offset.NewSeaweedMQStorage(brokers)
require.NoError(t, err)
storage := offset.NewSeaweedMQStorage()
defer storage.Close()
topicPartition := "test-persistence-topic-0"
// Create first ledger and add some entries
ledger1, err := offset.NewPersistentLedger(topicPartition, storage)
require.NoError(t, err)
ledger1 := offset.NewPersistentLedger(topicPartition, storage)
// Add test entries
testEntries := []struct {
@ -77,8 +75,7 @@ func testOffsetPersistenceAndRecovery(t *testing.T, brokers []string) {
time.Sleep(2 * time.Second)
// Create second ledger (simulating restart)
ledger2, err := offset.NewPersistentLedger(topicPartition, storage)
require.NoError(t, err)
ledger2 := offset.NewPersistentLedger(topicPartition, storage)
// Verify recovered state
assert.Equal(t, ledger1.GetHighWaterMark(), ledger2.GetHighWaterMark())

1
test/kafka/produce_consume_test.go

@ -16,7 +16,6 @@ func TestKafkaGoClient_DirectProduceConsume(t *testing.T) {
// Start the gateway server
srv := gateway.NewServer(gateway.Options{
Listen: ":0",
UseSeaweedMQ: false,
})
if err := srv.Start(); err != nil {

1
test/kafka/raw_protocol_test.go

@ -14,7 +14,6 @@ func TestRawProduceRequest(t *testing.T) {
// Start the gateway server
srv := gateway.NewServer(gateway.Options{
Listen: ":0",
UseSeaweedMQ: false,
})
if err := srv.Start(); err != nil {

13
test/kafka/seaweedmq_integration_test.go

@ -14,14 +14,9 @@ func TestSeaweedMQIntegration_E2E(t *testing.T) {
// Skip by default - requires real SeaweedMQ setup
t.Skip("Integration test requires real SeaweedMQ setup - run manually")
// Test configuration
agentAddress := "localhost:17777" // Default SeaweedMQ Agent address
// Start the gateway with SeaweedMQ backend
gatewayServer := gateway.NewServer(gateway.Options{
Listen: ":0", // random port
AgentAddress: agentAddress,
UseSeaweedMQ: true,
})
err := gatewayServer.Start()
@ -242,7 +237,6 @@ func TestSeaweedMQGateway_ModeSelection(t *testing.T) {
t.Run("InMemoryMode", func(t *testing.T) {
server := gateway.NewServer(gateway.Options{
Listen: ":0",
UseSeaweedMQ: false,
})
err := server.Start()
@ -263,8 +257,6 @@ func TestSeaweedMQGateway_ModeSelection(t *testing.T) {
t.Run("SeaweedMQModeFallback", func(t *testing.T) {
server := gateway.NewServer(gateway.Options{
Listen: ":0",
AgentAddress: "invalid:99999", // Invalid address
UseSeaweedMQ: true,
})
err := server.Start()
@ -293,7 +285,6 @@ func TestSeaweedMQGateway_ConfigValidation(t *testing.T) {
name: "ValidInMemory",
options: gateway.Options{
Listen: ":0",
UseSeaweedMQ: false,
},
shouldWork: true,
},
@ -301,8 +292,6 @@ func TestSeaweedMQGateway_ConfigValidation(t *testing.T) {
name: "ValidSeaweedMQWithAgent",
options: gateway.Options{
Listen: ":0",
AgentAddress: "localhost:17777",
UseSeaweedMQ: true,
},
shouldWork: true, // May fail if no agent, but config is valid
},
@ -310,8 +299,6 @@ func TestSeaweedMQGateway_ConfigValidation(t *testing.T) {
name: "SeaweedMQWithoutAgent",
options: gateway.Options{
Listen: ":0",
UseSeaweedMQ: true,
// AgentAddress is empty
},
shouldWork: true, // Should fall back to in-memory
},

Loading…
Cancel
Save