wakame-dolphin 0.0.2

data/spec/files/dolphin_spec.rb

@@ -0,0 +1,110 @@
+# -*- coding: utf-8 -*-
+
+require 'rubygems'
+require 'spec_helper'
+require 'net/http'
+
+require File.join(File.expand_path('../../../', __FILE__), 'lib/dolphin')
+
+describe 'Notification Service' do
+
+  describe Dolphin::RequestHandler do
+    context "リクエストパラメータを入力したとき" do
+      it "リクエストに成功すること" do
+      end
+
+      describe "リクエストに失敗すること" do
+      end
+    end
+  end
+
+  describe Dolphin::Manager do
+    it "開始できること" do
+    end
+
+    it "停止できること" do
+    end
+
+    it "Workerを起動できること" do
+    end
+
+    it "Workerを停止できること" do
+    end
+
+    it "RequestHandlerを起動できること" do
+    end
+
+    it "RequestHandlerを停止できること" do
+    end
+
+  end
+
+  describe Dolphin::Worker do
+
+    context "EventObjectを入力したとき" do
+      it "Senderにメッセージが送れること" do
+      end
+    end
+  end
+
+  describe Dolphin::QueryProcessor do
+
+    it "宛先情報を取得できること" do
+    end
+
+    it "イベントを記録できること" do
+    end
+  end
+
+  describe Dolphin::Sender do
+
+    context "ファイル出力を選択したとき" do
+      it "ファイルに保存できること" do
+      end
+    end
+
+    context "メールサーバーが設定されているとき", :smtp => true  do
+      it "送信できること" do
+      end
+    end
+
+  end
+
+  describe Dolphin::DataBase do
+
+    context "設定ファイルを入力したとき" do
+      it "Cassandraデータベースに接続できること" do
+      end
+    end
+
+    context "Cassandraデータベースに接続できたとき" do
+
+      it "Eventを保存できること" do
+      end
+
+      it "Eventを取得できること" do
+      end
+
+      it "Notificationを保存できること" do
+      end
+
+      it "Notificationを取得できること" do
+      end
+
+    end
+
+  end
+
+  describe Dolphin::Util do
+    context "ログタイプにstdoutを選択したとき" do
+      it "ログが書き込めること" do
+      end
+    end
+
+    context "ログタイプにltsvを選択したとき" do
+      it "ログが書き込めること" do
+      end
+    end
+  end
+
+end

data/spec/files/endpoint/event_spec.rb

@@ -0,0 +1,123 @@
+# -*- coding: utf-8 -*-
+
+require 'spec_helper'
+
+describe 'Event API' do
+
+  before(:all) do
+
+    # Mail Setting
+    output_mail_file_location = '/var/tmp/'
+
+    @mail_to = "test_to@example.com"
+    @temporary_mail = "#{output_mail_file_location}#{@mail_to}"
+
+    @connection = Dolphin::DataStore.current_store
+    @connection.connect
+
+    if @connection.closed?
+      pending "Failed cannect to database"
+    else
+      @notification_id = 'system'
+      @notification_values = {
+        "email"=> {
+          "to" => @mail_to,
+        }
+      }
+      @notification_id = 'test'
+      @connection.put_notification(@notification_id, @notification_values)
+
+      @message_type = "default"
+      @event_values = {
+        'notification_id' => @notification_id,
+        'message_type' => @message_type,
+        'messages' => {
+          "message"=>"Alert!!!!"
+        }
+      }
+
+      @connection.put_event({:messages => @event_values})
+    end
+  end
+
+  it 'expect to get events' do
+    res = get('/events', :headers => {
+      'Content-Type' =>'application/json',
+    })
+    expect(res['message']).to eql 'OK'
+  end
+
+  it 'expect to get last event' do
+    events = get('/events', :headers => {
+      'Content-Type' =>'application/json',
+    })
+    last_event = events['results'][-1]
+    last_event_id = last_event['id']
+
+    query_string = '?' + [
+      "count=1",
+      "start_id=#{SimpleUUID::UUID.new(last_event_id).to_guid}"
+    ].join('&')
+
+    res = get('/events' + query_string, :headers => {
+      'Content-Type' =>'application/json',
+    })['results'][0]
+    expect(res['id']).to eql last_event_id
+  end
+
+  it 'expect to post event and send email' do
+    message_subject = 'subject'
+    message_body = 'body'
+    res = post('/events',
+      :headers => {
+        'Content-Type' =>'application/json',
+        'X-Notification-Id' => @notification_id,
+        'X-Message-Type' => @message_type
+      },
+      :body => {
+        'subject' => message_subject,
+        'body' => message_body,
+        'message' => 'Alert!!!!'
+    }.to_json)
+
+    expect(res['message']).to eql 'OK'
+
+    wait_for_sending_mail_to_be_ready {
+      break if File.exists?(@temporary_mail)
+    }
+
+    mb = Dolphin::MessageBuilder::Mail.new
+    template = mb.build(@message_type, {
+      'messages' => {
+        'subject' => message_subject,
+        'body' => message_body,
+      },
+      'mail' => {
+        'from' => Dolphin.settings['mail']['from']
+      },
+      'to' => @mail_to
+    })
+
+    mail = Dolphin::Mailer.read_iso2022_jp_mail(@temporary_mail)
+    expect(mail[:from]).to eql Dolphin.settings['mail']['from']
+    expect(mail[:to]).to eql @mail_to
+    expect(mail[:subject]).to eql template.subject
+    expect(mail[:body]).to eql template.body
+  end
+
+  it 'expect to post event only' do
+    res = post('/events',
+      :headers => {
+        'Content-Type' =>'application/json',
+      },
+      :body => {
+        'message' => 'Alert!!!!'
+    }.to_json)
+
+    expect(res['message']).to eql 'OK'
+  end
+
+  before(:all) do
+    FileUtils.rm(@temporary_mail) if File.exists?(@temporary_mail)
+  end
+end

