linux_admin 0.1.1 → 0.1.2

data/lib/linux_admin/common.rb

@@ -1,6 +1,8 @@
 require 'shellwords'
 
 class LinuxAdmin
+  class CommandError < RuntimeError; end
+
   module Common
     def write(file, content)
       raise ArgumentError, "file and content can not be empty" if file.blank? || content.blank?
@@ -24,7 +26,7 @@ class LinuxAdmin
         elsif options[:return_exitstatus] || exitstatus == 0
           exitstatus
         else
-          raise "Error: Exit Code #{exitstatus}"
+          raise CommandError, "#{build_cmd(cmd, params)}: exit code: #{exitstatus}"
         end
       rescue
         return nil if options[:return_exitstatus]

data/lib/linux_admin/fstab.rb

@@ -0,0 +1,49 @@
+# LinuxAdmin fstab Representation
+#
+# Copyright (C) 2013 Red Hat Inc.
+# Licensed under the MIT License
+
+require 'singleton'
+
+class LinuxAdmin
+  class FSTabEntry < LinuxAdmin
+    attr_accessor :device
+    attr_accessor :mount_point
+    attr_accessor :fs_type
+    attr_accessor :mount_options
+    attr_accessor :dumpable
+    attr_accessor :fsck_order
+  end
+
+  class FSTab < LinuxAdmin
+    include Singleton
+
+    attr_accessor :entries
+
+    def initialize
+      refresh
+    end
+
+    private
+
+    def refresh
+      @entries = []
+      f = File.read('/etc/fstab')
+      f.each_line { |line|
+        first_char = line.strip[0] 
+        if first_char != '#' && first_char !~ /\s/
+          columns = line.split
+          entry   = FSTabEntry.new
+          entry.device         = columns[0]
+          entry.mount_point    = columns[1]
+          entry.fs_type        = columns[2]
+          entry.mount_options  = columns[3]
+          entry.dumpable       = columns[4].to_i
+          entry.fsck_order     = columns[5].to_i
+          @entries << entry
+        end
+      }
+      self
+    end
+  end
+end

data/lib/linux_admin/service.rb

@@ -20,21 +20,40 @@ class LinuxAdmin
     def enable
       run(cmd(:systemctl),
           :params => { nil => ["enable", "#{id}.service"] })
+      self
     end
 
     def disable
       run(cmd(:systemctl),
           :params => { nil => ["disable", "#{id}.service"] })
+      self
     end
 
     def start
       run(cmd(:service),
           :params => { nil => [id, "start"] })
+      self
     end
 
     def stop
       run(cmd(:service),
           :params => { nil => [id, "stop"] })
+      self
+    end
+
+    def restart
+      status =
+        run(cmd(:service),
+          :params => { nil => [id, "restart"] },
+          :return_exitstatus => true)
+
+      # attempt to manually stop/start if restart fails
+      if status != 0
+        self.stop
+        self.start
+      end
+
+      self
     end
   end
 end

data/lib/linux_admin/version.rb

@@ -1,3 +1,3 @@
 class LinuxAdmin
-  VERSION = "0.1.1"
+  VERSION = "0.1.2"
 end

data/lib/linux_admin.rb

@@ -13,6 +13,7 @@ require 'linux_admin/disk'
 require 'linux_admin/partition'
 require 'linux_admin/distro'
 require 'linux_admin/system'
+require 'linux_admin/fstab'
 
 class LinuxAdmin
   extend Common

data/spec/fstab_spec.rb

@@ -0,0 +1,27 @@
+require 'spec_helper'
+
+describe LinuxAdmin::FSTab do
+  it "creates FSTabEntry for each line in fstab" do
+    fstab = <<eos
+/dev/sda1 / ext4  defaults  1 1
+/dev/sda2 swap  swap  defaults  0 0
+eos
+    File.should_receive(:read).with('/etc/fstab').and_return(fstab)
+    entries = LinuxAdmin::FSTab.instance.entries
+    entries.size.should == 2
+
+    entries[0].device.should == '/dev/sda1'
+    entries[0].mount_point.should == '/'
+    entries[0].fs_type.should == 'ext4'
+    entries[0].mount_options.should == 'defaults'
+    entries[0].dumpable.should == 1
+    entries[0].fsck_order.should == 1
+
+    entries[1].device.should == '/dev/sda2'
+    entries[1].mount_point.should == 'swap'
+    entries[1].fs_type.should == 'swap'
+    entries[1].mount_options.should == 'defaults'
+    entries[1].dumpable.should == 0
+    entries[1].fsck_order.should == 0
+  end
+end

