[task] updated tests to extension builder