dr-rubbis 0.0.0

data/spec/spec_helper.rb

@@ -0,0 +1,88 @@
+require 'redis'
+
+require 'rubbis/server'
+
+module AcceptanceHelpers
+	def client
+		Redis.new(host: 'localhost', port: TEST_PORT)
+	end
+
+	def with_server(opts = {})
+		opts[:server_file] = opts.fetch(:server_file, true)
+		server = nil
+		server_thread = Thread.new do
+			begin
+				server = Rubbis::Server.new(
+					port: TEST_PORT,
+					server_file: opts[:server_file] ? test_db : nil,
+					aof_file: opts[:aof_file] ? aof_test_db : nil
+					)
+				server.listen
+			rescue
+				puts "[ERRS]"
+				puts $!
+				puts $!.backtrace
+			end
+		end
+
+		wait_for_open_port TEST_PORT
+
+		yield
+	rescue TimeoutError, Redis::CannotConnectError
+		server_thread.value unless server_thread.alive?
+
+		raise
+	ensure
+		server.shutdown if server
+	end
+
+	def wait_for_open_port(port)
+		time = Time.now
+		while !check_port(port) && time > Time.now - 3
+			sleep 0.01
+		end
+		raise TimeoutError unless check_port(port)
+	end
+
+	def check_port(port)
+		`nc -z localhost #{port}`
+		res = $?.success?
+		puts $?.inspect unless res
+		res
+	end	
+
+
+end
+
+class FakeClock
+	def initialize
+		@t = 0
+	end
+
+	def now
+		@t
+	end
+
+	def sleep(seconds)
+		@t += seconds
+	end
+end
+
+TEST_PORT = 6380
+
+RSpec.configure do |c|
+	def test_db
+		Dir.tmpdir + '/rubbis_test.dump'
+	end
+
+	def aof_test_db
+		Dir.tmpdir + '/rubbis_test.aof'
+	end
+
+	c.before(:each) do
+		FileUtils.rm_f(test_db)
+	end
+
+
+	c.include AcceptanceHelpers, acceptance: true
+end

data/spec/unit/protocol_spec.rb

@@ -0,0 +1,20 @@
+require 'spec_helper'
+require 'rubbis/protocol'
+
+describe Rubbis::Protocol, :unit do
+	describe '.marshal' do
+		def self.it_marshals(ruby, wire)
+			specify ruby.inspect do
+				expect(described_class.marshal ruby).to eq(wire)
+			end
+		end
+
+		it_marshals :ok , "+OK\r\n"
+		it_marshals nil, "$-1\r\n"
+		it_marshals 5, ":5\r\n"
+		it_marshals "hello", "$5\r\nhello\r\n"
+		it_marshals ["a","bc"], "*2\r\n$1\r\na\r\n$2\r\nbc\r\n"
+		it_marshals Rubbis::Error.incorrect_args('cmd'),\
+			"-ERR wrong number of arguments for 'cmd' command\r\n"
+	end
+end

data/spec/unit/state_spec.rb

