from __future__ import unicode_literals
from mogwai._compat import string_types, PY2
from nose.plugins.attrib import attr
from mogwai.tests import BaseMogwaiTestCase
from mogwai.models import Vertex, Edge
from mogwai import properties
from mogwai.tools import ImportStringError, import_string, cached_property, LazyImportClass, _Missing, Factory
import factory
from nose.tools import raises, nottest
from mogwai import connection
from mogwai.exceptions import MogwaiQueryError
from rexpro.exceptions import RexProScriptException
from mogwai.tools import BlueprintsWrapper, PartitionGraph
import eventlet


@attr('unit', 'tools')
class TestImportString(BaseMogwaiTestCase):

    def test_raises_import_string_error(self):
        with self.assertRaises(ImportStringError):
            import_string('does.not.exist')
        with self.assertRaises(ImportStringError):
            import_string('does.not:exist')

    def test_imports_correctly(self):
        cp = import_string('mogwai.tools:cached_property')
        self.assertIs(cp, cached_property)

    def test_import_string_error_class(self):
        ise = ImportStringError('mogwai.tools.cached_property', 'Should work')


class LazyImportObject(object):
    pass


@attr('unit', 'tools')
class TestLazyImportString(BaseMogwaiTestCase):

    def test_lazy_import(self):
        li = LazyImportClass('mogwai.tools._Missing')
        self.assertIsInstance(li, LazyImportClass)
        self.assertIs(li.klass, _Missing)
        self.assertIsInstance(li(), _Missing)


class TestFactoryVertex(Vertex):
    element_type = 'test_factory_vertex'

    name = properties.String(required=True, max_length=128)


class TestFactoryVertexFactory(Factory):
    FACTORY_FOR = TestFactoryVertex

    name = factory.Sequence(lambda n: 'Test Factory Vertex {}'.format(n))


class TestFactoryEdge(Edge):
    label = 'test_factory_edge'

    name = properties.String(required=True, max_length=128)


class TestFactoryEdgeFactory(Factory):
    FACTORY_FOR = TestFactoryEdge

    name = factory.Sequence(lambda n: 'Test Factory Edge {}'.format(n))


@attr('unit', 'tools', 'factory')
class TestModelFactory(BaseMogwaiTestCase):

    def test_vertex_factory(self):
        v1 = TestFactoryVertexFactory()
        self.assertIsInstance(v1, TestFactoryVertex)
        self.assertTrue(v1.name.startswith('Test Factory Vertex'))
        v1.delete()
        v2 = TestFactoryVertexFactory.create()
        self.assertIsInstance(v2, TestFactoryVertex)
        self.assertTrue(v2.name.startswith('Test Factory Vertex'))
        v2.delete()

    def test_edge_factory(self):
        v1 = TestFactoryVertexFactory()
        v2 = TestFactoryVertexFactory()
        e1 = TestFactoryEdgeFactory(outV=v1, inV=v2)
        e2 = TestFactoryEdgeFactory.create(outV=v1, inV=v2)
        self.assertIsInstance(e1, TestFactoryEdge)
        self.assertTrue(e1.name.startswith('Test Factory Edge'))
        self.assertIsInstance(e2, TestFactoryEdge)
        self.assertTrue(e2.name.startswith('Test Factory Edge'))
        e1.delete()
        e2.delete()
        v1.delete()
        v2.delete()


class BlueprintsWrapperVertex(Vertex):
    element_type = 'blueprints_wrapper_vertex'
    name = properties.String(required=True, max_length=128)


class BlueprintsWrapperEdge(Edge):
    element_type = 'blueprints_wrapper_edge'
    name = properties.String(required=True, max_length=128)


