[Extensions] - Refactoring for the testing needs

Signed-off-by: Oleg Nenashev <nenashev@synopsys.com>