Example #1
0
    def test_hb_transform(self):
        record = OrderedDict([('id', '1'), ('first_name', 'Maria'),
                              ('last_name', 'Russell'),
                              ('email', '*****@*****.**'),
                              ('gender', 'Female'),
                              ('ip_address', '141.48.134.32'),
                              ('dob', '5/26/1976')])
        expected = OrderedDict([('gender', 'female'),
                                ('accountid', '*****@*****.**'),
                                ('country', 'Germany'), ('age', 42),
                                ('game', 'hb'), ('extract_date', '2019-03-01'),
                                ('load_date', '2019-03-07')])

        extract_date = datetime.strptime('2019-03-01',
                                         transform.EXTRACT_DATE_FORMAT).date()
        load_date = datetime.strptime('2019-03-07',
                                      transform.EXTRACT_DATE_FORMAT).date()
        result = Transform.hb_transform(record, extract_date, load_date)
        self.assertEqual(result, expected)