mortar 0.1.0

data/spec/mortar/command/auth_spec.rb

@@ -0,0 +1,46 @@
+#
+# Copyright 2012 Mortar Data Inc.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+#
+# Portions of this code from heroku (https://github.com/heroku/heroku/) Copyright Heroku 2008 - 2012,
+# used under an MIT license (https://github.com/heroku/heroku/blob/master/LICENSE).
+#
+
+require "spec_helper"
+require "mortar/command/auth"
+
+describe Mortar::Command::Auth do
+  
+  describe "auth:key" do
+    it "displays the user's api key" do
+      mock(Mortar::Auth).password {"foo_api_key"}
+      stderr, stdout = execute("auth:key")
+      stderr.should == ""
+      stdout.should == <<-STDOUT
+foo_api_key
+STDOUT
+    end
+  end
+    
+  describe "auth:whoami" do
+    it "displays the user's email address" do
+      mock(Mortar::Auth).user {"sam@mortardata.com"}
+      stderr, stdout = execute("auth:whoami")
+      stderr.should == ""
+      stdout.should == <<-STDOUT
+sam@mortardata.com
+STDOUT
+    end
+  end
+end

data/spec/mortar/command/base_spec.rb

@@ -0,0 +1,82 @@
+#
+# Copyright 2012 Mortar Data Inc.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+#
+# Portions of this code from heroku (https://github.com/heroku/heroku/) Copyright Heroku 2008 - 2012,
+# used under an MIT license (https://github.com/heroku/heroku/blob/master/LICENSE).
+#
+
+require "spec_helper"
+require "mortar/command/base"
+
+module Mortar::Command
+  describe Base do
+    before do
+      @base = Base.new
+      stub(@base).display
+      @client = Object.new
+      stub(@client).host {'mortar.com'}
+    end
+    
+    context "error message context" do
+      it "get context for missing parameter error message" do
+        message = "Undefined parameter : INPUT"
+        @base.get_error_message_context(message).should == "Use -p, --parameter NAME=VALUE to set parameter NAME to value VALUE."
+      end
+      
+      it "get context for unhandled error message" do
+        message = "special kind of error"
+        @base.get_error_message_context(message).should == ""
+      end
+    end
+
+    context "detecting the project" do
+      it "read remotes from git config" do
+        stub(Dir).chdir
+        stub(@base.git).has_dot_git? {true}
+        mock(@base.git).git("remote -v").returns(<<-REMOTES)
+staging\tgit@github.com:mortarcode/4dbbd83cae8d5bf8a4000000_myproject-staging.git (fetch)
+staging\tgit@github.com:mortarcode/4dbbd83cae8d5bf8a4000000_myproject-staging.git (push)
+production\tgit@github.com:mortarcode/4dbbd83cae8d5bf8a4000000_myproject.git (fetch)
+production\tgit@github.com:mortarcode/4dbbd83cae8d5bf8a4000000_myproject.git (push)
+other\tgit@github.com:other.git (fetch)
+other\tgit@github.com:other.git (push)
+        REMOTES
+
+        @mortar = Object.new
+        stub(@mortar).host {'mortar.com'}
+        stub(@base).mortar { @mortar }
+
+        # need a better way to test internal functionality
+        @base.git.send(:remotes, 'mortarcode').should == { 'staging' => 'myproject-staging', 'production' => 'myproject' }
+      end
+
+      it "gets the project from remotes when there's only one project" do
+        stub(@base.git).has_dot_git? {true}
+        stub(@base.git).remotes {{ 'mortar' => 'myproject' }}
+        mock(@base.git).git("config mortar.remote", false).returns("")
+        @base.project.name.should == 'myproject'
+      end
+
+      it "accepts a --remote argument to choose the project from the remote name" do
+        stub(@base.git).has_dot_git?.returns(true)
+        stub(@base.git).remotes.returns({ 'staging' => 'myproject-staging', 'production' => 'myproject' })
+        stub(@base).options.returns(:remote => "staging")
+        @base.project.name.should == 'myproject-staging'
+      end
+
+    end
+
+  end
+end

data/spec/mortar/command/clusters_spec.rb

