def test_verifies_md5_sum_and_raises_exception_if_fails(self, mock_md5): conn = boto.connect_s3() conn.create_bucket('mybucket') with self.assertRaises(Exception): self.assertEqual(conn.get_bucket('mybucket').get_key(self.sgdid + "." + self.file_extension).get_contents_as_string(), self.file_content) upload_to_s3(self.tmpfilepath, self.sgdid, self.file_extension, "s3_access_key", "s3_secret_key", "mybucket") mock_md5.assert_called_with(self.tmpfilepath)
def test_deletes_file_from_filesystem(self, mock_os_remove, mock_search): mock_search.return_value = MockQuery(self.fdbentity) conn = boto.connect_s3() conn.create_bucket('mybucket') upload_to_s3(self.tmpfilepath, self.sgdid, self.file_extension, "s3_access_key", "s3_secret_key", "mybucket") self.assertTrue(mock_os_remove.called)
def test_saves_file_in_s3(self, mock_search): mock_search.return_value = MockQuery(self.fdbentity) conn = boto.connect_s3() conn.create_bucket('mybucket') upload_to_s3(self.tmpfilepath, self.sgdid, self.file_extension, "s3_access_key", "s3_secret_key", "mybucket") self.assertEqual(conn.get_bucket('mybucket').get_key(self.sgdid + "." + self.file_extension).get_contents_as_string(), self.file_content)
def test_updates_fildbentity_in_database(self, mock_search, mock_session, mock_transaction): fdbentity = factory.FiledbentityFactory.build(s3_url = None, md5sum = None) mock_search.return_value = MockQuery(fdbentity) local_md5 = md5(self.tmpfilepath) conn = boto.connect_s3() conn.create_bucket('mybucket') upload_to_s3(self.tmpfilepath, self.sgdid, self.file_extension, "s3_access_key", "s3_secret_key", "mybucket") self.assertEqual(fdbentity.s3_url, "https://mybucket.s3.amazonaws.com/" + self.sgdid + "." + self.file_extension) self.assertEqual(fdbentity.md5sum, local_md5) self.assertTrue(mock_session.called) self.assertTrue(mock_transaction.called)