@@ -0,0 +1,238 @@
+require 'spec_helper'
+require 'rubbis/state'
+
+describe Rubbis::State, :unit do
+	let(:clock) {FakeClock.new}
+	let(:state) {described_class.new(clock)}
+
+	shared_examples "passive expiry" do |set_cmd, get_cmd|
+		it 'expires a key passively' do
+			key = "abc"
+			
+			set_cmd.call(state, key)
+			state.expire(key, "1")
+
+			clock.sleep 0.9
+			get_cmd.call(state,key)
+			expect(state.exists(key)).to eq(1)
+
+			clock.sleep 0.1
+			get_cmd.call(state,key)
+			expect(state.exists(key)).to eq(0)
+		end
+	end
+
+	describe '#set' do 
+		it 'sets a value' do
+			expect(state.set("abc", "123")).to eq(:ok)
+			expect(state.get("abc")).to eq("123")
+		end
+		it 'does not overwrite n existing value with NX' do
+			expect(state.set("abc", "123", "NX")).to eq(:ok)
+			expect(state.set("abc", "456", "NX")).to eq(nil)
+			expect(state.get("abc")).to eq("123")
+		end
+
+		it 'doen not set a new value with XX' do
+			expect(state.set("abc", "123", "XX")).to eq(nil)
+			state.set("abc", "123")
+			expect(state.set("abc", "456", "XX")).to eq(:ok)
+			expect(state.get("abc")).to eq("456")
+		end
+
+		it 'raises error when wrong number of arguments' do
+			expect(state.set("abc")).to \
+			eq(Rubbis::Error.incorrect_args('set'))
+		end
+	end
+
+	describe '#hset' do
+		it 'sets a value' do 
+			expect(state.hset("myhash", "abc", "123")).to eq(:ok)
+			expect(state.hset("otherhash", "abc", "456")).to eq(:ok)
+			expect(state.hget("myhash", "abc")).to eq("123")
+		end
+	end
+
+	describe "#hget" do
+		include_examples 'passive expiry',
+		-> s,k { s.hset(k, "abc", "123")},
+		-> s,k { s.hincrby(k,"abc", "1")}
+	end
+
+	describe "#hmget" do
+		it 'gets multiple values' do
+			expect(state.hset("myhash", "abc", "123")).to eq(:ok)
+			expect(state.hset("myhash", "def", "456")).to eq(:ok)
+			expect(state.hmget("myhash", "abc", "def")).to eq(["123", "456"])
+		end
+
+		it 'returns error when not hash value' do
+			state.set("myhash", "bogus")
+			expect(state.hmget("myhash", "key")).to \
+			eq(Rubbis::Error.type_error)
+		end
+
+		it 'returns nils when empty' do
+			expect(state.hmget("myhash", "key")).to eq([nil])
+		end
+
+		include_examples 'passive expiry',
+		-> s,k { s.hset(k,"abc", "123")},
+		-> s,k { s.hmget(k, "abc")}
+	end
+
+	describe "#hincrby" do
+		it 'increments a counter stored in a hash' do
+			state.hset("myhash", "abc", "123")
+			expect(state.hincrby("myhash", "abc", "2")).to eq(125)
+		end
+
+		include_examples 'passive expiry',
+		-> s,k { s.hset(k, "abc", "123")},
+		-> s,k { s.hincrby(k,"abc", "1")}
+	end
+
+
+	describe '#get' do
+		include_examples 'passive expiry',
+		-> s,k { s.set(k,"123") },
+		-> s,k { s.get(k) }
+	end
+
+	describe "#exists" do
+		it 'returns the number of keys existing' do
+			expect(state.exists("abc")).to eq(0)
+			state.set("abc", "123")
+			expect(state.exists("abc")).to eq(1)
+		end
+	end 
+
+	describe '#keys' do
+		it 'returns all keys in the system for *' do
+			state.set("abc", "123")
+			state.set("def", "456")
+			expect(state.keys('*')).to eq(%w(abc def))
+		end
+	end
+
+	describe 'sorted sets' do
+		it 'fetches keys by rank' do
+			set_name = 'leaderboard'
+			state.zadd(set_name, "1000", "a")
+			state.zadd(set_name, "3000", "b")
+			state.zadd(set_name, "2000", "c")
+			expect(state.zrange(set_name, "0", "1")).to \
+			eq(%w(a c))
+		end
+
+		it 'fetches rank by member' do
+			set_name = 'leaderboard'
+			state.zadd(set_name, "1000", "a")
+			state.zadd(set_name, "3000", "b")
+			state.zadd(set_name, "2000", "c")
+			expect(state.zrank(set_name, "c")).to \
+			eq(1)
+		end
+
+		it 'fetches rank by member' do
+			set_name = 'leaderboard'
+			state.zadd(set_name, "1000", "a")
+			state.zadd(set_name, "3000", "b")
+			state.zadd(set_name, "2000", "c")
+			expect(state.zscore(set_name, "c")).to \
+			eq(2000)
+		end
+
+		it 'breaks ties based on score' do
+			set_name = 'leaderboard'
+			state.zadd(set_name, "1000", "a")
+			state.zadd(set_name, "1000", "b")
+			state.zadd(set_name, "1000", "c")
+			expect(state.zrange(set_name, "0", "1")).to \
+			eq(%w(a b))
+		end
+	end
+
+	describe 'lists' do
+		it 'supports basic operations' do
+			state.lpush("q", "a")
+			state.lpush("q", "b")
+			state.lpush("q", "c")
+
+			expect(state.llen("q")).to eq(3)
+			expect(state.lrange("q", "0","1")).to eq(%w(c b))
+
+			expect(state.rpop("q")).to eq("a")
+			expect(state.llen("q")).to eq(2)
+		end
+
+		it 'support rpoplpush' do
+			state.lpush("q", "a")
+			state.lpush("q", "b")
+			state.lpush("q", "c")
+
+			expect(state.rpoplpush("q", "p")).to eq("a")
+			expect(state.lrange("q", "0", "-1")).to eq(%w(c b))
+			expect(state.lrange("p", "0", "-1")).to eq(%w(a))
+		end
+
+		it 'support cyclic rpoplpush' do
+			state.lpush("q", "a")
+			state.lpush("q", "b")
+			state.lpush("q", "c")
+
+			expect(state.rpoplpush("q", "q")).to eq("a")
+			expect(state.lrange("q", "0", "-1")).to eq(%w(a c b))
+		end
+	end
+
+	describe '#pexpireat' do
+		it 'sets absolute expiry' do
+			state.set("abc", "123")
+			state.pexpireat("abc", "1000")
+			clock.sleep(0.9)
+			expect(state.get("abc")).to eq("123")
+			clock.sleep(0.1)
+			expect(state.get("abc")).to eq(nil)
+		end
+	end
+
+	describe '#minimal_log' do
+		it 'handles string values' do
+			state.set('abc', '123')
+			state.set('abc', '123')
+			state.set('abc', '456')
+			expect(state.minimal_log).to eq([%w(set abc 456)])
+		end
+
+		it 'handles lists' do
+			state.lpush('abc', '123')
+			state.lpush('abc', '456')
+			state.lpush('abc', '789')
+			state.rpop('abc')
+			expect(state.minimal_log).to eq([
+				%w(lpush abc 456),
+				%w(lpush abc 789),
+				])	
+		end
+
+		it 'handles expiry' do
+			state.set("abc", "123")
+			state.expire("abc", "1")
+
+			expect(state.minimal_log).to eq([
+				%w(set abc 123),
+				%w(pexpireat abc 1000),
+				])
+		end
+	end
+end
+
+
+
+
+
+
+
+

