feat: test the program with rspec

Author: jim80net

Gemfile

@@ -3,6 +3,9 @@ source 'https://rubygems.org'
 gem 'redis'
 gem 'dogapi'
 
-group :development do
+group :development, :test do
   gem 'pry'
+  gem 'rspec'
+  gem 'guard-rspec'
+  gem 'timecop'
 end

Gemfile.lock

@@ -2,22 +2,69 @@ GEM
   remote: https://rubygems.org/
   specs:
     coderay (1.1.2)
+    diff-lcs (1.3)
     dogapi (1.40.0)
       multi_json
+    ffi (1.12.2)
+    formatador (0.2.5)
+    guard (2.16.2)
+      formatador (>= 0.2.4)
+      listen (>= 2.7, < 4.0)
+      lumberjack (>= 1.0.12, < 2.0)
+      nenv (~> 0.1)
+      notiffany (~> 0.0)
+      pry (>= 0.9.12)
+      shellany (~> 0.0)
+      thor (>= 0.18.1)
+    guard-compat (1.2.1)
+    guard-rspec (4.7.3)
+      guard (~> 2.1)
+      guard-compat (~> 1.1)
+      rspec (>= 2.99.0, < 4.0)
+    listen (3.2.1)
+      rb-fsevent (~> 0.10, >= 0.10.3)
+      rb-inotify (~> 0.9, >= 0.9.10)
+    lumberjack (1.2.4)
     method_source (1.0.0)
     multi_json (1.14.1)
+    nenv (0.3.0)
+    notiffany (0.1.3)
+      nenv (~> 0.1)
+      shellany (~> 0.0)
     pry (0.13.1)
       coderay (~> 1.1)
       method_source (~> 1.0)
+    rb-fsevent (0.10.3)
+    rb-inotify (0.10.1)
+      ffi (~> 1.0)
     redis (4.1.3)
+    rspec (3.9.0)
+      rspec-core (~> 3.9.0)
+      rspec-expectations (~> 3.9.0)
+      rspec-mocks (~> 3.9.0)
+    rspec-core (3.9.1)
+      rspec-support (~> 3.9.1)
+    rspec-expectations (3.9.1)
+      diff-lcs (>= 1.2.0, < 2.0)
+      rspec-support (~> 3.9.0)
+    rspec-mocks (3.9.1)
+      diff-lcs (>= 1.2.0, < 2.0)
+      rspec-support (~> 3.9.0)
+    rspec-support (3.9.2)
+    shellany (0.0.1)
+    thor (1.0.1)
+    timecop (0.9.1)
 
 PLATFORMS
   ruby
 
 DEPENDENCIES
   dogapi
+  guard-rspec
   pry
   redis
+  rspec
+  timecop
 
 BUNDLED WITH
    1.17.3

Guardfile

@@ -0,0 +1,43 @@
+# A sample Guardfile
+# More info at https://github.com/guard/guard#readme
+
+## Uncomment and set this to only include directories you want to watch
+# directories %w(app lib config test spec features) \
+#  .select{|d| Dir.exist?(d) ? d : UI.warning("Directory #{d} does not exist")}
+
+## Note: if you are using the `directories` clause above and you are not
+## watching the project directory ('.'), then you will want to move
+## the Guardfile to a watched dir and symlink it back, e.g.
+#
+#  $ mkdir config
+#  $ mv Guardfile config/
+#  $ ln -s config/Guardfile .
+#
+# and, you'll have to watch "config/Guardfile" instead of "Guardfile"
+
+# Note: The cmd option is now required due to the increasing number of ways
+#       rspec may be run, below are examples of the most common uses.
+#  * bundler: 'bundle exec rspec'
+#  * bundler binstubs: 'bin/rspec'
+#  * spring: 'bin/rspec' (This will use spring if running and you have
+#                          installed the spring binstubs per the docs)
+#  * zeus: 'zeus rspec' (requires the server to be started separately)
+#  * 'just' rspec: 'rspec'
+
+guard :rspec, cmd: "bundle exec rspec" do
+  require "guard/rspec/dsl"
+  dsl = Guard::RSpec::Dsl.new(self)
+
+  # Feel free to open issues for suggestions and improvements
+
+  # RSpec files
+  rspec = dsl.rspec
+  watch(rspec.spec_helper) { rspec.spec_dir }
+  watch(rspec.spec_support) { rspec.spec_dir }
+  watch(rspec.spec_files)
+
+  # Ruby files
+  ruby = dsl.ruby
+  dsl.watch_spec_files_for(ruby.lib_files)
+
+end

lambda_function.rb

@@ -0,0 +1,55 @@
+#!/usr/bin/env ruby
+# Copyright 2020 Scribd, Inc.
+
+require 'logger'
+require 'date'
+require 'redis'
+require 'dogapi'
+require_relative 'lib/slowlog_check'
+
+LOGGER = Logger.new($stdout)
+LOGGER.level = Logger::DEBUG
+LOGGER.freeze
+
+def event_time
+  # DateTime because Time does not natively parse AWS CloudWatch Event time
+  DateTime.rfc3339(@event.fetch("time", DateTime.now.rfc3339))
+end
+
+def log_context
+  LOGGER.debug('## ENVIRONMENT VARIABLES')
+  LOGGER.debug(ENV.to_a)
+  LOGGER.debug('## EVENT')
+  LOGGER.debug(@event)
+  LOGGER.info "Event time: #{event_time}."
+end
+
+def lambda_handler(event: {}, context: {})
+  @event = event
+  log_context
+
+  @slowlog_check ||= SlowlogCheck.new(
+    ddog: Dogapi::Client.new(
+      ENV.fetch('DATADOG_API_KEY'),
+      ENV.fetch('DATADOG_APP_KEY')
+    ),
+    redis: Redis.new(
+      host: ENV.fetch('REDIS_HOST'),
+      ssl: :true
+    ),
+    namespace: ENV.fetch('NAMESPACE'),
+    env: ENV.fetch('ENV'),
+    metricname: 'scribddev.redis.slowlog.micros'
+  )
+
+  @slowlog_check.ship_slowlogs
+
+  nil
+end
+
+if __FILE__ == $0
+  lambda_handler
+
+  require 'pry'
+  binding.pry
+end