aws_test_dump 0.1.0

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
checksums.yaml ADDED
@@ -0,0 +1,7 @@
1
+ ---
2
+ SHA1:
3
+ metadata.gz: 87d2bbc3615e526da41dc76ac1a04e19e7450a4d
4
+ data.tar.gz: b7a72b2dbbb9008ebb010a988fb54e0698e384af
5
+ SHA512:
6
+ metadata.gz: 656e91bd7ed8052557b1fdbbffe8e7d5cfb42b8f6939cb6e9277310c92f9f51dff6e2e1d6be7d08ba2c025c8a82a04e3cc0eff60c656c6e8afe3381c05418d62
7
+ data.tar.gz: a5f0c31fabe01848ae225ff493c5a57a62c6d598ef45e722494841e4afedae87d29b411e5fe03325df709826bd853db4291c2c5132a8b148395084ccef64ca50
data/.dockerignore ADDED
@@ -0,0 +1,20 @@
1
+ /.bundle/
2
+ /.yardoc
3
+ /Gemfile.lock
4
+ /_yardoc/
5
+ /coverage/
6
+ /doc/
7
+ /pkg/
8
+ /spec/reports/
9
+ /tmp/
10
+ .idea/**
11
+ .DS_Store
12
+ *.swp
13
+ tags/
14
+ tags
15
+ coverage/
16
+ stats/
17
+
18
+ # rspec failure tracking
19
+ .rspec_status
20
+
data/.editorconfig ADDED
@@ -0,0 +1,19 @@
1
+ root = true
2
+
3
+ [*]
4
+ trim_trailing_whitespace = true
5
+ insert_final_newline = true
6
+
7
+ [*.sh]
8
+ indent_style = space
9
+ indent_size = 2
10
+
11
+ # 2 spaces for ruby https://github.com/bbatsov/ruby-style-guide
12
+ [*.rb]
13
+ indent_style = space
14
+ indent_size = 2
15
+
16
+ # 2 spaces
17
+ [*.{js,coffee,yml,html,css,sass,scss,yaml}]
18
+ indent_style = space
19
+ indent_size = 2
data/.gitignore ADDED
@@ -0,0 +1,21 @@
1
+ spec/tmp/
2
+ /.bundle/
3
+ /.yardoc
4
+ /Gemfile.lock
5
+ /_yardoc/
6
+ /coverage/
7
+ /doc/
8
+ /pkg/
9
+ /spec/reports/
10
+ /tmp/
11
+ .idea/**
12
+ .DS_Store
13
+ *.swp
14
+ tags/
15
+ tags
16
+ coverage/
17
+ stats/
18
+
19
+ # rspec failure tracking
20
+ .rspec_status
21
+
data/.rspec ADDED
@@ -0,0 +1,2 @@
1
+ --format documentation
2
+ --color
data/.travis.yml ADDED
@@ -0,0 +1,5 @@
1
+ sudo: false
2
+ language: ruby
3
+ rvm:
4
+ - 2.2.5
5
+ before_install: gem install bundler -v 1.14.6
@@ -0,0 +1,74 @@
1
+ # Contributor Covenant Code of Conduct
2
+
3
+ ## Our Pledge
4
+
5
+ In the interest of fostering an open and welcoming environment, we as
6
+ contributors and maintainers pledge to making participation in our project and
7
+ our community a harassment-free experience for everyone, regardless of age, body
8
+ size, disability, ethnicity, gender identity and expression, level of experience,
9
+ nationality, personal appearance, race, religion, or sexual identity and
10
+ orientation.
11
+
12
+ ## Our Standards
13
+
14
+ Examples of behavior that contributes to creating a positive environment
15
+ include:
16
+
17
+ * Using welcoming and inclusive language
18
+ * Being respectful of differing viewpoints and experiences
19
+ * Gracefully accepting constructive criticism
20
+ * Focusing on what is best for the community
21
+ * Showing empathy towards other community members
22
+
23
+ Examples of unacceptable behavior by participants include:
24
+
25
+ * The use of sexualized language or imagery and unwelcome sexual attention or
26
+ advances
27
+ * Trolling, insulting/derogatory comments, and personal or political attacks
28
+ * Public or private harassment
29
+ * Publishing others' private information, such as a physical or electronic
30
+ address, without explicit permission
31
+ * Other conduct which could reasonably be considered inappropriate in a
32
+ professional setting
33
+
34
+ ## Our Responsibilities
35
+
36
+ Project maintainers are responsible for clarifying the standards of acceptable
37
+ behavior and are expected to take appropriate and fair corrective action in
38
+ response to any instances of unacceptable behavior.
39
+
40
+ Project maintainers have the right and responsibility to remove, edit, or
41
+ reject comments, commits, code, wiki edits, issues, and other contributions
42
+ that are not aligned to this Code of Conduct, or to ban temporarily or
43
+ permanently any contributor for other behaviors that they deem inappropriate,
44
+ threatening, offensive, or harmful.
45
+
46
+ ## Scope
47
+
48
+ This Code of Conduct applies both within project spaces and in public spaces
49
+ when an individual is representing the project or its community. Examples of
50
+ representing a project or community include using an official project e-mail
51
+ address, posting via an official social media account, or acting as an appointed
52
+ representative at an online or offline event. Representation of a project may be
53
+ further defined and clarified by project maintainers.
54
+
55
+ ## Enforcement
56
+
57
+ Instances of abusive, harassing, or otherwise unacceptable behavior may be
58
+ reported by contacting the project team at TODO: Write your email address. All
59
+ complaints will be reviewed and investigated and will result in a response that
60
+ is deemed necessary and appropriate to the circumstances. The project team is
61
+ obligated to maintain confidentiality with regard to the reporter of an incident.
62
+ Further details of specific enforcement policies may be posted separately.
63
+
64
+ Project maintainers who do not follow or enforce the Code of Conduct in good
65
+ faith may face temporary or permanent repercussions as determined by other
66
+ members of the project's leadership.
67
+
68
+ ## Attribution
69
+
70
+ This Code of Conduct is adapted from the [Contributor Covenant][homepage], version 1.4,
71
+ available at [http://contributor-covenant.org/version/1/4][version]
72
+
73
+ [homepage]: http://contributor-covenant.org
74
+ [version]: http://contributor-covenant.org/version/1/4/
data/Dockerfile ADDED
@@ -0,0 +1,11 @@
1
+ FROM ruby:2.3
2
+
3
+ RUN mkdir -p /usr/src/app
4
+ WORKDIR /usr/src/app
5
+
6
+ COPY . /usr/src/app
7
+
8
+ RUN gem install bundler
9
+
10
+ RUN bundle install
11
+
data/Gemfile ADDED
@@ -0,0 +1,6 @@
1
+ source 'https://rubygems.org'
2
+ gem 'aws-sdk', '~>2'
3
+ gem 'rake'
4
+ gem 'rspec'
5
+ gem 'simplecov'
6
+ gem 'simplecov-console'
data/LICENSE.txt ADDED
@@ -0,0 +1,21 @@
1
+ The MIT License (MIT)
2
+
3
+ Copyright (c) 2017 TODO: Write your name
4
+
5
+ Permission is hereby granted, free of charge, to any person obtaining a copy
6
+ of this software and associated documentation files (the "Software"), to deal
7
+ in the Software without restriction, including without limitation the rights
8
+ to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
9
+ copies of the Software, and to permit persons to whom the Software is
10
+ furnished to do so, subject to the following conditions:
11
+
12
+ The above copyright notice and this permission notice shall be included in
13
+ all copies or substantial portions of the Software.
14
+
15
+ THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
16
+ IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
17
+ FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
18
+ AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
19
+ LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
20
+ OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
21
+ THE SOFTWARE.
data/README.md ADDED
@@ -0,0 +1,90 @@
1
+ # AwsTestDump
2
+
3
+ ## Installation
4
+
5
+ Add this line to your application's Gemfile:
6
+
7
+ ```ruby
8
+ gem 'aws_test_dump'
9
+ ```
10
+
11
+ And then execute:
12
+
13
+ $ bundle
14
+
15
+ Or install it yourself as:
16
+
17
+ $ gem install aws_test_dump
18
+
19
+ ## Usage
20
+
21
+ ### generating test data
22
+
23
+ #### dynamo data
24
+
25
+ The dynamo schema and test data is generated and restored using the aws_test_dump script.
26
+
27
+ To generate the schema, with real AWS credentials, run:
28
+
29
+ `aws_test_dump schema_dump`
30
+
31
+ To generate the test data, make sure the appropriate environment variables are set and an appropriate entry is entered in the spec/test_data_dump_definition.rb file.
32
+
33
+ An example entry:
34
+ ```ruby
35
+ {
36
+ table_name: 'relation-customer-toe-analysis',
37
+ key_conditions: {
38
+ 'customer_id' => {
39
+ :attribute_value_list => [ENV['CUSTOMER_ID']],
40
+ :comparison_operator => 'EQ'
41
+ },
42
+ },
43
+ replace_these: {
44
+ 'customer_id' => ENV['CUSTOMER_ID'],
45
+ },
46
+ replace_first: {
47
+ 'toe_id' => ENV['TOE_ID'],
48
+ 'analysis_id' => ENV['ANALYSIS_ID'],
49
+ },
50
+ },
51
+ ```
52
+ To create data dump files for each entry in the DATA_DUMP_DEFINITIONS, run:
53
+
54
+ `aws_test_dump data_dump`
55
+
56
+ and to create a dump of a specific table:
57
+
58
+ `aws_test_dump data_dump staging-export-analyses`
59
+
60
+
61
+ #### s3 data
62
+
63
+ To dump test s3 files, run the following in a racksh session with production aws credentials
64
+
65
+ ```ruby
66
+ require_relative 'aws_test_dump'
67
+
68
+ bucket_name = 'some_bucket_name'
69
+ key_name = 'some_s3_file.json'
70
+ s3_dump = AwsTestDump::S3FileDump.new(bucket_name, key_name)
71
+ s3_dump.run
72
+ ```
73
+
74
+ make sure any new buckets are added to the links mapping in the compose file so they are hitting the fakes3 service.
75
+
76
+ ## Development
77
+
78
+ After checking out the repo, run `bin/setup` to install dependencies. Then, run `rake spec` to run the tests. You can also run `bin/console` for an interactive prompt that will allow you to experiment.
79
+
80
+ To install this gem onto your local machine, run `bundle exec rake install`. To release a new version, update the version number in `version.rb`, and then run `bundle exec rake release`, which will create a git tag for the version, push git commits and tags, and push the `.gem` file to [rubygems.org](https://rubygems.org).
81
+
82
+ ## Contributing
83
+
84
+ Bug reports and pull requests are welcome on GitHub at https://github.com/DanHanson82/ruby-aws-test-dump. This project is intended to be a safe, welcoming space for collaboration, and contributors are expected to adhere to the [Contributor Covenant](http://contributor-covenant.org) code of conduct.
85
+
86
+
87
+ ## License
88
+
89
+ The gem is available as open source under the terms of the [MIT License](http://opensource.org/licenses/MIT).
90
+
data/Rakefile ADDED
@@ -0,0 +1,6 @@
1
+ require "bundler/gem_tasks"
2
+ require "rspec/core/rake_task"
3
+
4
+ RSpec::Core::RakeTask.new(:spec)
5
+
6
+ task :default => :spec
@@ -0,0 +1,27 @@
1
+ # coding: utf-8
2
+ lib = File.expand_path('../lib', __FILE__)
3
+ $LOAD_PATH.unshift(lib) unless $LOAD_PATH.include?(lib)
4
+ require 'aws_test_dump/version'
5
+
6
+ Gem::Specification.new do |spec|
7
+ spec.name = "aws_test_dump"
8
+ spec.version = AwsTestDump::VERSION
9
+ spec.authors = ["Daniel Hanson"]
10
+ spec.email = ["daniel.hanson82@gmail.com"]
11
+
12
+ spec.summary = %q{simple script for dumping and restoring aws test data for local testing}
13
+ spec.homepage = "https://github.com/DanHanson82/ruby-aws-test-dump"
14
+ spec.license = "MIT"
15
+
16
+ spec.files = `git ls-files -z`.split("\x0").reject do |f|
17
+ f.match(%r{^(test|spec|features)/})
18
+ end
19
+ spec.executables = ["aws_test_dump"]
20
+ spec.require_paths = ["lib"]
21
+
22
+ spec.add_runtime_dependency "aws-sdk", "~> 2"
23
+ spec.add_development_dependency "bundler", "~> 1.14"
24
+ spec.add_development_dependency "rake", "~> 10.0"
25
+ spec.add_development_dependency "rspec", "~> 3.0"
26
+ spec.add_development_dependency "simplecov", "~> 3.0"
27
+ end
data/bin/aws_test_dump ADDED
@@ -0,0 +1,24 @@
1
+ #!/usr/bin/env ruby
2
+
3
+ require 'optparse'
4
+ require 'aws_test_dump'
5
+
6
+
7
+ action = ARGV[0]
8
+ table_name = ARGV[1]
9
+ dump_file = nil
10
+
11
+ ARGV.options do |opts|
12
+ opts.on("-o", "--out=val", String) { |val| dump_file = val }
13
+ opts.parse!
14
+ end
15
+
16
+
17
+ if action == 'schema_dump'
18
+ processor = AwsTestDump::DynamoSchemaDump.new dump_file
19
+ elsif action == 'data_dump'
20
+ processor = AwsTestDump::DynamoDataDump.new table_name
21
+ else
22
+ raise NotValidOptionError
23
+ end
24
+ processor.run
data/bin/console ADDED
@@ -0,0 +1,14 @@
1
+ #!/usr/bin/env ruby
2
+
3
+ require "bundler/setup"
4
+ require "aws_test_dump"
5
+
6
+ # You can add fixtures and/or initialization code here to make experimenting
7
+ # with your gem easier. You can also use a different console, if you like.
8
+
9
+ # (If you use this, don't forget to add pry to your Gemfile!)
10
+ # require "pry"
11
+ # Pry.start
12
+
13
+ require "irb"
14
+ IRB.start(__FILE__)
data/bin/setup ADDED
@@ -0,0 +1,8 @@
1
+ #!/usr/bin/env bash
2
+ set -euo pipefail
3
+ IFS=$'\n\t'
4
+ set -vx
5
+
6
+ bundle install
7
+
8
+ # Do any other automated setup that you need to do here
@@ -0,0 +1,40 @@
1
+ version: "3"
2
+ services:
3
+ test:
4
+ build: .
5
+ command: rspec
6
+ volumes:
7
+ - .:/usr/src/app
8
+ depends_on:
9
+ - dynamo
10
+ - fakes3
11
+ links:
12
+ - fakes3
13
+ environment:
14
+ - DYNAMO_ENDPOINT=http://dynamo:5000
15
+ - FAKES3_ENDPOINT=http://fakes3:4000
16
+ - AWS_ACCESS_KEY_ID=potato
17
+ - AWS_SECRET_ACCESS_KEY=chorizo
18
+ - AWS_REGION=us-west-2
19
+ networks:
20
+ - fake-aws
21
+
22
+ dynamo:
23
+ image: fingershock/dynamodb-local
24
+ command: -inMemory -port 5000
25
+ ports:
26
+ - "5000:5000"
27
+ networks:
28
+ - fake-aws
29
+
30
+ fakes3:
31
+ image: jlynn/moto_server
32
+ command: s3 -p4000
33
+ ports:
34
+ - "4000:4000"
35
+ networks:
36
+ - fake-aws
37
+
38
+ networks:
39
+ fake-aws:
40
+ driver: bridge
@@ -0,0 +1,392 @@
1
+ require 'aws_test_dump/version'
2
+
3
+
4
+ class NotValidOptionError < StandardError
5
+ end
6
+
7
+
8
+ def keep_keys(key_names, some_hash)
9
+ some_hash.delete_if {|k, v| !key_names.include? k}
10
+ some_hash.each_pair do |k,v|
11
+ if v.is_a?(Hash)
12
+ keep_keys(key_names, v)
13
+ elsif v.is_a?(Array)
14
+ v.each { |x| keep_keys(key_names, x) if x.is_a?(Hash)}
15
+ end
16
+ end
17
+ some_hash
18
+ end
19
+
20
+
21
+ module AwsTestDump
22
+ require 'aws-sdk-core'
23
+ require 'fileutils'
24
+ require 'yaml'
25
+
26
+ DATA_DUMP_DEFINITION = ENV['DATA_DUMP_DEFINITION'] || File.join(Dir.pwd, 'spec', 'test_data_dump_definition.rb')
27
+ begin
28
+ require_relative DATA_DUMP_DEFINITION
29
+ rescue LoadError
30
+ DATA_DUMP_DEFINITIONS = []
31
+ end
32
+
33
+ Aws.config[:region] = ENV['AWS_REGION']
34
+
35
+ DEFAULT_DUMP_FILE = File.join(Dir.pwd, 'spec', 'dynamo_schema_dump.yml')
36
+ DEFAULT_DATA_DUMP_DIR = File.join(Dir.pwd, 'spec', 'dynamo_data_dumps')
37
+ DEFAULT_S3_DUMP_DIR = File.join(Dir.pwd, 'spec', 's3_test_files')
38
+ DYNAMO_TABLE_FIELDS = %i(
39
+ local_secondary_indexes
40
+ global_secondary_indexes
41
+
42
+ index_name
43
+ projection
44
+ projection_type
45
+ non_key_attributes
46
+
47
+ attribute_definitions
48
+ key_schema
49
+ provisioned_throughput
50
+
51
+ attribute_name
52
+ attribute_type
53
+ key_type
54
+
55
+ table_name
56
+ read_capacity_units
57
+ write_capacity_units
58
+ )
59
+
60
+ class BaseProcessor
61
+ def run
62
+ raise NotImplementedError
63
+ end
64
+ end
65
+
66
+ class BaseDynamoProcessor < BaseProcessor
67
+ attr_accessor :dump_file
68
+
69
+ def initialize(dump_file=nil)
70
+ dynamo_args = {}
71
+ dynamo_args[:endpoint] = ENV['DYNAMO_ENDPOINT'] if ENV['DYNAMO_ENDPOINT']
72
+ @dynamo_client = Aws::DynamoDB::Client.new(**dynamo_args)
73
+ @dump_file = dump_file
74
+ @dump_file ||= DEFAULT_DUMP_FILE
75
+ end
76
+
77
+ end
78
+
79
+ class S3BaseProcessor < BaseProcessor
80
+ attr_accessor :bucket_name, :key_name
81
+
82
+ def initialize(bucket_name, key_name)
83
+ s3_args = Hash.new
84
+ s3_args[:endpoint] = ENV['FAKES3_ENDPOINT'] if ENV['FAKES3_ENDPOINT']
85
+ @s3_client = Aws::S3::Client.new(**s3_args)
86
+ @bucket_name = bucket_name
87
+ @key_name = key_name
88
+ @dump_file = File.join(DEFAULT_S3_DUMP_DIR, bucket_name, key_name)
89
+ @file_contents = nil
90
+ end
91
+ end
92
+
93
+ class S3FileDump < S3BaseProcessor
94
+ def run
95
+ dump_data
96
+ end
97
+
98
+ def file_contents
99
+ if @file_contents.nil?
100
+ response = @s3_client.get_object(
101
+ bucket: @bucket_name, key: @key_name
102
+ ).body.read
103
+ @file_contents = @key_name.end_with?('.json') ? JSON.pretty_generate(JSON.parse(response)) : response
104
+ end
105
+ @file_contents
106
+ end
107
+
108
+ def dump_data
109
+ dirname = File.dirname(@dump_file)
110
+ unless File.directory?(dirname)
111
+ FileUtils.mkdir_p(dirname)
112
+ end
113
+ File.open(@dump_file, 'w') { |file| file.write file_contents }
114
+ end
115
+ end
116
+
117
+ class S3FileRestore < S3BaseProcessor
118
+ def run
119
+ restore
120
+ end
121
+
122
+ def file_contents
123
+ if @file_contents.nil?
124
+ @file_contents = File.read(@dump_file)
125
+ end
126
+ @file_contents
127
+ end
128
+
129
+ def restore
130
+ puts @bucket_name
131
+ puts @key_name
132
+ @s3_client.create_bucket({bucket: @bucket_name})
133
+ @s3_client.put_object({bucket: @bucket_name, key: @key_name, body: file_contents})
134
+ end
135
+ end
136
+
137
+ class S3Restore < BaseProcessor
138
+ def initialize
139
+ @s3_files = nil
140
+ @s3_restore_processors = Array.new
141
+ end
142
+
143
+ def run
144
+ s3_restore_processors.each { |x| x.run }
145
+ end
146
+
147
+ def s3_files
148
+ if @s3_files.nil?
149
+ @s3_files = Dir[ File.join(DEFAULT_S3_DUMP_DIR, '**', '*') ].reject { |p| File.directory? p }
150
+ end
151
+ @s3_files
152
+ end
153
+
154
+ def s3_restore_processors
155
+ if @s3_restore_processors.empty?
156
+ s3_files.each do |s3_file|
157
+ relative_path = s3_file.split(DEFAULT_S3_DUMP_DIR)[1][1..-1]
158
+ bucket = relative_path.split('/')[0]
159
+ key = relative_path.gsub(bucket, '')[1..-1]
160
+ @s3_restore_processors << S3FileRestore.new(bucket, key)
161
+ end
162
+ end
163
+ @s3_restore_processors
164
+ end
165
+ end
166
+
167
+
168
+ class DynamoTableDump < BaseDynamoProcessor
169
+ attr_accessor :table_name, :data_dump_definition
170
+
171
+ def initialize(data_dump_definition=nil, dump_dir=nil, table_name=nil)
172
+ super nil
173
+ @data_dump_definition = data_dump_definition
174
+ @data_dump_definition ||= {}
175
+ @table_name = table_name
176
+ @table_name ||= @data_dump_definition[:table_name]
177
+ @dump_dir = dump_dir
178
+ @dump_dir ||= DEFAULT_DATA_DUMP_DIR
179
+ @dump_file = File.join(@dump_dir, "#{@table_name}.yml")
180
+ @query_results = nil
181
+ end
182
+
183
+ def run
184
+ dump_data
185
+ end
186
+
187
+ def dump_data
188
+ data = {table_name: @table_name, data: query_results}
189
+ unless File.directory?(@dump_dir)
190
+ FileUtils.mkdir_p(@dump_dir)
191
+ end
192
+ File.open(@dump_file, 'w') { |file| file.write data.to_yaml }
193
+ end
194
+
195
+ def _query
196
+ @dynamo_client.query({
197
+ :table_name => @table_name,
198
+ :select => 'ALL_ATTRIBUTES',
199
+ :key_conditions => @data_dump_definition[:key_conditions]
200
+ })
201
+ end
202
+
203
+ def _scan
204
+ @dynamo_client.scan({table_name: @table_name})
205
+ end
206
+
207
+ def query_results
208
+ if @query_results.nil?
209
+ response = !@data_dump_definition[:key_conditions].nil? ? _query : _scan
210
+ @query_results = response.items
211
+ end
212
+ @query_results
213
+ end
214
+
215
+ end
216
+
217
+ class DynamoDataDump < BaseDynamoProcessor
218
+ attr_accessor :table_name, :data_dump_definitions
219
+
220
+ def initialize(table_name=nil, dump_dir=nil)
221
+ super nil
222
+ @dump_dir = dump_dir
223
+ @dump_dir ||= DEFAULT_DATA_DUMP_DIR
224
+ @table_name = table_name
225
+ @data_dump_definitions = nil
226
+ end
227
+
228
+ def data_dump_definitions
229
+ if @data_dump_definitions.nil?
230
+ if !@table_name.nil?
231
+ data_dump_definition = DATA_DUMP_DEFINITIONS.find { |x| x[:table_name] == table_name }
232
+ @data_dump_definitions = !data_dump_definition.nil? ? [data_dump_definition] : []
233
+ else
234
+ @data_dump_definitions = DATA_DUMP_DEFINITIONS
235
+ end
236
+ end
237
+ @data_dump_definitions
238
+ end
239
+
240
+ def run
241
+ if !data_dump_definitions.empty?
242
+ data_dump_definitions.each do |data_dump_definition|
243
+ dynamo_table_dump = DynamoTableDump.new data_dump_definition, @dump_dir
244
+ dynamo_table_dump.run
245
+ end
246
+ elsif !@table_name.nil?
247
+ dynamo_table_dump = DynamoTableDump.new nil, @dump_dir, @table_name
248
+ dynamo_table_dump.run
249
+ end
250
+ end
251
+ end
252
+
253
+ class DynamoTableDataRestore < BaseDynamoProcessor
254
+ def initialize(dump_file)
255
+ super dump_file
256
+ @table_name = nil
257
+ @data = nil
258
+ @data_dump_definition = nil
259
+ end
260
+
261
+ def run
262
+ data.each_with_index do |item, index|
263
+ if index == 0
264
+ item.merge!(data_dump_definition.fetch(:replace_first, {}))
265
+ end
266
+ item.merge!(data_dump_definition.fetch(:replace_these, {}))
267
+ @dynamo_client.put_item({:table_name => table_name, item: item})
268
+ end
269
+ end
270
+
271
+ def parse_file
272
+ file_contents = YAML.load(File.open(@dump_file))
273
+ @table_name = file_contents[:table_name]
274
+ @data = file_contents[:data]
275
+ end
276
+
277
+ def data_dump_definition
278
+ if @data_dump_definition.nil?
279
+ @data_dump_definition = DATA_DUMP_DEFINITIONS.find { |x| x[:table_name] == table_name} || {}
280
+ end
281
+ @data_dump_definition
282
+ end
283
+
284
+ def table_name
285
+ parse_file if @table_name.nil?
286
+ @table_name
287
+ end
288
+
289
+ def data
290
+ parse_file if @data.nil?
291
+ @data
292
+ end
293
+ end
294
+
295
+ class DynamoDataRestore < BaseDynamoProcessor
296
+
297
+ def initialize(dump_dir=nil)
298
+ super
299
+ @dump_dir = dump_dir
300
+ @dump_dir ||= DEFAULT_DATA_DUMP_DIR
301
+ @data_dump_files = Array.new
302
+ end
303
+
304
+ def data_dump_files
305
+ if @data_dump_files.empty?
306
+ Dir.entries(@dump_dir).each do |f|
307
+ full_path = File.join(@dump_dir, f)
308
+ @data_dump_files << full_path if File.file?(full_path)
309
+ end
310
+ end
311
+ @data_dump_files
312
+ end
313
+
314
+ def run
315
+ data_dump_files.each do |f|
316
+ dynamo_table_data_restore = DynamoTableDataRestore.new(f)
317
+ dynamo_table_data_restore.run
318
+ end
319
+ end
320
+ end
321
+
322
+ class DynamoSchemaDump < BaseDynamoProcessor
323
+ attr_accessor :table_names, :schemata
324
+
325
+ def initialize(dump_file=nil)
326
+ super dump_file
327
+ @table_names = nil
328
+ @schemata = nil
329
+ end
330
+
331
+ def run
332
+ dump_schemata
333
+ end
334
+
335
+ def dump_schemata
336
+ dirname = File.dirname(@dump_file)
337
+ unless File.directory?(dirname)
338
+ FileUtils.mkdir_p(dirname)
339
+ end
340
+ File.open(@dump_file, 'w') { |file| file.write schemata.to_yaml }
341
+ end
342
+
343
+ def table_names
344
+ if @table_names.nil?
345
+ @table_names = @dynamo_client.list_tables.table_names
346
+ end
347
+ @table_names
348
+ end
349
+
350
+ def schemata
351
+ if @schemata.nil?
352
+ @schemata = Array.new
353
+ table_names.each do |table_name|
354
+ table_schema = @dynamo_client.describe_table(table_name: table_name).to_h[:table]
355
+ keep_keys(DYNAMO_TABLE_FIELDS, table_schema)
356
+ @schemata << table_schema
357
+ end
358
+ end
359
+ @schemata
360
+ end
361
+
362
+ end
363
+
364
+ class DynamoSchemaRestore < BaseDynamoProcessor
365
+ attr_accessor :schemata
366
+
367
+ def initialize(dump_file=nil)
368
+ super dump_file
369
+ @schemata = nil
370
+ end
371
+
372
+ def run
373
+ create_tables
374
+ end
375
+
376
+ def create_tables
377
+ schemata.each do |schema|
378
+ begin
379
+ @dynamo_client.create_table(schema)
380
+ rescue Aws::DynamoDB::Errors::ResourceInUseException
381
+ end
382
+ end
383
+ end
384
+
385
+ def schemata
386
+ if @schemata.nil?
387
+ @schemata = YAML.load(File.open(@dump_file))
388
+ end
389
+ @schemata
390
+ end
391
+ end
392
+ end
@@ -0,0 +1,3 @@
1
+ module AwsTestDump
2
+ VERSION = "0.1.0"
3
+ end
metadata ADDED
@@ -0,0 +1,133 @@
1
+ --- !ruby/object:Gem::Specification
2
+ name: aws_test_dump
3
+ version: !ruby/object:Gem::Version
4
+ version: 0.1.0
5
+ platform: ruby
6
+ authors:
7
+ - Daniel Hanson
8
+ autorequire:
9
+ bindir: bin
10
+ cert_chain: []
11
+ date: 2017-10-09 00:00:00.000000000 Z
12
+ dependencies:
13
+ - !ruby/object:Gem::Dependency
14
+ name: aws-sdk
15
+ requirement: !ruby/object:Gem::Requirement
16
+ requirements:
17
+ - - "~>"
18
+ - !ruby/object:Gem::Version
19
+ version: '2'
20
+ type: :runtime
21
+ prerelease: false
22
+ version_requirements: !ruby/object:Gem::Requirement
23
+ requirements:
24
+ - - "~>"
25
+ - !ruby/object:Gem::Version
26
+ version: '2'
27
+ - !ruby/object:Gem::Dependency
28
+ name: bundler
29
+ requirement: !ruby/object:Gem::Requirement
30
+ requirements:
31
+ - - "~>"
32
+ - !ruby/object:Gem::Version
33
+ version: '1.14'
34
+ type: :development
35
+ prerelease: false
36
+ version_requirements: !ruby/object:Gem::Requirement
37
+ requirements:
38
+ - - "~>"
39
+ - !ruby/object:Gem::Version
40
+ version: '1.14'
41
+ - !ruby/object:Gem::Dependency
42
+ name: rake
43
+ requirement: !ruby/object:Gem::Requirement
44
+ requirements:
45
+ - - "~>"
46
+ - !ruby/object:Gem::Version
47
+ version: '10.0'
48
+ type: :development
49
+ prerelease: false
50
+ version_requirements: !ruby/object:Gem::Requirement
51
+ requirements:
52
+ - - "~>"
53
+ - !ruby/object:Gem::Version
54
+ version: '10.0'
55
+ - !ruby/object:Gem::Dependency
56
+ name: rspec
57
+ requirement: !ruby/object:Gem::Requirement
58
+ requirements:
59
+ - - "~>"
60
+ - !ruby/object:Gem::Version
61
+ version: '3.0'
62
+ type: :development
63
+ prerelease: false
64
+ version_requirements: !ruby/object:Gem::Requirement
65
+ requirements:
66
+ - - "~>"
67
+ - !ruby/object:Gem::Version
68
+ version: '3.0'
69
+ - !ruby/object:Gem::Dependency
70
+ name: simplecov
71
+ requirement: !ruby/object:Gem::Requirement
72
+ requirements:
73
+ - - "~>"
74
+ - !ruby/object:Gem::Version
75
+ version: '3.0'
76
+ type: :development
77
+ prerelease: false
78
+ version_requirements: !ruby/object:Gem::Requirement
79
+ requirements:
80
+ - - "~>"
81
+ - !ruby/object:Gem::Version
82
+ version: '3.0'
83
+ description:
84
+ email:
85
+ - daniel.hanson82@gmail.com
86
+ executables:
87
+ - aws_test_dump
88
+ extensions: []
89
+ extra_rdoc_files: []
90
+ files:
91
+ - ".dockerignore"
92
+ - ".editorconfig"
93
+ - ".gitignore"
94
+ - ".rspec"
95
+ - ".travis.yml"
96
+ - CODE_OF_CONDUCT.md
97
+ - Dockerfile
98
+ - Gemfile
99
+ - LICENSE.txt
100
+ - README.md
101
+ - Rakefile
102
+ - aws_test_dump.gemspec
103
+ - bin/aws_test_dump
104
+ - bin/console
105
+ - bin/setup
106
+ - docker-compose.yml
107
+ - lib/aws_test_dump.rb
108
+ - lib/aws_test_dump/version.rb
109
+ homepage: https://github.com/DanHanson82/ruby-aws-test-dump
110
+ licenses:
111
+ - MIT
112
+ metadata: {}
113
+ post_install_message:
114
+ rdoc_options: []
115
+ require_paths:
116
+ - lib
117
+ required_ruby_version: !ruby/object:Gem::Requirement
118
+ requirements:
119
+ - - ">="
120
+ - !ruby/object:Gem::Version
121
+ version: '0'
122
+ required_rubygems_version: !ruby/object:Gem::Requirement
123
+ requirements:
124
+ - - ">="
125
+ - !ruby/object:Gem::Version
126
+ version: '0'
127
+ requirements: []
128
+ rubyforge_project:
129
+ rubygems_version: 2.4.8
130
+ signing_key:
131
+ specification_version: 4
132
+ summary: simple script for dumping and restoring aws test data for local testing
133
+ test_files: []