class TestRuoteWorkitem(TestMixinWorkitem, unittest.TestCase): """Tests for RuoteWorkitems.""" def setUp(self): self.wi = RuoteWorkitem('application/x-ruote-workitem') wi = RuoteWorkitem('application/x-ruote-workitem') wi.loads('{"fields": {"params": {"worker_type": "test"}}, "fei": {}}') self.loaded_wi = wi def test_repr(self): """Test RuoteWorkitem.__repr__().""" self.wi._worker_type = "worker_type" self.assertEqual(self.wi.__repr__(), "<RuoteWorkitem([worker_type='worker_type'])>") def test_loads(self): """Tests RuoteWorkitem.loads().""" self.wi.loads('{"fields": {"params": {"worker_type": "test"}}}') self.assertRaises(RuoteWorkitemError, self.wi.loads, "{}") def test_dumps(self): """Test RuoteWorkitem.dumps().""" TestMixinWorkitem.test_dumps(self) self.wi._body = {} self.assertEqual("{}", self.wi.dumps()) def test_fei(self): """Test RuoteWorkitem.fei.""" fei = self.loaded_wi.fei fei["test"] = 1 self.assertFalse(self.loaded_wi._body["fei"].has_key("test")) def test_fields(self): """Test RuoteWorkitem.fields.""" fields = self.loaded_wi.fields fields["test"] = 1 self.assertTrue(self.loaded_wi._body["fields"].has_key("test"))
def setUp(self): self.wi = RuoteWorkitem('application/x-ruote-workitem') wi = RuoteWorkitem('application/x-ruote-workitem') wi.loads('{"fields": {"params": {"worker_type": "test"}}, "fei": {}}') self.loaded_wi = wi