def test_bulk_create_creates_schemata(self):
     schemata = ['first', 'second', 'third']
     Schema.objects.mass_create(*schemata)
     cursor = connection.cursor()
     for schema in schemata:
         activate_schema(schema)
         cursor.execute(SCHEMA_QUERY, [schema])
         data = cursor.fetchone()
         self.assertEqual((schema,), data)
     cursor.close()
 def test_bulk_create_creates_schemata(self):
     schemata = ["first", "second", "third"]
     Schema.objects.mass_create(*schemata)
     cursor = connection.cursor()
     for schema in schemata:
         activate_schema(schema)
         cursor.execute(SCHEMA_QUERY, [schema])
         data = cursor.fetchone()
         self.assertEqual((schema,), data)
     cursor.close()
    def test_activate_schema_function(self):
        self.assertRaises(TemplateSchemaActivation, activate_schema, '__template__')

        Schema.objects.mass_create('a', 'b')

        activate_schema('a')
        self.assertEqual('a', get_active_schema_name())

        activate_schema('b')
        self.assertEqual('b', get_active_schema_name())

        deactivate_schema()
        self.assertEqual(None, get_active_schema_name())
    def test_activate_schema_function(self):
        self.assertRaises(TemplateSchemaActivation, activate_schema, "__template__")

        Schema.objects.mass_create("a", "b")

        activate_schema("a")
        self.assertEqual("a", get_active_schema_name())

        activate_schema("b")
        self.assertEqual("b", get_active_schema_name())

        deactivate_schema()
        self.assertEqual(None, get_active_schema_name())
    def test_deleting_schema_removes_it_from_session(self):
        schema = Schema.objects.mass_create('a')[0]
        user = User.objects.create_user(**CREDENTIALS)
        user.schemata.add(schema)

        self.client.login(**CREDENTIALS)
        self.client.post('/__change_schema__/a/')
        self.assertTrue('a', self.client.session['schema'])

        schema.delete(drop=True)
        self.assertEqual(0, Schema.objects.count())

        with self.assertRaises(SchemaNotFound):
            activate_schema('a')

        self.client.get('/')
        self.assertFalse('schema' in self.client.session)
    def test_deleting_schema_removes_it_from_session(self):
        schema = Schema.objects.mass_create('a')[0]
        user = User.objects.create_user(**CREDENTIALS)
        user.schemata.add(schema)

        self.client.login(**CREDENTIALS)
        self.client.post('/__change_schema__/a/')
        self.assertTrue('a', self.client.session['schema'])

        schema.delete(drop=True)
        self.assertEqual(0, Schema.objects.count())

        with self.assertRaises(SchemaNotFound):
            activate_schema('a')

        self.client.get('/')
        self.assertFalse('schema' in self.client.session)
예제 #7
0
def activate_schema_view(request, schema):
    # Don't do this.
    activate_schema(schema)
    return HttpResponse(schema)
예제 #8
0
 def activate(self):
     activate_schema(self.schema)
예제 #9
0
def activate_schema_view(request, schema):
    # Don't do this.
    activate_schema(schema)
    return HttpResponse(schema)
예제 #10
0
 def activate(self):
     activate_schema(self.schema)
예제 #11
0
 def activate(self):
     if self.expired:
         raise DemoSchemaExpired()
     activate_schema(self.schema)
예제 #12
0
 def activate(self):
     if self.expired:
         raise DemoSchemaExpired()
     activate_schema(self.schema)