pg_audit_log 0.1.0 → 0.1.1

data/.rvmrc

@@ -1 +1 @@
-rvm 1.9.2-p136@pg_audit_log
+rvm 1.9.2-p136@pg_audit_log --create

data/lib/pg_audit_log/function.rb

@@ -1,5 +1,5 @@
 module PgAuditLog
-  class Function < ActiveRecord
+  class Function < PgAuditLog::ActiveRecord
     class << self
       def install
         execute <<-SQL

data/lib/pg_audit_log/triggers.rb

@@ -1,5 +1,5 @@
 module PgAuditLog
-  class Triggers < ActiveRecord
+  class Triggers < PgAuditLog::ActiveRecord
     class << self
       def tables
         connection.tables - (PgAuditLog::IGNORED_TABLES + [PgAuditLog::Entry.table_name])
@@ -17,6 +17,27 @@ module PgAuditLog
         end
       end
 
+      def enable
+        tables.each do |table|
+          enable_for_table(table)
+        end
+      end
+
+      def disable
+        tables.each do |table|
+          disable_for_table(table)
+        end
+      end
+
+      def without_triggers
+        begin
+          disable
+          yield
+        ensure
+          enable
+        end
+      end
+
       def create_for_table(table_name)
         execute <<-SQL
         CREATE TRIGGER audit_#{table_name}
@@ -28,9 +49,20 @@ module PgAuditLog
       end
 
       def drop_for_table(table_name)
-        execute "DROP TRIGGER audit_#{table_name} ON #{table_name};"
+        execute "DROP TRIGGER #{trigger_name_for_table(table_name)} ON #{table_name}"
+      end
+
+      def enable_for_table(table_name)
+        execute "ALTER TABLE #{table_name} DISABLE TRIGGER #{trigger_name_for_table(table_name)}"
+      end
+
+      def disable_for_table(table_name)
+        execute "ALTER TABLE #{table_name} ENABLE TRIGGER #{trigger_name_for_table(table_name)}"
       end
 
+      def trigger_name_for_table(table_name)
+        "audit_#{table_name}"
+      end
     end
   end
 end

data/lib/pg_audit_log/version.rb

@@ -1,3 +1,3 @@
 module PgAuditLog
-  VERSION = "0.1.0"
+  VERSION = "0.1.1"
 end

data/spec/triggers_spec.rb

@@ -0,0 +1,53 @@
+require File.expand_path(File.dirname(__FILE__) + '/spec_helper')
+
+describe PgAuditLog::Triggers do
+  before :each do
+    PgAuditLog::Triggers.uninstall rescue nil
+  end
+
+  with_model :table_with_triggers do
+    table {}
+  end
+
+  describe ".install" do
+    it "should work" do
+      ->{
+        PgAuditLog::Triggers.install
+      }.should_not raise_error
+    end
+  end
+
+  context "when triggers are installed" do
+    before do
+      PgAuditLog::Triggers.install
+    end
+
+    describe ".uninstall" do
+      it "should work" do
+        ->{
+          PgAuditLog::Triggers.uninstall
+        }.should_not raise_error
+      end
+    end
+
+    describe ".enable" do
+      it "should work" do
+        ->{
+          PgAuditLog::Triggers.enable
+        }.should_not raise_error
+      end
+    end
+
+    describe ".disable" do
+      it "should work" do
+        ->{
+          PgAuditLog::Triggers.disable
+        }.should_not raise_error
+      end
+    end
+
+  end
+
+
+end
+

metadata

@@ -2,7 +2,7 @@
 name: pg_audit_log
 version: !ruby/object:Gem::Version 
   prerelease: 
-  version: 0.1.0
+  version: 0.1.1
 platform: ruby
 authors: 
 - Case Commons, LLC
@@ -10,7 +10,7 @@ autorequire:
 bindir: bin
 cert_chain: []
 
-date: 2011-03-29 00:00:00 -04:00
+date: 2011-04-15 00:00:00 -04:00
 default_executable: 
 dependencies: 
 - !ruby/object:Gem::Dependency 
@@ -93,6 +93,7 @@ files:
 - spec/model_spec.rb
 - spec/pg_audit_log_spec.rb
 - spec/spec_helper.rb
+- spec/triggers_spec.rb
 has_rdoc: true
 homepage: https://github.com/Casecommons/pg_audit_log
 licenses: []
@@ -127,3 +128,4 @@ test_files:
 - spec/model_spec.rb
 - spec/pg_audit_log_spec.rb
 - spec/spec_helper.rb
+- spec/triggers_spec.rb