class TestRowUpdate(TestCase):
def setUp(self):
self.subject = RowUpdate(MagicMock)
@patch.object(RowUpdate, 'filtered_rows')
@patch.object(RowUpdate, 'session')
def test_call_updates_using_executor(self, _, filtered_rows):
executor = MagicMock()
self.subject(executor)
executor.return_value.update.assert_called_with(filtered_rows())
@patch('gazette.data.row_update.create_tables')
@patch('gazette.data.row_update.db_connect')
@patch('gazette.data.row_update.sessionmaker')
@patch('gazette.data.row_update.text')
def test_call_commits_executor_changes(self, _text, sessionmaker, _db, _create):
executor = MagicMock()
self.subject(executor)
sessionmaker.return_value.return_value.commit.assert_called_with()
@patch('gazette.data.row_update.create_tables')
@patch('gazette.data.row_update.db_connect')
@patch('gazette.data.row_update.sessionmaker')
@patch('gazette.data.row_update.text')
def test_session_creates_tables(self, _text, _maker, _db, create_tables):
self.subject.session()
create_tables.assert_called()
@patch('gazette.data.row_update.text')
@patch.object(RowUpdate, 'session')
def test_filtered_rows_queries_following_condition(self, session, text):
condition = text('column = 42')
self.subject.filtered_rows(condition)
session.return_value.query.return_value.filter.assert_called_with(condition)
def parse_bidding_exemptions():
row_update = RowUpdate(BiddingExemption)
row_update(BiddingExemptionParsing)
def parse_sections():
row_update = RowUpdate(Gazette)
row_update(SectionParsing)
parse_bidding_exemptions.delay()
def setup_method(self, method):
self.subject = RowUpdate(MagicMock)
def setUp(self):
self.subject = RowUpdate(MagicMock)