From e70e721a0d90e7b9b8b9bbf45b92fec2eb7be26b Mon Sep 17 00:00:00 2001
From: Eugen Rochko <eugen@zeonfederated.com>
Date: Sun, 29 Jan 2017 12:35:55 +0100
Subject: [PATCH] Add tests for new API

---
 .../api/v1/devices_controller_spec.rb         | 38 +++++++++++++++++++
 1 file changed, 38 insertions(+)
 create mode 100644 spec/controllers/api/v1/devices_controller_spec.rb

diff --git a/spec/controllers/api/v1/devices_controller_spec.rb b/spec/controllers/api/v1/devices_controller_spec.rb
new file mode 100644
index 00000000000..745a462e3f0
--- /dev/null
+++ b/spec/controllers/api/v1/devices_controller_spec.rb
@@ -0,0 +1,38 @@
+require 'rails_helper'
+
+RSpec.describe Api::V1::DevicesController, type: :controller do
+  let(:user)  { Fabricate(:user, account: Fabricate(:account, username: 'alice')) }
+  let(:token) { double acceptable?: true, resource_owner_id: user.id }
+
+  before do
+    allow(controller).to receive(:doorkeeper_token) { token }
+  end
+
+  describe 'POST #register' do
+    before do
+      post :register, params: { registration_id: 'foo123' }
+    end
+
+    it 'returns http success' do
+      expect(response).to have_http_status(:success)
+    end
+
+    it 'registers device' do
+      expect(Device.where(account: user.account, registration_id: 'foo123').first).to_not be_nil
+    end
+  end
+
+  describe 'POST #unregister' do
+    before do
+      post :unregister, params: { registration_id: 'foo123' }
+    end
+
+    it 'returns http success' do
+      expect(response).to have_http_status(:success)
+    end
+
+    it 'removes device' do
+      expect(Device.where(account: user.account, registration_id: 'foo123').first).to be_nil
+    end
+  end
+end