export for given user, organization or repo id

Author: karolselak

lib/backup.rb

@@ -20,11 +20,24 @@ def connect_db
     ActiveRecord::Base.establish_connection(@config.database_url)
   end
 
-  def export(owner_id = nil)
+  def export(args={})
+    if args[:user_id]
+      owner_id = args[:user_id]
+      owner_type = 'User'
+    elsif args[:org_id]
+      owner_id = args[:org_id]
+      owner_type = 'Organization'
+    elsif args[:repo_id]
+      repo_id = args[:repo_id]
+    end
+
     if owner_id
-      Repository.where('owner_id = ?', owner_id).order(:id).each do |repository|
+      Repository.where('owner_id = ? and owner_type = ?', owner_id, owner_type).order(:id).each do |repository|
         process_repo(repository)
       end
+    elsif repo_id
+      repository = Repository.find(repo_id)
+      process_repo(repository)
     else
       Repository.order(:id).each do |repository|
         process_repo(repository)

lib/models/organization.rb

@@ -0,0 +1,7 @@
+# frozen_string_literal: true
+
+require 'models/model'
+
+class Organization < Model
+  self.table_name = 'organizations'
+end

lib/models/user.rb

@@ -0,0 +1,7 @@
+# frozen_string_literal: true
+
+require 'models/model'
+
+class User < Model
+  self.table_name = 'users'
+end

spec/backup_spec.rb

@@ -4,37 +4,102 @@
 require 'models/repository'
 require 'models/build'
 require 'models/job'
+require 'models/organization'
+require 'models/user'
 require 'support/factories'
 require 'pry'
 
 describe Backup do
   before(:all) do
-    system("psql '#{Config.new.database_url}' -f db/schema.sql")
+    system("psql '#{Config.new.database_url}' -f db/schema.sql > /dev/null")
   end
 
-  after(:each) do
-    Repository.destroy_all
-    Build.destroy_all
-    Job.destroy_all
-  end
-
-  let!(:config) { Config.new }
   let(:files_location) { "dump/tests" }
   let!(:backup) { Backup.new(files_location: files_location, limit: 2) }
-  let(:datetime) { (config.delay + 1).months.ago.to_time.utc }
-  let(:org_id) { rand(100000) }
-  let(:com_id) { rand(100000) }
-  let(:private_org_id) { rand(100000) }
-  let(:private_com_id) { rand(100000) }
-  let!(:repository) {
-    FactoryBot.create(
-      :repository_with_builds,
-      created_at: datetime,
-      updated_at: datetime
-    )
-  }
+
+  describe 'export' do
+    let!(:unassigned_repositories) {
+      (1..3).to_a.map do
+        FactoryBot.create(:repository)    
+      end
+    }
+    let!(:user1) {
+      FactoryBot.create(:user_with_repos)
+    }
+    let!(:user2) {
+      FactoryBot.create(:user_with_repos)
+    }
+    let!(:organization1) {
+      FactoryBot.create(:organization_with_repos)
+    }
+    let!(:organization2) {
+      FactoryBot.create(:organization_with_repos)
+    }
+
+    context 'when no arguments are given' do
+      it 'processes every repository' do
+        Repository.all.each do |repository|
+          expect(backup).to receive(:process_repo).once.with(repository)
+        end
+        backup.export
+      end
+    end
+    context 'when user_id is given' do
+      it 'processes only the repositories of the given user' do
+        processed_repos_ids = []
+        allow(backup).to receive(:process_repo) {|repo| processed_repos_ids.push(repo.id)}
+        user_repos_ids = Repository.where(
+          'owner_id = ? and owner_type = ?',
+          user1.id,
+          'User'
+        ).map(&:id)
+        backup.export(user_id: user1.id)
+        expect(processed_repos_ids).to match_array(user_repos_ids)
+      end
+    end
+    context 'when org_id is given' do
+      it 'processes only the repositories of the given organization' do
+        processed_repos_ids = []
+        allow(backup).to receive(:process_repo) {|repo| processed_repos_ids.push(repo.id)}
+        organization_repos_ids = Repository.where(
+          'owner_id = ? and owner_type = ?',
+          organization1.id,
+          'Organization'
+        ).map(&:id)
+        backup.export(org_id: organization1.id)
+        expect(processed_repos_ids).to match_array(organization_repos_ids)
+      end
+    end
+    context 'when repo_id is given' do
+      it 'processes only the repository with the given id' do
+        repo = Repository.first
+        expect(backup).to receive(:process_repo).once.with(repo)
+        backup.export(repo_id: repo.id)
+      end
+    end
+  end
 
   describe 'process_repo' do
+    after(:each) do
+      Repository.destroy_all
+      Build.destroy_all
+      Job.destroy_all
+    end
+  
+    let!(:config) { Config.new }
+    let(:datetime) { (config.delay + 1).months.ago.to_time.utc }
+    let(:org_id) { rand(100000) }
+    let(:com_id) { rand(100000) }
+    let(:private_org_id) { rand(100000) }
+    let(:private_com_id) { rand(100000) }
+    let!(:repository) {
+      FactoryBot.create(
+        :repository_with_builds,
+        created_at: datetime,
+        updated_at: datetime
+      )
+    }
+
     let!(:exported_object) {
       [[
         {

spec/support/factories.rb

@@ -3,6 +3,38 @@
 require 'factory_bot'
 
 FactoryBot.define do
+  factory :organization do
+    factory :organization_with_repos do
+      transient do
+        repos_count { 3 }
+      end
+      after(:create) do |organization, evaluator|
+        create_list(
+          :repository,
+          evaluator.repos_count,
+          owner_id: organization.id,
+          owner_type: 'Organization'
+        )
+      end
+    end
+  end
+
+  factory :user do
+    factory :user_with_repos do
+      transient do
+        repos_count { 3 }
+      end
+      after(:create) do |user, evaluator|
+        create_list(
+          :repository,
+          evaluator.repos_count,
+          owner_id: user.id,
+          owner_type: 'User'
+        )
+      end
+    end
+  end
+
   factory :repository do
     factory :repository_with_builds do
       transient do