@@ -407,30 +407,20 @@ func TestDo_rateLimit(t *testing.T) {
407
407
w .Header ().Add (headerRateReset , "1372700873" )
408
408
})
409
409
410
- if got , want := client .Rate ().Limit , 0 ; got != want {
411
- t .Errorf ("Client rate limit = %v, want %v" , got , want )
412
- }
413
- if got , want := client .Rate ().Remaining , 0 ; got != want {
414
- t .Errorf ("Client rate remaining = %v, got %v" , got , want )
415
- }
416
- if ! client .Rate ().Reset .IsZero () {
417
- t .Errorf ("Client rate reset not initialized to zero value" )
418
- }
419
-
420
410
req , _ := client .NewRequest ("GET" , "/" , nil )
421
- _ , err := client .Do (req , nil )
411
+ resp , err := client .Do (req , nil )
422
412
if err != nil {
423
413
t .Errorf ("Do returned unexpected error: %v" , err )
424
414
}
425
- if got , want := client .Rate () .Limit , 60 ; got != want {
415
+ if got , want := resp .Rate .Limit , 60 ; got != want {
426
416
t .Errorf ("Client rate limit = %v, want %v" , got , want )
427
417
}
428
- if got , want := client .Rate () .Remaining , 59 ; got != want {
418
+ if got , want := resp .Rate .Remaining , 59 ; got != want {
429
419
t .Errorf ("Client rate remaining = %v, want %v" , got , want )
430
420
}
431
421
reset := time .Date (2013 , 7 , 1 , 17 , 47 , 53 , 0 , time .UTC )
432
- if client .Rate () .Reset .UTC () != reset {
433
- t .Errorf ("Client rate reset = %v, want %v" , client .Rate () .Reset , reset )
422
+ if resp .Rate .Reset .UTC () != reset {
423
+ t .Errorf ("Client rate reset = %v, want %v" , resp .Rate .Reset , reset )
434
424
}
435
425
}
436
426
@@ -447,23 +437,22 @@ func TestDo_rateLimit_errorResponse(t *testing.T) {
447
437
})
448
438
449
439
req , _ := client .NewRequest ("GET" , "/" , nil )
450
- _ , err := client .Do (req , nil )
451
-
440
+ resp , err := client .Do (req , nil )
452
441
if err == nil {
453
442
t .Error ("Expected error to be returned." )
454
443
}
455
444
if _ , ok := err .(* RateLimitError ); ok {
456
445
t .Errorf ("Did not expect a *RateLimitError error; got %#v." , err )
457
446
}
458
- if got , want := client .Rate () .Limit , 60 ; got != want {
447
+ if got , want := resp .Rate .Limit , 60 ; got != want {
459
448
t .Errorf ("Client rate limit = %v, want %v" , got , want )
460
449
}
461
- if got , want := client .Rate () .Remaining , 59 ; got != want {
450
+ if got , want := resp .Rate .Remaining , 59 ; got != want {
462
451
t .Errorf ("Client rate remaining = %v, want %v" , got , want )
463
452
}
464
453
reset := time .Date (2013 , 7 , 1 , 17 , 47 , 53 , 0 , time .UTC )
465
- if client .Rate () .Reset .UTC () != reset {
466
- t .Errorf ("Client rate reset = %v, want %v" , client .Rate () .Reset , reset )
454
+ if resp .Rate .Reset .UTC () != reset {
455
+ t .Errorf ("Client rate reset = %v, want %v" , resp .Rate .Reset , reset )
467
456
}
468
457
}
469
458
@@ -764,35 +753,6 @@ func TestError_Error(t *testing.T) {
764
753
}
765
754
}
766
755
767
- func TestRateLimit (t * testing.T ) {
768
- setup ()
769
- defer teardown ()
770
-
771
- mux .HandleFunc ("/rate_limit" , func (w http.ResponseWriter , r * http.Request ) {
772
- if m := "GET" ; m != r .Method {
773
- t .Errorf ("Request method = %v, want %v" , r .Method , m )
774
- }
775
- fmt .Fprint (w , `{"resources":{
776
- "core": {"limit":2,"remaining":1,"reset":1372700873},
777
- "search": {"limit":3,"remaining":2,"reset":1372700874}
778
- }}` )
779
- })
780
-
781
- rate , _ , err := client .RateLimit ()
782
- if err != nil {
783
- t .Errorf ("Rate limit returned error: %v" , err )
784
- }
785
-
786
- want := & Rate {
787
- Limit : 2 ,
788
- Remaining : 1 ,
789
- Reset : Timestamp {time .Date (2013 , 7 , 1 , 17 , 47 , 53 , 0 , time .UTC ).Local ()},
790
- }
791
- if ! reflect .DeepEqual (rate , want ) {
792
- t .Errorf ("RateLimit returned %+v, want %+v" , rate , want )
793
- }
794
- }
795
-
796
756
func TestRateLimits (t * testing.T ) {
797
757
setup ()
798
758
defer teardown ()
0 commit comments