Follow framework API additions for mock classes

Change-Id: Ia24f9efd23c7d75ae34e86bd164f37244945284e
1 file changed