aws_security_viz 0.2.1.pre.alpha.pre.269 → 0.2.1.pre.alpha.pre.278

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 42589763944aa2efaaf22a7a87106ceca1e36076f1ba99aa23d3860cf336b45e
-  data.tar.gz: 89cf97f11e19d7cc63e4d3d4085056beb7264518d6b56d3d7df829907fe6f412
+  metadata.gz: 035cd6c263db0a456f221d6f46790cdb6eae3dbdfcd17d5bf5ae776eb6e4fe0a
+  data.tar.gz: 2214846679c445cde0005086acd776f17d836c5d1b6970de951cfcc79d79e6d9
 SHA512:
-  metadata.gz: 980a68f9c4da1ea1a00a4ac29114848d32eb10778de8f93c3d5d193c76562d2d73ab4d8aa4a566da4218aef5abf51e72277c7eae8b811eba3db47fa338a3d36a
-  data.tar.gz: ee734c1d4883fa045823e6707e9f59306347962004a96e3f14b6205234025814b6f50d9c9e4328a49e2972a22f5a7eea8b91d8dfb384107afdff8b513a0fbe8b
+  metadata.gz: 04204b3106c983c0bfb103fd437441e7b015ef00890e8c3088ffab1ca84e95ba4e0a23d7eb78c5187ae8a41aaa96eae5f900b025e31c9cc7afe989c61365bea4
+  data.tar.gz: cd8fef57bdae83be03b0320939376abbdcbafba60ca6cd6c34eb27382e7aa15fd25a1ad3c935afbead7209fcbd4dab9515f1c8b5efc37021ff93da9c895e476d

data/spec/integration/visualize_aws_spec.rb

@@ -1,6 +1,12 @@
 require 'spec_helper'
 require 'tempfile'
 
+RSpec::Matchers.define :be_graph_with do |nodes|
+  match do |graphv|
+    graphv.nodes.keys == nodes
+  end
+end
+
 describe VisualizeAws do
   let(:opts) {
     {
@@ -21,24 +27,38 @@ describe VisualizeAws do
       VisualizeAws.new(config, opts).unleash(temp_file.path)
       expect(expected_content).to eq(actual_content)
     end
+    
+    it 'should parse json input with stubbed out graphviz' do
+      nodes = ["app", "8.8.8.8/32", "amazon-elb-sg", "*", "db"]
+      allow(Graphviz).to receive(:output).with(be_graph_with(nodes), path: temp_file.path, format: nil)
+      VisualizeAws.new(config, opts).unleash(temp_file.path)
+    end
   end
 
   context 'json to json file' do
     let(:expected_file) { File.join(File.dirname(__FILE__), 'expected.json') }
     let(:temp_file) { Tempfile.new(%w(aws .json)) }
 
-    it 'should parse json input', :integration => true do
+    it 'should parse json input' do
       expect(FileUtils).to receive(:copy)
       VisualizeAws.new(config, opts.merge(:renderer => 'json')).unleash(temp_file.path)
       expect(JSON.parse(expected_content)).to eq(JSON.parse(actual_content))
     end
+
+    it 'should parse json input with obfuscation' do
+      config = AwsConfig.new({groups: {'0.0.0.0/0' => '*'}, obfuscate: true})
+      expect(FileUtils).to receive(:copy)
+      VisualizeAws.new(config, opts.merge(:renderer => 'json')).unleash(temp_file.path)
+      expect(actual_content).not_to include('"amazon-elb-sg"', '"app"', '"db"')
+    end
+
   end
 
   context 'json to navigator file' do
     let(:expected_file) { File.join(File.dirname(__FILE__), 'navigator.json') }
     let(:temp_file) { Tempfile.new(%w(aws .json)) }
 
-    it 'should parse json input', :integration => true do
+    it 'should parse json input' do
       expect(FileUtils).to receive(:copy)
       VisualizeAws.new(config, opts.merge(:renderer => 'navigator')).unleash(temp_file.path)
       expect(JSON.parse(expected_content)).to eq(JSON.parse(actual_content))
@@ -53,7 +73,8 @@ describe VisualizeAws do
         {
             :filename => temp_file,
             :secret_key => ENV['TEST_SECRET_KEY'],
-            :access_key => ENV['TEST_ACCESS_KEY']
+            :access_key => ENV['TEST_ACCESS_KEY'],
+            :region => 'us-east-1'
         }
       }
 

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: aws_security_viz
 version: !ruby/object:Gem::Version
-  version: 0.2.1.pre.alpha.pre.269
+  version: 0.2.1.pre.alpha.pre.278
 platform: ruby
 authors:
 - Anay Nayak