def test_main_load_parser(self, read_file_mock): read_file_mock.return_value = '{"profilename": "prof_name"}' profile_parser.main(['load', '--file', 'prof.json']) self.api.add_profile.assert_called_with({'profilename': "prof_name"}) # invalid profile file self.api.add_profile.reset_mock() read_file_mock.return_value = 'not json input' self.assertRaises(SystemExit, profile_parser.main, ['load', '--file', 'prof.json'])
def test_main_load_parser(self, read_file_mock): read_file_mock.return_value = '{"profilename": "prof_name"}' profile_parser.main(['load', '--file', 'prof.json']) self.api.add_profile.assert_called_with('prof_name', {'profilename': "prof_name"}) # invalid profile file self.api.add_profile.reset_mock() read_file_mock.return_value = '{"not_profilename_field": null}' self.assertRaises(SystemExit, profile_parser.main, ['load', '--file', 'prof.json'])
def test_main_get_parser(self): profile_parser.main(['get', '--name', 'profile_name']) self.api.get_profile.assert_called_with('profile_name') profile_parser.main(['get', '--list']) self.api.get_profiles.assert_called_with(None) profile_parser.main(['get', '--list', '--archi', 'm3']) self.api.get_profiles.assert_called_with('m3')
def test_main_add_parser(self): # add simple add profile_parser.main(['addwsn430', '-n', 'profile_name', '-p', 'dc']) self.api.add_profile.assert_called_with( 'profile_name', profile.ProfileWSN430('profile_name', 'dc')) profile_parser.main(['addm3', '-n', 'profile_name', '-p', 'dc']) self.api.add_profile.assert_called_with( 'profile_name', profile.ProfileM3('profile_name', 'dc')) profile_parser.main(['adda8', '-n', 'profile_name', '-p', 'dc']) self.api.add_profile.assert_called_with( 'profile_name', profile.ProfileA8('profile_name', 'dc')) profile_parser.main(['addcustom', '-n', 'profile_name', '-p', 'dc']) self.api.add_profile.assert_called_with( 'profile_name', profile.ProfileCustom('profile_name', 'dc')) # invalid configuration 'power' without period and average self.assertRaises( SystemExit, profile_parser.main, ['addm3', '-n', 'profile_name', '-p', 'dc', '-power'])
def test_main_del_parser(self): profile_parser.main(['del', '--name', 'profile_name']) self.api.del_profile.assert_called_with('profile_name')
def test_main_get_parser(self): profile_parser.main(['get', '--name', 'profile_name']) self.api.get_profile.assert_called_with('profile_name') profile_parser.main(['get', '--list']) self.api.get_profiles.assert_called_with()