~unity-2d-team/unity-2d/Shell-MultiMonitor

Creating a keyspace and schema for each test (or even test run) is too slow -- it takes about 2s. Instead use a random archivename.

added added

removed removed

__metaclass__ = type

import datetime

import os

from tempfile import _RandomNameSequence

import time

import unittest

import fixtures

from pycassa.util import convert_uuid_to_time

from grackle.server.testing.cassandra import TemporaryDB

from grackle.server.model import (

CassandraConnection,

_parse_message,

class TestCassandra(fixtures.TestWithFixtures):

def test_add_message(self):

keyspace = self.useFixture(TemporaryDB()).keyspace

c = CassandraConnection(keyspace, ['localhost:9160'])

c = CassandraConnection(

os.environ['GRACKLE_TEST_KEYSPACE'], ['localhost:9160'])

archive = next(_RandomNameSequence())

# Write the message out to Cassandra, and read it back in.

key = c.add_message('foo', TEST_MESSAGE)

key = c.add_message(archive, TEST_MESSAGE)

cmsg = c.messages.get(key)

# The archive should contain a single message, a reference to

# our new key.

archive_messages = c.archive_messages.get('foo').items()

archive_messages = c.archive_messages.get(archive).items()

self.assertEqual(1, len(archive_messages))

self.assertEqual(key, archive_messages[0][1])