chronic_tree 1.0.0

data/lib/chronic_tree/railtie.rb

@@ -0,0 +1,9 @@
+module ChronicTree
+  class Railtie < Rails::Railtie
+    initializer "chronic_tree.inject_into_active_record_base" do
+      ActiveSupport.on_load :active_record do
+        ::ActiveRecord::Base.include(ChronicTree)
+      end
+    end
+  end
+end

data/lib/chronic_tree/travesal.rb

@@ -0,0 +1,64 @@
+module ChronicTree
+  module Travesal
+    def children
+      init_tree_args_when_nil
+      children_relation(current_time_at, current_scope_name).map do |el|
+        el.child.as_tree(current_time_at, current_scope_name)
+      end
+    end
+
+    def parent
+      init_tree_args_when_nil
+      child_element = parent_relation(current_time_at, current_scope_name).first
+      child_element.parent.as_tree(current_time_at, current_scope_name) if child_element
+    end
+
+    def root
+      init_tree_args_when_nil
+      child_element = existed_relation(current_time_at, current_scope_name).first
+      child_element.root.as_tree(current_time_at, current_scope_name) if child_element
+    end
+
+    def descendants
+      init_tree_args_when_nil
+      descendants_relation(current_time_at, current_scope_name).inject([]) do |result, el|
+        result[el.distance - 1] ||= []
+        result[el.distance - 1] << el.child.as_tree(current_time_at, current_scope_name)
+        result
+      end
+    end
+
+    def flat_descendants
+      init_tree_args_when_nil
+      descendants_relation(current_time_at, current_scope_name).map do |el|
+        el.child.as_tree(current_time_at, current_scope_name)
+      end
+    end
+
+    def ancestors
+      init_tree_args_when_nil
+      ancestors_relation(current_time_at, current_scope_name).map do |el|
+        el.parent.as_tree(current_time_at, current_scope_name)
+      end
+    end
+
+    alias_method :parent?, :parent
+
+    def tree_empty?(time_at = Time.now, scope_name = 'default')
+      ChronicTree::ActiveRecord::Element.
+        where(scope_name: scope_name).
+        at(time_at).
+        empty?
+    end
+
+    def existed_in_tree?(time_at = Time.now, scope_name = 'default')
+      existed_relation(time_at, scope_name).any?
+    end
+
+    private
+
+      def init_tree_args_when_nil
+        as_tree if current_time_at.nil? || current_scope_name.nil?
+      end
+  end
+end

data/lib/chronic_tree/version.rb

@@ -0,0 +1,3 @@
+module ChronicTree
+  VERSION = "1.0.0"
+end

data/lib/generators/chronic_tree/install_generator.rb

@@ -0,0 +1,15 @@
+module ChronicTree
+  class InstallGenerator < Rails::Generators::Base
+
+    desc "This generator creates the db schema for chronic_tree gem."
+
+    source_root File.expand_path("../templates", __FILE__)
+
+    def create_migration_file
+      if Dir["#{destination_root}/db/migrate/*_create_chronic_tree_elements.rb"].empty?
+        template "create_chronic_tree_elements.rb",
+          "db/migrate/#{Time.now.strftime('%Y%m%d%H%M%S')}_create_chronic_tree_elements.rb"
+      end
+    end
+  end
+end

data/lib/generators/chronic_tree/templates/create_chronic_tree_elements.rb

@@ -0,0 +1,22 @@
+class CreateChronicTreeElements < ActiveRecord::Migration
+  def change
+    create_table :chronic_tree_elements do |t|
+      t.string :tree_type, not_null: true
+      t.integer :parent_id, not_null: true
+      t.integer :child_id, not_null: true
+      t.integer :root_id, not_null: true
+      t.integer :distance, not_null: true
+      t.datetime :start_time, not_null: true
+      t.datetime :end_time, not_null: true
+      t.string :scope_name, not_null: true
+      t.integer :position
+
+      t.index :tree_type
+      t.index :parent_id
+      t.index :child_id
+      t.index :root_id
+      t.index [:start_time, :end_time]
+      t.index :scope_name
+    end
+  end
+end

data/spec/chronic_tree_spec.rb

