Add feature gate for snippets

Author: rossta

app/controllers/snippets_controller.rb

@@ -1,4 +1,6 @@
 class SnippetsController < ApplicationController
+  before_action :feature_enabled!
+
   # GET /snippets
   def index
     @snippets = Snippet.all
@@ -54,4 +56,11 @@ def destroy
   def snippet_params
     params.fetch(:snippet, {}).permit(:filename, :source, :url, :language)
   end
+
+  def feature_enabled!
+    return if user_signed_in? &&
+      Flipper.enabled?(:snippets, current_user)
+
+    raise ActionController::RoutingError.new("Not Found")
+  end
 end

spec/requests/snippets_spec.rb

@@ -3,6 +3,8 @@
 RSpec.describe "/snippets", type: :request do
   describe "GET /index" do
     it "renders a successful response" do
+      Flipper.enable(:snippets, login_as_user)
+
       FactoryBot.create(:snippet)
       get snippets_url
       expect(response).to be_successful
@@ -11,6 +13,8 @@
 
   describe "GET /show" do
     it "renders a successful response" do
+      Flipper.enable(:snippets, login_as_user)
+
       snippet = FactoryBot.create(:snippet)
       get snippet_url(snippet)
       expect(response).to be_successful
@@ -19,13 +23,15 @@
 
   describe "GET /new" do
     it "renders a successful response" do
+      Flipper.enable(:snippets, login_as_user)
       get new_snippet_url
       expect(response).to be_successful
     end
   end
 
   describe "GET /edit" do
     it "renders a successful response" do
+      Flipper.enable(:snippets, login_as_user)
       snippet = FactoryBot.create(:snippet)
       get edit_snippet_url(snippet)
       expect(response).to be_successful
@@ -34,6 +40,10 @@
 
   describe "POST /create" do
     context "with valid parameters" do
+      before do
+        Flipper.enable(:snippets, login_as_user)
+      end
+
       it "creates a new Snippet" do
         expect {
           post snippets_url, params: {snippet: {source: "puts \"Hello!\"", language: "ruby"}}
@@ -47,6 +57,10 @@
     end
 
     context "with invalid parameters" do
+      before do
+        Flipper.enable(:snippets, login_as_user)
+      end
+
       it "does not create a new Snippet" do
         expect {
           post snippets_url, params: {snippet: {}}
@@ -62,9 +76,9 @@
 
   describe "PATCH /update" do
     context "with valid parameters" do
-      let(:new_attributes) {
-        skip("Add a hash of attributes valid for your model")
-      }
+      before do
+        Flipper.enable(:snippets, login_as_user)
+      end
 
       it "updates the requested snippet" do
         snippet = FactoryBot.create(:snippet)
@@ -82,6 +96,10 @@
     end
 
     context "with invalid parameters" do
+      before do
+        Flipper.enable(:snippets, login_as_user)
+      end
+
       it "renders a response with 422 status (i.e. to display the 'edit' template)" do
         snippet = FactoryBot.create(:snippet)
         patch snippet_url(snippet), params: {snippet: {language: "does_not_exist"}}
@@ -92,13 +110,15 @@
 
   describe "DELETE /destroy" do
     it "destroys the requested snippet" do
+      Flipper.enable(:snippets, login_as_user)
       snippet = FactoryBot.create(:snippet)
       expect {
         delete snippet_url(snippet)
       }.to change(Snippet, :count).by(-1)
     end
 
     it "redirects to the snippets list" do
+      Flipper.enable(:snippets, login_as_user)
       snippet = FactoryBot.create(:snippet)
       delete snippet_url(snippet)
       expect(response).to redirect_to(snippets_url)