linux_admin 0.1.2 → 0.1.3

data/lib/linux_admin.rb

@@ -1,10 +1,11 @@
 require 'more_core_extensions/all'
 require 'active_support/core_ext'
 
+require 'linux_admin/registration_system'
+
 require 'linux_admin/common'
-require 'linux_admin/rhn'
+require 'linux_admin/exceptions'
 require 'linux_admin/rpm'
-require 'linux_admin/subscription_manager'
 require 'linux_admin/version'
 require 'linux_admin/yum'
 
@@ -14,22 +15,11 @@ require 'linux_admin/partition'
 require 'linux_admin/distro'
 require 'linux_admin/system'
 require 'linux_admin/fstab'
+require 'linux_admin/logical_volume'
+require 'linux_admin/physical_volume'
+require 'linux_admin/volume_group'
 
 class LinuxAdmin
   extend Common
   include Common
-
-  def self.registered?
-    !!self.registration_type
-  end
-
-  def self.registration_type
-    if SubscriptionManager.registered?
-      SubscriptionManager
-    elsif Rhn.registered?
-      Rhn
-    else
-      nil
-    end
-  end
 end

data/linux_admin.gemspec

@@ -4,10 +4,14 @@ $LOAD_PATH.unshift(lib) unless $LOAD_PATH.include?(lib)
 require 'linux_admin/version'
 
 Gem::Specification.new do |spec|
+
+  # Dynamically create the authors information {name => e-mail}
+  authors_hash = Hash[`git log --no-merges --reverse --format='%an,%ae'`.split("\n").uniq.collect {|i| i.split(",")}]
+
   spec.name          = "linux_admin"
   spec.version       = LinuxAdmin::VERSION
