uk_bin_collection 0.85.6__tar.gz → 0.85.7__tar.gz

{uk_bin_collection-0.85.6 → uk_bin_collection-0.85.7}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: uk_bin_collection
-Version: 0.85.6
+Version: 0.85.7
 Summary: Python Lib to collect UK Bin Data
 Author: Robert Bradley
 Author-email: robbrad182@gmail.com

{uk_bin_collection-0.85.6 → uk_bin_collection-0.85.7}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "uk_bin_collection"
-version = "0.85.6"
+version = "0.85.7"
 description = "Python Lib to collect UK Bin Data"
 readme = "README.md"
 authors = ["Robert Bradley <robbrad182@gmail.com>"]

uk_bin_collection-0.85.7/uk_bin_collection/tests/test_collect_data.py

@@ -0,0 +1,73 @@
+from unittest.mock import MagicMock, patch
+import argparse
+import pytest
+from uk_bin_collection.collect_data import UKBinCollectionApp, import_council_module
+
+
+
+# Test UKBinCollectionApp setup_arg_parser
+def test_setup_arg_parser():
+    app = UKBinCollectionApp()
+    app.setup_arg_parser()
+
+    # Assert that the argument parser has the correct arguments
+    assert isinstance(app.parser, argparse.ArgumentParser)
+    args = app.parser._actions
+    arg_names = [action.dest for action in args]
+
+    expected_args = [
+        "module",
+        "URL",
+        "postcode",
+        "number",
+        "skip_get_url",
+        "uprn",
+        "web_driver",
+        "headless",
+        "local_browser",
+        "dev_mode",
+    ]
+    assert all(arg in arg_names for arg in expected_args)
+
+
+# Test UKBinCollectionApp set_args
+def test_set_args():
+    app = UKBinCollectionApp()
+    app.setup_arg_parser()
+
+    # Test valid args
+    args = ["council_module", "http://example.com", "--postcode", "AB1 2CD"]
+    app.set_args(args)
+
+    assert app.parsed_args.module == "council_module"
+    assert app.parsed_args.URL == "http://example.com"
+    assert app.parsed_args.postcode == "AB1 2CD"
+
+
+# Test UKBinCollectionApp client_code method
+def test_client_code():
+    app = UKBinCollectionApp()
+    mock_get_bin_data_class = MagicMock()
+
+    # Run the client_code and ensure that template_method is called
+    app.client_code(mock_get_bin_data_class, "http://example.com", postcode="AB1 2CD")
+    mock_get_bin_data_class.template_method.assert_called_once_with(
+        "http://example.com", postcode="AB1 2CD"
+    )
+
+
+# Test the run() function with logging setup
+@patch("uk_bin_collection.collect_data.setup_logging")  # Correct patch path
+@patch("uk_bin_collection.collect_data.UKBinCollectionApp.run")  # Correct patch path
+@patch("sys.argv", ["uk_bin_collection.py", "council_module", "http://example.com"])
+def test_run_function(mock_app_run, mock_setup_logging):
+    from uk_bin_collection.collect_data import run
+
+    mock_setup_logging.return_value = MagicMock()
+    mock_app_run.return_value = None
+
+    run()
+
+    # Ensure logging was set up and the app run method was called
+    mock_setup_logging.assert_called_once()
+    mock_app_run.assert_called_once()

uk_bin_collection-0.85.6/uk_bin_collection/tests/test_collect_data.py → uk_bin_collection-0.85.7/uk_bin_collection/tests/test_get_data.py

@@ -1,9 +1,11 @@
-import json
+import os
 from unittest import mock
+import tempfile
 
 import pytest
 from requests import exceptions as req_exp
 from requests.models import Response
+from unittest.mock import patch
 from uk_bin_collection.get_bin_data import AbstractGetBinDataClass as agbdc
 from uk_bin_collection.get_bin_data import setup_logging
 import logging
@@ -108,6 +110,10 @@ class ConcreteGetBinDataClass(agbdc):
     def parse_data(self, page: str, **kwargs) -> dict:
         return {"mock_key": "mock_value"}
 
+    def update_dev_mode_data(self, council_module_str, this_url, **kwargs):
+        # You can implement the method or delegate it to the abstract class's method
+        super().update_dev_mode_data(council_module_str, this_url, **kwargs)
+
 @pytest.fixture
 def concrete_class_instance():
     return ConcreteGetBinDataClass()
@@ -133,4 +139,59 @@ def test_get_and_parse_data_skip_get_url(concrete_class_instance):
         result = concrete_class_instance.get_and_parse_data("http://example.com", skip_get_url=True)
 
         mock_parse_data.assert_called_once_with("", url="http://example.com", skip_get_url=True)
-        assert result == mock_parsed_data
+        assert result == mock_parsed_data
+        
+@pytest.fixture
+def setup_test_update_dev_mode_data():
+    """Fixture to set up and tear down the environment for test_update_dev_mode_data"""
+    # Create a temporary directory to simulate the working directory
+    test_dir = tempfile.TemporaryDirectory()
+
+    # Patch os.getcwd() to return the temporary directory
+    cwd_patch = patch('os.getcwd', return_value=test_dir.name)
+    mock_getcwd = cwd_patch.start()
+
+    # Ensure the nested directory structure exists
+    os.makedirs(os.path.join(test_dir.name, "uk_bin_collection", "tests"))
+
+    # Yield control back to the test, then clean up after the test
+    yield test_dir.name  # Provide the test with the temporary directory
+
+    # Teardown
+    test_dir.cleanup()
+    cwd_patch.stop()
+
+
+def test_update_dev_mode_data(setup_test_update_dev_mode_data):
+    """Test update_dev_mode_data method to ensure input.json is updated correctly"""
+    # The setup fixture returns the mocked current working directory
+    mock_cwd = setup_test_update_dev_mode_data
+
+    # Create an instance of the concrete class that inherits from AbstractGetBinDataClass
+    obj = ConcreteGetBinDataClass()
+
+    # Define input arguments for the method
+    council_module_str = 'test_council_module'
+    this_url = 'https://example.com'
+    kwargs = {
+        'postcode': '12345',
+        'paon': '1A',
+        'uprn': '100012345',
+        'usrn': '200012345',
+        'web_driver': 'mocked_web_driver',
+        'skip_get_url': True,
+    }
+
+    # Call the method being tested on the instance
+    obj.update_dev_mode_data(council_module_str, this_url, **kwargs)
+
+    # Verify that input.json was created in the correct location
+    input_file_path = os.path.join(mock_cwd, "uk_bin_collection", "tests", "input.json")
+    assert os.path.exists(input_file_path)
+
+    # Read the contents of the file and make necessary assertions
+    with open(input_file_path, 'r') as f:
+        file_content = f.read()
+
+    # Example assertion - check if certain values exist in the file content (based on your actual file format)
+    assert '100012345' in file_content  # Checking UPRN as an example