@@ -0,0 +1,223 @@
+require 'spec_helper'
+
+describe ChronicTree do
+
+  context "Tree Traversal" do
+    before(:all) do
+      init_simple_tree
+    end
+
+    after(:all) do
+      destroy_simple_tree
+    end
+
+    it "f: access non-existed scope" do
+      expect { @root_org.as_tree(Time.now, 'non-existed') }.to raise_error
+    end
+
+    it "f: access the future" do
+      expect { @root_org.as_tree(1.minute.from_now) }.to raise_error
+    end
+
+    it "s: access children" do
+      expect(@root_org.children.size).to be == 2
+      expect(@root_org.children.first.id).to be == @lv1_child_org.id
+      expect(@lv1_child_org.children.size).to be == 2
+      expect(@lv1_child_org.children.first.id).to be == @lv2_child_org.id
+      expect(@lv2_child_org.children.size).to be == 1
+    end
+
+    it "s: access parent" do
+      expect(@lv1_child_org.parent.id).to be == @root_org.id
+      expect(@lv2_child_org.parent.id).to be == @lv1_child_org.id
+      expect(@root_org.parent?).to be_falsy
+    end
+
+    it "s: access root" do
+      expect(@lv1_child_org.root.id).to be == @root_org.id
+      expect(@lv2_child_org.root.id).to be == @root_org.id
+      expect(@root_org.root.id).to be == @root_org.id
+    end
+
+    it "s: access descendants" do
+      expect(@root_org.descendants.size).to be == 5
+      expect(@root_org.flat_descendants.size).to be == 12
+      expect(@root_org.flat_descendants.first.id).to be == @lv1_child_org.id
+      expect(@root_org.flat_descendants.second.id).to be == @lv1_child_org2.id
+      expect(@lv1_child_org.descendants.size).to be == 4
+      expect(@lv1_child_org.flat_descendants.size).to be == 8
+      expect(@lv1_child_org.flat_descendants.first.id).to be == @lv2_child_org.id
+      expect(@lv2_child_org.descendants.size).to be == 3
+    end
+
+    it "s: access ancestors" do
+      expect(@root_org.ancestors.size).to be == 1
+      expect(@root_org.ancestors.first.id).to be == @root_org.id
+      expect(@lv1_child_org.ancestors.size).to be == 1
+      expect(@lv1_child_org.ancestors.first.id).to be == @root_org.id
+      expect(@lv2_child_org.ancestors.size).to be == 2
+      expect(@lv2_child_org.ancestors.first.id).to be == @lv1_child_org.id
+      expect(@lv2_child_org.ancestors.second.id).to be == @root_org.id
+      expect(@lv5_child_org.ancestors.size).to be == 5
+    end
+
+    it "s: play with history versions" do
+      @root_org.as_tree(5.minutes.ago)
+
+      expect(@root_org.children.size).to be == 1
+      expect(@root_org.flat_descendants.size).to be == 2
+      expect(@root_org.children.first.children.first.id).to be == @lv1_child_org.id
+      expect(@root_org.children.first.children.first.parent.id).to be == @lv2_child_org.id
+    end
+  end
+
+  context "Tree Operation" do
+    context "Tree is empty" do
+      before(:each) do
+        @root_org = Org.create(name: 'root')
+      end
+
+      after(:each) do
+        @root_org.destroy
+      end
+
+      it "s: add self as the root element" do
+        root_org = @root_org.add_as_root
+        expect(root_org.tree_empty?).to be_falsy
+        expect(root_org.children.size).to be == 0
+        expect(root_org.parent?).to be_falsy
+        expect(root_org.descendants.size).to be == 0
+        expect(root_org.ancestors.size).to be == 1
+      end
+    end
+
+    context "Tree isn't empty" do
+
+      before(:each) do
+        init_simple_tree
+        @new_org = Org.create(name: 'new')
+        @new_org2 = Org.create(name: 'new2')
+      end
+
+      after(:each) do
+        destroy_simple_tree
+      end
+
+      context "Self is in the tree" do
+        it "f: add self as the root element" do
+          expect { @root_org.add_as_root }.to raise_error
+        end
+
+        it "s: add children" do
+          @lv1_child_org.add_child(@new_org)
+          expect(@lv1_child_org.children.size).to be == 3
+          expect(@root_org.flat_descendants.size).to be == 13
+          expect(@root_org.descendants.second.size).to be == 5
+        end
+
+        it "f: adding children" do
+          expect { @lv1_child_org.add_child(Org.new) }.to raise_error
+          expect { @lv1_child_org.add_child(Struct.new) }.to raise_error
+          expect { @lv1_child_org.add_child(@lv2_child_org) }.to raise_error
+        end
+
+        it "s: remove self" do
+          @lv1_child_org.remove_self
+          expect(@lv1_child_org.existed_in_tree?).to be_falsy
+          expect(@lv2_child_org.existed_in_tree?).to be_falsy
+          expect(@root_org.flat_descendants.size).to be == 3
+        end
+
+        it "s: remove descendants" do
+          @lv1_child_org.remove_descendants
+          expect(@lv1_child_org.existed_in_tree?).to be_truthy
+          expect(@lv2_child_org.existed_in_tree?).to be_falsy
+          expect(@root_org.flat_descendants.size).to be == 4
+        end
+
+        it "s: change parent to another object" do
+          @lv2_child_org.change_parent(@root_org)
+          expect(@lv1_child_org.children.size).to be == 1
+          expect(@lv2_child_org.parent.id).to be == @root_org.id
+          expect(@lv5_child_org.ancestors.size).to be == 4
+          expect(@root_org.children.size).to be == 3
+        end
+
+        it "s: replace root by another object" do
+          @root_org.replace_by(@new_org)
+          expect(@root_org.existed_in_tree?).to be_falsy
+          expect(@lv1_child_org.parent.id).to be == @new_org.id
+          expect(@new_org.parent?).to be_falsy
+          expect(@new_org.children.size).to be == 2
+          expect(@new_org.descendants.size).to be == 5
+        end
+
+        it "s: replace non-root by another object" do
+          @lv1_child_org.replace_by(@new_org)
+          expect(@lv1_child_org.existed_in_tree?).to be_falsy
+          expect(@lv2_child_org.parent.id).to be == @new_org.id
+          expect(@new_org.parent.id).to be == @root_org.id
+          expect(@root_org.children.size).to be == 2
+          expect(@root_org.descendants.size).to be == 5
+        end
+
+        it "f: change parent to another object" do
+          expect { @lv1_child_org.change_parent(Org.new) }.to raise_error
+          expect { @lv1_child_org.change_parent(Struct.new) }.to raise_error
+          expect { @lv1_child_org.change_parent(@new_org) }.to raise_error
+          expect { @lv1_child_org.change_parent(@lv2_child_org) }.to raise_error
+          expect { @lv1_child_org.change_parent(@lv1_child_org) }.to raise_error
+        end
+
+        it "f: replace self by another object" do
+          expect { @lv1_child_org.replace_by(Org.new) }.to raise_error
+          expect { @lv1_child_org.replace_by(Struct.new) }.to raise_error
+          expect { @lv1_child_org.replace_by(@lv2_child_org) }.to raise_error
+        end
+
+        it "s: play with multiple scopes" do
+          # operate with special scope
+          @root_org.as_tree('special').add_as_root.add_child(@lv1_child_org)
+          @lv1_child_org.add_child(@lv2_child_org4)
+
+          # force refreshing the timestamp
+          @root_org.as_tree('special')
+          expect(@root_org.descendants.size).to be == 2
+          expect(@lv2_child_org4.parent.id).to be == @lv1_child_org.id
+
+          # change the scope back to default
+          @root_org.as_tree
+          @lv2_child_org4.as_tree
+          expect(@root_org.flat_descendants.size).to be == 12
+          expect(@lv2_child_org4.parent.id).to be == @lv1_child_org2.id
+        end
+      end
+
+      context "Self isn't in the tree" do
+        it "f: add self as the root element" do
+          expect { @new_org.add_as_root }.to raise_error
+        end
+
+        it "f: add children" do
+          expect { @new_org.add_child(@new_org2) }.to raise_error
+        end
+
+        it "f: remove self" do
+          expect { @new_org.remove_self }.to raise_error
+        end
+
+        it "f: remove descendants" do
+          expect { @new_org.remove_descendants }.to raise_error
+        end
+
+        it "f: change parent to another object" do
+          expect { @new_org.change_parent(@root_org) }.to raise_error
+        end
+
+        it "f: replace self by another object" do
+          expect { @new_org.replace_by(@root_org) }.to raise_error
+        end
+      end
+    end
+  end
+end

