# encoding: utf-8
require 'spec_helper'

describe Dataset do
  context "when built" do
    subject{ build(:dataset) }
    its(:saved?){ should be_false }
    its(:valid?){ should be_true }
  end
  
  context "when created" do
    context "without token" do
  
      before(:all){ Quandl::Client.token = '' }
      
      let(:dataset){ create(:dataset) }
      subject{ dataset }
  
      its(:saved?){ should be_false }
      its(:status){ should eq 401 }
    
    end
    context "with token" do

      before(:all){ Quandl::Client.token = ENV['QUANDL_AUTH_TOKEN'] }
      
      let(:source){ create(:source) }
      let(:dataset){ create(:dataset, source_code: source.code ) }
      subject{ dataset }
  
      its(:saved?){ should be_true }
      its(:status){ should eq 201 }
  
    end
    context "with data" do

      before(:all){ Quandl::Client.token = ENV['QUANDL_AUTH_TOKEN'] }
      
      let(:source){ create(:source) }
      let(:dataset){ create(:dataset, source_code: source.code, data: Quandl::Data::Random.table(rows: 20, columns: 2, nils: false) ) }
      subject{ dataset }
  
      its(:saved?){ should be_true }
      its(:status){ should eq 201 }
  
    end
  end
  
  context "when updated" do

    let(:source){ create(:source) }
    let(:dataset){ create(:dataset, source_code: source.code, data: Quandl::Data::Random.table(rows: 20, columns: 2, nils: false).to_csv ) }
    subject{ Dataset.find(dataset.id) }
    
    
    it "should update data" do
      # update the dataset
      new_row = [ subject.data_table[0][0], 1.0, 2.0]
      subject.data = [ new_row ]
      subject.save
      # check the data
      Dataset.find(dataset.id).data_table.sort_descending[0].should eq new_row
    end
    
    it "should update column_spec" do
      subject.column_spec = "[0,[\"Date \\n\",{}],[\"Column 1 \",{}],[\"New Column Name \",{}]]"
      subject.save
      Dataset.find(dataset.id).column_spec.should eq subject.column_spec
    end
    
    context "with new rows" do
      
      it "should include new row" do
        new_data = 10.times.collect{|i| [Date.parse(subject.to_date) + i + 1, rand(12), rand(12) ] }
        new_data = Quandl::Data::Table.new(new_data).sort_descending
        subject.data = new_data
        subject.save
        updated_dataset = Dataset.find(subject.id)
        updated_dataset.data_table.to_date[0].should eq new_data.to_date[0]
      end
      
      it "should include old rows" do
        new_data = 10.times.collect{|i| [Date.parse(subject.to_date) + i + 2, rand(12), rand(12) ] }
        new_data = Quandl::Data::Table.new(new_data).sort_descending
        subject.data = new_data
        subject.save
        updated_dataset = Dataset.find(subject.id)
        updated_dataset.data_table.count.should eq 30
      end
    end
    
  end
  
end