metadata

@@ -0,0 +1,65 @@
+--- !ruby/object:Gem::Specification
+name: dr-rubbis
+version: !ruby/object:Gem::Version
+  version: 0.0.0
+platform: ruby
+authors:
+- Dror Nir
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2017-08-06 00:00:00.000000000 Z
+dependencies: []
+description: A ruby redis server based on online tutorial
+email: dror.nir0@gmail.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- Gemfile
+- Gemfile.lock
+- README.md
+- bin/rspec
+- bin/server
+- lib/dr_rubbis.rb
+- lib/rubbis/handler.rb
+- lib/rubbis/protocol.rb
+- lib/rubbis/server.rb
+- lib/rubbis/state.rb
+- lib/rubbis/transaction.rb
+- lib/rubbis/zset.rb
+- spec/acceptance/expiry_spec.rb
+- spec/acceptance/hash_spec.rb
+- spec/acceptance/list_spec.rb
+- spec/acceptance/persistence_spec.rb
+- spec/acceptance/pubsub_spec.rb
+- spec/acceptance/skeleton_spec.rb
+- spec/acceptance/transactions_spec.rb
+- spec/spec_helper.rb
+- spec/unit/protocol_spec.rb
+- spec/unit/state_spec.rb
+homepage: ''
+licenses:
+- MIT
+metadata: {}
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 2.4.6
+signing_key: 
+specification_version: 4
+summary: Mock redis server
+test_files: []