Add Behavior unit test