lib/bup/t/tclient.py

   1 import sys, os, stat, time, random, subprocess
   2 from bup import client, git
   3 from wvtest import *
   4
   5 def randbytes(sz):
   6     s = ''
   7     for i in xrange(sz):
   8         s += chr(random.randrange(0,256))
   9     return s
  10
  11 s1 = randbytes(10000)
  12 s2 = randbytes(10000)
  13 s3 = randbytes(10000)
  14
  15 @wvtest
  16 def test_server_split_with_indexes():
  17     os.environ['BUP_MAIN_EXE'] = '../../../bup'
  18     os.environ['BUP_DIR'] = bupdir = 'buptest_tclient.tmp'
  19     subprocess.call(['rm', '-rf', bupdir])
  20     git.init_repo(bupdir)
  21     lw = git.PackWriter()
  22     c = client.Client(bupdir, create=True)
  23     rw = c.new_packwriter()
  24
  25     lw.new_blob(s1)
  26     lw.close()
  27
  28     rw.new_blob(s2)
  29     rw.breakpoint()
  30     rw.new_blob(s1)
  31
  32
  33 @wvtest
  34 def test_multiple_suggestions():
  35     os.environ['BUP_MAIN_EXE'] = '../../../bup'
  36     os.environ['BUP_DIR'] = bupdir = 'buptest_tclient.tmp'
  37     subprocess.call(['rm', '-rf', bupdir])
  38     git.init_repo(bupdir)
  39
  40     lw = git.PackWriter()
  41     lw.new_blob(s1)
  42     lw.close()
  43     lw = git.PackWriter()
  44     lw.new_blob(s2)
  45     lw.close()
  46     WVPASSEQ(len(os.listdir(git.repo('objects/pack'))), 4)
  47
  48     c = client.Client(bupdir, create=True)
  49     WVPASSEQ(len(os.listdir(c.cachedir)), 0)
  50     rw = c.new_packwriter()
  51     rw.new_blob(s1)
  52     rw.new_blob(s2)
  53     rw.new_blob(s3)
  54     WVPASSEQ(len(os.listdir(c.cachedir)), 2)
  55     rw.close()
  56     WVPASSEQ(len(os.listdir(c.cachedir)), 3)
  57
  58
  59 @wvtest
  60 def test_dumb_client_server():
  61     os.environ['BUP_MAIN_EXE'] = '../../../bup'
  62     os.environ['BUP_DIR'] = bupdir = 'buptest_tclient.tmp'
  63     subprocess.call(['rm', '-rf', bupdir])
  64     git.init_repo(bupdir)
  65     open(git.repo('bup-dumb-server'), 'w').close()
  66
  67     lw = git.PackWriter()
  68     lw.new_blob(s1)
  69     lw.close()
  70
  71     c = client.Client(bupdir, create=True)
  72     rw = c.new_packwriter()
  73     WVPASSEQ(len(os.listdir(c.cachedir)), 1)
  74     rw.new_blob(s1)
  75     WVPASSEQ(len(os.listdir(c.cachedir)), 1)
  76     rw.new_blob(s2)
  77     rw.close()
  78     WVPASSEQ(len(os.listdir(c.cachedir)), 2)
  79
  80
  81 @wvtest
  82 def test_midx_refreshing():
  83     os.environ['BUP_MAIN_EXE'] = bupmain = '../../../bup'
  84     os.environ['BUP_DIR'] = bupdir = 'buptest_tmidx.tmp'
  85     subprocess.call(['rm', '-rf', bupdir])
  86     git.init_repo(bupdir)
  87     lw = git.PackWriter()
  88     lw.new_blob(s1)
  89     lw.breakpoint()
  90     lw.new_blob(s2)
  91     del lw
  92     pi = git.PackIdxList(bupdir + '/objects/pack')
  93     WVPASSEQ(len(pi.packs), 2)
  94     pi.refresh()
  95     WVPASSEQ(len(pi.packs), 2)
  96     subprocess.call([bupmain, 'midx', '-f'])
  97     pi.refresh()
  98     WVPASSEQ(len(pi.packs), 1)
  99     pi.refresh(skip_midx=True)
 100     WVPASSEQ(len(pi.packs), 2)
 101     pi.refresh(skip_midx=False)
 102     WVPASSEQ(len(pi.packs), 1)
 103
 104 @wvtest
 105 def test_remote_parsing():
 106     tests = (
 107         (':/bup', ('file', None, None, '/bup')),
 108         ('file:///bup', ('file', None, None, '/bup')),
 109         ('192.168.1.1:/bup', ('ssh', '192.168.1.1', None, '/bup')),
 110         ('ssh://192.168.1.1:2222/bup', ('ssh', '192.168.1.1', '2222', '/bup')),
 111         ('ssh://[ff:fe::1]:2222/bup', ('ssh', 'ff:fe::1', '2222', '/bup')),
 112         ('bup://foo.com:1950', ('bup', 'foo.com', '1950', None)),
 113         ('bup://foo.com:1950/bup', ('bup', 'foo.com', '1950', '/bup')),
 114         ('bup://[ff:fe::1]/bup', ('bup', 'ff:fe::1', None, '/bup')),
 115     )
 116     for remote, values in tests:
 117         WVPASSEQ(client.parse_remote(remote), values)
 118     try:
 119         client.parse_remote('http://asdf.com/bup')
 120         WVFAIL()
 121     except AssertionError:
 122         WVPASS()