Function TestPartitionRingPageHandler_ViewPage

ring/partition_ring_http_test.go:24–138

(t *testing.T)

)

func TestPartitionRingPageHandler_ViewPage(t *testing.T) {
	partRing, err := NewPartitionRing(PartitionRingDesc{
		Partitions: map[int32]PartitionDesc{
			1: {
				State:          PartitionActive,
				StateTimestamp: time.Now().Unix(),
				Tokens:         []uint32{1000000, 3000000, 6000000},
			},
			2: {
				State:          PartitionInactive,
				StateTimestamp: time.Now().Unix(),
				Tokens:         []uint32{2000000, 4000000, 5000000, 7000000},
			},
		},
		Owners: map[string]OwnerDesc{
			"ingester-zone-a-0": {
				OwnedPartition: 1,
			},
			"ingester-zone-a-1": {
				OwnedPartition: 2,
			},
			"ingester-zone-b-0": {
				OwnedPartition: 1,
			},
			"ingester-zone-b-1": {
				OwnedPartition: 2,
			},

			// Simulate a corrupted partition, with a dangling owner but no partition.
			"ingester-zone-b-2": {
				OwnedPartition: 3,
			},
		},
	})
	require.NoError(t, err)

	handler := NewPartitionRingPageHandler(
		newStaticPartitionRingReader(partRing),
		nil,
	)

	t.Run("displays expected partition info", func(t *testing.T) {
		recorder := httptest.NewRecorder()
		handler.ServeHTTP(recorder, httptest.NewRequest(http.MethodGet, "/partition-ring", nil))

		assert.Equal(t, http.StatusOK, recorder.Code)
		assert.Equal(t, "text/html", recorder.Header().Get("Content-Type"))

		assert.Regexp(t, regexp.MustCompile(fmt.Sprintf("(?m)%s", strings.Join([]string{
			"<td>", "1", "</td>",
			"<td>", "Active", "</td>",
			"<td>", "[^<]+", "</td>",
			"<td>", "ingester-zone-a-0", "<br />", "ingester-zone-b-0", "<br />", "</td>",
			"<td>", "3", "</td>",
			"<td>", "99.9%", "</td>",
		}, `\s*`))), recorder.Body.String())

		assert.Regexp(t, regexp.MustCompile(fmt.Sprintf("(?m)%s", strings.Join([]string{