Changed ConnectionObject and updated unit tests accordingly.

[xestiaab/.git] / source / connobject / ConnectionObject.h

diff --git a/source/connobject/ConnectionObject.h b/source/connobject/ConnectionObject.h
index 9ffbe4d..4ba84b8 100644 (file)
--- a/source/connobject/ConnectionObject.h
+++ b/source/connobject/ConnectionObject.h
@@ -21,10 +21,12 @@
 
 #include <string>
 #include <vector>
+#include <iostream>
 
 enum COConnectResult {
        COCONNECT_UNITTESTFAIL = -1,
        COCONNECT_OK,
+       COCONNECT_SSLFAIL,
        COCONNECT_INVALID,
        COCONNECT_TIMEOUT,
        COCONNECT_AUTHFAIL
@@ -73,12 +75,12 @@ class ConnectionObject{
        
                // Virtual functions to be setup by the inheriting classes.
        
-               virtual void SetupConnectionObject() {};
+               virtual void SetupConnectionObject() { };
                //virtual ~ConnectionObject() {};
                
                virtual bool IsTaskCompleted() {};
                
-               virtual COConnectResult Connect() {};
+               virtual COConnectResult Connect(bool DoAuthentication) {};
                
                virtual std::string GetDefaultPrefix() {};
                virtual COServerResponse AddContact(std::string Location, std::string Data) {};
@@ -93,6 +95,8 @@ class ConnectionObject{
                virtual COSSLVerified SSLVerify() {};
                virtual bool AbleToLogin() {};
                virtual bool HasValidResponse() {};
+               virtual bool IsSelfSigned() {};
+               virtual std::string GetErrorMessage() {};
                
        protected:
                // Test Mode.
@@ -109,7 +113,6 @@ class ConnectionObject{
                std::string ServerAccount = "";
                bool ServerSSL = true;
                std::string ErrorMessage = "";
-               std::string ErrorBufferMessage = "";
        
                // Connect results.