(require-extension srfi-1)
(load "../uri-generic.scm")
(import uri-generic)
(require-extension test)

;; test cases from Python URI implementation

(define  path-cases
  '(("foo:xyz" "bar:abc" "bar:abc")
     ("http://example/x/y/z" "http://example/x/abc" "../abc")
     ("http://example2/x/y/z" "http://example/x/abc" "//example/x/abc")
     ("http://ex/x/y/z" "http://ex/x/r" "../r")

    ("http://ex/x/y"  "http://ex/x/q/r" "./q/r")
    ("http://ex/x/y"  "http://ex/x/q/r#s" "./q/r#s")
    ("http://ex/x/y"  "http://ex/x/q/r#s/t" "./q/r#s/t")
    ("http://ex/x/y"  "ftp://ex/x/q/r" "ftp://ex/x/q/r")
    ("http://ex/x/y"  "http://ex/x/y"   "")
    ("http://ex/x/y/" "http://ex/x/y/"  "")
    ("http://ex/x/y/pdq" "http://ex/x/y/pdq" "")
    ("http://ex/x/y/" "http://ex/x/y/z/" "./z/")
    ("file:/swap/test/animal.rdf" "file:/swap/test/animal.rdf#Animal" "#Animal")
    ("file:/e/x/y/z" "file:/e/x/abc" "../abc")
    ("file:/example2/x/y/z" "file:/example/x/abc" "/example/x/abc")   
    ("file:/ex/x/y/z" "file:/ex/x/r" "../r")
    ("file:/ex/x/y/z" "file:/r" "/r")        
    ("file:/ex/x/y" "file:/ex/x/q/r" "./q/r")
    ("file:/ex/x/y" "file:/ex/x/q/r#s" "./q/r#s")
    ("file:/ex/x/y" "file:/ex/x/q/r#" "./q/r#")
    ("file:/ex/x/y" "file:/ex/x/q/r#s/t" "./q/r#s/t")
    ("file:/ex/x/y" "ftp://ex/x/q/r" "ftp://ex/x/q/r")
    ("file:/ex/x/y" "file:/ex/x/y" "")
    ("file:/ex/x/y/" "file:/ex/x/y/" "")
    ("file:/ex/x/y/pdq" "file:/ex/x/y/pdq" "")
    ("file:/ex/x/y/" "file:/ex/x/y/z/" "./z/")
    ("file:/devel/WWW/2000/10/swap/test/reluri-1.n3" "file://meetings.example.com/cal#m1" 
     "//meetings.example.com/cal#m1")
    ("file:/home/connolly/w3ccvs/WWW/2000/10/swap/test/reluri-1.n3" "file://meetings.example.com/cal#m1" 
     "//meetings.example.com/cal#m1")
    ("file:/some/dir/foo" "file:/some/dir/#blort" "./#blort")
    ("file:/some/dir/foo" "file:/some/dir/#" "./#")
    ;; From Graham Klyne Thu 20 Feb 2003 18:08:17 +0000
   ("http://example/x/y%2Fz"  "http://example/x/abc"     "./abc")
   ("http://example/x/y/z"    "http://example/x%2Fabc"   "/x%2Fabc")
   ("http://example/x/y%2Fz"  "http://example/x%2Fabc"   "/x%2Fabc")
   ("http://example/x%2Fy/z"  "http://example/x%2Fy/abc" "./abc")
   ;; Ryan Lee
   ("http://example/x/abc.efg" "http://example/x/" "./")
   ))

(define base "http://a/b/c/d;p?q")