@@ -0,0 +1,61 @@
+#
+# Copyright 2012 Mortar Data Inc.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+#
+
+require 'spec_helper'
+require 'mortar/command/clusters'
+
+module Mortar::Command
+  describe Clusters do
+    
+    before(:each) do
+      stub_core
+    end
+        
+    context("index") do
+      
+      it "lists running and recently started clusters" do
+        # stub api request
+        clusters = [{"cluster_id" => "50fbe5a23004292547fc2224", 
+                     "size" => 2,
+                     "status_description" => "Running",
+                     "start_timestamp" => "2012-08-27T21:27:15.669000+00:00",
+                     "duration" => "2 mins"},
+                    {"cluster_id" => "50fbe5a23004292547fc2225", 
+                                 "size" => 10,
+                                 "status_description" => "Shut Down",
+                                 "start_timestamp" => "2011-08-27T21:27:15.669000+00:00",
+                                 "duration" => "20 mins"}]
+         mock(Mortar::Auth.api).get_clusters().returns(Excon::Response.new(:body => {"clusters" => clusters}))
+         stderr, stdout = execute("clusters", nil, nil)
+         stdout.should == <<-STDOUT
+cluster_id                Size (# of Nodes)  Status     Type  Start Timestamp                   Elapsed Time
+------------------------  -----------------  ---------  ----  --------------------------------  ------------
+50fbe5a23004292547fc2224                  2  Running          2012-08-27T21:27:15.669000+00:00  2 mins
+50fbe5a23004292547fc2225                 10  Shut Down        2011-08-27T21:27:15.669000+00:00  20 mins
+STDOUT
+      end
+
+      it "handles no clusters running" do
+        mock(Mortar::Auth.api).get_clusters().returns(Excon::Response.new(:body => {"clusters" => []}))
+         stderr, stdout = execute("clusters", nil, nil)
+         stdout.should == <<-STDOUT
+cluster_id  Size (# of Nodes)  Status  Type  Start Timestamp  Elapsed Time
+----------  -----------------  ------  ----  ---------------  ------------
+STDOUT
+      end
+    end
+  end
+end

data/spec/mortar/command/describe_spec.rb

@@ -0,0 +1,135 @@
+#
+# Copyright 2012 Mortar Data Inc.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+#
+
+require 'spec_helper'
+require 'fakefs/spec_helpers'
+require 'mortar/command/describe'
+require 'mortar/api/describe'
+require 'launchy'
+
+module Mortar::Command
+  describe Describe do
+    
+    before(:each) do
+      stub_core      
+      @git = Mortar::Git::Git.new
+    end
+        
+    context("index") do
+      
+      it "errors when an alias is not provided" do
+        with_git_initialized_project do |p|
+          write_file(File.join(p.pigscripts_path, "my_script.pig"))
+          stderr, stdout = execute("describe my_script", p)
+          stderr.should == <<-STDERR
+ !    Usage: mortar describe PIGSCRIPT ALIAS
+ !    Must specify PIGSCRIPT and ALIAS.
+STDERR
+        end
+      end
+
+      it "errors when no remote exists in the project" do
+        with_git_initialized_project do |p|
+          @git.git('remote rm mortar')
+          p.remote = nil
+          write_file(File.join(p.pigscripts_path, "my_script.pig"))
+          stderr, stdout = execute("describe my_script my_alias", p, @git)
+          stderr.should == <<-STDERR
+ !    Unable to find git remote for project myproject
+STDERR
+        end
+      end
+
+      it "errors when requested pigscript cannot be found" do
+        with_git_initialized_project do |p|
+          stderr, stdout = execute("describe does_not_exist my_alias", p, @git)
+          stderr.should == <<-STDERR
+ !    Unable to find pigscript does_not_exist
+ !    No pigscripts found
+ STDERR
+        end
+      end
+      
+      it "requests and reports on a successful describe" do
+        with_git_initialized_project do |p|
+          # stub api requests
+          describe_id = "c571a8c7f76a4fd4a67c103d753e2dd5"
+          describe_url = "https://api.mortardata.com/describe/#{describe_id}"
+          parameters = ["name"=>"key", "value"=>"value" ]
+          
+          mock(Mortar::Auth.api).post_describe("myproject", "my_script", "my_alias", is_a(String), :parameters => parameters) {Excon::Response.new(:body => {"describe_id" => describe_id})}
+          mock(Mortar::Auth.api).get_describe(describe_id, :exclude_result => true).returns(Excon::Response.new(:body => {"status_code" => Mortar::API::Describe::STATUS_QUEUED, "status_description" => "Pending"})).ordered
+          mock(Mortar::Auth.api).get_describe(describe_id, :exclude_result => true).returns(Excon::Response.new(:body => {"status_code" => Mortar::API::Describe::STATUS_GATEWAY_STARTING, "status_description" => "Gateway starting"})).ordered
+          mock(Mortar::Auth.api).get_describe(describe_id, :exclude_result => true).returns(Excon::Response.new(:body => {"status_code" => Mortar::API::Describe::STATUS_PROGRESS, "status_description" => "Starting pig"})).ordered
+          mock(Mortar::Auth.api).get_describe(describe_id, :exclude_result => true).returns(Excon::Response.new(:body => {"status_code" => Mortar::API::Describe::STATUS_SUCCESS, "status_description" => "Success", "web_result_url" => describe_url})).ordered
+          
+          # stub launchy
+          mock(Launchy).open(describe_url) {Thread.new {}}
+                    
+          write_file(File.join(p.pigscripts_path, "my_script.pig"))
+          stderr, stdout = execute("describe my_script my_alias --polling_interval 0.05 -p key=value", p, @git)
+          stdout.should == <<-STDOUT
+Taking code snapshot... done
+Sending code snapshot to Mortar... done
+Starting describe... done
+
+\r\e[0KStatus: Pending... /\r\e[0KStatus: Gateway starting... -\r\e[0KStatus: Starting pig... \\\r\e[0KStatus: Success  
+
+Results available at https://api.mortardata.com/describe/c571a8c7f76a4fd4a67c103d753e2dd5
+Opening web browser to show results... done
+STDOUT
+        end
+      end
+      
+      it "requests and reports on a failed describe" do
+        with_git_initialized_project do |p|
+          # stub api requests
+          describe_id = "c571a8c7f76a4fd4a67c103d753e2dd5"
+          
+          error_message = "This is my error message\nWith multiple lines."
+          line_number = 23
+          column_number = 32
+          error_type = 'PigError'
+          
+          mock(Mortar::Auth.api).post_describe("myproject", "my_script", "my_alias", is_a(String), :parameters => []) {Excon::Response.new(:body => {"describe_id" => describe_id})}
+          mock(Mortar::Auth.api).get_describe(describe_id, :exclude_result => true).returns(Excon::Response.new(:body => {"status_code" => Mortar::API::Describe::STATUS_QUEUED, "status_description" => "Pending"})).ordered
+          mock(Mortar::Auth.api).get_describe(describe_id, :exclude_result => true).returns(Excon::Response.new(:body => {"status_code" => Mortar::API::Describe::STATUS_FAILURE, "status_description" => "Failed",
+            "error_message" => error_message,
+            "line_number" => line_number,
+            "column_number" => column_number,
+            "error_type" => error_type})).ordered
+
+          write_file(File.join(p.pigscripts_path, "my_script.pig"))
+          stderr, stdout = execute("describe my_script my_alias --polling_interval 0.05", p, @git)
+          stdout.should == <<-STDOUT
+Taking code snapshot... done
+Sending code snapshot to Mortar... done
+Starting describe... done
+
+\r\e[0KStatus: Pending... /\r\e[0KStatus: Failed  
+
+STDOUT
+          stderr.should == <<-STDERR
+ !    Describe failed with PigError at Line 23, Column 32:
+ !    
+ !    This is my error message
+ !    With multiple lines.
+STDERR
+        end
+      end
+    end
+  end
+end

data/spec/mortar/command/generate_spec.rb

@@ -0,0 +1,139 @@
+#
+# Copyright 2012 Mortar Data Inc.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+#
+
+require "spec_helper"
+require "mortar/command/generate"
+require "fileutils"
+require "tmpdir"
+
+describe Mortar::Command::Generate do
+
+  before(:each) do
+    @tmpdir = Dir.mktmpdir
+    Dir.chdir(@tmpdir)
+  end
+
+  describe "generate:project" do
+    it "creates new project" do
+      stderr, stdout = execute("generate:project Test")
+      File.exists?("Test").should be_true
+      File.exists?("Test/macros").should be_true
+      File.exists?("Test/pigscripts").should be_true
+      File.exists?("Test/udfs").should be_true
+      File.exists?("Test/README.md").should be_true
+      File.exists?("Test/Gemfile").should be_true
+      #File.exists?("Test/Gemfile.lock").should be_true
+      File.exists?("Test/macros/.gitkeep").should be_true
+      File.exists?("Test/pigscripts/Test.pig").should be_true
+      File.exists?("Test/udfs/python/Test.py").should be_true
+
+      File.read("Test/pigscripts/Test.pig").each_line { |line| line.match(/<%.*%>/).should be_nil }
+    end
+    it "error when name isn't provided" do
+      stderr, stdout = execute("generate:project")
+      stderr.should == <<-STDERR
+ !    Usage: mortar new PROJECTNAME
+ !    Must specify PROJECTNAME.
+STDERR
+    end
+  end
+
+  describe "new" do
+    it "create new project using alias" do
+      stderr, stdout = execute("new Test")
+      File.exists?("Test").should be_true
+      File.exists?("Test/macros").should be_true
+      File.exists?("Test/pigscripts").should be_true
+      File.exists?("Test/udfs").should be_true
+      File.exists?("Test/README.md").should be_true
+      File.exists?("Test/Gemfile").should be_true
+      #File.exists?("Test/Gemfile.lock").should be_true
+      File.exists?("Test/macros/.gitkeep").should be_true
+      File.exists?("Test/pigscripts/Test.pig").should be_true
+      File.exists?("Test/udfs/python/Test.py").should be_true
+
+      File.read("Test/pigscripts/Test.pig").each_line { |line| line.match(/<%.*%>/).should be_nil }
+    end
+
+    it "error when name isn't provided" do
+      stderr, stdout = execute("new")
+      stderr.should == <<-STDERR
+ !    Usage: mortar new PROJECTNAME
+ !    Must specify PROJECTNAME.
+STDERR
+    end
+  end
+
+  describe "generate:pigscript" do
+    it "Generate a new pigscript in a project" do
+      with_blank_project do |p| 
+        stderr, stdout = execute("generate:pigscript Oink", p)
+        File.exists?(File.join(p.root_path, "pigscripts/Oink.pig")).should be_true
+        File.read("pigscripts/Oink.pig").each_line { |line| line.match(/<%.*%>/).should be_nil }
+      end
+    end
+
+    it "error when pigscript name isn't provided" do
+      with_blank_project do |p|
+        stderr, stdout = execute("generate:pigscript")
+        stderr.should == <<-STDERR
+ !    Usage: mortar generate:pigscript SCRIPTNAME
+ !    Must specify SCRIPTNAME.
+ STDERR
+      end
+    end
+  end
+
+  describe "generate:python_udf" do
+    it "Generate a new python udf in a project" do
+      with_blank_project do |p| 
+        stderr, stdout = execute("generate:python_udf slither", p)
+        File.exists?(File.join(p.root_path, "udfs/python/slither.py")).should be_true
+        File.read("udfs/python/slither.py").each_line { |line| line.match(/<%.*%>/).should be_nil }
+      end
+    end
+
+    it "error when udf name isn't provided" do
+      with_blank_project do |p|
+        stderr, stdout = execute("generate:python_udf")
+        stderr.should == <<-STDERR
+ !    Usage: mortar generate:python_udf UDFNAME
+ !    Must specify UDFNAME.
+ STDERR
+      end
+    end
+  end
+
+  describe "generate:macro" do
+    it "Generate a new macro in a project" do
+      with_blank_project do |p| 
+        stderr, stdout = execute("generate:macro big_mac", p)
+        File.exists?(File.join(p.root_path, "macros/big_mac.pig")).should be_true
+        File.read("macros/big_mac.pig").each_line { |line| line.match(/<%.*%>/).should be_nil }
+      end
+    end
+
+    it "error when udf name isn't provided" do
+      with_blank_project do |p|
+        stderr, stdout = execute("generate:macro")
+        stderr.should == <<-STDERR
+ !    Usage: mortar generate:macro MACRONAME
+ !    Must specify MACRONAME.
+ STDERR
+      end
+    end
+  end
+end