예제 #1
0
    def test_insert(self) -> None:
        table = CassandraTable(self.mock_config, self.keyspace)
        table.insert({
            'col1': 'mock_id',
            'col2': 1546304400000,
            'col3': 26.919388,
            'col4': -8.932613
        }, keyspace='mock_keyspace')

        self.assertEqual(1, len(table.statements))
        self.assertEqual(
            "INSERT INTO mock_keyspace.mock_table "
            "JSON '{\"col1\": \"mock_id\", \"col2\": 1546304400000, "
            "\"col3\": 26.919388, \"col4\": -8.932613, \"day\": 1546300800000, "
            "\"h3\": \"835525fffffffff\"}' ;",
            table.statements[0]
        )

        for _ in range(4):
            table = table.insert({
                'col1': 'mock_id',
                'col2': 1546304400000,
                'col3': 26.919388,
                'col4': -8.932613
            }, keyspace='mock_keyspace')
        self.assertEqual(5, len(table.statements))
        for i in range(5):
            self.assertEqual(
                "INSERT INTO mock_keyspace.mock_table "
                "JSON '{\"col1\": \"mock_id\", \"col2\": 1546304400000, "
                "\"col3\": 26.919388, \"col4\": -8.932613, "
                "\"day\": 1546300800000, \"h3\": \"835525fffffffff\"}' ;",
                table.statements[i]
            )
예제 #2
0
    def test_insert_with_split(self) -> None:
        self.mock_config['split'] = 'day'
        table = CassandraTable(self.mock_config, self.keyspace)
        table.insert({
            'col1': 'mock_id',
            'col2': 1546304400000,
            'col3': 26.919388,
            'col4': -8.932613
        }, keyspace='mock_keyspace')

        self.assertEqual(1, len(table.statements))
        self.assertEqual(
            "INSERT INTO mock_keyspace.mock_table_01_01_2019 "
            "JSON '{\"col1\": \"mock_id\", \"col2\": 1546304400000, "
            "\"col3\": 26.919388, \"col4\": -8.932613, "
            "\"day\": 1546300800000, \"h3\": \"835525fffffffff\"}' ;",
            table.statements[0]
        )