def test_get_context(self): c = CreatedTestContent() data = Mock(spec_set=[]) c.get_context_data = Mock(spec_set=[], return_value=data) ctx = c.get_context() self.assertEqual(type(ctx), Context) self.assertEqual(len(ctx.dicts), 2) self.assertEqual(ctx.dicts[1], data)
def test_get_context_request(self): c = CreatedTestContent() request = Mock(spec_set=[]) ctx = c.get_context(request=request) self.assertEqual(type(ctx), RequestContext)