Retry interface implementation must exactly match (#142)

fixes #141

src/retry/endpointDiscoveryRetryPolicy.ts

@@ -36,13 +36,17 @@ export class EndpointDiscoveryRetryPolicy implements IRetryPolicy {
    */
   public async shouldRetry(
     err: ErrorResponse,
-    retryContext: RetryContext,
+    retryContext?: RetryContext,
-    locationEndpoint: string
+    locationEndpoint?: string
   ): Promise<boolean | [boolean, string]> {
     if (!err) {
       return false;
     }
 
+    if (!retryContext || !locationEndpoint) {
+      return false;
+    }
+
     if (!this.globalEndpointManager.enableEndpointDiscovery) {
       return false;
     }

src/retry/sessionRetryPolicy.ts

@@ -34,11 +34,15 @@ export class SessionRetryPolicy implements IRetryPolicy {
    * @param {function} callback - The callback function which takes bool argument which specifies whether the request\
    * will be retried or not.
    */
-  public async shouldRetry(err: ErrorResponse, retryContext: RetryContext): Promise<boolean> {
+  public async shouldRetry(err: ErrorResponse, retryContext?: RetryContext): Promise<boolean> {
     if (!err) {
       return false;
     }
 
+    if (!retryContext) {
+      return false;
+    }
+
     if (!this.connectionPolicy.EnableEndpointDiscovery) {
       return false;
     }