;; disfluid, implementation of the Solid specification
;; Copyright (C) 2020, 2021  Vivien Kraus

;; This program is free software: you can redistribute it and/or modify
;; it under the terms of the GNU Affero General Public License as
;; published by the Free Software Foundation, either version 3 of the
;; License, or (at your option) any later version.

;; This program is distributed in the hope that it will be useful,
;; but WITHOUT ANY WARRANTY; without even the implied warranty of
;; MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
;; GNU Affero General Public License for more details.

;; You should have received a copy of the GNU Affero General Public License
;; along with this program.  If not, see <https://www.gnu.org/licenses/>.

(use-modules (webid-oidc refresh-token)
             (webid-oidc testing)
             (webid-oidc errors)
             (webid-oidc jwk)
             (web uri)
             (srfi srfi-19)
             (web response)
             (ice-9 optargs)
             (ice-9 receive))

(with-test-environment
 "refresh-token"
 (lambda ()
   (define sub-a (string->uri "https://subject.a"))
   (define sub-b (string->uri "https://subject.b"))
   (define aud-a (string->uri "https://client.a"))
   (define aud-b (string->uri "https://client.b"))
   (define key-a (generate-key #:n-size 2048))
   (define key-b (generate-key #:n-size 2048))
   (define refresh-a (issue-refresh-token sub-a aud-a (jkt key-a)))
   (define refresh-b (issue-refresh-token sub-b aud-b (jkt key-b)))
   (unless (string? refresh-a)
     (exit 2))
   (unless (string? refresh-b)
     (exit 3))
   (unless
       (with-refresh-token refresh-a key-a
                           (lambda (sub aud)
                             (unless (equal? sub-a sub)
                               (exit 4))
                             (unless (equal? aud-a aud)
                               (exit 5))))
     (exit 6))
   (unless
       (with-refresh-token refresh-b key-b
                           (lambda (sub aud)
                             (unless (equal? sub-b sub)
                               (exit 7))
                             (unless (equal? aud-b aud)
                               (exit 8))))
     (exit 9))
   (remove-refresh-token sub-b aud-b)
   (with-exception-handler
       (lambda (error)
         (unless (invalid-refresh-token? error)
           (exit 10)))
     (lambda ()
       (with-refresh-token refresh-b key-b
                           (lambda (sub aud)
                             (exit 11))))
     #:unwind? #t
     #:unwind-for-type &invalid-refresh-token)))