cloud_formation 0.1.2 → 0.1.3

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 5c28700bca1d3dad061e12b10d3587e60e2248f1
-  data.tar.gz: 2faf4c04795ecdcf492a00ebe4665aa9e51ef75a
+  metadata.gz: e28fac1fb5095b00790bf590942d272275cace0a
+  data.tar.gz: 618a1852d7398f4009b9415d7a47d2b8ef308b35
 SHA512:
-  metadata.gz: 8b2174afbd62a8509242de4f1dd17c95e0fdf75e85dbb9a1d4f8990edeba4470bb2e2a65506b1cd39c252f4015e9119e86be15dabc9598aae99c6143571a4787
-  data.tar.gz: 1adad111f96331fb195abfe6685dfd2f4001a8dc50aff8458917b4a1b20f62f311b6c7115c635748ec9ee10b30f253911c86138cb541b7b99c165c8e276b38bc
+  metadata.gz: bd96b5b1052c3dbb4d619ffc1c65d1fc19cc16c4042ae070499bc152f1538b742ae308806ef3b676fbcad1f85e0d6b06b4352080e898139d5451fa3af306b2e8
+  data.tar.gz: 259f215e6b17f3189f38b1ec4554891a6f17260cbe2d0d08c7b86472839ee0f906c2b574a61111bbc83be57cbc52a897d4dd5ea6ccbdfc8c985422916344ff85

data/lib/cloud_formation/dynamo_db.rb

@@ -19,6 +19,12 @@ module CloudFormation
       class Table < Base
         attribute_list :name, :read_capacity_units, :write_capacity_units, :hash_key, :range_key
 
+        attr_accessor :attribute_definitions
+
+        def initialize
+          @attribute_definitions = []
+        end
+
         def serialize
           properties = {
             "Type" => "AWS::DynamoDB::Table",
@@ -30,12 +36,17 @@ module CloudFormation
             }
           }
           properties["Properties"].merge! add_key_schemas
+          properties["Properties"]["AttributeDefinitions"] = @attribute_definitions unless @attribute_definitions.empty?
           properties
         end
 
+        def add_attribute attr_name, attr_value
+          @attribute_definitions << { "AttributeName" => attr_name.to_s, "AttributeType" => type_mapping(attr_value) }
+        end
+
         private
 
-          def key_type_mapping type
+          def type_mapping type
             case type
               when :string then "S"
               when :number then "N"
@@ -52,7 +63,7 @@ module CloudFormation
 
           def add_key_schema_for key
             k = key.keys.first
-            { "AttributeName" => k.to_s, "AttributeType" => key_type_mapping(key[k])  }
+            { "AttributeName" => k.to_s, "AttributeType" => type_mapping(key[k])  }
           end
 
       end

data/lib/cloud_formation/version.rb

@@ -1,3 +1,3 @@
 module CloudFormation
-  VERSION = "0.1.2"
+  VERSION = "0.1.3"
 end

data/test/dynamo_db_test.rb

@@ -6,6 +6,11 @@ class DynamoDBTest < Minitest::Test
     @table = build_dynamo_table_instance
   end
 
+  def test_basic
+    expected = @table.serialize
+    assert_equal load_fixture("dynamo_db/basic"), expected
+  end
+
   def test_build_table_should_return_table_class
     dynamo_db = CloudFormation::DynamoDB.new
     table = build_dynamo_table_instance
@@ -20,9 +25,16 @@ class DynamoDBTest < Minitest::Test
     assert_equal 1, dynamo_db.tables.length
   end
 
-  def test_basic
-    expected = @table.serialize
-    assert_equal load_fixture("dynamo_db/basic"), expected
-  end
+  def test_add_attribute_should_generate_attribute_definition_schema
+    dynamo_db = CloudFormation::DynamoDB.new
+
+    table = build_dynamo_table_instance
 
+    table.add_attribute :first_name, :string
+    table.add_attribute :last_name, :string
+    table.add_attribute :age, :number
+
+    assert_equal table.serialize, load_fixture('dynamo_db/add_attributes')
+
+  end
 end

data/test/templates/dynamo_db/add_attributes.json

@@ -0,0 +1,33 @@
+{
+  "Type": "AWS::DynamoDB::Table",
+  "Properties": {
+    "ProvisionedThroughput": {
+      "WriteCapacityUnits": "5",
+      "ReadCapacityUnits": "10"
+    },
+    "KeySchema": {
+      "RangeKeyElement": {
+        "AttributeName": "column_two_id",
+        "AttributeType": "N"
+      },
+      "HashKeyElement": {
+        "AttributeName": "column_one_id",
+        "AttributeType": "N"
+      }
+    },
+    "AttributeDefinitions": [
+      {
+        "AttributeName": "first_name",
+        "AttributeType": "S"
+      },
+      {
+        "AttributeName": "last_name",
+        "AttributeType": "S"
+      },
+      {
+        "AttributeName": "age",
+        "AttributeType": "N"
+      }
+    ]
+  }
+}

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: cloud_formation
 version: !ruby/object:Gem::Version
-  version: 0.1.2
+  version: 0.1.3
 platform: ruby
 authors:
 - David White
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2014-03-15 00:00:00.000000000 Z
+date: 2014-03-18 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: bundler
@@ -60,6 +60,7 @@ files:
 - test/dynamo_db_test.rb
 - test/factories/factory_helpers.rb
 - test/template_test.rb
+- test/templates/dynamo_db/add_attributes.json
 - test/templates/dynamo_db/basic.json
 - test/templates/template/adding_multiple_resources.json
 - test/templates/template/adding_single_resource.json
@@ -94,6 +95,7 @@ test_files:
 - test/dynamo_db_test.rb
 - test/factories/factory_helpers.rb
 - test/template_test.rb
+- test/templates/dynamo_db/add_attributes.json
 - test/templates/dynamo_db/basic.json
 - test/templates/template/adding_multiple_resources.json
 - test/templates/template/adding_single_resource.json