data/spec/files/endpoint/notification_spec.rb

@@ -0,0 +1,54 @@
+# -*- coding: utf-8 -*-
+
+require 'spec_helper'
+
+describe 'Notification API' do
+  before(:all) do
+
+    @connection = Dolphin::DataStore.current_store
+    @connection.connect
+    if @connection.closed?
+      pending "Failed cannect to database"
+    else
+      @notification_values = {
+        "email"=> {
+          "to" => "foo@example.com,bar@example.com",
+          "cc" => "foo@example.com,bar@example.com",
+          "bcc" =>"foofoo@example.com,barbar@example.com"
+        }
+      }
+      @notification_id = 'system'
+      @connection.put_notification(@notification_id, @notification_values)
+    end
+  end
+
+  it 'expect to get notifications' do
+    res = get('/notifications', :headers => {
+      'Content-Type' =>'application/json',
+      'X-Notification-Id' => @notification_id,
+    })
+
+    expect(res['results']).to eql @notification_values
+    expect(res['message']).to eql 'OK'
+  end
+
+  it 'expect to post notification' do
+    res = post('/notifications',
+      :headers => {
+        'Content-Type' =>'application/json',
+        'X-Notification-Id' => @notification_id
+      },
+      :body => @notification_values.to_json
+    )
+    expect(res['message']).to eql 'OK'
+  end
+
+  it 'expect to delete notification' do
+    res = delete('/notifications', :headers => {
+      'Content-Type' =>'application/json',
+      'X-Notification-Id' => @notification_id,
+    })
+    expect(res['message']).to eql 'OK'
+  end
+
+end

data/spec/files/message_builder_spec.rb

@@ -0,0 +1,15 @@
+# -*- coding: utf-8 -*-
+
+require 'spec_helper'
+
+describe Dolphin::TemplateBuilder do
+  describe ".build" do
+    it 'expect to build template' do
+      tb = Dolphin::TemplateBuilder.new
+
+      template_str = 'hello <%= @foo %>'
+      bind_params = {:foo => 'world'}
+      expect(tb.build(template_str, bind_params)).to eql 'hello world'
+    end
+  end
+end

data/spec/helpers/test_helper.rb

@@ -0,0 +1,21 @@
+module TestHelper
+  def wait_for_sending_mail_to_be_ready(timeout = 5, sleep_time = 1, &block)
+    if block_given? &block
+      waiting_count = 0
+      loop do
+        block.call
+
+        if waiting_count > timeout
+          puts "wait timeout #{timeout}"
+          break
+        end
+
+        puts "sleep #{sleep_time} wait..#{waiting_count}"
+        sleep sleep_time
+        waiting_count +=1
+      end
+    else
+      sleep sleep_time
+    end
+  end
+end

