fake_dynamo 0.1.1 → 0.1.2

data/README.md

@@ -20,6 +20,12 @@ gem install fake_dynamo
 fake_dynamo --port 4567
 ````
 
+send a DELETE request to reset the database. eg
+
+````
+curl -X DELETE http://localhost:4567
+````
+
 # Clients
 
 * aws-sdk

data/bin/fake_dynamo

@@ -19,8 +19,35 @@ OptionParser.new do |opts|
   opts.on("-c", "--compact", "Default: #{options[:compact]}") do |v|
     options[:compact] = v
   end
+
+  opts.on "-P", "--pid PIDFILE", "Where to write the pid" do |pid|
+    options[:pid] = pid
+  end
+
+  opts.on "-d", "--[no-]daemonize", "Detaches the process" do |daemonize|
+    options[:daemonize] = daemonize
+  end
+
 end.parse!
 
+if options[:daemonize]
+  Process.daemon(true, true)
+end
+
+if (pid = options[:pid])
+  if File.exist?(pid)
+    existing_pid = File.open(pid, 'r').read.chomp.to_i
+    running = Process.getpgid(existing_pid) rescue false
+    if running
+      warn "Error, Process #{existing_pid} already running"
+      exit 1
+    else
+      warn "Cleaning up stale pid at #{pid}"
+    end
+  end
+  File.open(pid, 'w') { |f| f.write(Process.pid) }
+end
+
 FakeDynamo::Storage.db_path = options[:db]
 
 if options[:compact]

data/lib/fake_dynamo/attribute.rb

@@ -5,7 +5,7 @@ module FakeDynamo
     def initialize(name, value, type)
       @name, @value, @type = name, value, type
 
-      if @type == 'B'
+      if @type == 'B' and value
         @value = Base64.decode64(value)
       end
 

data/lib/fake_dynamo/db.rb

@@ -15,6 +15,10 @@ module FakeDynamo
       @tables = {}
     end
 
+    def reset
+      @tables = {}
+    end
+
     def process(operation, data)
       validate_payload(operation, data)
       operation = operation.underscore

data/lib/fake_dynamo/server.rb

@@ -26,6 +26,12 @@ module FakeDynamo
       [status, response.to_json]
     end
 
+    delete '/' do
+      db.reset
+      storage.reset
+      {success: true}.to_json
+    end
+
     def db
       DB.instance
     end

data/lib/fake_dynamo/storage.rb

@@ -41,6 +41,13 @@ module FakeDynamo
       FileUtils.rm(db_path)
     end
 
+    def reset
+      puts "resetting database ..."
+      @aof.close if @aof
+      @aof = nil
+      delete_db
+    end
+
     def db
       DB.instance
     end

data/lib/fake_dynamo/version.rb

@@ -1,3 +1,3 @@
 module FakeDynamo
-  VERSION = "0.1.1"
+  VERSION = "0.1.2"
 end

data/spec/fake_dynamo/server_spec.rb

@@ -34,5 +34,20 @@ module FakeDynamo
       last_response.status.should eq(400)
     end
 
+    it "should reset database" do
+      post '/', {}.to_json, 'HTTP_X_AMZ_TARGET' => 'DynamoDB_20111205.ListTables'
+      JSON.parse(last_response.body)["TableNames"].size.should == 1
+
+      delete '/'
+      last_response.should be_ok
+
+      post '/', {}.to_json, 'HTTP_X_AMZ_TARGET' => 'DynamoDB_20111205.ListTables'
+      JSON.parse(last_response.body)["TableNames"].size.should == 0
+
+      post '/', data.to_json, 'HTTP_X_AMZ_TARGET' => 'DynamoDB_20111205.CreateTable'
+      post '/', {}.to_json, 'HTTP_X_AMZ_TARGET' => 'DynamoDB_20111205.ListTables'
+      JSON.parse(last_response.body)["TableNames"].size.should == 1
+    end
+
   end
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: fake_dynamo
 version: !ruby/object:Gem::Version
-  version: 0.1.1
+  version: 0.1.2
   prerelease: 
 platform: ruby
 authors:
@@ -9,11 +9,11 @@ authors:
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2012-11-11 00:00:00.000000000Z
+date: 2013-03-23 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: sinatra
-  requirement: &70342224155740 !ruby/object:Gem::Requirement
+  requirement: &70338580048120 !ruby/object:Gem::Requirement
     none: false
     requirements:
     - - ! '>='
@@ -21,10 +21,10 @@ dependencies:
         version: '0'
   type: :runtime
   prerelease: false
-  version_requirements: *70342224155740
+  version_requirements: *70338580048120
 - !ruby/object:Gem::Dependency
   name: activesupport
-  requirement: &70342224155280 !ruby/object:Gem::Requirement
+  requirement: &70338580047540 !ruby/object:Gem::Requirement
     none: false
     requirements:
     - - ! '>='
@@ -32,10 +32,10 @@ dependencies:
         version: '0'
   type: :runtime
   prerelease: false
-  version_requirements: *70342224155280
+  version_requirements: *70338580047540
 - !ruby/object:Gem::Dependency
   name: json
-  requirement: &70342224154860 !ruby/object:Gem::Requirement
+  requirement: &70338580046980 !ruby/object:Gem::Requirement
     none: false
     requirements:
     - - ! '>='
@@ -43,7 +43,7 @@ dependencies:
         version: '0'
   type: :runtime
   prerelease: false
-  version_requirements: *70342224154860
+  version_requirements: *70338580046980
 description: 
 email:
 - ananthakumaran@gmail.com