def test_is_beta_user(self): source = Source(id='x') self.assertFalse(source.is_beta_user()) self.mox.stubs.Set(util, 'BETA_USER_PATHS', set()) self.assertFalse(source.is_beta_user()) self.mox.stubs.Set(util, 'BETA_USER_PATHS', set([source.bridgy_path()])) self.assertTrue(source.is_beta_user())