2 files changed, 2 insertions, 0 deletions

diff --git a/test/idiscover.cc b/test/idiscover.cc
index 4b1e90c..8a5a3fb 100644
--- a/test/idiscover.cc
+++ b/test/idiscover.cc
@@ -1,41 +1,42 @@
 #include <iostream>
 #include <stdexcept>
 #include <iterator>
 #include <algorithm>
+#include <unistd.h>
 using namespace std;
 #include <opkele/exception.h>
 #include <opkele/discovery.h>
 #include <opkele/util.h>
 #include <opkele/util-internal.h>
 
 namespace opkele {
     ostream& operator<<(ostream& o,const opkele::openid_endpoint_t& oep) {
         o
             << " URI:        " << oep.uri << endl
             << " Claimed ID: " << oep.claimed_id << endl
             << " Local ID:   " << oep.local_id << endl;
         return o;
     }
 }
 
 int main(int argc,char **argv) {
     try {
         if(argc<2)
             throw opkele::exception(OPKELE_CP_ "Please, give me something to resolve");
         for(int a=1;a<argc;++a) {
             cout << "==============================================================" << endl
                 << "User-supplied ID: " << argv[a] << endl
                 << "Endpoints:" << endl
                 << " --" << endl;
             string normalized = opkele::idiscover(
                         ostream_iterator<opkele::openid_endpoint_t>(cout," --\n")
                      ,argv[a]);
             cout << "Normalized ID:   " << normalized << endl;
         }
     }catch(exception& e) {
         cerr << "oops, caught " << opkele::util::abi_demangle(typeid(e).name()) << endl
             << " .what(): " << e.what() << endl;
         _exit(1);
     }
     _exit(0);
 }
diff --git a/test/test.cc b/test/test.cc
index 2143ac0..770e0c6 100644
--- a/test/test.cc
+++ b/test/test.cc
@@ -1,50 +1,51 @@
 #include <iostream>
 #include <stdexcept>
+#include <unistd.h>
 using namespace std;
 #include <opkele/exception.h>
 #include <opkele/util.h>
 
 #include "config.h"
 
 class failed_test : public opkele::exception {
     public:
         failed_test(OPKELE_E_PARS)
             : exception(OPKELE_E_CONS) { }
 };
 
 void test_rfc_3986_normalize_uri(const string &ouri,bool success,const string& nuri="") {
     try {
         string n = opkele::util::rfc_3986_normalize_uri(ouri);
         if(!success)
             throw failed_test(OPKELE_CP_ "Normalized URI when it shouldn't ('"+ouri+"' normalization resulted in '"+n+"')");
         if(n!=nuri)
             throw failed_test(OPKELE_CP_ "rfc_3986_test_failed for '"+ouri+"' failed, expected '"+nuri+"', got '"+n+"'");
     }catch(opkele::bad_input& obi) {
         if(success)
             throw failed_test(OPKELE_CP_ "Test '"+ouri+"' failed due to 'bad_input'["+obi.what()+"]");
     }catch(opkele::not_implemented& oni) {
         if(success)
             throw failed_test(OPKELE_CP_ "Test '"+ouri+"' failed due to 'not_implemented'["+oni.what()+"]");
     }
 }
 
 void test_rfc_3986_normalize_uri() {
     test_rfc_3986_normalize_uri(
             "invalid", false );
     test_rfc_3986_normalize_uri(
             "http://", false );
     test_rfc_3986_normalize_uri(
             "http:/hacker.klever.net/", false );
     test_rfc_3986_normalize_uri(
             "hTTp://hacker.klever.net#uh?oh", true, "http://hacker.klever.net/#uh?oh" );
     test_rfc_3986_normalize_uri(
             "http://hacker.klever.net?uh#oh", true, "http://hacker.klever.net/?uh#oh" );
     test_rfc_3986_normalize_uri(
             "http://hacker.klever.net:80/", true, "http://hacker.klever.net/" );
     test_rfc_3986_normalize_uri(
             "http://hacker.klever.net:80?uh", true, "http://hacker.klever.net/?uh" );
     test_rfc_3986_normalize_uri(
             "http://hacker.klever.net:80#uh", true, "http://hacker.klever.net/#uh" );
     test_rfc_3986_normalize_uri(
             "https://hacker.klever.net:443", true, "https://hacker.klever.net/" );
     test_rfc_3986_normalize_uri(