def _default(self): args = self.app.pargs GDocDown(credentials=self.app.credentials).download( args.google_file, format=args.format, out_path=args.out_path, extension=args.extension)
def test_api_txt(self): GDocDown(credentials=self.credentials).download(self.GDOC_FILE, format='txt', out_path=os.path.join(self.out_dir, 'example-out.text')) # check that file downloaded self.assertTrue(os.path.isfile(os.path.join(self.out_dir, 'example-out.text'))) # check that file has correct content with open(os.path.join(self.out_dir, 'example-out.text'), 'r') as file: self.assertRegex(file.read().strip(), 'gdoc_down example file')
def test_cli_2odt(self): with cli(argv=['-f', 'odt', '-o', self.out_dir, self.FIXTURE_FILE], credentials=self.credentials) as app: app.run() # check that file downloaded self.assertTrue(os.path.isfile(os.path.join(self.out_dir, 'example.odt'))) # check that file has correct content doc = opendocument.load(os.path.join(self.out_dir, 'example.odt')) root = ElementTree.fromstring(doc.toXml().encode('utf-8')) self.assertRegexpMatches(GDocDown.get_element_text(root), 'gdoc_down example file')
def test_cli_gdoc_2odt(self): with cli(argv=['-f', 'odt', '-o', self.out_dir, self.GDOC_FILE], credentials=self.credentials) as app: app.run() # check that file downloaded self.assertTrue(os.path.isfile(os.path.join(self.out_dir, 'example.odt'))) # check that file has correct content doc = opendocument.load(os.path.join(self.out_dir, 'example.odt')) root = ElementTree.fromstring(doc.toXml().encode('utf-8')) self.assertRegex(GDocDown.get_element_text(root), 'gdoc_down example file')
def test_cli_2rtf(self): with cli(argv=['-f', 'rtf', '-o', self.out_dir, self.FIXTURE_FILE], credentials=self.credentials) as app: app.run() # check that file downloaded self.assertTrue(os.path.isfile(os.path.join(self.out_dir, 'example.rtf'))) # check that file has correct content if sys.version_info < (3, 0, 0): rtf2xml.ParseRtf.ParseRtf( in_file=os.path.join(self.out_dir, 'example.rtf'), out_file=os.path.join(self.out_dir, 'example.xml'), ).parse_rtf() root = ElementTree.parse(os.path.join(self.out_dir, 'example.xml')) self.assertRegexpMatches(GDocDown.get_element_text(root.getroot()), 'gdoc_down example file')
def test_cli_gdoc_2rtf(self): with cli(argv=['-f', 'rtf', '-o', self.out_dir, self.GDOC_FILE], credentials=self.credentials) as app: app.run() # check that file downloaded self.assertTrue(os.path.isfile(os.path.join(self.out_dir, 'example.rtf'))) # check that file has correct content if sys.version_info < (3, 0, 0): rtf2xml.ParseRtf.ParseRtf( in_file=os.path.join(self.out_dir, 'example.rtf'), out_file=os.path.join(self.out_dir, 'example.xml'), ).parse_rtf() root = ElementTree.parse(os.path.join(self.out_dir, 'example.xml')) self.assertRegex(GDocDown.get_element_text(root.getroot()), 'gdoc_down example file')
def test_get_gdoc_id(self): self.assertEqual( GDocDown.get_gdoc_id(self.FIXTURE_FILE), '1mgPojZVReTAMBIVvt6LSQ59AGTsxx2-myLR9oIYIJ2s')
def test_get_credentials(self): with mock.patch.object(oauth2client.file.Storage, 'get', return_value=None): with mock.patch.object(oauth2client.tools, 'run_flow', return_value=self.credentials): gdoc_down = GDocDown() self.assertEqual(gdoc_down.credentials, self.credentials)
def test_get_google_id(self): self.assertEqual(GDocDown.get_google_id(self.GDOC_FILE), '1mgPojZVReTAMBIVvt6LSQ59AGTsxx2-myLR9oIYIJ2s')