add end to end test using unittest