def decode(self, data):
return AgreementRecord(record_type=data['record_type'],
transaction_sequence_n=data[
'transaction_sequence_n'],
record_sequence_n=data['record_sequence_n'],
submitter_agreement_n=data[
'submitter_agreement_n'],
agreement_type=data['agreement_type'],
agreement_start_date=data[
'agreement_start_date'],
prior_royalty_status=data[
'prior_royalty_status'],
post_term_collection_status=data[
'post_term_collection_status'],
number_of_works=data['number_of_works'],
society_assigned_agreement_n=data[
'society_assigned_agreement_n'],
international_standard_code=data[
'international_standard_code'],
sales_manufacture_clause=data[
'sales_manufacture_clause'],
agreement_end_date=data['agreement_end_date'],
date_of_signature=data['date_of_signature'],
retention_end_date=data['retention_end_date'],
prior_royalty_start_date=data[
'prior_royalty_start_date'],
post_term_collection_end_date=data[
'post_term_collection_end_date'],
shares_change=data['shares_change'],
advance_given=data['advance_given'])
def _get_agreement(self):
return AgreementRecord(
record_type='ACK',
transaction_sequence_n=3,
record_sequence_n=15,
submitter_agreement_n='AB12',
agreement_type='OS',
agreement_start_date=datetime.datetime.strptime(
'20030215', '%Y%m%d').date(),
number_of_works=12,
prior_royalty_status='D',
post_term_collection_status='D',
international_standard_code='DFG135',
society_assigned_agreement_n='DF35',
sales_manufacture_clause='M',
agreement_end_date=datetime.datetime.strptime(
'20030216', '%Y%m%d').date(),
date_of_signature=datetime.datetime.strptime('20030217',
'%Y%m%d').date(),
retention_end_date=datetime.datetime.strptime(
'20030218', '%Y%m%d').date(),
prior_royalty_start_date=datetime.datetime.strptime(
'20030219', '%Y%m%d').date(),
post_term_collection_end_date=datetime.datetime.strptime(
'20030220', '%Y%m%d').date(),
shares_change=True,
advance_given=True)
def test_encoded(self):
data = AgreementRecord(record_type='ACK',
transaction_sequence_n=3,
record_sequence_n=15,
submitter_agreement_n='AB12',
agreement_type='OS',
agreement_start_date=datetime.datetime.strptime(
'20030215', '%Y%m%d').date(),
number_of_works=12,
prior_royalty_status='D',
post_term_collection_status='D',
international_standard_code='DFG135',
society_assigned_agreement_n='DF35',
sales_manufacture_clause='M',
agreement_end_date=datetime.datetime.strptime(
'20030216', '%Y%m%d').date(),
date_of_signature=datetime.datetime.strptime(
'20030217', '%Y%m%d').date(),
retention_end_date=datetime.datetime.strptime(
'20030218', '%Y%m%d').date(),
prior_royalty_start_date=datetime.datetime.strptime(
'20030219', '%Y%m%d').date(),
post_term_collection_end_date=datetime.datetime.strptime(
'20030220', '%Y%m%d').date(),
shares_change=True,
advance_given=True)
encoded = self._encoder.encode(data)
self.assertEqual('ACK', encoded['record_type'])
self.assertEqual(3, encoded['transaction_sequence_n'])
self.assertEqual(15, encoded['record_sequence_n'])
self.assertEqual('AB12', encoded['submitter_agreement_n'])
self.assertEqual('OS', encoded['agreement_type'])
self.assertEqual(
datetime.datetime.strptime('20030215', '%Y%m%d').date(),
encoded['agreement_start_date'])
self.assertEqual(12, encoded['number_of_works'])
self.assertEqual('D', encoded['prior_royalty_status'])
self.assertEqual('D', encoded['post_term_collection_status'])
self.assertEqual('DFG135', encoded['international_standard_code'])
self.assertEqual('DF35', encoded['society_assigned_agreement_n'])
self.assertEqual('M', encoded['sales_manufacture_clause'])
self.assertEqual(
datetime.datetime.strptime('20030216', '%Y%m%d').date(),
encoded['agreement_end_date'])
self.assertEqual(
datetime.datetime.strptime('20030217', '%Y%m%d').date(),
encoded['date_of_signature'])
self.assertEqual(
datetime.datetime.strptime('20030218', '%Y%m%d').date(),
encoded['retention_end_date'])
self.assertEqual(
datetime.datetime.strptime('20030219', '%Y%m%d').date(),
encoded['prior_royalty_start_date'])
self.assertEqual(
datetime.datetime.strptime('20030220', '%Y%m%d').date(),
encoded['post_term_collection_end_date'])
self.assertEqual(True, encoded['shares_change'])
self.assertEqual(True, encoded['advance_given'])