mirror of
https://github.com/apache/superset.git
synced 2024-09-19 03:59:49 -04:00
4675ca31c5
* refactor: replace the way the birth_names data is loaded to DB * fix failed unit test * fix failed unit test * fix failed tests * fix pass wrong flag of support datetime type * remove unused fixture
36 lines
1.5 KiB
Python
36 lines
1.5 KiB
Python
# Licensed to the Apache Software Foundation (ASF) under one
|
|
# or more contributor license agreements. See the NOTICE file
|
|
# distributed with this work for additional information
|
|
# regarding copyright ownership. The ASF licenses this file
|
|
# to you under the Apache License, Version 2.0 (the
|
|
# "License"); you may not use this file except in compliance
|
|
# with the License. You may obtain a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing,
|
|
# software distributed under the License is distributed on an
|
|
# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
|
|
# KIND, either express or implied. See the License for the
|
|
# specific language governing permissions and limitations
|
|
# under the License.
|
|
from unittest.mock import Mock, patch
|
|
|
|
from tests.example_data.data_generator.string_generator import StringGenerator
|
|
|
|
|
|
@patch("tests.example_data.data_generator.string_generator.choices")
|
|
@patch("tests.example_data.data_generator.string_generator.randint")
|
|
def test_string_generator(randint_mock: Mock, choices_mock: Mock):
|
|
letters = "abcdets"
|
|
min_len = 3
|
|
max_len = 5
|
|
randomized_string_len = 4
|
|
string_generator = StringGenerator(letters, min_len, max_len)
|
|
randint_mock.return_value = randomized_string_len
|
|
choices_mock.return_value = ["t", "e", "s", "t"]
|
|
|
|
assert string_generator.generate() == "test"
|
|
randint_mock.assert_called_once_with(min_len, max_len)
|
|
choices_mock.assert_called_with(letters, k=randomized_string_len)
|