require "pg"

module Support
  module Helpers
    class RoyaltiesPG
      def clear_dags
        conn = PG.connect(ENV["PG_HOST"],
                          ENV["PG_PORT"],
                          "",
                          "",
                          ENV["PG_DB_NAME"],
                          ENV["PG_USER"],
                          ENV["PG_PASS"])
        conn.exec("DELETE FROM dag_run WHERE state='running'")
      end
    end
  end
end
