Add application logic tests