baleen 0.1.1 → 0.1.2

data/lib/baleen/task/cucumber.rb

@@ -0,0 +1,44 @@
+require "baleen/task/task"
+
+module Baleen
+  module Task
+    class Cucumber < Baleen::Task::Base
+
+      include Serializable
+      include Baleen::Default
+
+      attr_reader :target_files
+
+      def initialize(opt)
+        super()
+        @params[:shell]          = opt[:shell]          || "/bin/bash"
+        @params[:opt]            = opt[:opt]            || "-c"
+        @params[:exe]            = opt[:exe]            || "bundle exec cucumber"
+        @params[:work_dir]       = opt[:work_dir]       || default_work_dir
+        @params[:files]          = opt[:files]          || default_features
+        @params[:concurrency]    = opt[:concurrency]    || default_concurrency
+        @params[:before_command] = opt[:before_command] || default_before_command
+        @params[:image]          = opt[:image]
+        @params[:command]        = opt[:command]
+        @params[:results]        = opt[:results]
+        @params[:status]         = opt[:status]
+        @params[:commit]         = nil
+      end
+
+      def prepare
+        task = Generic.new(
+          shell:    shell,
+          opt:      opt,
+          work_dir: work_dir,
+          image:    image,
+          command:  %{find #{files} | grep "\\.feature"}
+        )
+        runner = Baleen::Runner.new(task)
+        result = runner.run
+        @target_files = result[:log].split("\n")
+      end
+
+    end
+
+  end
+end

data/lib/baleen/task/generic.rb

@@ -0,0 +1,17 @@
+require "baleen/task/task"
+
+module Baleen
+  module Task
+    class Generic < Baleen::Task::Base
+
+      def initialize(opt)
+        super()
+        @params[:shell]          = opt[:shell]       ||="/bin/bash"
+        @params[:opt]            = opt[:opt]         ||="-c"
+        @params[:work_dir]       = opt[:work_dir]    ||="./"
+        @params[:image]          = opt[:image]
+        @params[:command]        = opt[:command]
+      end
+    end
+  end
+end

data/lib/baleen/task/image_update.rb

@@ -0,0 +1,22 @@
+require "baleen/task/task"
+
+module Baleen
+  module Task
+    class ImageUpdate < Base
+
+      include Serializable
+
+      def initialize(opt)
+        super()
+        @params[:shell]          = opt[:shell]       ||="/bin/bash"
+        @params[:opt]            = opt[:opt]         ||="-c"
+        @params[:image]          = opt[:image]       ||="kimh/baleen-poc"
+        @params[:command]        = opt[:command]
+        @params[:work_dir]       = opt[:work_dir]
+        @params[:files]          = "" # Without this, #start_runner raises exception. Need to think what to do.
+        @params[:concurrency]    = 1
+        @params[:commit]         = true
+      end
+    end
+  end
+end

data/lib/baleen/task/run_project.rb

@@ -0,0 +1,14 @@
+require "baleen/task/task"
+
+module Baleen
+  module Task
+    class RunProject < Baleen::Task::Base
+
+      def initialize(opt)
+        super()
+        @params[:project] = opt[:project]
+      end
+
+    end
+  end
+end

data/lib/baleen/task/task.rb

@@ -0,0 +1,5 @@
+require "baleen/task/base"
+require "baleen/task/generic"
+require "baleen/task/cucumber"
+require "baleen/task/image_update"
+require "baleen/task/run_project"

data/lib/baleen/utils/{colored_puts.rb → highlighter.rb}

@@ -3,15 +3,15 @@ require "colorize"
 module Kernel
   private
 
-  def info(msg)
+  def hl_info(msg)
     puts msg.green
   end
 
-  def notice(msg)
+  def hl_warn(msg)
     puts msg.yellow
   end
 
-  def warning(msg)
+  def hl_error(msg)
     puts msg.red
   end
 end

data/lib/baleen/utils/logger.rb

@@ -0,0 +1,74 @@
+require 'singleton'
+require 'colorize'
+require 'logger'
+
+module Baleen
+
+  # End of Log
+  class EoL
+    def initialize(proc)
+      @proc = proc
+    end
+
+    def eol
+      @proc.call
+    end
+  end
+
+  ERROR   = Logger::ERROR
+  WARN    = Logger::WARN
+  INFO    = Logger::INFO
+  DEBUG   = Logger::DEBUG
+
+  class BL
+    include Singleton
+
+    attr_reader :log
+
+    def initialize
+      config = Baleen::Configuration.instance
+
+      if config.debug == DEBUG
+        device = STDOUT
+      else
+        dir = File.join(config.dir, "log")
+        FileUtils.mkdir_p dir
+        device = File.join(dir, "baleen.log")
+        Celluloid.logger = nil
+      end
+      @log = Logger.new(device)
+      @log.level = config.log_level
+    end
+
+    def self.error(msg)
+      instance.log.error(msg.red)
+      EoL.new(Proc.new{instance.log.error("")})
+    end
+
+    def self.warn(msg)
+      instance.log.warn(msg.yellow)
+      EoL.new(Proc.new{instance.log.warn("")})
+    end
+
+    def self.info(msg)
+      instance.log.info(msg)
+      EoL.new(Proc.new{instance.log.info("")})
+    end
+
+    def self.debug(msg)
+      instance.log.debug(msg.magenta)
+      EoL.new(Proc.new{instance.log.debug("")})
+    end
+
+    def self.notice(msg)
+      instance.log.info(msg.green)
+      EoL.new(Proc.new{instance.log.info("")})
+    end
+
+    def puts(msg)
+      log.info(msg)
+    end
+
+  end
+end
+

data/lib/baleen/validator.rb

@@ -0,0 +1,115 @@
+require "yaml"
+
+module Baleen
+  module Validation
+    class Validator
+
+      include Baleen::Serializable
+
+      def self.check(project)
+        sections = [:runner, :framework, :ci]
+
+        sections.each do |sect|
+          validator = Baleen::Validation.const_get(sect.to_s.capitalize)
+          unless validator.new(project).validate
+            return false
+          end
+        end
+      end
+    end
+
+    class Common
+      def initialize(yaml)
+        @section = self.class.to_s.split("::").last.downcase.to_sym
+        @project  = yaml[@section]
+      end
+
+      def attributes
+        mandatory_attributes + optional_attributes
+      end
+
+      def validate
+        unless @project
+          hl_error "Your baleen.yml is missing the following mandatory section"
+          hl_warn  " :#{@section}"
+          raise Baleen::Error::Validator::MandatoryMissing
+        end
+
+        mandatory = mandatory_attributes
+        @project.keys.each do |k|
+          mandatory.delete k
+          unless attributes.include? k
+            hl_error "Your baleen.yml has the following invalid attribute at :#{@section} section"
+            hl_warn " :#{k}"
+            return false
+          end
+        end
+
+        unless mandatory.empty?
+          hl_error "Following attributes are mandatory at :#{@section} section of baleen.yml"
+          mandatory.each {|m| hl_warn " :#{m}"}
+          raise Baleen::Error::Validator::MandatoryMissing
+        end
+
+        true
+      end
+
+    end
+
+    class Runner < Common
+      def mandatory_attributes
+        [
+          :image,
+        ]
+      end
+
+      def optional_attributes
+        [
+          :work_dir,
+          :concurrency,
+          :before_command,
+        ]
+      end
+
+    end
+
+    class Framework < Common
+
+      def mandatory_attributes
+        [
+          :type,
+        ]
+      end
+
+      def optional_attributes
+        [
+          target_files,
+        ]
+      end
+
+      private
+
+      def target_files
+        case @project[:type]
+          when "cucumber"; :features
+        end
+      end
+    end
+
+    class Ci < Common
+      def mandatory_attributes
+        [
+          :url,
+          :repo,
+        ]
+      end
+
+      def optional_attributes
+        [
+          :branch,
+          :build,
+        ]
+      end
+    end
+  end
+end

data/lib/baleen/version.rb

@@ -1,3 +1,3 @@
 module Baleen
-  VERSION = "0.1.1"
+  VERSION = "0.1.2"
 end

data/lib/baleen.rb

@@ -1,13 +1,21 @@
 require "baleen/version"
 
+require 'baleen/default'
+require 'baleen/configuration'
+require 'baleen/utils/logger'
+require 'baleen/serializable'
 require 'baleen/client'
-require 'baleen/message/message'
+require 'baleen/task/task'
 require 'baleen/task'
 require 'baleen/job'
-
 require 'baleen/server'
-require 'baleen/container'
 require 'baleen/runner'
+require 'baleen/ci_server'
+require 'baleen/connection'
+require 'baleen/message/message'
+require 'baleen/validator'
+require 'baleen/project'
+require 'baleen/builder'
 
 module Baleen
   # Your code goes here...

data/spec/lib/baleen/spec_result_display.rb

@@ -0,0 +1,47 @@
+require_relative '../../spec_helper'
+
+describe Baleen::ResultDisplay do
+  describe "#summary" do
+    context "when all test passes" do
+      it "should display Pass" do
+        result = [
+          {
+            "status_code" => 0,
+            "container_id" => "aaaaaa",
+            "log" => "brabrabra",
+            "file" => "feature/t1.feature"
+          },
+          {
+            "status_code" => 0,
+            "container_id" => "bbbbbb",
+            "log" => "brabrabra",
+            "file" => "feature/t2.feature"
+          }
+        ]
+        displayer = Baleen::ResultDisplay.new(result, Time.now, Time.now+10)
+        capture(:stdout) { displayer.summary }.should include 'Pass'
+      end
+    end
+
+    context "when some test fails" do
+      it "should display Fail" do
+         result = [
+          {
+            "status_code" => 0,
+            "container_id" => "aaaaaa",
+            "log" => "brabrabra",
+            "file" => "feature/t1.feature"
+          },
+          {
+            "status_code" => 1,
+            "container_id" => "bbbbbb",
+            "log" => "brabrabra",
+            "file" => "feature/t2.feature"
+          }
+        ]
+        displayer = Baleen::ResultDisplay.new(result, Time.now, Time.now+10)
+        capture(:stdout) { displayer.summary }.should include 'Fail'
+      end
+    end
+  end
+end

data/spec/lib/baleen/spec_serializable.rb

@@ -0,0 +1,45 @@
+require_relative '../../spec_helper'
+
+describe "serialization" do
+  context Baleen::Task::Generic do
+    it "should be serializable" do
+      task = Baleen::Task::Generic.new({})
+      json = task.to_json
+      deserialized = Baleen::Serializable.deserialize(json)
+
+      expect(deserialized.class).to eq Baleen::Task::Generic
+    end
+
+  end
+
+  context Baleen::Task::Cucumber do
+    it "should be serializable" do
+      task = Baleen::Task::Cucumber.new({})
+      json = task.to_json
+      deserialized = Baleen::Serializable.deserialize(json)
+
+      expect(deserialized.class).to eq Baleen::Task::Cucumber
+    end
+  end
+
+  context Baleen::Task::ImageUpdate do
+    it "should be serializable" do
+      task = Baleen::Task::ImageUpdate.new({})
+      json = task.to_json
+      deserialized = Baleen::Serializable.deserialize(json)
+
+      expect(deserialized.class).to eq Baleen::Task::ImageUpdate
+    end
+  end
+
+  context Baleen::Message::ToClient do
+    it "should be serializable" do
+      task = Baleen::Message::ToClient.new({})
+      json = task.to_json
+      deserialized = Baleen::Serializable.deserialize(json)
+
+      expect(deserialized.class).to eq Baleen::Message::ToClient
+    end
+  end
+
+end

data/spec/lib/baleen/spec_task.rb

@@ -0,0 +1,62 @@
+require_relative '../../spec_helper'
+
+include Baleen::Task
+
+describe Baleen::Task do
+
+  describe Generic do
+    it "executes command" do
+      string = "OK"
+      task = Baleen::Task::Generic.new(
+        image: base_image,
+        command: "echo #{string}",
+      )
+      runner = Baleen::Runner.new(task)
+      result = runner.run
+      expect(result[:log]).to include string
+    end
+  end
+
+  describe Cucumber do
+    it "runs cucumber test" do
+      task = Baleen::Task::Cucumber.new(
+        image: base_image,
+        work_dir: "/poc",
+        files: "features/t0.feature",
+        before_command: "source /etc/profile",
+        concurrency: 1,
+      )
+      runner = Baleen::Runner.new(task)
+      result = runner.run
+      expect(result[:log]).to include "Scenario"
+    end
+  end
+
+  describe ImageUpdate do
+    before :each do
+      Docker::Image.all.each do |i|
+        if i.json["container_config"]["Image"] == test_image
+          i.remove
+        end
+      end
+      @image = Docker::Image.build("from #{base_image}")
+      @image.tag('repo' => test_image, 'force' => true)
+    end
+
+    it "commit changes" do
+      before_id = @image.json["id"]
+      task = Baleen::Task::ImageUpdate.new(
+        image: test_image,
+        command: "touch ./new_file.txt",
+      )
+
+      runner = Baleen::Runner.new(task)
+      runner.run
+      @image = Docker::Image.build("from #{test_image}")
+      after_id = @image.json["id"]
+
+      expect(after_id).not_to eq(before_id)
+    end
+  end
+
+end

data/spec/lib/baleen/spec_validator.rb

@@ -0,0 +1,53 @@
+require_relative '../../spec_helper'
+
+include Baleen::Config
+
+describe Baleen::Config do
+  context "when all mandatory attributes are given" do
+    it "should pass the check" do
+      config = {
+        :base => {
+          :baleen_server=>"127.0.0.1",
+        },
+        :runner => {
+          :image=>"kimh/baleen-poc",
+        },
+        :framework => {
+          :type => "cucumber"
+        }
+      }
+      expect(Baleen::Validation::Validator.check(config)).to be_true
+    end
+  end
+
+  context "when invalid attributes are given" do
+    it "should not pass the check" do
+      config = {
+        :base => {
+          :baleen_server=>"127.0.0.1",
+        },
+        :runner => {
+          :image=>"kimh/baleen-poc",
+        },
+        :framework => {
+          :type => "cucumber",
+          :bad =>"MJ", # This is invalid
+        }
+      }
+      expect(Baleen::Validation::Validator.check(config)).to be_false
+    end
+  end
+
+  context "when mandatory attributes are not given" do
+    it "should raise Baleen::Error::Validator::MandatoryMissing" do
+      config = {
+        :base => {
+          #:baleen_server is missing
+          :port=>5533,
+        }
+      }
+      expect{Baleen::Validation::Validator.check(config)}.to raise_error Baleen::Error::Validator::MandatoryMissing
+    end
+  end
+
+end

data/spec/spec_helper.rb

@@ -0,0 +1,31 @@
+require "rspec"
+require 'stringio'
+require File.expand_path('../../lib/baleen.rb', __FILE__)
+
+RSpec.configure do |config|
+  unless Docker.url = ENV["docker_url"]
+    hl_error "You have to set 'docker_url' environment variable before running test
+    Ex: export docker_url=\"http://192.168.56.4:4243\""
+    exit 1
+  end
+end
+
+def base_image
+  ENV["test_base"] ||= "baleen/test_base"
+end
+
+def test_image
+  "baleen/#{example.description.gsub("\s", "-")}"
+end
+
+def capture(stream)
+  begin
+    stream = stream.to_s
+    eval "$#{stream} = StringIO.new"
+    yield
+    result = eval("$#{stream}").string
+  ensure
+    eval "$#{stream} = #{stream.upcase}"
+  end
+  result
+end