Adding easy-mock usage for square-test.