(define rfc-cases
  `((,base "g:h" "g:h")
    (,base "g" "http://a/b/c/g")
    (,base "./g" "http://a/b/c/g")
    (,base "g/" "http://a/b/c/g/")
    (,base "/g" "http://a/g")
    (,base "//g" "http://g")
    (,base "?y" "http://a/b/c/?y") 
    (,base "g?y" "http://a/b/c/g?y")
    (,base "#s" "http://a/b/c/d;p?q#s") 
    (,base "g#s" "http://a/b/c/g#s")
    (,base "g?y#s" "http://a/b/c/g?y#s")
    (,base ";x" "http://a/b/c/;x")
    (,base "g;x" "http://a/b/c/g;x")
    (,base "g;x?y#s" "http://a/b/c/g;x?y#s")
    (,base "."  "http://a/b/c/")
    (,base "./" "http://a/b/c/")
    (,base ".." "http://a/b/")
    (,base "../" "http://a/b/")
    (,base "../g" "http://a/b/g")
    (,base "../.." "http://a/")
    (,base "../../" "http://a/")
    (,base "../../g" "http://a/g")
    ))

(define extra-cases
  `((,base "" ,base)
    (,base "../../../g" "http://a/g") 
    (,base "../../../../g" "http://a/g")
    (,base "../../../.." "http://a/") ; Is this correct? Or http://a ?
    (,base "../../../../" "http://a/")
    (,base "/./g" "http://a/g")
    (,base "/../g" "http://a/g")
    (,base "g.." "http://a/b/c/g..")
    (,base "..g" "http://a/b/c/..g")
    
    (,base "./../g" "http://a/b/g")
    (,base "./g/." "http://a/b/c/g/") 
    (,base "g/./h" "http://a/b/c/g/h") 
    (,base "g/../h" "http://a/b/c/h")
    (,base "g;x=1/./y" "http://a/b/c/g;x=1/y") 
    (,base "g;x=1/../y" "http://a/b/c/y")  
    
    (,base "g?y/./x" "http://a/b/c/g?y/./x")
    (,base "g?y/../x" "http://a/b/c/g?y/../x")
    (,base "g#s/./x" "http://a/b/c/g#s/./x")
    (,base "g#s/../x" "http://a/b/c/g#s/../x")
    
    ("?a=b&c=d" "" "?a=b&c=d")
    (,base "" "http://a/b/c/d;p?q")
    ("" ,base "http://a/b/c/d;p?q")
    (,base "http:" "http:")
    (,base "..%2f" "http://a/b/c/..%2f")
    ("http://a/b/c/d/" ".." "http://a/b/c/")
    ("http://a/b/c/d/" "../e" "http://a/b/c/e")
    ("http://a/b/c/d/" "../e/" "http://a/b/c/e/")
    ("http://a/b//c///d///" "..//.." "http://a/b/")
    ("http://a" "b" "http://a/b") ; RFC3986, section 5.2.3, first bullet point
    ))

