X-Git-Url: https://git.giacomolli.uk/onechat/blobdiff_plain/d647f33b3e70f9419f6c0740e1d01b210439a519..97729fc59a2fbd73e3f8a424cae4c825f5ba885b:/responses_test.go?ds=sidebyside

diff --git a/responses_test.go b/responses_test.go
index 6017ad7..9a1355a 100644
--- a/responses_test.go
+++ b/responses_test.go
@@ -197,6 +197,24 @@ func TestClientRecordsEachCall(t *testing.T) {
 	require.Equal(t, os.FileMode(0o600), info.Mode().Perm())
 }
 
+func TestClientRefusesCallsOverTheLimit(t *testing.T) {
+	server := newFakeResponses(t)
+	client := newTestClient(t, server)
+	client.LimitCalls(2)
+
+	for range 2 {
+		require.False(t, client.LimitReached())
+		_, err := client.Create(t.Context(), testRequest())
+		require.NoError(t, err)
+	}
+
+	require.True(t, client.LimitReached())
+	_, err := client.Create(t.Context(), testRequest())
+	require.ErrorIs(t, err, errCallLimit)
+	require.Equal(t, 2, server.Calls())
+	require.Len(t, readUsage(t, client.usage.path), 2)
+}
+
 func TestClientFailsWhenTheResponseFails(t *testing.T) {
 	testCases := []struct {
 		name   string