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)
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()