cartodb/spec/lib/tasks/data_observatory_rake_spec.rb
2020-06-15 10:58:47 +08:00

87 lines
3.0 KiB
Ruby

require 'spec_helper_min'
require 'rake'
describe 'data_observatory.rake' do
before(:all) do
Rake.application.rake_require('tasks/data_observatory')
Rake::Task.define_task(:environment)
@user = FactoryGirl.create(:valid_user, username: 'fulano')
end
after(:all) do
@user.destroy
end
describe '#purchase_datasets' do
before(:each) do
Rake::Task['cartodb:data_observatory:purchase_datasets'].reenable
end
after(:each) do
File.unstub(:open)
end
it 'throws an error if username is not provided' do
expect {
Rake::Task['cartodb:data_observatory:purchase_datasets'].invoke
}.to raise_error(RuntimeError, 'USAGE: data_observatory:purchase_datasets["username","path/datasets.csv"]')
end
it 'throws an error if the datasets path is not provided' do
expect {
Rake::Task['cartodb:data_observatory:purchase_datasets'].invoke('fulano')
}.to raise_error(RuntimeError, 'USAGE: data_observatory:purchase_datasets["username","path/datasets.csv"]')
end
it 'calls subscribe from Carto::DoLicensingService with the expected parameters' do
File.stubs(:open).returns(csv_example)
expected_datasets = [
{ dataset_id: 'dataset1', available_in: ['bq', 'spanner'], price: 100.0,
expires_at: Time.new(2020, 9, 27, 8, 0, 0) },
{ dataset_id: 'dataset2', available_in: ['spanner'], price: 200.0,
expires_at: Time.new(2020, 12, 31, 12, 0, 0) }
]
service_mock = mock
service_mock.expects(:subscribe).once.with(expected_datasets)
Carto::DoLicensingService.expects(:new).once.with('fulano').returns(service_mock)
Rake::Task['cartodb:data_observatory:purchase_datasets'].invoke('fulano', 'datasets.csv')
end
end
describe '#remove_purchase' do
before(:each) do
Rake::Task['cartodb:data_observatory:remove_purchase'].reenable
end
it 'throws an error if username is not provided' do
expect {
Rake::Task['cartodb:data_observatory:remove_purchase'].invoke
}.to raise_error(RuntimeError, 'USAGE: data_observatory:remove_purchase["username","project.schema.table"]')
end
it 'throws an error if dataset_id is not provided' do
expect {
Rake::Task['cartodb:data_observatory:remove_purchase'].invoke('fulano')
}.to raise_error(RuntimeError, 'USAGE: data_observatory:remove_purchase["username","project.schema.table"]')
end
it 'calls unsubscribe from Carto::DoLicensingService with the expected parameters' do
service_mock = mock
service_mock.expects(:unsubscribe).once.with('carto.abc.dataset')
Carto::DoLicensingService.expects(:new).once.with('fulano').returns(service_mock)
Rake::Task['cartodb:data_observatory:remove_purchase'].invoke('fulano', 'carto.abc.dataset')
end
end
def csv_example
CSV.generate do |csv|
csv << ["dataset1", "bq;spanner", "100", "2020-09-27T08:00:00"]
csv << ["dataset2", "spanner", "200", "2020-12-31T12:00:00"]
end
end
end