Index: firmware/App/Modes/ModeDrain.h
===================================================================
diff -u -rdcd360fb4dc37db2dcbeb7fb14fb327fe68235f4 -r7d4711edd7b40cd3e29f43e766f79a8a09586fe9
--- firmware/App/Modes/ModeDrain.h	(.../ModeDrain.h)	(revision dcd360fb4dc37db2dcbeb7fb14fb327fe68235f4)
+++ firmware/App/Modes/ModeDrain.h	(.../ModeDrain.h)	(revision 7d4711edd7b40cd3e29f43e766f79a8a09586fe9)
@@ -1,14 +1,14 @@
 /**************************************************************************
 *
-* Copyright (c) 2019-2021 Diality Inc. - All Rights Reserved.
+* Copyright (c) 2019-2024 Diality Inc. - All Rights Reserved.
 *
 * THIS CODE MAY NOT BE COPIED OR REPRODUCED IN ANY FORM, IN PART OR IN
 * WHOLE, WITHOUT THE EXPLICIT PERMISSION OF THE COPYRIGHT OWNER.
 *
 * @file    ModeDrain.h
 *
-* @author  (last)      Dara Navaei
-* @date    (last)      24-Oct-2021
+* @author  (last)      Michael Garthwaite
+* @date    (last)      09-Aug-2023
 *
 * @author  (original)  Leonardo Baloa
 * @date    (original)  20-Dec-2019
@@ -31,6 +31,15 @@
 
 // ********** public definitions **********
 
+/// Drain mode data structure
+
+typedef struct
+{
+    U32 drainModeState;         ///< Drain Mode state.
+    U32 badFillSignal;          ///< Gen Idle bad fill signal.
+    U32 badFillState;           ///< Gen Idle bad fill substate.
+} DG_DRAIN_MODE_DATA_T;
+
 // ********** public function prototypes **********
 
 void initDrainMode( void );         // initialize this module
@@ -40,6 +49,9 @@
 DG_DRAIN_STATE_T getCurrentDrainState( void );  // get the current state of the drain mode.
 void signalDrainModeRinseConcentrateLines( BOOL rinse );    // Signal drain mode to rinse concentrate lines.
 
+BOOL testSetDrainModeDataPublishIntervalOverride( U32 value );
+BOOL testResetDrainModeDataPublishIntervalOverride( void );
+
 /**@}*/
 
 #endif