config_test.rb in pups-1.2.0

- old
+ new

@@ -1,37 +1,37 @@
 # frozen_string_literal: true
 
-require 'test_helper'
-require 'tempfile'
+require "test_helper"
+require "tempfile"
 
 module Pups
-  class ConfigTest < MiniTest::Test
+  class ConfigTest < ::Minitest::Test
     def test_config_from_env
-      ENV['HELLO'] = 'world'
+      ENV["HELLO"] = "world"
       config = Config.new({})
-      assert_equal('world', config.params['$ENV_HELLO'])
+      assert_equal("world", config.params["$ENV_HELLO"])
     end
 
     def test_env_param
-      ENV['FOO'] = 'BAR'
+      ENV["FOO"] = "BAR"
       config = <<~YAML
         env:
           BAR: baz
           hello: WORLD
           one: 1
       YAML
 
       config = Config.new(YAML.safe_load(config))
       %w[BAR hello one].each { |e| ENV.delete(e) }
-      assert_equal('BAR', config.params['$ENV_FOO'])
-      assert_equal('baz', config.params['$ENV_BAR'])
-      assert_equal('WORLD', config.params['$ENV_hello'])
-      assert_equal('1', config.params['$ENV_one'])
+      assert_equal("BAR", config.params["$ENV_FOO"])
+      assert_equal("baz", config.params["$ENV_BAR"])
+      assert_equal("WORLD", config.params["$ENV_hello"])
+      assert_equal("1", config.params["$ENV_one"])
     end
 
     def test_env_with_template
-      ENV['FOO'] = 'BAR'
+      ENV["FOO"] = "BAR"
       config = <<~YAML
         env:
           greeting: "{{hello}}, {{planet}}!"
           one: 1
           other: "where are we on {{planet}}?"
@@ -41,14 +41,14 @@
       YAML
       config_hash = YAML.safe_load(config)
 
       config = Config.new(config_hash)
       %w[greeting one other].each { |e| ENV.delete(e) }
-      assert_equal('hola, pluto!', config.params['$ENV_greeting'])
-      assert_equal('1', config.params['$ENV_one'])
-      assert_equal('BAR', config.params['$ENV_FOO'])
-      assert_equal('where are we on pluto?', config.params['$ENV_other'])
+      assert_equal("hola, pluto!", config.params["$ENV_greeting"])
+      assert_equal("1", config.params["$ENV_one"])
+      assert_equal("BAR", config.params["$ENV_FOO"])
+      assert_equal("where are we on pluto?", config.params["$ENV_other"])
     end
 
     def test_label_with_template
       ENV["FOO"] = "BAR"
       config = <<~YAML
@@ -65,15 +65,15 @@
       YAML
       config_hash = YAML.load(config)
 
       config = Config.new(config_hash)
       %w[greeting one other].each { |e| ENV.delete(e) }
-      assert_equal("various_discourse", config.config['labels']['app_name'])
+      assert_equal("various_discourse", config.config["labels"]["app_name"])
     end
 
     def test_env_with_ENV_templated_variable
-      ENV['env_template_config'] = 'my_application'
+      ENV["env_template_config"] = "my_application"
       config = <<~YAML
         env:
           greeting: "{{hello}}, {{planet}}!"
           one: 1
           other: "building {{config}}"
@@ -83,18 +83,18 @@
       YAML
       config_hash = YAML.safe_load(config)
 
       config = Config.new(config_hash)
       %w[greeting one other].each { |e| ENV.delete(e) }
-      assert_equal('hola, pluto!', config.params['$ENV_greeting'])
-      assert_equal('1', config.params['$ENV_one'])
-      assert_equal('building my_application', config.params['$ENV_other'])
+      assert_equal("hola, pluto!", config.params["$ENV_greeting"])
+      assert_equal("1", config.params["$ENV_one"])
+      assert_equal("building my_application", config.params["$ENV_other"])
       ENV["env_template_config"] = nil
     end
 
     def test_integration
-      f = Tempfile.new('test')
+      f = Tempfile.new("test")
       f.close
 
       config = <<~YAML
         env:
           PLANET: world
@@ -104,12 +104,12 @@
         run:
           - exec: echo $greeting $PLANET >> #{f.path}
       YAML
 
       Config.new(YAML.safe_load(config)).run
-      ENV.delete('PLANET')
-      assert_equal('hello world', File.read(f.path).strip)
+      ENV.delete("PLANET")
+      assert_equal("hello world", File.read(f.path).strip)
     ensure
       f.unlink
     end
 
     def test_hooks