data/spec/service_spec.rb

@@ -42,6 +42,11 @@ describe LinuxAdmin::Service do
               :params => { nil => [ 'enable', 'foo.service']})
       @service.enable
     end
+
+    it "returns self" do
+      @service.should_receive(:run) # stub out cmd invocation
+      @service.enable.should == @service
+    end
   end
 
   describe "#disable" do
@@ -51,6 +56,11 @@ describe LinuxAdmin::Service do
               :params => { nil => [ 'disable', 'foo.service']})
       @service.disable
     end
+
+    it "returns self" do
+      @service.should_receive(:run)
+      @service.disable.should == @service
+    end
   end
 
   describe "#start" do
@@ -60,6 +70,11 @@ describe LinuxAdmin::Service do
               :params => { nil => [ 'foo', 'start']})
       @service.start
     end
+
+    it "returns self" do
+      @service.should_receive(:run)
+      @service.start.should == @service
+    end
   end
 
   describe "#stop" do
@@ -69,5 +84,35 @@ describe LinuxAdmin::Service do
               :params => { nil => [ 'foo', 'stop']})
       @service.stop
     end
+
+    it "returns self" do
+      @service.should_receive(:run)
+      @service.stop.should == @service
+    end
   end
+
+  describe "#restart" do
+    it "stops service" do
+      @service.should_receive(:run).
+         with(@service.cmd(:service),
+              :params => { nil => [ 'foo', 'restart']},
+              :return_exitstatus => true).and_return(0)
+      @service.restart
+    end
+
+    context "service restart fails" do
+      it "manually stops/starts service" do
+        @service.should_receive(:run).and_return(1)
+        @service.should_receive(:stop)
+        @service.should_receive(:start)
+        @service.restart
+      end
+    end
+
+    it "returns self" do
+      @service.should_receive(:run).and_return(0)
+      @service.restart.should == @service
+    end
+  end
+
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: linux_admin
 version: !ruby/object:Gem::Version
-  version: 0.1.1
+  version: 0.1.2
   prerelease: 
 platform: ruby
 authors:
@@ -9,7 +9,7 @@ authors:
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2013-07-08 00:00:00.000000000 Z
+date: 2013-07-11 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: bundler
@@ -165,6 +165,7 @@ files:
 - lib/linux_admin/common.rb
 - lib/linux_admin/disk.rb
 - lib/linux_admin/distro.rb
+- lib/linux_admin/fstab.rb
 - lib/linux_admin/logical_volume.rb
 - lib/linux_admin/partition.rb
 - lib/linux_admin/rhn.rb
@@ -185,6 +186,7 @@ files:
 - spec/data/yum/second.repo
 - spec/disk_spec.rb
 - spec/distro_spec.rb
+- spec/fstab_spec.rb
 - spec/linux_admin_spec.rb
 - spec/partition_spec.rb
 - spec/rhn_spec.rb
@@ -207,18 +209,12 @@ required_ruby_version: !ruby/object:Gem::Requirement
   - - ! '>='
     - !ruby/object:Gem::Version
       version: '0'
-      segments:
-      - 0
-      hash: -2093528175376818552
 required_rubygems_version: !ruby/object:Gem::Requirement
   none: false
   requirements:
   - - ! '>='
     - !ruby/object:Gem::Version
       version: '0'
-      segments:
-      - 0
-      hash: -2093528175376818552
 requirements: []
 rubyforge_project: 
 rubygems_version: 1.8.25
@@ -236,6 +232,7 @@ test_files:
 - spec/data/yum/second.repo
 - spec/disk_spec.rb
 - spec/distro_spec.rb
+- spec/fstab_spec.rb
 - spec/linux_admin_spec.rb
 - spec/partition_spec.rb
 - spec/rhn_spec.rb