MXS-1461 Modify some mock-classes

server/modules/filter/dbfwfilter/test/maxscale/mock/dcb.hh

@@ -43,6 +43,11 @@ public:
         const char* zHost);
     ~Dcb();
 
+private:
+    int32_t write(GWBUF* pData);
+
+    static int32_t write(DCB* pDcb, GWBUF* pData);
+
 private:
     std::string m_user;
     std::string m_host;

server/modules/filter/dbfwfilter/test/maxscale/mock/session.hh

@@ -14,6 +14,7 @@
 
 #include "mock.hh"
 #include <maxscale/session.h>
+#include <maxscale/protocol/mysql.h>
 #include "dcb.hh"
 
 namespace maxscale
@@ -43,7 +44,8 @@ public:
     ~Session();
 
 private:
-    Dcb m_client_dcb;
+    Dcb           m_client_dcb;
+    MYSQL_session m_mysql_session;
 };
 
 }

server/modules/filter/dbfwfilter/test/mock_backend.cc

@@ -73,7 +73,6 @@ bool BufferBackend::idle(const RouterSession* pSession) const
     bool rv = true;
 
     SessionResponses::const_iterator i = m_session_responses.find(pSession);
-    ss_dassert(i != m_session_responses.end());
 
     if (i != m_session_responses.end())
     {

server/modules/filter/dbfwfilter/test/mock_dcb.cc

@@ -47,12 +47,27 @@ Dcb::Dcb(MXS_SESSION* pSession,
     pDcb->session = pSession;
     pDcb->remote = const_cast<char*>(zHost);
     pDcb->user = const_cast<char*>(zUser);
+
+    pDcb->func.write = &Dcb::write;
 }
 
 Dcb::~Dcb()
 {
 }
 
+int32_t Dcb::write(GWBUF* pData)
+{
+    // TODO: Should be routed somewhere
+    gwbuf_free(pData);
+    return 1;
+}
+
+//static
+int32_t Dcb::write(DCB* pDcb, GWBUF* pData)
+{
+    return static_cast<Dcb*>(pDcb)->write(pData);
+}
+
 }
 
 }

server/modules/filter/dbfwfilter/test/mock_session.cc

@@ -32,6 +32,12 @@ Session::Session(const char* zUser,
     pSession->ses_chk_tail = CHK_NUM_SESSION;
 
     pSession->client_dcb = &m_client_dcb;
+
+    memset(&m_mysql_session, 0, sizeof(m_mysql_session));
+
+    strcpy(m_mysql_session.db, "dummy");
+
+    m_client_dcb.data = &m_mysql_session;
 }
 
 Session::~Session()