s3:smbd/open: use Builtin_Administrators as owner of files (if possible)
[kai/samba.git] / source3 / stf / stf.py
1 #!/usr/bin/python
2 #
3 # Samba Testing Framework for Unit-testing
4 #
5
6 import os, string, re
7 import osver
8
9 def get_server_list_from_string(s):
10
11     server_list = []
12     
13     # Format is a list of server:domain\username%password separated
14     # by commas.
15
16     for entry in string.split(s, ","):
17
18         # Parse entry 
19
20         m = re.match("(.*):(.*)(\\\\|/)(.*)%(.*)", entry)
21         if not m:
22             raise "badly formed server list entry '%s'" % entry
23
24         server = m.group(1)
25         domain = m.group(2)
26         username = m.group(4)
27         password = m.group(5)
28
29         # Categorise servers
30
31         server_list.append({"platform": osver.os_version(server),
32                             "hostname": server,
33                             "administrator": {"username": username,
34                                               "domain": domain,
35                                               "password" : password}})
36
37     return server_list
38
39 def get_server_list():
40     """Iterate through all sources of server info and append them all
41     in one big list."""
42     
43     server_list = []
44
45     # The $STF_SERVERS environment variable
46
47     if os.environ.has_key("STF_SERVERS"):
48         server_list = server_list + \
49                       get_server_list_from_string(os.environ["STF_SERVERS"])
50
51     return server_list
52
53 def get_server(platform = None):
54     """Return configuration information for a server.  The platform
55     argument can be a string either 'nt4' or 'nt5' for Windows NT or
56     Windows 2000 servers, or just 'nt' for Windows NT and higher."""
57     
58     server_list = get_server_list()
59
60     for server in server_list:
61         if platform:
62             p = server["platform"]
63             if platform == "nt":
64                 if (p == osver.PLATFORM_NT4 or p == osver.PLATFORM_NT5):
65                     return server
66             if platform == "nt4" and p == osver.PLATFORM_NT4:
67                 return server
68             if platform == "nt5" and p == osver.PLATFORM_NT5:
69                 return server
70         else:
71             # No filter defined, return first in list
72             return server
73         
74     return None
75
76 def dict_check(sample_dict, real_dict):
77     """Check that real_dict contains all the keys present in sample_dict
78     and no extras.  Also check that common keys are of them same type."""
79     tmp = real_dict.copy()
80     for key in sample_dict.keys():
81         # Check existing key and type
82         if not real_dict.has_key(key):
83             raise ValueError, "dict does not contain key '%s'" % key
84         if type(sample_dict[key]) != type(real_dict[key]):
85             raise ValueError, "dict has differing types (%s vs %s) for key " \
86                   "'%s'" % (type(sample_dict[key]), type(real_dict[key]), key)
87         # Check dictionaries recursively
88         if type(sample_dict[key]) == dict:
89             dict_check(sample_dict[key], real_dict[key])
90         # Delete visited keys from copy
91         del(tmp[key])
92     # Any keys leftover are present in the real dict but not the sample
93     if len(tmp) == 0:
94         return
95     result = "dict has extra keys: "
96     for key in tmp.keys():
97         result = result + key + " "
98     raise ValueError, result
99
100 if __name__ == "__main__":
101     print get_server(platform = "nt")