@attr('unit', 'tools', 'blueprintswrapper')
class TestBlueprintsWrapper(BaseMogwaiTestCase):

    def test_blueprints_wrapper(self):
        k = 10
        wrapper_config = {
            'class_name': "ReadOnlyGraph",
            'bindings': {'k': k}
        }
        with BlueprintsWrapper(**wrapper_config):
            gsk = connection.execute_query('"powers of ${k}"')
            pysk = "powers of {}".format(k)
            self.assertEqual(gsk, pysk)
            kk = connection.execute_query("k * k")
            self.assertEqual(kk, k * k)

    @raises(RexProScriptException, MogwaiQueryError)
    def test_wrapper_isolation(self):
        connection.execute_query("k")

    def test_partition_graph(self):
        with PartitionGraph(write='a'):
            v1 = BlueprintsWrapperVertex.create(name="only in a")
            v3 = BlueprintsWrapperVertex.create(name="started in a")

        with PartitionGraph(write='b', read=['a']):
            v2 = BlueprintsWrapperVertex.create(name="only in b")
            e1 = BlueprintsWrapperEdge.create(outV=v2, inV=v1, name="only in b")
            v3.name = "still in a"
            v3.save()

        with PartitionGraph(write='a'):
            v1 = BlueprintsWrapperVertex.get(v1._id)
            self.assertEqual(len(v1.bothE()), 0)
            with self.assertRaises(BlueprintsWrapperVertex.DoesNotExist):
                v2 = BlueprintsWrapperVertex.get(v2._id)

        self.assertEqual(len(v1.bothE()), 1)
        self.assertIsInstance(v1.bothE()[0], BlueprintsWrapperEdge)
        self.assertEqual(v1.bothE()[0]._id, e1._id)
        v3 = BlueprintsWrapperVertex.get(v3._id)
        self.assertEqual(v3['_partition'], 'a')
        e1.delete()
        v1.delete()
        v2.delete()
        v3.delete()


@nottest
def isolation_query(scope):
    wrapper_config = {
        'class_name': "ReadOnlyGraph",
        'bindings': {'scope': scope},
        'pool_size': 5
    }
    scope_values = []
    with BlueprintsWrapper(**wrapper_config) as pool:
        for i in range(7):
            scope_val = connection.execute_query(
                "sleep sleep_length\nreturn scope",
                params={'sleep_length': 100 * (i % 2 + 1) - scope*10},
                pool=pool
            )
            scope_values.append(scope_val)

    return scope, scope_values

@nottest
def nested_wrappers(scope):
    partition_config = {
        'write': "characters",
        'bindings': {'scope': scope},
        'pool_size': 3
    }
    with PartitionGraph(**partition_config) as pool:
        pile = eventlet.GreenPile()
        [pile.spawn(isolation_query, i) for i in range(3)]
        scope_val = connection.execute_query('scope', pool=pool)
        scope_v = BlueprintsWrapperVertex.create(name=scope_val, pool=pool)
        scope_v['nested_values'] = list(pile)
        scope_v.save(pool=pool)

    return scope, scope_v

@attr('unit', 'tools', 'blueprintswrapper', 'concurrency')
class TestWrapperConcurrency(BaseMogwaiTestCase):

    @classmethod
    def setUpClass(cls):
        super(TestWrapperConcurrency, cls).setUpClass()
        host_params = connection.HOST_PARAMS.copy()
        host_params.pop('port')
        connection.setup(concurrency='eventlet', **host_params)

    @classmethod
    def tearDownClass(cls):
        host_params = connection.HOST_PARAMS.copy()
        host_params.pop('port')
        connection.setup(concurrency='sync', **host_params)
        super(TestWrapperConcurrency, cls).tearDownClass()

    def test_wrapper_isolation(self):
        pile = eventlet.GreenPile()
        [pile.spawn(isolation_query, i) for i in range(10)]

        for scope, scope_values in pile:
            for val in scope_values:
                self.assertEqual(scope, val)

    def test_nested_values(self):
        pile = eventlet.GreenPile()
        [pile.spawn(nested_wrappers, chr(i)) for i in range(97, 103)]

        for scope, scope_v in pile:
            self.assertEqual(scope, scope_v.name)
            for inner_scope, scope_values in scope_v['nested_values']:
                for val in scope_values:
                    self.assertEqual(inner_scope, val)
            scope_v.delete()