data/spec/spec_helper.rb

@@ -0,0 +1,438 @@
+require "codeclimate-test-reporter"
+CodeClimate::TestReporter.start
+
+require 'coveralls'
+Coveralls.wear!
+
+require 'active_record'
+require 'logger'
+
+require 'chronic_tree'
+
+# This connection will do for database-independent bug reports.
+ActiveRecord::Base.establish_connection(adapter: 'sqlite3', database: ':memory:')
+ActiveRecord::Base.logger = Logger.new(STDOUT)
+
+# Comment this line to view the debug information in the console.
+ActiveRecord::Base.logger.level = Logger::INFO
+
+ActiveRecord::Schema.define do
+  create_table :orgs do |t|
+    t.string :name
+  end
+
+  create_table :chronic_tree_elements do |t|
+    t.string :tree_type, not_null: true
+    t.integer :parent_id, not_null: true
+    t.integer :child_id, not_null: true
+    t.integer :root_id, not_null: true
+    t.integer :distance, not_null: true
+    t.datetime :start_time, not_null: true
+    t.datetime :end_time, not_null: true
+    t.string :scope_name, not_null: true
+    t.integer :position
+
+    t.index :tree_type
+    t.index :parent_id
+    t.index :child_id
+    t.index :root_id
+    t.index [:start_time, :end_time]
+    t.index :scope_name
+  end
+end
+
+class Org < ActiveRecord::Base
+  include ChronicTree
+
+  chronic_tree
+  chronic_tree('special')
+end
+
+# Provide a tree with the hierarchy below:
+#
+# Now:
+#   root
+#     -- lv1
+#       -- lv2
+#         -- lv3
+#           -- lv4
+#             -- lv5
+#       -- lv2-2
+#         -- lv3-2
+#           -- lv4-2
+#             -- lv5-2
+#     -- lv1-2
+#       -- lv2-3
+#       -- lv2-4
+#
+# 10 minutes ago:
+#
+#   root
+#     -- lv2
+#       -- lv1
+def init_simple_tree
+  @root_org = Org.create(name: 'root')
+  @lv1_child_org = Org.create(name: 'lv1')
+  @lv1_child_org2 = Org.create(name: 'lv1-2')
+  @lv2_child_org = Org.create(name: 'lv2')
+  @lv2_child_org2 = Org.create(name: 'lv2-2')
+  @lv2_child_org3 = Org.create(name: 'lv2-3')
+  @lv2_child_org4 = Org.create(name: 'lv2-4')
+  @lv3_child_org = Org.create(name: 'lv3')
+  @lv3_child_org2 = Org.create(name: 'lv3-2')
+  @lv4_child_org = Org.create(name: 'lv4')
+  @lv4_child_org2 = Org.create(name: 'lv4-2')
+  @lv5_child_org = Org.create(name: 'lv5')
+  @lv5_child_org2 = Org.create(name: 'lv5-2')
+
+  current_time = Time.now
+
+  @root_org.elements_under_default_root.create(
+    parent_id: @root_org.id,
+    child_id: @root_org.id,
+    distance: 0,
+    start_time: 10.minutes.ago(current_time),
+    end_time: 1000.years.from_now,
+    scope_name: 'default'
+  )
+
+  # 10 minutes ago
+  @root_org.elements_under_default_root.create(
+    parent_id: @root_org.id,
+    child_id: @lv2_child_org.id,
+    distance: 1,
+    start_time: 10.minutes.ago(current_time),
+    end_time: current_time,
+    scope_name: 'default'
+  )
+
+  @root_org.elements_under_default_root.create(
+    parent_id: @root_org.id,
+    child_id: @lv1_child_org.id,
+    distance: 2,
+    start_time: 10.minutes.ago(current_time),
+    end_time: current_time,
+    scope_name: 'default'
+  )
+
+  @root_org.elements_under_default_root.create(
+    parent_id: @lv2_child_org.id,
+    child_id: @lv1_child_org.id,
+    distance: 1,
+    start_time: 10.minutes.ago(current_time),
+    end_time: current_time,
+    scope_name: 'default'
+  )
+
+  # now
+
+  @root_org.elements_under_default_root.create(
+    parent_id: @root_org.id,
+    child_id: @lv1_child_org.id,
+    distance: 1,
+    start_time: current_time,
+    end_time: 1000.years.since(current_time),
+    scope_name: 'default'
+  )
+
+  @root_org.elements_under_default_root.create(
+    parent_id: @root_org.id,
+    child_id: @lv1_child_org2.id,
+    distance: 1,
+    start_time: current_time,
+    end_time: 1000.years.since(current_time),
+    scope_name: 'default'
+  )
+
+  @root_org.elements_under_default_root.create(
+    parent_id: @root_org.id,
+    child_id: @lv2_child_org.id,
+    distance: 2,
+    start_time: current_time,
+    end_time: 1000.years.since(current_time),
+    scope_name: 'default'
+  )
+
+  @root_org.elements_under_default_root.create(
+    parent_id: @lv1_child_org.id,
+    child_id: @lv2_child_org.id,
+    distance: 1,
+    start_time: current_time,
+    end_time: 1000.years.since(current_time),
+    scope_name: 'default'
+  )
+
+  @root_org.elements_under_default_root.create(
+    parent_id: @root_org.id,
+    child_id: @lv2_child_org2.id,
+    distance: 2,
+    start_time: current_time,
+    end_time: 1000.years.since(current_time),
+    scope_name: 'default'
+  )
+
+  @root_org.elements_under_default_root.create(
+    parent_id: @lv1_child_org.id,
+    child_id: @lv2_child_org2.id,
+    distance: 1,
+    start_time: current_time,
+    end_time: 1000.years.since(current_time),
+    scope_name: 'default'
+  )
+
+  @root_org.elements_under_default_root.create(
+    parent_id: @root_org.id,
+    child_id: @lv2_child_org3.id,
+    distance: 2,
+    start_time: current_time,
+    end_time: 1000.years.since(current_time),
+    scope_name: 'default'
+  )
+
+  @root_org.elements_under_default_root.create(
+    parent_id: @lv1_child_org2.id,
+    child_id: @lv2_child_org3.id,
+    distance: 1,
+    start_time: current_time,
+    end_time: 1000.years.since(current_time),
+    scope_name: 'default'
+  )
+
+  @root_org.elements_under_default_root.create(
+    parent_id: @root_org.id,
+    child_id: @lv2_child_org4.id,
+    distance: 2,
+    start_time: current_time,
+    end_time: 1000.years.since(current_time),
+    scope_name: 'default'
+  )
+
+  @root_org.elements_under_default_root.create(
+    parent_id: @lv1_child_org2.id,
+    child_id: @lv2_child_org4.id,
+    distance: 1,
+    start_time: current_time,
+    end_time: 1000.years.since(current_time),
+    scope_name: 'default'
+  )
+
+  @root_org.elements_under_default_root.create(
+    parent_id: @root_org.id,
+    child_id: @lv3_child_org.id,
+    distance: 3,
+    start_time: current_time,
+    end_time: 1000.years.since(current_time),
+    scope_name: 'default'
+  )
+
+  @root_org.elements_under_default_root.create(
+    parent_id: @lv1_child_org.id,
+    child_id: @lv3_child_org.id,
+    distance: 2,
+    start_time: current_time,
+    end_time: 1000.years.since(current_time),
+    scope_name: 'default'
+  )
+
+  @root_org.elements_under_default_root.create(
+    parent_id: @lv2_child_org.id,
+    child_id: @lv3_child_org.id,
+    distance: 1,
+    start_time: current_time,
+    end_time: 1000.years.since(current_time),
+    scope_name: 'default'
+  )
+
+  @root_org.elements_under_default_root.create(
+    parent_id: @root_org.id,
+    child_id: @lv4_child_org.id,
+    distance: 4,
+    start_time: current_time,
+    end_time: 1000.years.since(current_time),
+    scope_name: 'default'
+  )
+
+  @root_org.elements_under_default_root.create(
+    parent_id: @lv1_child_org.id,
+    child_id: @lv4_child_org.id,
+    distance: 3,
+    start_time: current_time,
+    end_time: 1000.years.since(current_time),
+    scope_name: 'default'
+  )
+
+  @root_org.elements_under_default_root.create(
+    parent_id: @lv2_child_org.id,
+    child_id: @lv4_child_org.id,
+    distance: 2,
+    start_time: current_time,
+    end_time: 1000.years.since(current_time),
+    scope_name: 'default'
+  )
+
+  @root_org.elements_under_default_root.create(
+    parent_id: @lv3_child_org.id,
+    child_id: @lv4_child_org.id,
+    distance: 1,
+    start_time: current_time,
+    end_time: 1000.years.since(current_time),
+    scope_name: 'default'
+  )
+
+  @root_org.elements_under_default_root.create(
+    parent_id: @root_org.id,
+    child_id: @lv5_child_org.id,
+    distance: 5,
+    start_time: current_time,
+    end_time: 1000.years.since(current_time),
+    scope_name: 'default'
+  )
+
+  @root_org.elements_under_default_root.create(
+    parent_id: @lv1_child_org.id,
+    child_id: @lv5_child_org.id,
+    distance: 4,
+    start_time: current_time,
+    end_time: 1000.years.since(current_time),
+    scope_name: 'default'
+  )
+
+  @root_org.elements_under_default_root.create(
+    parent_id: @lv2_child_org.id,
+    child_id: @lv5_child_org.id,
+    distance: 3,
+    start_time: current_time,
+    end_time: 1000.years.since(current_time),
+    scope_name: 'default'
+  )
+
+  @root_org.elements_under_default_root.create(
+    parent_id: @lv3_child_org.id,
+    child_id: @lv5_child_org.id,
+    distance: 2,
+    start_time: current_time,
+    end_time: 1000.years.since(current_time),
+    scope_name: 'default'
+  )
+
+  @root_org.elements_under_default_root.create(
+    parent_id: @lv4_child_org.id,
+    child_id: @lv5_child_org.id,
+    distance: 1,
+    start_time: current_time,
+    end_time: 1000.years.since(current_time),
+    scope_name: 'default'
+  )
+
+  @root_org.elements_under_default_root.create(
+    parent_id: @root_org.id,
+    child_id: @lv3_child_org2.id,
+    distance: 3,
+    start_time: current_time,
+    end_time: 1000.years.since(current_time),
+    scope_name: 'default'
+  )
+
+  @root_org.elements_under_default_root.create(
+    parent_id: @lv1_child_org.id,
+    child_id: @lv3_child_org2.id,
+    distance: 2,
+    start_time: current_time,
+    end_time: 1000.years.since(current_time),
+    scope_name: 'default'
+  )
+
+  @root_org.elements_under_default_root.create(
+    parent_id: @lv2_child_org2.id,
+    child_id: @lv3_child_org2.id,
+    distance: 1,
+    start_time: current_time,
+    end_time: 1000.years.since(current_time),
+    scope_name: 'default'
+  )
+
+  @root_org.elements_under_default_root.create(
+    parent_id: @root_org.id,
+    child_id: @lv4_child_org2.id,
+    distance: 4,
+    start_time: current_time,
+    end_time: 1000.years.since(current_time),
+    scope_name: 'default'
+  )
+
+  @root_org.elements_under_default_root.create(
+    parent_id: @lv1_child_org.id,
+    child_id: @lv4_child_org2.id,
+    distance: 3,
+    start_time: current_time,
+    end_time: 1000.years.since(current_time),
+    scope_name: 'default'
+  )
+
+  @root_org.elements_under_default_root.create(
+    parent_id: @lv2_child_org2.id,
+    child_id: @lv4_child_org2.id,
+    distance: 2,
+    start_time: current_time,
+    end_time: 1000.years.since(current_time),
+    scope_name: 'default'
+  )
+
+  @root_org.elements_under_default_root.create(
+    parent_id: @lv3_child_org2.id,
+    child_id: @lv4_child_org2.id,
+    distance: 1,
+    start_time: current_time,
+    end_time: 1000.years.since(current_time),
+    scope_name: 'default'
+  )
+
+  @root_org.elements_under_default_root.create(
+    parent_id: @root_org.id,
+    child_id: @lv5_child_org2.id,
+    distance: 5,
+    start_time: current_time,
+    end_time: 1000.years.since(current_time),
+    scope_name: 'default'
+  )
+
+  @root_org.elements_under_default_root.create(
+    parent_id: @lv1_child_org.id,
+    child_id: @lv5_child_org2.id,
+    distance: 4,
+    start_time: current_time,
+    end_time: 1000.years.since(current_time),
+    scope_name: 'default'
+  )
+
+  @root_org.elements_under_default_root.create(
+    parent_id: @lv2_child_org2.id,
+    child_id: @lv5_child_org2.id,
+    distance: 3,
+    start_time: current_time,
+    end_time: 1000.years.since(current_time),
+    scope_name: 'default'
+  )
+
+  @root_org.elements_under_default_root.create(
+    parent_id: @lv3_child_org2.id,
+    child_id: @lv5_child_org2.id,
+    distance: 2,
+    start_time: current_time,
+    end_time: 1000.years.since(current_time),
+    scope_name: 'default'
+  )
+
+  @root_org.elements_under_default_root.create(
+    parent_id: @lv4_child_org2.id,
+    child_id: @lv5_child_org2.id,
+    distance: 1,
+    start_time: current_time,
+    end_time: 1000.years.since(current_time),
+    scope_name: 'default'
+  )
+end
+
+def destroy_simple_tree
+  Org.destroy_all
+end