Esempio n. 1
0
 def test_fake_block_simple_oneline(self, write_row_mock):
     yaml = """
     - object: OBJ
       fields:
         first_name:
             fake: first_name
     """
     generate(StringIO(yaml), {}, None)
     assert row_values(write_row_mock, 0, "first_name")
Esempio n. 2
0
 def test_constants(self, write_row_mock):
     yaml = """
     - plugin: snowfakery.standard_plugins.Math
     - object: OBJ
       fields:
         pi: ${{Math.pi}}
     """
     generate(StringIO(yaml), {})
     assert row_values(write_row_mock, 0, "pi") == math.pi
Esempio n. 3
0
 def test_random_number_with_step(self, generated_rows):
     yaml = """
     - object : A
       fields:
         number: ${{random_number(min=15, max=200, step=5)}}
     """
     generate(StringIO(yaml), {}, None)
     assert 15 <= int(generated_rows.row_values(0, "number")) <= 200
     assert int(generated_rows.row_values(0, "number")) % 5 == 0
 def test_math(self, write_row_mock):
     yaml = """
     - plugin: snowfakery.standard_plugins.Math
     - object: OBJ
       fields:
         twelve: ${{Math.sqrt(144)}}
     """
     generate(StringIO(yaml), {})
     assert row_values(write_row_mock, 0, "twelve") == 12
Esempio n. 5
0
 def test_stops_if_criteria_misspelled(self, write_row):
     yaml = """
     - object: foo
       just_once: True
     - object: bar
     """
     with pytest.raises(DataGenError):
         generate(StringIO(yaml),
                  stopping_criteria=StoppingCriteria("baz", 3))
Esempio n. 6
0
    def test_gen_npsp_standard_objects(self, write_row):
        with open(npsp_standard_objects) as open_yaml_file:
            generate(open_yaml_file, {}, None)

        calls = write_row.mock_calls

        assert find_row("Account", {}, calls)
        assert find_row("Contact", {}, calls)
        assert find_row("Opportunity", {}, calls)
Esempio n. 7
0
 def test_stops_on_no_progress(self, write_row):
     yaml = """
     - object: foo
       just_once: True
     - object: bar
     """
     with pytest.raises(RuntimeError):
         generate(StringIO(yaml),
                  stopping_criteria=StoppingCriteria("foo", 3))
Esempio n. 8
0
def generate_twice(yaml):
    continuation_file = StringIO()
    generate(StringIO(yaml), generate_continuation_file=continuation_file)
    next_contination_file = StringIO()
    generate(
        StringIO(yaml),
        continuation_file=StringIO(continuation_file.getvalue()),
        generate_continuation_file=next_contination_file,
    )
Esempio n. 9
0
 def test_missing_options_from_yaml(self):
     yaml = """
     - option: total_data_imports
       default: 16
     - option: xyzzy
     """
     with pytest.raises(DataGenNameError) as e:
         generate(StringIO(yaml), {"qwerty": "EBCDIC"})
     assert "xyzzy" in str(e.value)
Esempio n. 10
0
 def test_functions_inline(self, write_row):
     yaml = """
     - object : A
       fields:
         wedding: Our wedding date is ${{date_between(start_date="2012-01-31", end_date="2012-12-31")}}
         number: The number is ${{random_number(min=15, max=19)}}
     """
     generate(StringIO(yaml), {}, None)
     assert "2012" in write_row.mock_calls[0][1][1]["wedding"]
     assert "1" in write_row.mock_calls[0][1][1]["number"]
Esempio n. 11
0
 def test_wrong_special_object(self):
     recipe_data = """
       - plugin: snowfakery.standard_plugins.Salesforce
       - object: Account
         fields:
           Foo:
             Salesforce.SpecialObject: Bar
       """
     with pytest.raises(exc.DataGenError, match="Bar"):
         generate(StringIO(recipe_data), {}, None)
Esempio n. 12
0
 def test_reference_really_wrong_type(self):
     yaml = """
     - object: B
       count: 2
       fields:
           A_ref:
             reference: 5"""
     with pytest.raises(DataGenError) as e:
         generate(StringIO(yaml))
     assert "can't get reference to object" in str(e).lower()
Esempio n. 13
0
 def test_missing_orgname(self, sf):
     yaml = """
         - plugin: snowfakery.standard_plugins.Salesforce.SalesforceQuery
         - object: Contact
           fields:
             AccountId:
                 SalesforceQuery.random_record: Account
     """
     with pytest.raises(DataGenError):
         generate(StringIO(yaml), {})
Esempio n. 14
0
 def test_i18n(self, write_row_mock):
     yaml = """
     - object: foo
       fields:
         japanese_name:
             i18n_fake:
                 locale: ja_JP
                 fake: name"""
     generate(StringIO(yaml), {})
     assert isinstance(row_values(write_row_mock, 0, "japanese_name"), str)
Esempio n. 15
0
 def test_top_field_unknown(self):
     yaml = """
     - bobject: OBJ
       fields:
         first_name:
             fake:
                 first_name
     """
     with self.assertRaises(DataGenError):
         generate(StringIO(yaml), {}, None)
