From 0b9f0de0a10dda9bb24dc9b4be1009e8ff84e746 Mon Sep 17 00:00:00 2001
From: Aliaksandr Valialkin <valyala@gmail.com>
Date: Fri, 28 May 2021 01:32:50 +0300
Subject: [PATCH] lib/promscrape: fix tests after
 f0c21b6300b6df4f4d661dc90cd718bf7b6d28b5

---
 lib/promscrape/scrapework_test.go | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/lib/promscrape/scrapework_test.go b/lib/promscrape/scrapework_test.go
index 79e6cb6ce..843e8dde1 100644
--- a/lib/promscrape/scrapework_test.go
+++ b/lib/promscrape/scrapework_test.go
@@ -115,7 +115,9 @@ func TestScrapeWorkScrapeInternalSuccess(t *testing.T) {
 
 		timestamp := int64(123000)
 		if err := sw.scrapeInternal(timestamp, timestamp); err != nil {
-			t.Fatalf("unexpected error: %s", err)
+			if !strings.Contains(err.Error(), "sample_limit") {
+				t.Fatalf("unexpected error: %s", err)
+			}
 		}
 		if pushDataErr != nil {
 			t.Fatalf("unexpected error: %s", pushDataErr)