def test_dict_params(self): change = AddUnitChange(1, [], params={"application": "application", "to": "to"}) self.assertEqual({"change_id": 1, "requires": [], "application": "application", "to": "to"}, change.__dict__)
async def test_run(self, event_loop): change = AddUnitChange(1, [], params={"application": "application", "to": "to"}) app = mock.Mock() app.add_unit = base.AsyncMock(return_value="unit1") model = MockModel({"application1": app}) context = mock.Mock() context.resolve = mock.Mock(side_effect=['application1', 'to1']) context._units_by_app = {} context.model = model result = await change.run(context) assert result == "unit1" model.applications["application1"].add_unit.assert_called_once() model.applications["application1"].add_unit.assert_called_with(count=1, to="to1")
def test_method(self): self.assertEqual("addUnit", AddUnitChange.method())