Added integration test for main

Signed-off-by: Phan Le <ple@pivotallabs.com>