def test_diff_no_templates(self): """ Test diff with no template """ api = self.API() file_contents = {"gup.py": True} self.mock(api.diff_model, "get_by_id").using(staticmethod(lambda keyId: False)) key = ndb.Key(models.User, 1) obj = self.obj().set(get_messages=lambda: {"file_contents": file_contents}, key=key) obj.assignment = self._assign obj.assignment.get = lambda: self.obj().set(templates=None) templates = obj.assignment.get().templates self.assertFalse(api.diff_model.get_by_id(obj.key.id())) self.assertTrue(not templates or templates == {}) with self.assertRaises(BadValueError): api.diff(obj, self.accounts["dummy_admin"], {})
def test_diff_no_templates(self): """ Test diff with no template """ api = self.API() file_contents = {'gup.py': True} self.mock(api.diff_model, 'get_by_id').using(staticmethod(lambda keyId: False)) key = ndb.Key(models.User, 1) obj = self.obj().set( get_messages=lambda: {'file_contents': file_contents}, key=key) obj.assignment = self._assign obj.assignment.get = lambda: self.obj().set(templates=None) templates = obj.assignment.get().templates self.assertFalse(api.diff_model.get_by_id(obj.key.id())) self.assertTrue(not templates or templates == {}) with self.assertRaises(BadValueError): api.diff(obj, self.accounts['dummy_admin'], {})
def test_diff_obj(self): """ Tests that existing diff is just returned """ api, fake = self.API(), True file_contents = {"gup.py": True} self.mock(api.diff_model, "get_by_id").using(staticmethod(lambda keyId: fake)) key = ndb.Key(models.User, 1) obj = self.obj().set( get_messages=lambda: {"file_contents": file_contents}, key=key, file_contents=file_contents ) diff = api.diff(obj, self.accounts["dummy_admin"], {}) self.assertEqual(fake, diff) return obj
def test_diff_obj(self): """ Tests that existing diff is just returned """ api, fake = self.API(), True file_contents = {'gup.py': True} self.mock(api.diff_model, 'get_by_id').using(staticmethod(lambda keyId: fake)) key = ndb.Key(models.User, 1) obj = self.obj().set( get_messages=lambda: {'file_contents': file_contents}, key=key, file_contents=file_contents) diff = api.diff(obj, self.accounts['dummy_admin'], {}) self.assertEqual(fake, diff) return obj