}
let(:incorrect_hosts) { ['100','[100,100]','1:0:0:1','1/0/0/1','0.0.0.1','1.0.0.0','0.0.0.0','240.0.0.1','240.255.255.254','0.256.0.1','0.0.256.1','239.255.255.255','-1.0.0.0','0.-1.0.0','0.0.-1.0','0.0.0.-1'] }
before do
allow(regist_test).to receive(:store).and_return(true)
end
describe '#create' do
subject {regist_test.create(localhost, test_key) }
before do
regist_test.create(localhost, test_key)
end
it { is_expected.to eq(false) }
end
describe '#read_hosts' do
subject {regist_test.read_hosts }
context '正常に登録されている場合' do
it { is_expected.to eq([]) }
it 'localhost を1つ登録すると、[127.0.0.1]' do
regist_test.create(localhost, test_key)
is_expected.to eq([localhost])
end
it '複数のアドレスを登録した場合は、複数のアドレス' do
correct_hosts.each { |host|regist_test.create(host, test_key) }
is_expected.to match_array(correct_hosts)
end
end
context '不正なIPアドレスの場合' do
it '有効範囲外のIPアドレスを登録しても登録されず、空の配列' do
incorrect_hosts.each { |host|regist_test.create(host, test_key) }
is_expected.to match_array([])
end
end
end
describe '#read_hostkeys' do
it '登録しなければ、空' do
expect(regist_test.hosts).to eq([])
expect(regist_test.read_hostkeys).to eq({})
expect(regist_test.read_hostkeys.size).to eq(0)
end
it 'ハッシュ化された接続キーの一覧' do
correct_hosts.each { |host|regist_test.create(host, test_key) }
expect(regist_test.read_hostkeys.size).to eq(3)
end
end
describe '#update' do
before {regist_test.create(localhost, test_key) }
context '正常に更新できる場合' do
it '有効なIPアドレスへ変更すると host が変更' do
before_host = localhost
correct_hosts.each do |host|
expect {regist_test.update(before_host, host) }.to change {regist_test.read_hosts }.from([before_host]).to([host])
before_host = host
end
end
end
context '更新できない場合' do
it '有効範囲外のIPアドレスへ変更すると false' do
incorrect_hosts.each do |host|
expect(regist_test.update(localhost, host)).to eq(false)
end
end
it '未登録のIPアドレスを指定すると false' do
expect(regist_test.update('192.168.0.1','172.16.0.1')).to eq(false)
end
it 'すでに存在するIPアドレスへ変更すると false' do
expect(regist_test.update('192.168.0.1',localhost)).to eq(false)
end
end
end
describe '#delete' do
before {regist_test.create("192.168.0.1", test_key)}
subject {regist_test.delete(localhost) }
context '正常に削除できる場合' do
it '登録済みのアドレスを削除すると、そのアドレス' do
regist_test.create(localhost, test_key)
is_expected.to eq(localhost)
end
end
context '削除できない場合' do
it { is_expected.to eq(nil) }
end
end
end
# TaskControl class のテスト
describe 'TaskControl' do
before { @server = TCPServer.new(24368) }
after { @server.close }
let(:tasks) { Reins::TaskControl.new }
describe '#connect' do
context 'クライアントへ接続できる場合' do
subject { tasks.connect }
it 'モックを使って正常接続のコール' do
allow(tasks).to receive(:connect).and_return(true)
is_expected.to eq(true)
end
end
context 'クライアントへ接続できない場合' do
it '接続先が存在しないと Standard Error' do
# TODO: 実際は Raise ではなく、成否を True/False でもらうこととする
expect{ Reins::TaskControl.new('localhost', 65000) }.to raise_error 'Not Connect'
end
it 'クライアントが停止していたら false' do
allow(tasks).to receive(:connect).and_return(false)
expect(tasks.connect).to eq(false)
end
end
end
describe '#disconnect' do
context '正常に切断できた場合' do
subject { tasks.disconnect }
it 'こちらから、クライアントとの接続を切断して、成功すると nil' do
is_expected.to eq(nil)
end
end
end
end
# Dispatcher class のテスト
describe 'Dispatch' do
let(:test_key) { "TestKey" }
let(:correct_host) { Reins::Dispatch.new("192.168.0.10", test_key)}
describe '#command' do
context 'ホストを追加する場合' do
it 'ホストを正常に登録' do
allow(Reins::regist_host).to receive(:create).and_return(true)
expect(correct_host.command("add", "")).to eq(true)
end
end
context 'ホスト一覧を出力する場合' do
it 'ホスト一覧が出力された' do
allow(Reins::regist_host).to receive(:read_hosts).and_return([])
expect(correct_host.command("list", "")).to eq([])
end
end
context 'アドレスを更新する場合' do
it '正常に移行先アドレスへ更新' do
allow(Reins::regist_host).to receive(:update).and_return(true)
expect(correct_host.command("update", "172.16.0.1")).to eq(true)
end
end
context 'アドレスを削除する場合' do
it 'アドレスを正常に削除' do
allow(Reins::regist_host).to receive(:delete).and_return("192.168.0.10")
expect(correct_host.command("delete","")).to eq("192.168.0.10")
end
end
end
end
end
</pre]
<pre class="toolbar:1 scroll:true lang:ruby decode:true" title="＜プログラム5 lib/reins/dispatcher.rb＞"># coding: utf-8
# filename: dispatcher.rb
module Reins
class Dispatch
# 対象となるクライアントのIPアドレスを保持
# == パラメータ
# ip:: 対象となるクライアントのIPアドレス
# == 返り値
# 特になし
def initialize(ip, key)
@ip_address = ip
@keycode = key
end
# コマンドを受け取って、対象の機能へ振り分ける
# == パラメータ
# comm:: クライアントからの要求コマンド
# value:: コマンドの実行に必要な引数
# == 返り値
# exception:: 失敗した場合
# exception以外:: コマンドの実行結果
def command(comm, value)
case comm
when /^add/ then
Reins::logger.info("#{@ip_address} を追加するよう指示がありました")
Reins::regist_host.create(@ip_address, @key)
when /^list/ then
Reins::logger.info("一覧表示コマンドを受け付けました")
Reins::regist_host.read_hosts
when /^update/ then
Reins::logger.info("IPアドレス更新依頼がありました")
Reins::regist_host.update(@ip_address, value)
when /^delete/ then
Reins::logger.info("削除依頼が発生しました")
Reins::regist_host.delete(@ip_address)
end
end
end
end
