def test_get_data_except_exception(self, mock_api): mock_api.environment_get.side_effect = Exception('foo_error') data_table = tables.UpdateEnvironmentRow(self.mock_data_table) with self.assertRaisesRegexp(Exception, 'foo_error'): data_table.get_data(None, 'foo_environment_id')
def test_get_data_except_http_not_found(self, mock_api): mock_api.environment_get.side_effect = exc.HTTPNotFound data_table = tables.UpdateEnvironmentRow(self.mock_data_table) with self.assertRaisesRegexp(django_http.Http404, None): data_table.get_data(None, 'foo_environment_id')
def test_get_data(self, mock_api): mock_api.environment_get.side_effect = None mock_api.environment_get.return_value = 'test_environment' data_table = tables.UpdateEnvironmentRow(self.mock_data_table) environment = data_table.get_data(None, 'foo_environment_id') self.assertEqual('test_environment', environment) mock_api.environment_get.assert_called_once_with( None, 'foo_environment_id')
def test_update_environment_row(self): data_table = tables.UpdateEnvironmentRow(self.mock_data_table, self.mock_datum) self.assertEqual('foo_status', data_table.attrs['status'])