from __future__ import unicode_literals
from mogwai._compat import string_types, PY2
from nose.plugins.attrib import attr
from mogwai.tests import BaseMogwaiTestCase
from mogwai.models import Vertex, Edge
from mogwai import properties
from mogwai.tools import ImportStringError, import_string, cached_property, LazyImportClass, _Missing, Factory
import factory
from nose.tools import raises, nottest
from mogwai import connection
from mogwai.exceptions import MogwaiQueryError
from rexpro.exceptions import RexProScriptException
from mogwai.tools import BlueprintsWrapper, PartitionGraph
import eventlet


@attr('unit', 'tools')
class TestImportString(BaseMogwaiTestCase):

    def test_raises_import_string_error(self):
        with self.assertRaises(ImportStringError):
            import_string('does.not.exist')
        with self.assertRaises(ImportStringError):
            import_string('does.not:exist')

    def test_imports_correctly(self):
        cp = import_string('mogwai.tools:cached_property')
        self.assertIs(cp, cached_property)

    def test_import_string_error_class(self):
        ise = ImportStringError('mogwai.tools.cached_property', 'Should work')


class LazyImportObject(object):
    pass


@attr('unit', 'tools')
class TestLazyImportString(BaseMogwaiTestCase):

    def test_lazy_import(self):
        li = LazyImportClass('mogwai.tools._Missing')
        self.assertIsInstance(li, LazyImportClass)
        self.assertIs(li.klass, _Missing)
        self.assertIsInstance(li(), _Missing)


class TestFactoryVertex(Vertex):
    element_type = 'test_factory_vertex'

    name = properties.String(required=True, max_length=128)


class TestFactoryVertexFactory(Factory):
    FACTORY_FOR = TestFactoryVertex

    name = factory.Sequence(lambda n: 'Test Factory Vertex {}'.format(n))


class TestFactoryEdge(Edge):
    label = 'test_factory_edge'

    name = properties.String(required=True, max_length=128)


class TestFactoryEdgeFactory(Factory):
    FACTORY_FOR = TestFactoryEdge

    name = factory.Sequence(lambda n: 'Test Factory Edge {}'.format(n))


@attr('unit', 'tools', 'factory')
class TestModelFactory(BaseMogwaiTestCase):

    def test_vertex_factory(self):
        v1 = TestFactoryVertexFactory()
        self.assertIsInstance(v1, TestFactoryVertex)
        self.assertTrue(v1.name.startswith('Test Factory Vertex'))
        v1.delete()
        v2 = TestFactoryVertexFactory.create()
        self.assertIsInstance(v2, TestFactoryVertex)
        self.assertTrue(v2.name.startswith('Test Factory Vertex'))
        v2.delete()

    def test_edge_factory(self):
        v1 = TestFactoryVertexFactory()
        v2 = TestFactoryVertexFactory()
        e1 = TestFactoryEdgeFactory(outV=v1, inV=v2)
        e2 = TestFactoryEdgeFactory.create(outV=v1, inV=v2)
        self.assertIsInstance(e1, TestFactoryEdge)
        self.assertTrue(e1.name.startswith('Test Factory Edge'))
        self.assertIsInstance(e2, TestFactoryEdge)
        self.assertTrue(e2.name.startswith('Test Factory Edge'))
        e1.delete()
        e2.delete()
        v1.delete()
        v2.delete()


class BlueprintsWrapperVertex(Vertex):
    element_type = 'blueprints_wrapper_vertex'
    name = properties.String(required=True, max_length=128)


class BlueprintsWrapperEdge(Edge):
    element_type = 'blueprints_wrapper_edge'
    name = properties.String(required=True, max_length=128)


