1 files changed, 68 insertions, 0 deletions

diff --git a/test/test.cc b/test/test.cc
new file mode 100644
index 0000000..f0526c6
--- a/dev/null
+++ b/test/test.cc
@@ -0,0 +1,68 @@
+#include <iostream>
+#include <stdexcept>
+using namespace std;
+#include <opkele/exception.h>
+#include <opkele/consumer.h>
+
+#include "config.h"
+
+class failed_test : public opkele::exception {
+    public:
+        failed_test(OPKELE_E_PARS)
+            : exception(OPKELE_E_CONS) { }
+};
+
+class dummy_consumer_t : public opkele::consumer_t {
+    public:
+        virtual opkele::assoc_t store_assoc(const string& server,const string& handle,const opkele::secret_t& secret,int expires_in) {
+            throw opkele::not_implemented(OPKELE_CP_ "Not implemented");
+        }
+        virtual opkele::assoc_t retrieve_assoc(const string& server,const string& handle) {
+            throw opkele::not_implemented(OPKELE_CP_ "Not implemented");
+        }
+        virtual void invalidate_assoc(const string& server,const string& handle) {
+            throw opkele::not_implemented(OPKELE_CP_ "Not implemented");
+        }
+};
+
+
+void test_retrieve_links(const string& f,bool success,const string& s="",const string& d="") {
+    dummy_consumer_t dc;
+    string server, delegate;
+    try {
+        dc.retrieve_links("file://" OPKELE_SRC_DIR "/test/html/"+f,server,delegate);
+        if(!success)
+            throw failed_test(OPKELE_CP_ "Retrieved links when it shouldn't");
+        if(server!=s)
+            throw failed_test(OPKELE_CP_ "retrieve_links test failed, expected server '"+s+"', got '"+server+"'");
+        if(delegate!=d)
+            throw failed_test(OPKELE_CP_ "retrieve_links test failed, expected delegate '"+d+"', got '"+delegate+"'");
+    }catch(opkele::bad_input& obi) {
+        if(success)
+            throw failed_test(OPKELE_CP_ "Test failed");
+    }catch(opkele::failed_assertion& ofa) {
+        if(success)
+            throw failed_test(OPKELE_CP_ "Test failed");
+    }
+}
+
+void test_retrieve_links() {
+    test_retrieve_links("empty.html",false);
+    test_retrieve_links("in-body.html",false);
+    test_retrieve_links("head-in-body.html",false);
+    test_retrieve_links("hkn.html",true,"http://www.klever.net/openid.server","http://hacker.klever.net/");
+    test_retrieve_links("hkn-server.html",true,"http://www.klever.net/openid.server");
+    test_retrieve_links("hkn-delegate.html",false);
+}
+
+main() {
+    try {
+        test_retrieve_links();
+    }catch(failed_test& ft) {
+        cerr << "Test failed: " << ft.what() << endl;
+    }catch(exception& e) {
+        cerr << "oops: " << e.what() << endl;
+        _exit(1);
+    }
+    _exit(0);
+}