def test_load_source(self): conf = Configuration() test_file_type = 'json' test_contents = 'test contents' test_parsed = {'foo': 'bar'} conf._get_file_type = mock.Mock(return_value=test_file_type) conf._get_file_contents = mock.Mock(return_value=test_contents) conf._parse_contents = mock.Mock(return_value=(test_parsed, None)) conf.load_source(self.test_origin) self.assertEqual(len(conf.sources), 1) built_source = conf.sources[0] self.assertEqual(built_source['origin'], self.test_origin) self.assertEqual(built_source['location'], self.test_origin) self.assertEqual(built_source['type'], test_file_type) self.assertEqual(built_source['contents'], test_contents) self.assertEqual(built_source['data'], test_parsed)
def test_file_type(self): test_path = '/path/to/some/foo.bar.json' conf = Configuration() t = conf._get_file_type(test_path) self.assertEqual(t, 'json')