Merge pull request saymedia#13 from kshay/master

Enhance the overriden $location service's search() method.

Author: kshay

lib/ngoverrides.js

@@ -103,12 +103,12 @@ function registerModule(context) {
                 };
             };
             var parsedUrl = function (code) {
-                return function (set) {
+                return function (set, value) {
                     if (request) {
                         if (! requestUrlParts) {
                             requestUrlParts = url.parse(request.url, true);
                         }
-                        return code(requestUrlParts, set);
+                        return code(requestUrlParts, set, value);
                     }
                     else {
                         throw new Error('location not available yet');
@@ -170,7 +170,19 @@ function registerModule(context) {
                         }
                     ),
                     search: parsedUrl(
-                        function (parts) {
+                        function (parts, set, paramValue) {
+                            if (set) {
+                                if (paramValue === null) {
+                                    delete parts.query[paramValue];
+                                }
+                                else if (paramValue) {
+                                    parts.query[set] = paramValue;
+                                }
+                                else {
+                                    parts.query = set;
+                                }
+                                return this;
+                            }
                             return parts.query;
                         }
                     ),