def setUp(self):
self.uow = MagicMock()
self.repo = SavedReportConfigurationRepository(self.uow)
class TestSavedReportConfigurationRepository(unittest.TestCase):
def setUp(self):
self.uow = MagicMock()
self.repo = SavedReportConfigurationRepository(self.uow)
def test_get_configurations_for_user(self):
user_id = Mock()
table_mock = self.uow.tables.saved_report_configurations
rv = self.repo.get_configurations_for_user(user_id)
table_mock.get_all.assert_called_with(user_id, index='user_id')
self.uow.run_list.assert_called_with(table_mock.get_all.return_value)
self.assertEqual(rv, self.uow.run_list.return_value)
def test_insert_configuration(self):
configuration = {
'report_type': Mock(),
'user_id': Mock(),
'configuration': Mock(),
'name': Mock()
}
table_mock = self.uow.tables.saved_report_configurations
rv = self.repo.insert_configuration(configuration)
table_mock.insert.assert_called_with(configuration)
self.uow.run.assert_called_with(table_mock.insert.return_value)
def test_get_configuration_by_name(self):
user_id = Mock()
name = Mock()
table_mock = self.uow.tables.saved_report_configurations
rv = self.repo.get_configuration_by_name(user_id, name)
# table.get_all()
table_mock.get_all.assert_called_with(user_id, index='user_id')
return_value = table_mock.get_all.return_value
#table.get_all().filter()
return_value.filter.assert_called_with({'name': name})
return_value = return_value.filter.return_value
self.uow.run_list.assert_called_with(return_value)
self.assertEqual(rv, self.uow.run_list.return_value)
def test_get_configuration_by_id(self):
config_id = Mock()
table_mock = self.uow.tables.saved_report_configurations
rv = self.repo.get_configuration_by_id(config_id)
table_mock.get.assert_called_with(config_id)
self.uow.run.assert_called_with(table_mock.get.return_value)
self.assertEqual(rv, self.uow.run.return_value)
def test_remove_configuration(self):
config_id = Mock()
table_mock = self.uow.tables.saved_report_configurations
self.repo.delete_configuration(config_id)
# table.get()
table_mock.get.assert_called_with(config_id)
return_value = table_mock.get.return_value
#table.get().delete()
return_value.delete.assert_called_with()
return_value = return_value.delete.return_value
self.uow.run.assert_called_with(return_value)
def test_get_by_type(self):
config_type = Mock()
user_id = Mock()
table_mock = self.uow.tables.saved_report_configurations
rv = self.repo.get_configuration_by_type(user_id, config_type)
# table.get_all()
table_mock.get_all.assert_called_with(user_id, index='user_id')
return_value = table_mock.get_all.return_value
# table.get_all().filter()
return_value.filter.assert_called_with({'report_type':config_type})
return_value = return_value.filter.return_value
self.uow.run_list.assert_called_with(return_value)
self.assertEqual(rv, self.uow.run_list.return_value)
