aboutsummaryrefslogtreecommitdiff
path: root/weed/query/engine/real_namespace_test.go
blob: 6c88ef612bd1d74b9086ce754d381afa24913867 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
package engine

import (
	"context"
	"testing"
)

// TestRealNamespaceDiscovery tests the real namespace discovery functionality
func TestRealNamespaceDiscovery(t *testing.T) {
	engine := NewSQLEngine("localhost:8888")

	// Test SHOW DATABASES with real namespace discovery
	result, err := engine.ExecuteSQL(context.Background(), "SHOW DATABASES")
	if err != nil {
		t.Fatalf("SHOW DATABASES failed: %v", err)
	}

	// Should have Database column
	if len(result.Columns) != 1 || result.Columns[0] != "Database" {
		t.Errorf("Expected 1 column 'Database', got %v", result.Columns)
	}

	// With no fallback sample data, result may be empty if no real MQ cluster
	t.Logf("Discovered %d namespaces (no fallback data):", len(result.Rows))
	if len(result.Rows) == 0 {
		t.Log("  (No namespaces found - requires real SeaweedFS MQ cluster)")
	} else {
		for _, row := range result.Rows {
			if len(row) > 0 {
				t.Logf("  - %s", row[0].ToString())
			}
		}
	}
}

// TestRealTopicDiscovery tests the real topic discovery functionality
func TestRealTopicDiscovery(t *testing.T) {
	engine := NewSQLEngine("localhost:8888")

	// Test SHOW TABLES with real topic discovery (use double quotes for PostgreSQL)
	result, err := engine.ExecuteSQL(context.Background(), "SHOW TABLES FROM \"default\"")
	if err != nil {
		t.Fatalf("SHOW TABLES failed: %v", err)
	}

	// Should have table name column
	expectedColumn := "Tables_in_default"
	if len(result.Columns) != 1 || result.Columns[0] != expectedColumn {
		t.Errorf("Expected 1 column '%s', got %v", expectedColumn, result.Columns)
	}

	// With no fallback sample data, result may be empty if no real MQ cluster or namespace doesn't exist
	t.Logf("Discovered %d topics in 'default' namespace (no fallback data):", len(result.Rows))
	if len(result.Rows) == 0 {
		t.Log("  (No topics found - requires real SeaweedFS MQ cluster with 'default' namespace)")
	} else {
		for _, row := range result.Rows {
			if len(row) > 0 {
				t.Logf("  - %s", row[0].ToString())
			}
		}
	}
}

// TestNamespaceDiscoveryNoFallback tests behavior when filer is unavailable (no sample data)
func TestNamespaceDiscoveryNoFallback(t *testing.T) {
	// This test demonstrates the no-fallback behavior when no real MQ cluster is running
	engine := NewSQLEngine("localhost:8888")

	// Get broker client to test directly
	brokerClient := engine.catalog.brokerClient
	if brokerClient == nil {
		t.Fatal("Expected brokerClient to be initialized")
	}

	// Test namespace listing (should fail without real cluster)
	namespaces, err := brokerClient.ListNamespaces(context.Background())
	if err != nil {
		t.Logf("ListNamespaces failed as expected: %v", err)
		namespaces = []string{} // Set empty for the rest of the test
	}

	// With no fallback sample data, should return empty lists
	if len(namespaces) != 0 {
		t.Errorf("Expected empty namespace list with no fallback, got %v", namespaces)
	}

	// Test topic listing (should return empty list)
	topics, err := brokerClient.ListTopics(context.Background(), "default")
	if err != nil {
		t.Fatalf("ListTopics failed: %v", err)
	}

	// Should have no fallback topics
	if len(topics) != 0 {
		t.Errorf("Expected empty topic list with no fallback, got %v", topics)
	}

	t.Log("No fallback behavior - returns empty lists when filer unavailable")
}