[VOL-2061] OpenOLT should wait till BAL is ready before it can configure
the PON schedulers
Fixed unit test cases per the new code.
Change-Id: I51e49ee57b64d62558b3c9f0faa8fe8653916a6b
diff --git a/agent/test/inc/bal_mocker.h b/agent/test/inc/bal_mocker.h
index 85bdb22..72cbbc4 100644
--- a/agent/test/inc/bal_mocker.h
+++ b/agent/test/inc/bal_mocker.h
@@ -18,7 +18,10 @@
#ifndef __BAL_MOCKER_H__
#define __BAL_MOCKER_H__
#include <cmock/cmock.h>
+#include <gmock-global/gmock-global.h>
#include <cstdlib>
+#include <grpc++/grpc++.h>
+using grpc::Status;
extern "C" {
#include <bcmos_system.h>
#include <bcmolt_api.h>
@@ -41,5 +44,7 @@
MOCK_METHOD1(bcmolt_host_init, bcmos_errno(bcmolt_host_init_parms*));
MOCK_METHOD2(bcmolt_cfg_get, bcmos_errno(bcmolt_oltid, bcmolt_cfg*));
MOCK_METHOD2(bcmolt_oper_submit, bcmos_errno(bcmolt_oltid, bcmolt_oper*));
+ // Add more here
};
+
#endif