diff options
| author | Daniel Worrall <Daniel.Worrall@hotmail.co.uk> | 2019-05-22 17:47:52 +0100 |
|---|---|---|
| committer | Daniel Worrall <Daniel.Worrall@hotmail.co.uk> | 2019-05-22 17:47:52 +0100 |
| commit | ffe9604fc9f9e54ffa14c99ab45b6f8e4457f6f9 (patch) | |
| tree | a0cd4da34b5cfd0e09751457fc563a31b637d58b | |
| parent | 3132621245630ac2ae345d331bb742ddeb88bfa2 (diff) | |
| download | twitch-ffe9604fc9f9e54ffa14c99ab45b6f8e4457f6f9.tar.gz twitch-ffe9604fc9f9e54ffa14c99ab45b6f8e4457f6f9.zip | |
Rate Limit Spec and less 1 letter vars
| -rw-r--r-- | spec/rate_limiter_spec.cr | 20 | ||||
| -rw-r--r-- | src/twitch/irc/connection.cr | 4 |
2 files changed, 22 insertions, 2 deletions
diff --git a/spec/rate_limiter_spec.cr b/spec/rate_limiter_spec.cr new file mode 100644 index 0000000..345aa1a --- /dev/null +++ b/spec/rate_limiter_spec.cr @@ -0,0 +1,20 @@ +require "./spec_helper"
+
+describe RateLimiter do
+ describe "#rate_limited?" do
+ it "does the thing" do
+ limiter = RateLimiter(String).new
+ limiter.bucket(:foo, 5_u32, 1.seconds)
+ limiter.rate_limited?(:foo, "z64")
+ sleep 0.9
+ 3.times do
+ limiter.rate_limited?(:foo, "z64")
+ end
+ sleep 0.2
+ 3.times do
+ limiter.rate_limited?(:foo, "z64")
+ end
+ limiter.rate_limited?(:foo, "z64").should be_truthy
+ end
+ end
+end
\ No newline at end of file diff --git a/src/twitch/irc/connection.cr b/src/twitch/irc/connection.cr index daf6a05..b2647f7 100644 --- a/src/twitch/irc/connection.cr +++ b/src/twitch/irc/connection.cr @@ -38,8 +38,8 @@ class Twitch::IRC::Connection else socket.on_message do |message| reader = FastIRC::Reader.new(IO::Memory.new(message)) - while m = reader.next - @on_message.try &.call(m) + while msg = reader.next + @on_message.try &.call(msg) end end socket.run |