Method TestResiliencyUnary

pkg/api/grpc/proxy/handler_test.go:418–547

()

}

func (s *proxyTestSuite) TestResiliencyUnary() {
	const message = "Ciao mamma guarda come mi diverto"

	// Set the resiliency policy and the number of simulated failures in a row
	s.setupResiliency()
	defer func() {
		s.policyDef = nil
	}()

	s.T().Run("retries", func(t *testing.T) {
		s.service.simulatePingFailures.Store(2)
		defer func() {
			s.service.simulatePingFailures.Store(0)
		}()

		meter := setupMetrics(s)
		t.Cleanup(func() {
			meter.Stop()
		})

		ctx, cancel := s.ctx()
		defer cancel()
		ctx = metadata.NewOutgoingContext(ctx, metadata.Pairs(diagConsts.GRPCProxyAppIDKey, testAppID))

		// Reset callCount before this test
		s.service.pingCallCount.Store(0)

		res, err := s.testClient.Ping(ctx, &pb.PingRequest{Value: message})
		require.NoError(t, err, "Ping should succeed after retrying")
		require.NotNil(t, res, "Response should not be nil")
		require.Equal(t, int32(3), s.service.pingCallCount.Load())
		require.Equal(t, message, res.GetValue())

		assertRequestSentMetrics(t, meter, "unary", 3, nil)

		rows, err := meter.RetrieveData(serviceInvocationResponseRecvName)
		require.NoError(t, err)
		assert.Len(t, rows, 2)
		// 2 Ping failures
		assert.Equal(t, int64(2), diag.GetCountValueForObservationWithTagSet(
			rows, map[tag.Tag]bool{diag.NewTag("status", strconv.Itoa(int(codes.Internal))): true}))
		// 1 success
		assert.Equal(t, int64(1), diag.GetCountValueForObservationWithTagSet(
			rows, map[tag.Tag]bool{diag.NewTag("status", strconv.Itoa(int(codes.OK))): true}))
	})

	s.T().Run("timeouts", func(t *testing.T) {
		// The delay is longer than the timeout set in the resiliency policy (500ms)
		s.service.simulateDelay.Store(800)
		defer func() {
			s.service.simulateDelay.Store(0)
		}()

		// Reset callCount before this test
		s.service.pingCallCount.Store(0)

		meter := setupMetrics(s)
		t.Cleanup(func() {