Esempio n. 16
0
 def test_reference_unknown_object(self):
     yaml = """
     - object: B
       count: 2
       fields:
           A_ref:
             reference: AA"""
     with pytest.raises(DataGenError) as e:
         generate(StringIO(yaml))
     assert "cannot find" in str(e).lower()
Esempio n. 17
0
 def test_bogus_plugin(self):
     yaml = """
     - plugin: tests.test_custom_plugins_and_providers.TestCustomPlugin
     - object: OBJ
       fields:
         service_name: saascrmlightning
     """
     with pytest.raises(TypeError) as e:
         generate(StringIO(yaml), {})
     assert "TestCustomPlugin" in str(e.value)
Esempio n. 18
0
 def test_secondary_field_unknown(self):
     yaml = """
     - object: OBJ
       bfields:
         first_name:
             fake:
                 first_name
     """
     with pytest.raises(DataGenError):
         generate(StringIO(yaml), {}, None)
Esempio n. 19
0
 def test_missing_plugin(self):
     yaml = """
     - plugin: xyzzy.test_custom_plugins_and_providers.TestCustomPlugin
     - object: OBJ
       fields:
         service_name: saascrmlightning
     """
     with pytest.raises(DataGenImportError) as e:
         generate(StringIO(yaml), {})
     assert "xyzzy" in str(e.value)
Esempio n. 20
0
 def test_fake_block_one_param(self, write_row_mock):
     yaml = """
     - object: OBJ
       fields:
         country:
             fake.country_code:
                 representation: alpha-2
     """
     generate(StringIO(yaml), {})
     assert len(row_values(write_row_mock, 0, "country")) == 2
Esempio n. 21
0
 def test_fake_two_params_flat(self, write_row_mock):
     yaml = """
     - object: OBJ
       fields:
         date: ${{fake.date(pattern="%Y-%m-%d", end_datetime=None)}}
     """
     generate(StringIO(yaml), {}, None)
     date = row_values(write_row_mock, 0, "date")
     assert type(date) == str, write_row_mock.mock_calls
     assert len(date.split("-")) == 3, date
Esempio n. 22
0
 def test_faker_internals_are_invisible(self):
     yaml = """
     - object: A
       fields:
         xyzzy:
           fake: seed
     """
     with pytest.raises(exc.DataGenError) as e:
         generate(StringIO(yaml), {}, None)
     assert "seed" in str(e.value)
Esempio n. 23
0
 def test_did_you_mean(self, write_row_mock):
     yaml = """
     - object: A
       fields:
         xyzzy:
           fake: frst_name
     """
     with pytest.raises(exc.DataGenError) as e:
         generate(StringIO(yaml), {}, None)
     assert "first_name" in str(e.value)
Esempio n. 24
0
 def test_custom_faker_provider(self, write_row_mock):
     yaml = """
     - plugin: faker_microservice.Provider
     - object: OBJ
       fields:
         service_name:
             fake:
                 microservice
     """
     generate(StringIO(yaml), {})
     assert row_values(write_row_mock, 0, "service_name")
Esempio n. 25
0
 def test_weird_types(self):
     yaml = """
     - plugin: tests.test_custom_plugins_and_providers.WrongTypePlugin  # 2
     - object: B                             #3
       fields:                               #4
         foo:                                #5
             WrongTypePlugin.return_bad_type: 5  #6
     """
     with pytest.raises(DataGenError) as e:
         generate(StringIO(yaml))
     assert 6 > e.value.line_num >= 3
Esempio n. 26
0
 def test_non_overlapping_dates(self, write_row_mock):
     yaml = """
     - object: OBJ
       fields:
         date:
             date_between:
                 start_date: today
                 end_date: 2000-01-01
     """
     generate(StringIO(yaml), {}, None)
     assert row_values(write_row_mock, 0, "date") is None
Esempio n. 27
0
 def test_fake_two_params_nested(self, write_row_mock):
     yaml = """
     - object: OBJ
       fields:
         date:
             fake.date_between:
                 start_date: -10y
                 end_date: today
     """
     generate(StringIO(yaml), {}, None)
     assert row_values(write_row_mock, 0, "date").year
Esempio n. 28
0
 def test_error_handling(self, write_row_mock):
     yaml = """
     - object: A
       fields:
         xyzzy:
           fake: xyzzy
     """
     with pytest.raises(exc.DataGenError) as e:
         generate(StringIO(yaml), {}, None)
     assert "xyzzy" in str(e.value)
     assert "fake" in str(e.value)
Esempio n. 29
0
 def test_faker_kwargs(self, write_row_mock):
     yaml = """
     - object: A
       fields:
         neg:
           fake.random_int:
             min: -5
             max: -1
     """
     generate(StringIO(yaml), {}, None)
     assert -5 <= row_values(write_row_mock, 0, "neg") <= -1
Esempio n. 30
0
    def test_simple_parent_list_child(self, write_row):
        generate(StringIO(simple_parent_list), {}, None)

        a_values = find_row("A", {}, write_row.mock_calls)
        b_values = find_row("B", {}, write_row.mock_calls)
        id_a = a_values["id"]
        reference_b = a_values["B"]
        id_b = b_values["id"]
        reference_a = b_values["A_ref"]
        assert f"A({id_a})" == reference_a
        assert f"B({id_b})" == reference_b