(define reverse-extra-cases
  `((,base ,base "")
    (,base "http://a/b/c/e" "./e")
    (,base "http://a/b/e" "../e")
    (,base "http://a/" "/") ;; or "../../"
    (,base "http://a" "//a") ; No relative representation possible
    (,base "http://b" "//b")
    (,base "http://b/" "//b/")
    (,base "http://b/c" "//b/c")
    (,base "ftp://a/b/c/d;p?q" "ftp://a/b/c/d;p?q")
    (,base "ftp://x/y/z;a?b" "ftp://x/y/z;a?b")))

(test-group "uri test"
  (for-each (lambda (p)
              (let ((ubase (uri-reference (first p)))
                    (urabs  (uri-reference (second p)))
                    (uex   (uri-reference (third p))))
                (let* ((from (uri-relative-from urabs ubase))
                       (to    (uri-relative-to from ubase)))
                  (test (apply sprintf "~S * ~S -> ~S" p) uex from)
                  (test (apply sprintf "~S * ~S -> ~S" p) urabs to)
                  (unless (uri-fragment urabs)
                    (let ((uabs  (absolute-uri (second p))))
                      (test (sprintf "~S = ~S" uabs urabs) urabs uabs)))
                  ))
              (for-each
               (lambda (s)
                 (test (sprintf "~S = ~S" s (uri->string (uri-reference s)))
                       s (uri->string (uri-reference s))))
               p))
            path-cases))

(test-group "rfc test"
  (for-each (lambda (p)
              (let ((ubase (uri-reference (first p)))
                    (urabs  (uri-reference (second p)))
                    (uex   (uri-reference (third p))))
                (let* ((to    (uri-relative-to urabs ubase)))
                  (test (apply sprintf "~S * ~S -> ~S" p) uex to)
                  ))
              (for-each
               (lambda (s)
                 (test (sprintf "~S = ~S" s (uri->string (uri-reference s)))
                       s (uri->string (uri-reference s))))
               p))
            rfc-cases))

(test-group "extra-test"
  (for-each (lambda (p)
              (let ((ubase (uri-reference (first p)))
                    (urabs  (uri-reference (second p)))
                    (uex   (uri-reference (third p))))
                (let* ((to    (uri-relative-to urabs ubase)))
                  (test (apply sprintf "~S * ~S -> ~S" p) uex to)
                  )))
            extra-cases))

(test-group "reverse-extra-test"
  (for-each (lambda (p)
              (let ((ubase (uri-reference (first p)))
                    (urabs  (uri-reference (second p)))
                    (uex   (uri-reference (third p))))
                (let* ((to    (uri-relative-from urabs ubase)))
                  (test (apply sprintf "~S * ~S -> ~S" p) uex to)
                  )))
            reverse-extra-cases))

(define encode/decode-cases
  '(("foo?bar" "foo%3Fbar")
    ("foo&bar" "foo%26bar")
    ("foo%20bar" "foo%2520bar")
    ("foo\x00bar\n" "foo%00bar%0A")))

(test-group "uri-encode-string test"
  (for-each (lambda (p)
              (let ((expected (second p))
                    (encoded (uri-encode-string (first p))))
                  (test (sprintf "~S -> ~S" (first p) expected) expected encoded)))
            encode/decode-cases))

(test-group "uri-decode-string test"
  (for-each (lambda (p)
              (let ((expected (first p))
                    (decoded (uri-decode-string (second p))))
                  (test (sprintf "~S -> ~S" (second p) expected) expected decoded)))
            encode/decode-cases))

(define normalize-case-cases
  '(("http://exa%2fmple/FOO%2fbar" "http://exa%2Fmple/FOO%2Fbar")
    ("http://EXA%2fMPLE/FOO%2fbar" "http://exa%2Fmple/FOO%2Fbar")
    ("HTTP://example/" "http://example/")
    ("http://user:PASS@example/FOO%2fbar" "http://user:PASS@example/FOO%2Fbar")
    ("http://uS%2fer:PA%2fSS@example/FOO%2fbar" "http://uS%2Fer:PA%2FSS@example/FOO%2Fbar")
    ("HTTP://example/?mooH=MUMBLe%2f" "http://example/?mooH=MUMBLe%2F")
    ("http://example/#baR%2f" "http://example/#baR%2F")))

(test-group "normalize-case test"
  (for-each (lambda (p)
              (let ((case-normalized (uri-normalize-case (uri-reference (first p))))
                    (expected (second p)))
                  (test (sprintf "~S -> ~S" (first p) (second p)) expected (uri->string case-normalized (lambda (user pass) (conc user ":" pass))))))
            normalize-case-cases))

(define internal-representation-cases
  `(("scheme" ,uri-scheme
     ;; pct-encoding not allowed in scheme
     ("http.:" http.)
     ("http+:" http+)
     ("http-:" http-)
     ("HTTP:" HTTP)
     ("" #f)
     ("/foo" #f)
     (":" ||))
    ("username" ,uri-username
     ("//foo" #f)
     ("//@" "")
     ("//foo@" "foo")
     ("//foo:bar@" "foo")
     ("//foo:bar:qux@" "foo")
     ("//foo%20bar@" "foo%20bar")
     ("//foo%3Abar:qux@" "foo%3Abar") ;; %3A = ':'
     ("//foo%2Ebar@" "foo.bar" "//foo.bar@"))
    ("password ",uri-password
     ("//foo" #f)
     ("//@" #f)
     ("//foo@" #f)
     ("//foo:bar@" "bar")
     ("//foo:bar:qux@" "bar:qux")
     ("//foo:bar%20qux@" "bar%20qux")
     ("//foo:bar%2Equx@" "bar.qux" "//foo:bar.qux@"))
    ("path" ,uri-path
     ("//foo" ())   ; Can path ever be #f?
     ("foo%20bar" ("foo%20bar"))
     ("foo%2Fbar" ("foo/bar"))
     ("foo%2ebar" ("foo.bar") "foo.bar")
     ("foo/bar%2Fqux" ("foo" "bar/qux"))
     ("foo/" ("foo" ""))
     ("foo/bar:qux" ("foo" "bar:qux"))
     ("/foo%2Fbar" (/ "foo/bar"))
     ("/foo/" (/ "foo" ""))
     ("/" (/ ""))
     ("/?foo" (/ ""))
     ("/#foo" (/ ""))
     ("/foo:bar" (/ "foo:bar")))
    ("query ",uri-query
     ("//" #f)
     ("/?foo" "foo")
     ("?foo" "foo")
     ("?foo?bar" "foo?bar")
     ("?foo/bar" "foo/bar")
     ("?foo%3Fbar" "foo%3Fbar")
     ("?foo%2Ebar" "foo.bar" "?foo.bar"))
    ("fragment" ,uri-fragment
     ("?foo" #f)
     ("#bar" "bar")
     ("/#bar" "bar")
     ("?foo#bar" "bar")
     ("/?foo#bar" "bar")
     ("#foo?bar" "foo?bar")
     ("#foo/bar" "foo/bar")
     ("#foo%3Fbar" "foo%3Fbar")
     ("#foo%2Ebar" "foo.bar" "#foo.bar"))))

(test-group "internal representations"
  (for-each (lambda (p)
              (test-group (car p)
               (for-each (lambda (u)
                           (let ((in (first u))
                                 (internal (second u))
                                 (out (if (null? (cddr u))
                                          (first u)
                                          (third u)))
                                 (uri (uri-reference (first u))))
                             (test (sprintf "~S decoded as ~S" in internal)
                                   internal ((cadr p) uri))
                             (test (sprintf "~S encoded to ~S" internal out)
                                   out (uri->string uri
                                                    (lambda (u p)
                                                      (if p (conc u ":" p) u))))))
                         (cddr p))))
            internal-representation-cases))

;; I wonder if there's a term for this :)
(define non-relative-non-absolute-uri-references
  '("http://foo#frag"
    "http://foo?a=b#frag"
    "http://foo/bar#frag"
    "http://foo/bar?a=b#frag"))

(define absolute-uris
  '("http://foo"
    "http://foo?a=b"
    "http://foo/bar"
    "http://foo/bar?a=b"))

(define relative-refs
  `(""
    "bar"
    "bar?a=b"
    "bar#frag"
    "bar?a=b#frag"
    "/"
    "/bar"
    "/bar?a=b"
    "/bar#frag"
    "/bar?a=b#frag"
    "//foo"
    "//foo?a=b"
    "//foo#frag"
    "//foo?a=b#frag"
    "//foo/bar"
    "//foo/bar?a=b"
    "//foo/bar#frag"
    "//foo/bar?a=b#frag"))

(test-group "absolute/relative distinction"
  (for-each (lambda (s)
              (test-assert (sprintf "~S is a relative ref" s)
                           (relative-ref? (uri-reference s)))
              (test-assert (sprintf "~S is not an URI" s)
                           (not (uri? (uri-reference s))))
              (test-assert (sprintf "~S is not an absolute URI" s)
                           (not (absolute-uri? (uri-reference s))))
              (test-error (absolute-uri s)))
            relative-refs)
  (for-each (lambda (s)
              (test-assert (sprintf "~S is not a relative ref" s)
                           (not (relative-ref? (uri-reference s))))
              (test-assert (sprintf "~S is an URI" s)
                           (uri? (uri-reference s)))
              (test-assert (sprintf "~S is an absolute URI" s)
                           (absolute-uri? (uri-reference s)))
              (test (uri-reference s) (absolute-uri s)))
            absolute-uris)
  (for-each (lambda (s)
              (test-assert (sprintf "~S is not a relative ref" s)
                           (not (relative-ref? (uri-reference s))))
              (test-assert (sprintf "~S is an URI" s)
                           (uri? (uri-reference s)))
              (test-assert (sprintf "~S is not an absolute URI" s)
                           (not (absolute-uri? (uri-reference s))))
              ;; Should this give an error in the fragment case?
              (test-error (absolute-uri s)))
            non-relative-non-absolute-uri-references))

(define absolute-paths
  '("/"
    "/foo"
    "//foo/"
    "http://foo/bar"
    "http://foo/"
    "http://foo/#qux"
    "http://foo/?bar=qux"))

(define relative-paths
  '(""
    "http://foo"
    "//foo"
    "http://foo#bar"
    "http://foo?bar=qux"))

(test-group "absolute/relative path distinction"
  (for-each (lambda (s)
              (test-assert (sprintf "~S is not a relative path" s)
                           (not (uri-path-relative? (uri-reference s))))
              (test-assert (sprintf "~S is an absolute path" s)
                           (uri-path-absolute? (uri-reference s))))
            absolute-paths)
  (for-each (lambda (s)
              (test-assert (sprintf "~S is a relative path" s)
                           (uri-path-relative? (uri-reference s)))
              (test-assert (sprintf "~S is not an absolute path" s)
                           (not (uri-path-absolute? (uri-reference s)))))
            relative-paths))

(test-group "miscellaneous"
  ;; Special case, see section 4.2
  (test "./foo:bar" (uri->string (update-uri (uri-reference "") path: '("foo:bar")))))