]> git.giacomolli.uk - onechat/blobdiff - responses_test.go
Convert reply Markdown to Telegram entities
[onechat] / responses_test.go
index 6017ad791e03f64859d5331ffeb911e80b4885c0..9a1355a8d0098bb2ee18b01a1d0b9e3d0a8d665e 100644 (file)
@@ -197,6 +197,24 @@ func TestClientRecordsEachCall(t *testing.T) {
        require.Equal(t, os.FileMode(0o600), info.Mode().Perm())
 }
 
        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
 func TestClientFailsWhenTheResponseFails(t *testing.T) {
        testCases := []struct {
                name   string