Exemple #1
0
    def test_get_bucket_creates_bucket_connection(self, Config):
        config = Mock()
        Config.return_value = config
        mock_bucket = Mock()
        truck = ImageTruck(None, None, None)
        truck._stored_bucket = mock_bucket

        bucket = truck._bucket()
        eq_(bucket, mock_bucket)
        eq_(config.table.call_count, 0)
Exemple #2
0
    def test_get_bucket_creates_bucket_connection(self, Config):
        config = Mock()
        Config.return_value = config
        mock_bucket = Mock()
        config.bucket.return_value = mock_bucket

        truck = ImageTruck(None, None, None)
        bucket = truck._bucket()
        eq_(bucket, mock_bucket)
        eq_(truck._stored_bucket, mock_bucket)
        config.bucket.assert_called_with()