def test_restore_from_cluster_snapshot_returns_dict_with_cluster_data(self):
     self._create_clusters()
     hook = RedshiftHook(aws_conn_id='aws_default')
     hook.create_cluster_snapshot('test_snapshot', 'test_cluster')
     self.assertEqual(
         hook.restore_from_cluster_snapshot('test_cluster_3', 'test_snapshot')['ClusterIdentifier'],
         'test_cluster_3',
     )
    def test_create_cluster_snapshot_returns_snapshot_data(self):
        self._create_clusters()
        hook = RedshiftHook(aws_conn_id='aws_default')

        snapshot = hook.create_cluster_snapshot('test_snapshot_2',
                                                'test_cluster')
        self.assertNotEqual(snapshot, None)