-  spec.authors       = ["Brandon Dunne"]
-  spec.email         = ["brandondunne@hotmail.com"]
+  spec.authors       = authors_hash.keys
+  spec.email         = authors_hash.values
   spec.description   = %q{
 LinuxAdmin is a module to simplify management of linux systems.
 It should be a single place to manage various system level configurations,

data/spec/common_spec.rb

@@ -27,15 +27,9 @@ describe LinuxAdmin::Common do
 
   subject { TestClass }
 
-  context ".write" do
-    it "no file no content" do
-      expect { subject.write("", "") }.to raise_error(ArgumentError)
-    end
-  end
-
   context ".cmd" do
     it "looks up local command from id" do
-      d = stub(LinuxAdmin::Distro)
+      d = double(LinuxAdmin::Distro)
       d.class::COMMANDS = { :sh => '/bin/sh' }
       LinuxAdmin::Distro.should_receive(:local).and_return(d)
       subject.cmd(:sh).should == '/bin/sh'
@@ -45,17 +39,17 @@ describe LinuxAdmin::Common do
   context ".run" do
     context "with params" do
       it "sanitizes crazy params" do
-        subject.should_receive(:launch).once.with("true --user bob --pass P@\\$sw0\\^\\&\\ \\|\\<\\>/-\\+\\*d\\% --db --desc=Some\\ Description pkg1 some\\ pkg --pool 123 --pool 456")
+        subject.should_receive(:launch).once.with("true --user bob --pass P@\\$sw0\\^\\&\\ \\|\\<\\>/-\\+\\*d\\% --db --desc=Some\\ Description pkg1 some\\ pkg --pool 123 --pool 456", {})
         subject.run("true", :params => modified_params, :return_exitstatus => true)
       end
 
       it "as empty hash" do
-        subject.should_receive(:launch).once.with("true")
+        subject.should_receive(:launch).once.with("true", {})
         subject.run("true", :params => {}, :return_exitstatus => true)
       end
 
       it "as nil" do
-        subject.should_receive(:launch).once.with("true")
+        subject.should_receive(:launch).once.with("true", {})
         subject.run("true", :params => nil, :return_exitstatus => true)
       end
 
@@ -66,44 +60,55 @@ describe LinuxAdmin::Common do
       end
     end
 
-    it "command ok exit ok" do
-      expect(subject.run("true")).to be_true
-    end
-
-    it "command ok exit bad" do
-      expect { subject.run("false") }.to raise_error
-    end
-
-    it "command bad" do
-      expect { subject.run("XXXXX") }.to raise_error
-    end
+    context "with real execution" do
+      before do
+        Kernel.should_receive(:spawn).and_call_original
+      end
 
-    context "with :return_exitstatus => true" do
       it "command ok exit ok" do
-        expect(subject.run("true", :return_exitstatus => true)).to eq(0)
+        expect(subject.run("true")).to be_true
       end
 
       it "command ok exit bad" do
-        expect(subject.run("false", :return_exitstatus => true)).to eq(1)
+        expect { subject.run("false") }.to raise_error(CommandError)
       end
 
       it "command bad" do
-        expect(subject.run("XXXXX", :return_exitstatus => true)).to be_nil
+        expect { subject.run("XXXXX") }.to raise_error(Errno::ENOENT)
       end
-    end
 
-    context "with :return_output => true" do
-      it "command ok exit ok" do
-        expect(subject.run("echo \"Hello World\"", :return_output => true)).to eq("Hello World\n")
-      end
+      context "with :return_exitstatus => true" do
+        it "command ok exit ok" do
+          expect(subject.run("true", :return_exitstatus => true)).to eq(0)
+        end
 
-      it "command ok exit bad" do
-        expect { subject.run("false", :return_output => true) }.to raise_error
+        it "command ok exit bad" do
+          expect(subject.run("false", :return_exitstatus => true)).to eq(1)
+        end
+
+        it "command bad" do
+          expect(subject.run("XXXXX", :return_exitstatus => true)).to be_nil
+        end
       end
 
-      it "command bad" do
-        expect { subject.run("XXXXX", :return_output => true) }.to raise_error
+      context "with :return_output => true" do
+        it "command ok exit ok" do
+          expect(subject.run("echo \"Hello World\"", :return_output => true)).to eq("Hello World\n")
+        end
+
+        it "command ok exit bad" do
+          expect { subject.run("false", :return_output => true) }.to raise_error(CommandError)
+        end
+
+        it "command bad" do
+          expect { subject.run("XXXXX", :return_output => true) }.to raise_error(Errno::ENOENT)
+        end
       end
     end
+
+    it "supports spawn's chdir option" do
+      subject.should_receive(:launch).once.with("true", {:chdir => ".."})
+      subject.run("true", :chdir => "..", :return_exitstatus => true)
+    end
   end
 end

data/spec/data/subscription_manager/output_orgs

@@ -0,0 +1,6 @@
++-------------------------------------------+
+          myUser Organizations
++-------------------------------------------+
+
+Name:   SomeOrg
+Key:  1234567

data/spec/disk_spec.rb

@@ -12,16 +12,56 @@ describe LinuxAdmin::Disk do
     end
   end
 
+  describe "#size" do
+    it "uses fdisk" do
+      disk = LinuxAdmin::Disk.new :path => '/dev/hda'
+      disk.should_receive(:run).
+        with(disk.cmd(:fdisk),
+             :return_output => true,
+             :params => {"-l" => nil}).
+        and_return("")
+      disk.size
+    end
+
+    it "returns disk size" do
+      fdisk = <<eos
+Disk /dev/hda: 500.1 GB, 500107862016 bytes
+255 heads, 63 sectors/track, 60801 cylinders, total 976773168 sectors
+Units = sectors of 1 * 512 = 512 bytes
+Sector size (logical/physical): 512 bytes / 512 bytes
+I/O size (minimum/optimal): 512 bytes / 512 bytes
+Disk identifier: 0x3ddb508b
+
+   Device Boot      Start         End      Blocks   Id  System
+ 1      1259MB  81.8GB  80.5GB  primary   ntfs
+ 2      81.8GB  162GB   80.5GB  primary   ext4
+ 3      162GB   163GB   1074MB  logical   linux-swap(v1)
+eos
+
+      disk = LinuxAdmin::Disk.new :path => '/dev/hda'
+      disk.stub(:run).and_return(fdisk)
+      disk.size.should == 500.1.gigabytes
+    end
+  end
+
   describe "#partitions" do
     it "uses parted" do
       disk = LinuxAdmin::Disk.new :path => '/dev/hda'
       disk.should_receive(:run).
         with(disk.cmd(:parted),
+             :return_exitstatus => true,
              :return_output => true,
              :params => { nil => ['/dev/hda', 'print'] }).and_return ""
       disk.partitions
     end
 
+    it "returns [] on non-zero parted rc" do
+      disk = LinuxAdmin::Disk.new :path => '/dev/hda'
+      disk.stub(:exitstatus => 1)
+      disk.stub(:launch)
+      disk.partitions.should == []
+    end
+
     it "sets partitons" do
       partitions = <<eos
 Model: ATA TOSHIBA MK5061GS (scsi)
@@ -41,15 +81,92 @@ eos
       disk.partitions[0].id.should == 1
       disk.partitions[0].disk.should == disk
       disk.partitions[0].size.should == 80.5.gigabytes
+      disk.partitions[0].start_sector.should == 1259.megabytes
+      disk.partitions[0].end_sector.should == 81.8.gigabytes
+      disk.partitions[0].partition_type.should == 'primary'
       disk.partitions[0].fs_type.should == 'ntfs'
       disk.partitions[1].id.should == 2
       disk.partitions[1].disk.should == disk
       disk.partitions[1].size.should == 80.5.gigabytes
+      disk.partitions[1].start_sector.should == 81.8.gigabytes
+      disk.partitions[1].end_sector.should == 162.gigabytes
+      disk.partitions[1].partition_type.should == 'primary'
       disk.partitions[1].fs_type.should == 'ext4'
       disk.partitions[2].id.should == 3
       disk.partitions[2].disk.should == disk
       disk.partitions[2].size.should == 1074.megabytes
+      disk.partitions[2].start_sector.should == 162.gigabytes
+      disk.partitions[2].end_sector.should == 163.gigabytes
+      disk.partitions[2].partition_type.should == 'logical'
       disk.partitions[2].fs_type.should == 'linux-swap(v1)'
     end
   end
+
+  describe "#create_partition" do
+    before(:each) do
+      # test disk w/ existing partition
+      @disk = LinuxAdmin::Disk.new :path => '/dev/hda'
+      @disk.instance_variable_set(:@partitions,
+                                  [LinuxAdmin::Partition.new(:id => 1,
+                                                 :end_sector => 1024)])
+    end
+
+    it "uses parted" do
+      @disk.should_receive(:run).
+        with(@disk.cmd(:parted),
+             :params => { nil => ['/dev/hda', 'mkpart', 'primary', 1024, 2048] })
+      @disk.create_partition 'primary', 1024
+    end
+
+    it "returns partition" do
+      @disk.should_receive(:run) # stub out call to parted
+      partition = @disk.create_partition 'primary', 1024
+      partition.should be_an_instance_of(LinuxAdmin::Partition)
+    end
+
+    it "increments partition id" do
+      @disk.should_receive(:run) # stub out call to parted
+      partition = @disk.create_partition 'primary', 1024
+      partition.id.should == 2
+    end
+
+    it "sets partition start to first unused sector on disk" do
+      @disk.should_receive(:run) # stub out call to parted
+      partition = @disk.create_partition 'primary', 1024
+      partition.start_sector.should == 1024
+    end
+
+    it "stores new partition locally" do
+      @disk.should_receive(:run) # stub out call to parted
+      lambda {
+        @disk.create_partition 'primary', 1024
+      }.should change{@disk.partitions.size}.by(1)
+    end
+  end
+
+  describe "#clear!" do
+    it "clears partitions" do
+      disk = LinuxAdmin::Disk.new :path => '/dev/hda'
+      disk.stub(:run).and_return("") # stub out call to cmds
+      disk.partitions << LinuxAdmin::Partition.new
+      disk.clear!
+      disk.partitions.should be_empty
+    end
+
+    it "uses dd to clear partition table" do
+      disk = LinuxAdmin::Disk.new :path => '/dev/hda'
+      disk.should_receive(:run).
+           with(disk.cmd(:dd),
+                :params => {'if=' => '/dev/zero', 'of=' => '/dev/hda',
+                            'bs=' => 512, 'count=' => 1})
+      disk.clear!
+    end
+
+    it "returns self" do
+      disk = LinuxAdmin::Disk.new :path => '/dev/hda'
+      disk.stub(:run) # stub out call to dd
+      disk.clear!.should == disk
+    end
+  end
+
 end

data/spec/distro_spec.rb

@@ -45,7 +45,7 @@ describe LinuxAdmin::Distro do
     end
 
     it "returns Distros.generic" do
-      File.stub!(:exists?).and_return(false)
+      File.stub(:exists?).and_return(false)
       LinuxAdmin::Distro.local.should == LinuxAdmin::Distros.generic
     end
   end

data/spec/fstab_spec.rb

@@ -1,4 +1,5 @@
 require 'spec_helper'
+require 'stringio'
 
 describe LinuxAdmin::FSTab do
   it "creates FSTabEntry for each line in fstab" do
@@ -24,4 +25,21 @@ eos
     entries[1].dumpable.should == 0
     entries[1].fsck_order.should == 0
   end
+
+  describe "#write!" do
+    it "writes entries to /etc/fstab" do
+      # maually set fstab
+      entry = LinuxAdmin::FSTabEntry.new
+      entry.device        = '/dev/sda1'
+      entry.mount_point   = '/'
+      entry.fs_type       = 'ext4'
+      entry.mount_options = 'defaults'
+      entry.dumpable      = 1
+      entry.fsck_order    = 1
+      LinuxAdmin::FSTab.instance.entries = [entry]
+
+      File.should_receive(:write).with('/etc/fstab', "/dev/sda1 / ext4 defaults 1 1\n")
+      LinuxAdmin::FSTab.instance.write!
+    end
+  end
 end

data/spec/linux_admin_spec.rb

@@ -1,42 +1,4 @@
 require 'spec_helper'
 
 describe LinuxAdmin do
-  context ".registered?" do
-    it "when registered Subscription Manager" do
-      stub_registered_to_system(:sm)
-      expect(described_class.registered?).to be_true
-    end
-
-    it "when registered RHN" do
-      stub_registered_to_system(:rhn)
-      expect(described_class.registered?).to be_true
-    end
-
-    it "when unregistered" do
-      stub_registered_to_system(nil)
-      expect(described_class.registered?).to be_false
-    end
-  end
-
-  context ".registration_type" do
-    it "when registered Subscription Manager" do
-      stub_registered_to_system(:sm)
-      expect(described_class.registration_type).to eq(LinuxAdmin::SubscriptionManager)
-    end
-
-    it "when registered RHN" do
-      stub_registered_to_system(:rhn)
-      expect(described_class.registration_type).to eq(LinuxAdmin::Rhn)
-    end
-
-    it "when unregistered" do
-      stub_registered_to_system(nil)
-      expect(described_class.registration_type).to be_nil
-    end
-  end
-
-  def stub_registered_to_system(system)
-    described_class::SubscriptionManager.stub(:registered? => (system == :sm))
-    described_class::Rhn.stub(:registered? => (system == :rhn))
-  end
 end

data/spec/logical_volume_spec.rb

@@ -0,0 +1,108 @@
+require 'spec_helper'
+
+describe LinuxAdmin::LogicalVolume do
+  before(:each) do
+    LinuxAdmin::Distro.stub(:local => LinuxAdmin::Distros::Test.new)
+
+    @logical_volumes = <<eos
+/dev/vg_foobar/lv_swap:vg_foobar:3:1:-1:2:4128768:63:-1:0:-1:253:0
+/dev/vg_foobar/lv_root:vg_foobar:3:1:-1:1:19988480:305:-1:0:-1:253:1
+eos
+
+    @groups = <<eos
+vg_foobar:r/w:772:-1:0:2:2:-1:0:1:1:12058624:32768:368:368:0:tILZUF-IspH-H90I-pT5j-vVFl-b76L-zWx3CW
+eos
+  end
+
+  after(:each) do
+    # reset local copies of volumes / groups
+    described_class.instance_variable_set(:@lvs, nil)
+    LinuxAdmin::PhysicalVolume.instance_variable_set(:@pvs, nil)
+    LinuxAdmin::VolumeGroup.instance_variable_set(:@vgs, nil)
+  end
+
+  describe "#extend_with" do
+    it "uses lvextend" do
+      lv = described_class.new :name => 'lv'
+      vg = LinuxAdmin::VolumeGroup.new :name => 'vg'
+      lv.should_receive(:run).
+         with(vg.cmd(:lvextend),
+              :params => ['lv', 'vg'])
+      lv.extend_with(vg)
+    end
+
+    it "returns self" do
+      lv = described_class.new :name => 'lv'
+      vg = LinuxAdmin::VolumeGroup.new :name => 'vg'
+      lv.stub(:run)
+      lv.extend_with(vg).should == lv
+    end
+  end
+
+  describe "#create" do
+    before(:each) do
+      @vg = LinuxAdmin::VolumeGroup.new :name => 'vg'
+    end
+
+    it "uses lvcreate" do
+      described_class.instance_variable_set(:@lvs, [])
+      described_class.should_receive(:run).
+                                with(LinuxAdmin.cmd(:lvcreate),
+                                     :params => { '-n' => 'lv',
+                                                   nil => 'vg',
+                                                  '-L' => '256G' })
+      described_class.create 'lv', @vg, '256G'
+    end
+
+    it "returns new logical volume" do
+      LinuxAdmin::VolumeGroup.stub(:run => "")
+      described_class.stub(:run => "")
+      lv = described_class.create 'lv', @vg, '256G'
+      lv.should be_an_instance_of(described_class)
+      lv.name.should == 'lv'
+    end
+
+    it "adds logical volume to local registry" do
+      LinuxAdmin::VolumeGroup.stub(:run => "")
+      described_class.stub(:run => "")
+      lv = described_class.create 'lv', @vg, '256G'
+      described_class.scan.should include(lv)
+    end
+  end
+
+  describe "#scan" do
+    it "uses lvdisplay" do
+      described_class.should_receive(:run).
+                                with(LinuxAdmin.cmd(:lvdisplay),
+                                     :return_output => true,
+                                     :params => { '-c' => nil}).
+                                and_return(@logical_volumes)
+      LinuxAdmin::VolumeGroup.should_receive(:run).and_return(@groups) # stub out call to vgdisplay
+      described_class.scan
+    end
+
+    it "returns local logical volumes" do
+      described_class.should_receive(:run).and_return(@logical_volumes)
+      LinuxAdmin::VolumeGroup.should_receive(:run).and_return(@groups)
+      lvs = described_class.scan
+
+      lvs[0].should be_an_instance_of(described_class)
+      lvs[0].name.should == '/dev/vg_foobar/lv_swap'
+      lvs[0].sectors.should == 4128768
+
+      lvs[1].should be_an_instance_of(described_class)
+      lvs[1].name.should == '/dev/vg_foobar/lv_root'
+      lvs[1].sectors.should == 19988480
+    end
+
+    it "resolves volume group references" do
+      described_class.should_receive(:run).and_return(@logical_volumes)
+      LinuxAdmin::VolumeGroup.should_receive(:run).and_return(@groups)
+      lvs = described_class.scan
+      lvs[0].volume_group.should be_an_instance_of(LinuxAdmin::VolumeGroup)
+      lvs[0].volume_group.name.should == 'vg_foobar'
+      lvs[1].volume_group.should be_an_instance_of(LinuxAdmin::VolumeGroup)
+      lvs[1].volume_group.name.should == 'vg_foobar'
+    end
+  end
+end

data/spec/partition_spec.rb

@@ -16,6 +16,21 @@ describe LinuxAdmin::Partition do
     end
   end
 
+  describe "#format_to" do
+    it "uses mke2fs" do
+      @partition.should_receive(:run).
+         with(@partition.cmd(:mke2fs),
+              :params => { '-t' => 'ext4', nil => '/dev/sda2'})
+      @partition.format_to('ext4')
+    end
+
+    it "sets fs type" do
+      @partition.should_receive(:run) # ignore actual formatting cmd
+      @partition.format_to('ext4')
+      @partition.fs_type.should == 'ext4'
+    end
+  end
+
   describe "#mount" do
     it "sets mount point" do
       @partition.should_receive(:run) # ignore actual mount cmd

data/spec/physical_volume_spec.rb

@@ -0,0 +1,110 @@
+require 'spec_helper'
+
+describe LinuxAdmin::PhysicalVolume do
+  before(:each) do
+    LinuxAdmin::Distro.stub(:local => LinuxAdmin::Distros::Test.new)
+
+    @physical_volumes = <<eos
+/dev/vda2:vg_foobar:24139776:-1:8:8:-1:32768:368:0:368:pxR32D-YkC2-PfHe-zOwb-eaGD-9Ar0-mAOl9u
+eos
+
+    @groups = <<eos
+vg_foobar:r/w:772:-1:0:2:2:-1:0:1:1:12058624:32768:368:368:0:tILZUF-IspH-H90I-pT5j-vVFl-b76L-zWx3CW
+eos
+  end
+
+  after(:each) do
+    # reset local copies of volumes / groups
+    LinuxAdmin::LogicalVolume.instance_variable_set(:@lvs, nil)
+    described_class.instance_variable_set(:@pvs, nil)
+    LinuxAdmin::VolumeGroup.instance_variable_set(:@vgs, nil)
+  end
+
+  describe "#attach_to" do
+    it "uses vgextend" do
+      vg = LinuxAdmin::VolumeGroup.new :name => 'vg'
+      pv = described_class.new :device_name => '/dev/hda'
+      pv.should_receive(:run).
+         with(pv.cmd(:vgextend),
+              :params => ['vg', '/dev/hda'])
+      pv.attach_to(vg)
+    end
+
+    it "assigns volume group to physical volume" do
+      vg = LinuxAdmin::VolumeGroup.new :name => 'vg'
+      pv = described_class.new :device_name => '/dev/hda'
+      pv.stub(:run)
+      pv.attach_to(vg)
+      pv.volume_group.should == vg
+    end
+
+    it "returns self" do
+      vg = LinuxAdmin::VolumeGroup.new :name => 'vg'
+      pv = described_class.new :device_name => '/dev/hda'
+      pv.stub(:run)
+      pv.attach_to(vg).should == pv
+    end
+  end
+
+  describe "#create" do
+    before do
+      @disk = LinuxAdmin::Disk.new :path => '/dev/hda'
+      @disk.stub(:size)
+    end
+
+    let(:disk) {@disk}
+
+    it "uses pvcreate" do
+      described_class.instance_variable_set(:@pvs, [])
+      described_class.should_receive(:run).
+                                 with(LinuxAdmin.cmd(:pvcreate),
+                                      :params => { nil => '/dev/hda'})
+      described_class.create disk
+    end
+
+    it "returns new physical volume" do
+      LinuxAdmin::VolumeGroup.stub(:run => "")
+      described_class.stub(:run => "")
+      pv = described_class.create disk
+      pv.should be_an_instance_of(described_class)
+      pv.device_name.should == '/dev/hda'
+    end
+
+    it "adds physical volume to local registry" do
+      LinuxAdmin::VolumeGroup.stub(:run => "")
+      described_class.stub(:run => "")
+      pv = described_class.create disk
+      described_class.scan.should include(pv)
+    end
+  end
+
+  describe "#scan" do
+    it "uses pvdisplay" do
+      described_class.should_receive(:run).
+                                 with(LinuxAdmin.cmd(:pvdisplay),
+                                     :return_output => true,
+                                     :params => { '-c' => nil}).
+                                 and_return(@physical_volumes)
+      LinuxAdmin::VolumeGroup.should_receive(:run).and_return(@groups) # stub out call to vgdisplay
+      described_class.scan
+    end
+
+    it "returns local physical volumes" do
+      described_class.should_receive(:run).and_return(@physical_volumes)
+      LinuxAdmin::VolumeGroup.should_receive(:run).and_return(@groups)
+      pvs = described_class.scan
+
+      pvs[0].should be_an_instance_of(described_class)
+      pvs[0].device_name.should == '/dev/vda2'
+      pvs[0].size.should == 24139776
+    end
+
+    it "resolves volume group references" do
+      described_class.should_receive(:run).and_return(@physical_volumes)
+      LinuxAdmin::VolumeGroup.should_receive(:run).and_return(@groups)
+      pvs = described_class.scan
+      pvs[0].volume_group.should be_an_instance_of(LinuxAdmin::VolumeGroup)
+      pvs[0].volume_group.name.should == 'vg_foobar'
+    end
+  end
+end