marty 0.5.15 → 0.5.16

data/spec/lib/xl_spec.rb

@@ -0,0 +1,198 @@
+require 'spec_helper'
+require 'marty'
+require 'delorean_lang'
+
+CODE =<<EOS
+M:
+    cfmt = {
+        "type": ":cellIs",
+        "priority": 1,
+        "operator": ":greaterThan",
+        "dxfId" : {
+            "fg_color": "FFF8696B",
+            "type": ":dxf",
+            "b": true,
+            },
+        "formula": "$B$1",
+        }
+
+    hdr_style = {
+        "font_name": "Calibri",
+        "color": "FFFFFF",
+        "b": true,
+        "sz": 14,
+        "style": {
+            "bg_color": "0D056F",
+            "fg_color": "FFFFFF",
+            "alignment": {
+                    "horizontal": ":center",
+                    }
+            },
+        }
+
+    #handles cell style
+    c_style = {"style": [{"bg_color": "C5D9F1"}]}
+
+    hi_style = {
+        "b": true,
+        "bg_color": "0D056F",
+        "fg_color": "FFFFFF",
+        }
+
+    threshold_rows = [
+        ["row", ["Threshold", 5], {"style": [hi_style, {}]}],
+        ]
+
+    title_header = [
+        ["row", ["Sec Inst Name","Market Change Sec Inst"], hdr_style],
+        ]
+    rdata = [
+         ["title", 2, 4, 5, 6, 8 ],
+         ["text", 12, 4, 15, 6, 18 ],
+         ["text", 22, 4, 25, 6, 28 ],
+         ["text", 32, 4, 35, 6, 38 ],
+         ["text", 42, 4, 45, 6, 48 ],
+         ["text", 52, 4, 55, 6, 58 ],
+         ["text", 62, 4, 65, 6, 68 ],
+         ["text", 72, 4, 75, 6, 78 ]
+         ]
+
+    height = rdata.length()
+    width = rdata[0].length()
+
+    rows = [ ["row", r, c_style] for r in rdata ]
+
+    raw = title_header +
+        [["merge",  [1, {"off": 0}, width - 1, {"off": 0}]]] +
+        [["conditional_formatting",
+            [1, {"off": 1}, width, {"off": height }], cfmt]] +
+        rows
+
+    ws = ["A Sheet",
+           [
+             ["pos", [0, 0], threshold_rows ],
+             ["pos", C1, raw ],
+             ["pos", C2, raw ],
+           ]
+         ]
+
+    title = "Per-Secutiry Market Change Report"
+    form = []
+
+    result = [ws]
+    format = "xlsx"
+EOS
+
+
+describe Marty::Xl do
+
+  let(:engine) {
+    Delorean::Engine.new "YYY"
+  }
+
+  def worksheet(ind, c)
+    code = CODE.clone
+    map = {'C1' => c[0].to_s, 'C2' => c[1].to_s}
+    map.each { |k,v| code.sub!(k,v)  }
+    engine.parse(code)
+    engine.evaluate_attrs("M", ["result"]).flatten(1)
+  end
+
+  before(:all) do
+    @coords = [
+               [[9, 5], [2, 5]],   # coords for non-overlaping datasets
+               [[5, 8], [2, 5]],   # coords for overlaping datasets
+              ]
+  end
+
+  it "should be able to create a spreadsheet that includes multiple datasets that don't overlap " do
+    ws = worksheet(0, @coords[0])
+    sp = Marty::Xl.spreadsheet(ws)
+    file = Tempfile.new('file.xlsx')
+    lambda { sp.serialize(file) }.should_not raise_error()
+
+    sp.workbook.worksheets[0].rows[0].cells[0..1].map { |c| c.value }.should ==
+      ["Threshold", 5]
+
+    sp.workbook.worksheets[0].rows[1].cells[0..1].map { |c| c.value }.should ==
+      ["", ""]
+
+    sp.workbook.worksheets[0].rows[5].cells[0..15].map { |c| c.value }.should == 
+      ["", "", "Sec Inst Name", "Market Change Sec Inst", "", "", "", "", "", "Sec Inst Name", "Market Change Sec Inst", "","", "", ""]
+
+    sp.workbook.worksheets[0].rows[6].cells[0..15].map { |c| c.value }.should == 
+      ["", "", "title", 2, 4, 5, 6, 8, "", "title", 2, 4, 5, 6, 8]
+
+    sp.workbook.worksheets[0].rows[7].cells[0..15].map { |c| c.value }.should == 
+      ["", "", "text", 12, 4, 15, 6, 18, "", "text", 12, 4, 15, 6, 18]
+
+    sp.workbook.worksheets[0].rows[8].cells[0..15].map { |c| c.value }.should == 
+      ["", "", "text", 22, 4, 25, 6, 28, "", "text", 22, 4, 25, 6, 28]
+
+    sp.workbook.worksheets[0].rows[9].cells[0..15].map { |c| c.value }.should == 
+      ["", "", "text", 32, 4, 35, 6, 38, "", "text", 32, 4, 35, 6, 38]
+
+    sp.workbook.worksheets[0].rows[10].cells[0..15].map { |c| c.value }.should == 
+      ["", "", "text", 42, 4, 45, 6, 48, "", "text", 42, 4, 45, 6, 48]
+
+    sp.workbook.worksheets[0].rows[11].cells[0..15].map { |c| c.value }.should == 
+      ["", "", "text", 52, 4, 55, 6, 58, "", "text", 52, 4, 55, 6, 58]
+
+    sp.workbook.worksheets[0].rows[12].cells[0..15].map { |c| c.value }.should == 
+      ["", "", "text", 62, 4, 65, 6, 68, "", "text", 62, 4, 65, 6, 68]
+
+    sp.workbook.worksheets[0].rows[13].cells[0..15].map { |c| c.value }.should == 
+      ["", "", "text", 72, 4, 75, 6, 78, "", "text", 72, 4, 75, 6, 78]
+
+  end
+
+  it "should be able to create a spreadsheet that includes multiple datasets that overlap " do
+    ws = worksheet(1, @coords[1])
+    sp = Marty::Xl.spreadsheet(ws)
+    file = Tempfile.new('file.xlsx')
+    lambda { sp.serialize(file) }.should_not raise_error()
+
+    sp.workbook.worksheets[0].rows[0].cells[0..1].map { |c| c.value }.should ==
+      ["Threshold", 5]
+
+    sp.workbook.worksheets[0].rows[1].cells[0..1].map { |c| c.value }.should ==
+      ["", ""]
+
+    sp.workbook.worksheets[0].rows[5].cells[0..11].map { |c| c.value }.should == 
+      ["", "", "Sec Inst Name", "Market Change Sec Inst", "", "", "", "", "", "", ""]
+
+    sp.workbook.worksheets[0].rows[6].cells[0..11].map { |c| c.value }.should == 
+      ["", "", "title", 2, 4, 5, 6, 8, "", "", ""]
+
+    sp.workbook.worksheets[0].rows[7].cells[0..11].map { |c| c.value }.should == 
+      ["", "", "text", 12, 4, 15, 6, 18, "", "", ""]
+
+    sp.workbook.worksheets[0].rows[8].cells[0..11].map { |c| c.value }.should == 
+      ["", "", "text", 22, 4, "Sec Inst Name", "Market Change Sec Inst", 28, "", "", ""]
+
+    sp.workbook.worksheets[0].rows[9].cells[0..11].map { |c| c.value }.should == 
+      ["", "", "text", 32, 4, "title", 2, 4, 5, 6, 8]
+
+    sp.workbook.worksheets[0].rows[10].cells[0..11].map { |c| c.value }.should == 
+      ["", "", "text", 42, 4, "text", 12, 4, 15, 6, 18]
+
+    sp.workbook.worksheets[0].rows[11].cells[0..11].map { |c| c.value }.should == 
+      ["", "", "text", 52, 4, "text", 22, 4, 25, 6, 28]
+
+    sp.workbook.worksheets[0].rows[12].cells[0..11].map { |c| c.value }.should == 
+      ["", "", "text", 62, 4, "text", 32, 4, 35, 6, 38]
+
+    sp.workbook.worksheets[0].rows[13].cells[0..11].map { |c| c.value }.should == 
+      ["", "", "text", 72, 4, "text", 42, 4, 45, 6, 48]
+
+    sp.workbook.worksheets[0].rows[14].cells[0..11].map { |c| c.value }.should == 
+      ["", "", "", "", "", "text", 52, 4, 55, 6, 58]
+
+    sp.workbook.worksheets[0].rows[15].cells[0..11].map { |c| c.value }.should == 
+      ["", "", "", "", "", "text", 62, 4, 65, 6, 68]
+
+    sp.workbook.worksheets[0].rows[16].cells[0..11].map { |c| c.value }.should == 
+      ["", "", "", "", "", "text", 72, 4, 75, 6, 78]
+  end
+
+end

