828e56912e
When invoking the audit broadcast command, provide a few environment variables so that people can customize the format of the message if they want. We currently provide `MRSK_PERFORMER`, `MRSK_ROLE`, `MRSK_DESTINATION` and `MRSK_EVENT`. Also adds the destination to the default message, which we continue to send as the first argument as before.
61 lines
1.7 KiB
Ruby
61 lines
1.7 KiB
Ruby
require "test_helper"
|
|
|
|
class CommandsAuditorTest < ActiveSupport::TestCase
|
|
setup do
|
|
@config = {
|
|
service: "app", image: "dhh/app", registry: { "username" => "dhh", "password" => "secret" }, servers: [ "1.1.1.1" ],
|
|
audit_broadcast_cmd: "bin/audit_broadcast"
|
|
}
|
|
end
|
|
|
|
test "record" do
|
|
assert_match \
|
|
/echo '.* app removed container' >> mrsk-app-audit.log/,
|
|
new_command.record("app removed container").join(" ")
|
|
end
|
|
|
|
test "record with destination" do
|
|
@destination = "staging"
|
|
|
|
assert_match \
|
|
/echo '.* app removed container' >> mrsk-app-staging-audit.log/,
|
|
new_command.record("app removed container").join(" ")
|
|
end
|
|
|
|
test "record with role" do
|
|
@role = "web"
|
|
|
|
assert_match \
|
|
/echo '.* \[web\] app removed container' >> mrsk-app-audit.log/,
|
|
new_command.record("app removed container").join(" ")
|
|
end
|
|
|
|
test "broadcast" do
|
|
Mrsk::Commands::Auditor.any_instance.stubs(:performer).returns("bob")
|
|
@role = "web"
|
|
@destination = "staging"
|
|
|
|
assert_equal \
|
|
["bin/audit_broadcast", "'[bob] [web] [staging] app removed container'"],
|
|
new_command.broadcast("app removed container")
|
|
end
|
|
|
|
test "broadcast environment" do
|
|
Mrsk::Commands::Auditor.any_instance.stubs(:performer).returns("bob")
|
|
@role = "web"
|
|
@destination = "staging"
|
|
|
|
env = new_command.broadcast_environment("app removed container")
|
|
|
|
assert_equal "bob", env["MRSK_PERFORMER"]
|
|
assert_equal "web", env["MRSK_ROLE"]
|
|
assert_equal "staging", env["MRSK_DESTINATION"]
|
|
assert_equal "app removed container", env["MRSK_EVENT"]
|
|
end
|
|
|
|
private
|
|
def new_command
|
|
Mrsk::Commands::Auditor.new(Mrsk::Configuration.new(@config, destination: @destination, version: "123"), role: @role)
|
|
end
|
|
end
|