Update test class for SDK api integration