add support for testing with users with roles

Signed-off-by: Jaka Hudoklin <[email protected]>

Author: offlinehacker

lib/helpers.js

@@ -109,6 +109,54 @@ _beforeEach.givenModel = function(modelName, attrs, optionalHandler) {
 
 _beforeEach.givenUser = function(attrs, optionalHandler) {
   _beforeEach.givenModel('user', attrs, optionalHandler);
+} 
+
+_beforeEach.givenUserWithRole = function (attrs, role, optionalHandler) {
+  _beforeEach.givenUser(attrs, function (done) {
+    var test = this;
+    test.app.models.Role.create({name: role}, function (err, result) {
+      if(err) {
+        console.error(err.message);
+        if(err.details) console.error(err.details);
+        return done(err);
+      }
+
+      test.userRole = result;
+      test.app.models.roleMapping.create(
+        {principalId: test.user.id,
+         principalType: test.app.models.roleMapping.USER,
+         roleId: result.id},
+        function (err, result) {
+          if(err) {
+            console.error(err.message);
+            if(err.details) console.error(err.details);
+            return done(err);
+          }
+    
+          test.userRoleMapping = result;
+          done();
+        }
+      );
+    });
+  });
+
+  if(typeof optionalHandler === 'function') {
+    beforeEach(optionalHandler);
+  }
+
+  afterEach(function(done) {
+    var test = this;
+    this.userRole.destroy(function(err) {
+      if(err) return done(err);
+      test.userRole = undefined;
+
+      test.userRoleMapping.destroy(function(err) {
+        if(err) return done(err);
+        test.userRoleMapping = undefined;
+        done();
+      });
+    });
+  });
 }
 
 _beforeEach.givenLoggedInUser = function(credentials, optionalHandler) {
@@ -134,6 +182,29 @@ _beforeEach.givenLoggedInUser = function(credentials, optionalHandler) {
   });
 }
 
+_beforeEach.givenLoggedInUserWithRole = function(credentials, role, optionalHandler){
+  _beforeEach.givenUserWithRole(credentials, role, function(done) {
+    var test = this;
+    this.user.constructor.login(credentials, function(err, token) {
+      if(err) {
+        done(err);
+      } else {
+        test.loggedInAccessToken = token;
+        done();
+      }
+    });
+  });
+
+  afterEach(function(done) {
+    var test = this;
+    this.loggedInAccessToken.destroy(function(err) {
+      if(err) return done(err);
+      test.loggedInAccessToken = undefined;
+      done();
+    });
+  });
+}
+
 _beforeEach.givenAnUnauthenticatedToken = function(attrs, optionalHandler) {
   _beforeEach.givenModel('accessToken', attrs, optionalHandler);
 }
@@ -197,13 +268,27 @@ _describe.whenLoggedInAsUser = function(credentials, cb) {
   });
 }
 
+_describe.whenLoggedInAsUserWithRole = function(credentials, role, cb) {
+  describe('when logged in as user', function () {
+    _beforeEach.givenLoggedInUser(credentials, role);
+    cb();
+  });
+}
+
 _describe.whenCalledByUser = function(credentials, verb, url, data, cb) {
   describe('when called by logged in user', function () {
     _beforeEach.givenLoggedInUser(credentials);
     _describe.whenCalledRemotely(verb, url, data, cb);
   });
 }
 
+_describe.whenCalledByUserWithRole = function (credentials, role, verb, url, data, cb) {
+   describe('when called by logged in user with role ' + role, function () {
+    _beforeEach.givenLoggedInUserWithRole(credentials, role);
+    _describe.whenCalledRemotely(verb, url, data, cb);
+  }); 
+}
+
 _describe.whenCalledAnonymously = function(verb, url, data, cb) {
   describe('when called anonymously', function () {
     _beforeEach.givenAnAnonymousToken();
@@ -285,3 +370,17 @@ function(credentials, verb, url) {
     _it.shouldBeDenied();
   });
 }
+
+_it.shouldBeAllowedWhenCalledByUserWithRole =
+function(credentials, role, verb, url, data) {
+  _describe.whenCalledByUserWithRole(credentials, role, verb, url, data, function() {
+    _it.shouldBeAllowed();
+  });
+}
+
+_it.shouldBeDeniedWhenCalledByUserWithRole =
+function(credentials, role, verb, url) {
+  _describe.whenCalledByUserWithRole(credentials, role, verb, url, function() {
+    _it.shouldBeDenied();
+  });
+}