data/spec/lib/xl_styles_spec.rb

@@ -0,0 +1,115 @@
+require 'spec_helper'
+require 'marty'
+require 'delorean_lang'
+
+
+STYLE_CODE =<<EOS
+S:
+    data =
+        [
+         [1, 2],
+         [3, 4],
+        ]
+
+    raw =   [ [ "border", [1,0,1,2], {
+              "style" : ":thin",
+              "color" : "000000"
+              } ]
+            ] +
+            [ [ "border", [0,1,2,1], {
+              "style" : ":thick",
+              "color" : "FF0000"
+              } ]
+            ] +
+            [
+             ["row", r]
+             for r in data
+            ]
+
+    ws    = ["BorderExample", [ ["pos", [1, 1], raw] ] ]
+
+    title = "Border Example"
+    form = []
+    format = "xlsx"
+    result = [ws]
+EOS
+
+describe Marty::Xl do
+
+  let(:engine) {
+    Delorean::Engine.new "YYY"
+  }
+
+  before(:each) do
+    code = STYLE_CODE.clone
+    engine.parse(code)
+    @ws =  engine.evaluate_attrs("S", ["result"]).flatten(1)
+  end
+
+  it "should be able to create a spreadsheet with overlapping border styles" do
+    sp = Marty::Xl.spreadsheet(@ws)
+    wb = sp.workbook
+
+    file = Tempfile.new('file2.xlsx')
+    lambda { sp.serialize(file) }.should_not raise_error()
+
+    def border_details(b)
+      edges = []
+      b.prs.each do |pr|
+        edges << [pr.name, pr.style, pr.color.rgb]
+      end
+      edges.sort_by{ |k| k[0] }
+    end
+
+    sp.workbook.worksheets[0].rows[0].cells[0..1].map { |c| c.value }.should ==
+      ["", ""]
+
+    sp.workbook.worksheets[0].rows[1].cells[0..2].map { |c| c.value }.should ==
+      ["", 1, 2]
+
+    sp.workbook.worksheets[0].rows[2].cells[0..2].map { |c| c.value }.should ==
+      ["", 3, 4]
+
+    wb.worksheets[0].styles.borders.count.should >= 4
+
+    wb.worksheets[0].styles.borders.each_index do |i|
+      b = border_details(wb.worksheets[0].styles.borders[i])
+
+      case i
+      when 0
+        b.should == []
+      when 2
+        b.should == [[:left, :thin, "FF000000"]]
+      when 3
+        b.should == [[:top, :thick, "FFFF0000"]]
+      when 4
+        b.should == [[:left, :thin, "FF000000"], [:top, :thick, "FFFF0000"]]
+      else
+        next
+      end
+    end
+
+    wb.worksheets[0].styles.cellXfs.count.should >= 8
+
+    wb.worksheets[0].styles.cellXfs.each_index do |i|
+      c = wb.worksheets[0].styles.cellXfs[i]
+      case i
+      when 3
+        c.borderId.should == 0
+      when 4
+        c.borderId.should == 0
+      when 5
+        c.borderId.should == 2
+      when 6
+        c.borderId.should == 0
+      when 7
+        c.borderId.should == 3
+      when 8
+        c.borderId.should == 4
+      else
+        next
+      end
+    end
+
+  end
+end