@attr('unit', 'tools', 'blueprintswrapper')
class TestBlueprintsWrapper(BaseMogwaiTestCase):

    def test_blueprints_wrapper(self):
        k = 10
        wrapper_config = {
            'class_name': "ReadOnlyGraph",
            'bindings': {'k': k}
        }
        with BlueprintsWrapper(**wrapper_config):
            gsk = connection.execute_query('"powers of ${k}"')
            pysk = "powers of {}".format(k)
            self.assertEqual(gsk, pysk)
            kk = connection.execute_query("k * k")
            self.assertEqual(kk, k * k)

    @raises(RexProScriptException, MogwaiQueryError)
    def test_wrapper_isolation(self):
        connection.execute_query("k")

    def test_partition_graph(self):
        with PartitionGraph(write='a'):
            v1 = BlueprintsWrapperVertex.create(name="only in a")
            v3 = BlueprintsWrapperVertex.create(name="started in a")

        with PartitionGraph(write='b', read=['a']):
            v2 = BlueprintsWrapperVertex.create(name="only in b")
            e1 = BlueprintsWrapperEdge.create(outV=v2, inV=v1, name="only in b")
            v3.name = "still in a"
            v3.save()

        with PartitionGraph(write='a'):
            v1 = BlueprintsWrapperVertex.get(v1._id)
            self.assertEqual(len(v1.bothE()), 0)
            with self.assertRaises(BlueprintsWrapperVertex.DoesNotExist):
                v2 = BlueprintsWrapperVertex.get(v2._id)

        self.assertEqual(len(v1.bothE()), 1)
        self.assertIsInstance(v1.bothE()[0], BlueprintsWrapperEdge)
        self.assertEqual(v1.bothE()[0]._id, e1._id)
        v3 = BlueprintsWrapperVertex.get(v3._id)
        self.assertEqual(v3['_partition'], 'a')
        e1.delete()
        v1.delete()
        v2.delete()
        v3.delete()


@nottest
def isolation_query(scope):
    wrapper_config = {
        'class_name': "ReadOnlyGraph",
        'bindings': {'scope': scope},
        'pool_size': 5
    }
    scope_values = []
    with BlueprintsWrapper(**wrapper_config) as pool:
        for i in range(7):
            scope_val = connection.execute_query(
                "sleep sleep_length\nreturn scope",
                params={'sleep_length': 100 * (i % 2 + 1) - scope*10},
                pool=pool
            )
            scope_values.append(scope_val)

    return scope, scope_values

@nottest
def nested_wrappers(scope):
    partition_config = {
        'write': "characters",
        'bindings': {'scope': scope},
        'pool_size': 3
    }
    with PartitionGraph(**partition_config) as pool:
        pile = eventlet.GreenPile()
        [pile.spawn(isolation_query, i) for i in range(3)]
        scope_val = connection.execute_query('scope', pool=pool)
        scope_v = BlueprintsWrapperVertex.create(name=scope_val, pool=pool)
        scope_v['nested_values'] = list(pile)
        scope_v.save(pool=pool)

    return scope, scope_v

@attr('unit', 'tools', 'blueprintswrapper', 'concurrency')
class TestWrapperConcurrency(BaseMogwaiTestCase):

    @classmethod
    def setUpClass(cls):
        super(TestWrapperConcurrency, cls).setUpClass()
        host_params = connection.HOST_PARAMS.copy()
        host_params.pop('port')
        connection.setup(concurrency='eventlet', **host_params)

    @classmethod
    def tearDownClass(cls):
        host_params = connection.HOST_PARAMS.copy()
        host_params.pop('port')
        connection.setup(concurrency='sync', **host_params)
        super(TestWrapperConcurrency, cls).tearDownClass()

    def test_wrapper_isolation(self):
        pile = eventlet.GreenPile()
        [pile.spawn(isolation_query, i) for i in range(10)]

        for scope, scope_values in pile:
            for val in scope_values:
                self.assertEqual(scope, val)

    def test_nested_values(self):
        pile = eventlet.GreenPile()
        [pile.spawn(nested_wrappers, chr(i)) for i in range(97, 103)]

        for scope, scope_v in pile:
            self.assertEqual(scope, scope_v.name)
            for inner_scope, scope_values in scope_v['nested_values']:
                for val in scope_values:
                    self.assertEqual(inner_scope, val)
            scope_v.delete()