@@ -126,16 +126,16 @@
           before_middle:
             - exec: 1.9
       YAML
 
       config = Config.load_config(yaml).config
-      assert_equal({ 'exec' => 1.9 }, config['run'][1])
-      assert_equal({ 'exec' => 2.1 }, config['run'][3])
+      assert_equal({ "exec" => 1.9 }, config["run"][1])
+      assert_equal({ "exec" => 2.1 }, config["run"][3])
     end
 
     def test_ignored_elements
-      f = Tempfile.new('test')
+      f = Tempfile.new("test")
       f.close
 
       yaml = <<~YAML
         env:
           PLANET: world
@@ -155,19 +155,25 @@
             - exec: 1.9
       YAML
 
       conf = Config.load_config(yaml, %w[hooks params])
       config = conf.config
-      assert_equal({ 'exec' => 1 }, config['run'][0])
-      assert_equal({ 'exec' => { 'hook' => 'middle', 'cmd' => 2 } }, config['run'][1])
-      assert_equal({ 'exec' => 3 }, config['run'][2])
-      assert_equal({ 'exec' => "echo $greeting $PLANET >> #{f.path}" }, config['run'][3])
+      assert_equal({ "exec" => 1 }, config["run"][0])
+      assert_equal(
+        { "exec" => { "hook" => "middle", "cmd" => 2 } },
+        config["run"][1]
+      )
+      assert_equal({ "exec" => 3 }, config["run"][2])
+      assert_equal(
+        { "exec" => "echo $greeting $PLANET >> #{f.path}" },
+        config["run"][3]
+      )
 
       # $greet from params will be an empty var as it was ignored
       conf.run
-      ENV.delete('PLANET')
-      assert_equal('world', File.read(f.path).strip)
+      ENV.delete("PLANET")
+      assert_equal("world", File.read(f.path).strip)
     end
 
     def test_generate_docker_run_arguments
       yaml = <<~YAML
       env:
@@ -210,8 +216,72 @@
       expected << "--link postgres:postgres --link foo:bar"
       expected << "--label monitor=true --label app_name=my_app_discourse"
       expected.sort!
 
       assert_equal(expected.join(" "), args)
+    end
+
+    def test_tag_filtering
+      f = Tempfile.new("test")
+      f.close
+
+      yaml = <<~YAML
+        run:
+          - exec: 1
+          - exec:
+              hook: middle
+              cmd: 2
+              tag: one_tag
+          - exec:
+              cmd: 3
+              tag: two_tag
+        hooks:
+          after_middle:
+            - exec: 2.1
+          before_middle:
+            - exec: 1.9
+      YAML
+
+      # No tagging loads everything
+      conf = Config.load_config(yaml)
+      config = conf.config
+      assert_equal({ "exec" => 1 }, config["run"][0])
+      assert_equal({ "exec" => 1.9 }, config["run"][1])
+      assert_equal(
+        { "exec" => { "hook" => "middle", "cmd" => 2, "tag" => "one_tag" } },
+        config["run"][2]
+      )
+      assert_equal({ "exec" => 2.1 }, config["run"][3])
+      assert_equal(
+        { "exec" => { "cmd" => 3, "tag" => "two_tag" } },
+        config["run"][4]
+      )
+
+      # hooks get applied if hook command is not filtered
+      conf = Config.load_config(yaml, tags: ["one_tag"])
+      config = conf.config
+      assert_equal({ "exec" => 1.9 }, config["run"][0])
+      assert_equal(
+        { "exec" => { "hook" => "middle", "cmd" => 2, "tag" => "one_tag" } },
+        config["run"][1]
+      )
+      assert_equal({ "exec" => 2.1 }, config["run"][2])
+
+      # hooks get filtered out if the main hook command is filtered
+      conf = Config.load_config(yaml, tags: ["two_tag"])
+      config = conf.config
+      assert_equal(
+        { "exec" => { "cmd" => 3, "tag" => "two_tag" } },
+        config["run"][0]
+      )
+
+      # skip tags filter out commands with tags
+      conf = Config.load_config(yaml, skip_tags: ["one_tag"])
+      config = conf.config
+      assert_equal({ "exec" => 1 }, config["run"][0])
+      assert_equal(
+        { "exec" => { "cmd" => 3, "tag" => "two_tag" } },
+        config["run"][1]
+      )
     end
   end
 end