import pytest
from unittest.mock import MagicMock, patch
from infra_cli.samba import SambaManager

class MockConfig:
    def get_node(self, name):
        return {"host": "10.32.2.1", "pass": "secret"}
    def get(self, key, default=None):
        mapping = {
            'samba.host': '10.32.1.101',
            'samba.user': 'Administrator',
            'samba.domain': 'FE.LOOPAWARE.COM'
        }
        return mapping.get(key, default)

@patch('infra_cli.samba.SSHClient')
def test_samba_add_user_logic(mock_ssh):
    mock_instance = mock_ssh.return_value
    mock_instance.run.return_value.returncode = 0
    
    mgr = SambaManager(MockConfig())
    mgr.add_user("testuser", "testpass")
    
    assert any("samba-tool user create testuser testpass" in str(call) 
               for call in mock_instance.run.call_args_list)

@patch('infra_cli.samba.SSHClient')
def test_samba_add_to_group_logic(mock_ssh):
    mock_instance = mock_ssh.return_value
    mock_instance.run.return_value.returncode = 0
    
    mgr = SambaManager(MockConfig())
    mgr.add_to_group("testgroup", "testuser")
    
    assert any("samba-tool group addmembers testgroup testuser" in str(call) 
               for call in mock_instance.run.call_args_list)