data/spec/helpers/web_request_helper.rb

@@ -0,0 +1,40 @@
+require "net/http"
+require "uri"
+
+module WebRequestHelper
+  def get(path, params)
+    uri = build_uri(path)
+    net = Net::HTTP::Get.new(uri.request_uri, params[:headers])
+    request(net, uri)
+  end
+
+  def post(path, params)
+    uri = build_uri(path)
+    net = Net::HTTP::Post.new(uri.request_uri, params[:headers])
+    net.body = params[:body]
+    request(net, uri)
+  end
+
+  def delete(path, params)
+    uri = build_uri(path)
+    net = Net::HTTP::Delete.new(uri.request_uri, params[:headers])
+    request(net, uri)
+  end
+
+  private
+
+  def build_uri(path)
+    server = Dolphin.settings['server']
+    endpoint = "http://#{server['host']}:#{server['port']}"
+    URI.parse(endpoint + path)
+  end
+
+  def request(net, uri)
+    http = Net::HTTP.new(uri.host, uri.port)
+    # http.set_debug_output $stderr
+    http.start do |h|
+      response = h.request(net)
+      JSON.parse(response.body)
+    end
+  end
+end

data/spec/spec_helper.rb

@@ -0,0 +1,18 @@
+# -*- coding: utf-8 -*-
+
+require 'rubygems'
+require 'rspec'
+require File.join(File.expand_path('../../', __FILE__), 'lib/dolphin')
+
+# Setup Bundler
+ENV['BUNDLE_GEMFILE'] ||= File.expand_path(File.dirname(__FILE__) + '/Gemfile')
+require 'bundler/setup' if File.exists?(ENV['BUNDLE_GEMFILE'])
+Bundler.require :default if defined?(Bundler)
+
+# Load helper files
+Dir.glob('./spec/helpers/*_helper.rb').each {|f| require File.expand_path(f) }
+
+RSpec.configure do |c|
+  c.include WebRequestHelper
+  c.include TestHelper
+end

data/templates/email/alert_port.erb

@@ -0,0 +1,24 @@
+監視アラート
+----------------------------------------
+
+監視アラート
+
+<%= @message %>
+
+【コンポーネントID】
+<%= @instance_name %>(<%= @instance_id %>)
+
+【IPアドレス】
+<%= @instance_ip %>
+
+【ステータス】
+異常
+
+【検知日時】
+<%= @event_datetime %>
+
+【補足】
+監視概要：監視間隔5分、応答待ち時間30秒、リトライ回数2回
+
+※このメールは送信専用アドレスから送信されています。
+　本メールにご返信いただいてもお答えできませんのでご了承下ください。

data/templates/email/default.erb

@@ -0,0 +1,3 @@
+<%= @subject %>
+----------------------------------------
+<%= @body %>

data/tests/test_dolphin

@@ -0,0 +1,10 @@
+#!/bin/bash
+#
+# Example:
+# DOLPHIN_HOST=127.0.0.1 DOLPHIN_PORT=9004 ./test_dolphin.sh
+#
+
+./test_post_notification
+./test_post_event
+./test_get_event
+./test_delete_notification

data/tests/test_get_event

@@ -0,0 +1,31 @@
+#!/usr/bin/env ruby
+# -*- coding: utf-8 -*-
+
+require 'net/http'
+require 'uri'
+require 'json'
+require 'time'
+
+start_time = URI.encode((Time.now - 60).iso8601)
+
+host = ENV['DOLPHIN_HOST'] || '127.0.0.1'
+port = ENV['DOLPHIN_PORT'] || 9004
+
+path = "http://#{host}:#{port}/events"
+# path = "http://#{host}:#{port}/events?limit=10&start_time=#{start_time}"
+# path = 'http://#{host}:#{port}/events?start_id=12174a14-87c1-11e2-927c-31035db8d436'
+
+uri = URI.parse(path)
+
+headers = {
+  'Content-Type' =>'application/json',
+}
+
+request = Net::HTTP::Get.new(uri.request_uri, headers)
+http = Net::HTTP.new(uri.host, uri.port)
+http.set_debug_output $stderr
+http.start do |h|
+  response = h.request(request)
+  res = JSON.parse(response.body)
+  p res
+end