def test_missing_mailchimp_api_key_line(self):
mailchimp_api_key = ('%s-us18' % ('').join(['1'] * 32))
def mock_getpass(prompt): # pylint: disable=unused-argument
return mailchimp_api_key
getpass_swap = self.swap(getpass, 'getpass', mock_getpass)
temp_feconf_path = tempfile.NamedTemporaryFile().name
feconf_text = (
'REDISHOST = \'192.13.2.1\'\n'
'# When the site terms were last updated, in UTC.\n'
'REGISTRATION_PAGE_LAST_UPDATED_UTC = '
'datetime.datetime(2015, 10, 14, 2, 40, 0)\n'
'# Format of string for dashboard statistics logs.\n'
'# NOTE TO DEVELOPERS: This format should not be changed, '
'since it is used in\n'
'# the existing storage models for UserStatsModel.\n'
'DASHBOARD_STATS_DATETIME_STRING_FORMAT = \'%Y-%m-%d\'\n')
with python_utils.open_file(temp_feconf_path, 'w') as f:
f.write(feconf_text)
with getpass_swap, self.assertRaisesRegexp(
AssertionError, 'Missing mailchimp API key'):
update_configs.add_mailchimp_api_key(temp_feconf_path)
def test_invalid_mailchimp_api_key(self):
check_prompts = {
'Enter mailchimp api key from the release process doc.':
False,
'You have entered an invalid mailchimp api key: invalid, '
'please retry.':
False
}
expected_check_prompts = {
'Enter mailchimp api key from the release process doc.':
True,
'You have entered an invalid mailchimp api key: invalid, '
'please retry.':
True
}
mailchimp_api_key = ('%s-us18' % ('').join(['1'] * 32))
def mock_getpass(prompt):
check_prompts[prompt] = True
if 'invalid' in prompt:
return mailchimp_api_key
return 'invalid'
getpass_swap = self.swap(getpass, 'getpass', mock_getpass)
temp_feconf_path = tempfile.NamedTemporaryFile().name
feconf_text = (
'REDISHOST = \'192.13.2.1\'\n'
'MAILGUN_API_KEY = None\n'
'MAILCHIMP_API_KEY = None\n'
'# When the site terms were last updated, in UTC.\n'
'REGISTRATION_PAGE_LAST_UPDATED_UTC = '
'datetime.datetime(2015, 10, 14, 2, 40, 0)\n'
'# Format of string for dashboard statistics logs.\n'
'# NOTE TO DEVELOPERS: This format should not be changed, '
'since it is used in\n'
'# the existing storage models for UserStatsModel.\n'
'DASHBOARD_STATS_DATETIME_STRING_FORMAT = \'YY-mm-dd\'\n')
expected_feconf_text = (
'REDISHOST = \'192.13.2.1\'\n'
'MAILGUN_API_KEY = None\n'
'MAILCHIMP_API_KEY = \'%s\'\n'
'# When the site terms were last updated, in UTC.\n'
'REGISTRATION_PAGE_LAST_UPDATED_UTC = '
'datetime.datetime(2015, 10, 14, 2, 40, 0)\n'
'# Format of string for dashboard statistics logs.\n'
'# NOTE TO DEVELOPERS: This format should not be changed, '
'since it is used in\n'
'# the existing storage models for UserStatsModel.\n'
'DASHBOARD_STATS_DATETIME_STRING_FORMAT = \'YY-mm-dd\'\n' %
(mailchimp_api_key))
with python_utils.open_file(temp_feconf_path, 'w') as f:
f.write(feconf_text)
with getpass_swap:
update_configs.add_mailchimp_api_key(temp_feconf_path)
self.assertEqual(check_prompts, expected_check_prompts)
with python_utils.open_file(temp_feconf_path, 'r') as f:
self.assertEqual(f.read(), expected_feconf_text)
def test_addition_of_mailchimp_api_key(self):
mailchimp_api_key = ('%s-us18' % ('').join(['1'] * 32))
def mock_getpass(prompt): # pylint: disable=unused-argument
return mailchimp_api_key
getpass_swap = self.swap(getpass, 'getpass', mock_getpass)
temp_feconf_path = tempfile.NamedTemporaryFile().name
feconf_text = (
'REDISHOST = \'192.13.2.1\'\n'
'MAILGUN_API_KEY = None\n'
'MAILCHIMP_API_KEY = None\n'
'# When the site terms were last updated, in UTC.\n'
'REGISTRATION_PAGE_LAST_UPDATED_UTC = '
'datetime.datetime(2015, 10, 14, 2, 40, 0)\n'
'# Format of string for dashboard statistics logs.\n'
'# NOTE TO DEVELOPERS: This format should not be changed, '
'since it is used in\n'
'# the existing storage models for UserStatsModel.\n'
'DASHBOARD_STATS_DATETIME_STRING_FORMAT = \'YY-mm-dd\'\n')
expected_feconf_text = (
'REDISHOST = \'192.13.2.1\'\n'
'MAILGUN_API_KEY = None\n'
'MAILCHIMP_API_KEY = \'%s\'\n'
'# When the site terms were last updated, in UTC.\n'
'REGISTRATION_PAGE_LAST_UPDATED_UTC = '
'datetime.datetime(2015, 10, 14, 2, 40, 0)\n'
'# Format of string for dashboard statistics logs.\n'
'# NOTE TO DEVELOPERS: This format should not be changed, '
'since it is used in\n'
'# the existing storage models for UserStatsModel.\n'
'DASHBOARD_STATS_DATETIME_STRING_FORMAT = \'YY-mm-dd\'\n' %
(mailchimp_api_key))
with python_utils.open_file(temp_feconf_path, 'w') as f:
f.write(feconf_text)
with getpass_swap:
update_configs.add_mailchimp_api_key(temp_feconf_path)
with python_utils.open_file(temp_feconf_path, 'r') as f:
self.assertEqual(f.read(), expected_feconf_text)