def test_counts(self, write_row): generate_cli.callback( yaml_file=sample_yaml, target_number=("Account", 2), option={}, debug_internals=None, generate_cci_mapping_file=None, ) assert write_row.mock_calls == [ mock.call( "Account", { "id": 1, "name": "Default Company Name", "ShippingCountry": "Canada" }, ), mock.call( "Account", { "id": 2, "name": "Default Company Name", "ShippingCountry": "Canada" }, ), ]
def test_with_debug_flags_on(self, write_row): generate_cli.callback( yaml_file=sample_yaml, option={}, debug_internals=True, mapping_file=None, )
def test_from_cli(self): x = StringIO() with redirect_stdout(x): generate_cli.callback(yaml_file=sample_yaml, output_format="json") data = json.loads(x.getvalue()) print(data) assert data == sample_output
def test_with_option(self, write_row): with pytest.warns(UserWarning): generate_cli.callback( yaml_file=sample_yaml, option={"xyzzy": "abcd"}, debug_internals=None, generate_cci_mapping_file=None, )
def test_external_output_stream(self): x = StringIO() with redirect_stdout(x): generate_cli.callback(yaml_file=sample_yaml, output_format="package1.TestOutputStream") assert (x.getvalue() == """A - {'id': 1, 'B': 'B(1)'} B - {'id': 1, 'A': 'A(1)'} """)
def test_with_bad_dburl(self, write_row): with pytest.raises(Exception): generate_cli.callback( yaml_file=sample_yaml, option={}, dburl="xyzzy:////foo/bar/baz.com", debug_internals=None, generate_cci_mapping_file=None, )
def test_from_cli__unknown_extension(self, capsys): with pytest.raises(ClickException) as e: generate_cli.callback( yaml_file=str(sample_yaml), target_number=("Account", 5), output_format="xyzzy", output_files=["foo.txt"], ) assert "xyzzy" in str(e.value)
def test_exception_with_debug_flags_off(self, write_row): with named_temporary_file_path(suffix=".yml") as t: with pytest.raises(ClickException): generate_cli.callback( yaml_file=bad_sample_yaml, option={}, debug_internals=False, generate_cci_mapping_file=t, ) assert yaml.safe_load(t)
def test_from_cli(self): x = StringIO() with redirect_stdout(x): generate_cli.callback(yaml_file=sample_yaml, output_format="json") data = json.loads(x.getvalue()) assert data == [{ "_table": "Account", "id": 1, "name": "Default Company Name", "ShippingCountry": "Canada", }]
def test_external_output_stream_yaml(self): x = StringIO() with redirect_stdout(x): generate_cli.callback(yaml_file=sample_yaml, output_format="examples.YamlOutputStream") expected = """- object: A nickname: row_1 fields: id: 1 B: reference: row_1 - object: B nickname: row_1 fields: id: 1 A: reference: row_1 """ print(x.getvalue()) assert x.getvalue() == expected
def test_external_output_stream__failure(self): with pytest.raises(ClickException, match="no.such.output.Stream"): generate_cli.callback( yaml_file=sample_yaml, output_format="no.such.output.Stream" )
def test_from_cli(self): x = StringIO() with redirect_stdout(x): generate_cli.callback( yaml_file=sample_yaml, output_format="json", output_files=["-"] )