use flexmock for now

since we're not moving away from flexmock just yet, I updated my tests to use
flexmock to be consistent. cc @mtodd