data/spec/models/api_auth_spec.rb

@@ -0,0 +1,187 @@
+require 'spec_helper'
+
+module Marty
+  describe ApiAuth do
+    before(:each) do
+      Marty::Script.load_script_bodies({
+                           "Script1" => "A:\n    a = 1\n",
+                           "NewScript1" => "B:\n    b = 0\n",
+                         }, Date.today)
+
+      @api = ApiAuth.new
+      @api.app_name = 'TestApp'
+      @api.script_name = 'Script1'
+      @api.save!
+      @api.reload
+    end
+
+    describe "validations" do
+      it "should require app name, api key and script name" do
+        api = ApiAuth.new
+        expect(api).to_not be_valid
+        expect(api.errors[:app_name].any?).to be true
+        expect(api.errors[:api_key].any?).to be false # auto-set if nil
+        expect(api.errors[:script_name].any?).to be true
+      end
+
+      it "should require unique app name/script name" do
+        api = @api.dup
+        expect(api).to_not be_valid
+        expect(api.errors[:app_name].to_s).to include('already been taken')
+        api.script_name = 'NewScript1'
+        expect(api).to be_valid
+      end
+
+      it "should require unique api key/script name" do
+        api = @api.dup
+        expect(api).to_not be_valid
+        expect(api.errors[:api_key].to_s).to include('must be unique')
+        api.script_name = 'NewScript1'
+        expect(api).to be_valid
+      end
+
+      it "should require a valid associated script name" do
+        api = ApiAuth.new
+        api.app_name = 'NewApp'
+        api.script_name = @api.script_name + 'Bad'
+        expect(api).to_not be_valid
+        expect(api.errors[:base].to_s).to include('reference a valid script')
+      end
+
+      it "should allow a tagged script version to be associated when a DEV " +
+        "version of that script also exists" do
+        s = Marty::Script.lookup('infinity', 'Script1')
+        s.body = "A:\n    a = 3\n"
+        s.save!
+
+        api = ApiAuth.new
+        api.app_name = 'NewApp'
+        api.script_name = s.name
+        expect(api).to be_valid
+      end
+
+      it "should not allow a DEV script to be associated if there is no " +
+        "tagged version of that script" do
+        s = Marty::Script.new
+        s.name = 'TestScript'
+        s.body = '-- Test3'
+        # Creates a script in the future - no tag exists
+        s.created_dt = Time.now + 1.minute
+        s.save!
+
+        api = ApiAuth.new
+        api.app_name = 'NewApp'
+        api.script_name = s.name
+        expect(api).to_not be_valid
+        expect(api.errors[:base].to_s).to include('reference a valid script')
+      end
+    end
+
+    describe "key management" do
+      it "should require a 38 character key" do
+        expect(@api.api_key.length).to eq(38)
+
+        @api.api_key = "123456789"
+        expect(@api).to_not be_valid
+        expect(@api.errors[:base].to_s).to include('length must be 38')
+      end
+
+      it "should create the api key if necessary when the record is created" do
+        a = ApiAuth.new
+        a.app_name = 'MyApp'
+        a.script_name = 'NewScript1'
+        expect(a.api_key).to be_nil
+        a.save!
+        expect(a.api_key).to_not be_nil
+      end
+
+      it "should allow api key to be updated for an existing record" do
+        old = @api.api_key
+        @api.api_key = ApiAuth.generate_key
+        new = @api.api_key
+        expect(@api.api_key).to_not eq(old)
+        expect(@api).to be_valid
+        @api.save!
+        # Verifies that validation and saving does not create new key
+        expect(@api.api_key).to eq(new)
+      end
+
+      it "should generate new api key if old one is cleared" do
+        old = @api.api_key
+        @api.api_key = ''
+        @api.app_name += 'x'
+        @api.save!
+        expect(@api.api_key).to_not eq(old)
+      end
+
+      it "should generate new api key if old one is cleared (2)" do
+        old = @api.api_key
+        @api.api_key = nil
+        @api.app_name += 'x'
+        @api.save!
+        expect(@api.api_key).to_not eq(old)
+      end
+    end
+
+    describe "authorization" do
+      it "should pass when script is not secured" do
+        # Script is not secured by any entries
+        expect(ApiAuth.authorized?('SomeScript','SomeKey')).to be true
+      end
+
+      it "should pass when script is secured and key is valid" do
+        expect(@api.script_name).to include('Script1')
+        expect(ApiAuth.authorized?(@api.script_name,@api.api_key)).to be true
+      end
+
+      it "should pass when script is secured and key is valid 2" do
+        a = ApiAuth.new
+        a.app_name = @api.app_name + 'x'
+        a.script_name = 'NewScript1'
+        a.save!
+        expect(ApiAuth.authorized?('NewScript1',a.api_key)).to be true
+      end
+
+      it "should match on exact script name" do
+        a = ApiAuth.new
+        a.app_name = @api.app_name + 'x'
+        a.script_name = 'NewScript1'
+        a.save!
+        expect(ApiAuth.authorized?('Script1',a.api_key)).to be false
+      end
+
+      it "should fail when script is secured and key is invalid" do
+        expect(ApiAuth.authorized?('Script1','SomeKey')).to be false
+      end
+
+      it "should fail when script is secured and key is invalid (2)" do
+        a = ApiAuth.new
+        a.app_name = @api.app_name + 'x'
+        a.script_name = 'NewScript1'
+        a.save!
+        expect(ApiAuth.authorized?('NewScript1',@api.api_key)).to be false
+      end
+
+      it "should fail when script is secured and key is not specified" do
+        expect(ApiAuth.authorized?('Script1',nil)).to be false
+        expect(ApiAuth.authorized?('Script1','')).to be false
+      end
+
+      it "should pass when api_auth is deleted and no other auths exist" do
+        expect(ApiAuth.authorized?(@api.script_name,@api.api_key)).to be true
+        @api.delete
+        expect(ApiAuth.authorized?(@api.script_name,@api.api_key)).to be true
+      end
+
+      it "should fail when api_auth is deleted and another auth exists" do
+        api = @api.dup
+        api.app_name += 'x'
+        api.api_key = nil
+        api.save!
+        expect(ApiAuth.authorized?(@api.script_name,@api.api_key)).to be true
+        @api.delete
+        expect(ApiAuth.authorized?(@api.script_name,@api.api_key)).to be false
+      end
+    end
+  end
+end