def test_rest_calls_value_for_inputfield(self): field_value = 3 inputfield = InputFieldF.build(type=InputField.TYPE_INTEGER) with mock.patch("flooding_lib.models.Scenario.value_for_inputfield", return_value=field_value) as patched: headers = ({}, {"inputfield": inputfield}) scenariorow = eie.ScenarioRow(ScenarioF.build(), headers) columns = scenariorow.columns() columns.next() # Skip scenario id self.assertEquals(columns.next().value, field_value) patched.assert_called_with(inputfield)
def test_rest_calls_value_for_inputfield(self): field_value = 3 inputfield = InputFieldF.build(type=InputField.TYPE_INTEGER) with mock.patch('flooding_lib.models.Scenario.value_for_inputfield', return_value=field_value) as patched: headers = ({}, {'inputfield': inputfield}) scenariorow = eie.ScenarioRow(ScenarioF.build(), headers) columns = scenariorow.columns() columns.next() # Skip scenario id self.assertEquals(columns.next().value, field_value) patched.assert_called_with(inputfield)
def test_scenarios(self): scenario = ScenarioF.build() fieldinfo = eie.FieldInfo([scenario]) for s in fieldinfo.scenarios(): self.assertTrue(s is scenario)