mirror of
https://github.com/facebook/sapling.git
synced 2024-10-10 00:45:18 +03:00
0e6b93a0e0
Summary: - Add support for RocksDB engine (developed as a drop in replacement for innodb) to hgsql to allow new xdb.hgsql.1-10 shards to host hg repos - Prefer MySQL test DBs in same region - Run all hgsql unit tests also for RocksDB engine - Allow for nested ifs to make that possible (downside if you switch off rockdb tests, innodb tests are run twice) Reviewed By: quark-zju Differential Revision: D7014064 fbshipit-source-id: 073c36176aa7eaf74252ef33c3f47da594920b28
83 lines
2.5 KiB
Raku
83 lines
2.5 KiB
Raku
#testcases case-innodb case-rocksdb
|
|
|
|
#if case-rocksdb
|
|
$ DBENGINE=rocksdb
|
|
#else
|
|
$ DBENGINE=innodb
|
|
#endif
|
|
|
|
$ . "$TESTDIR/hgsql/library.sh"
|
|
|
|
# Populate the db with an initial commit
|
|
|
|
$ initclient client
|
|
$ cd client
|
|
$ echo x > x
|
|
$ hg commit -qAm x
|
|
$ cd ..
|
|
|
|
$ initserver master masterrepo
|
|
|
|
$ cat >> $TESTTMP/uilog.py <<EOF
|
|
> from mercurial import extensions
|
|
> from mercurial import ui as uimod
|
|
> def uisetup(ui):
|
|
> extensions.wrapfunction(uimod.ui, 'log', mylog)
|
|
> def mylog(orig, self, service, *msg, **opts):
|
|
> if service in ['sqllock']:
|
|
> kwstr = ", ".join("%s=%s" % (k, v) for k, v in
|
|
> sorted(opts.iteritems()))
|
|
> msgstr = msg[0] % msg[1:]
|
|
> self.warn('%s: %s (%s)\n' % (service, msgstr, kwstr))
|
|
> return orig(self, service, *msg, **opts)
|
|
> EOF
|
|
$ cat >> master/.hg/hgrc <<EOF
|
|
> [extensions]
|
|
> uilog=$TESTTMP/uilog.py
|
|
> EOF
|
|
|
|
# Verify timeouts are logged
|
|
$ cat >> $TESTTMP/forcetimeout.py <<EOF
|
|
> from mercurial import error, extensions
|
|
> def uisetup(ui):
|
|
> hgsql = extensions.find('hgsql')
|
|
> extensions.wrapfunction(hgsql.sqlcontext, '__enter__', fakeenter)
|
|
> def fakeenter(orig, self):
|
|
> if self.takelock:
|
|
> extensions.wrapfunction(self.repo.__class__, 'sqllock', lockthrow)
|
|
> return orig(self)
|
|
> def lockthrow(*args, **kwargs):
|
|
> raise error.Abort("fake timeout")
|
|
> EOF
|
|
|
|
$ cp master/.hg/hgrc $TESTTMP/orighgrc
|
|
$ cat >> master/.hg/hgrc <<EOF
|
|
> [extensions]
|
|
> forcetimeout=$TESTTMP/forcetimeout.py
|
|
> EOF
|
|
$ cd client
|
|
$ hg push ssh://user@dummy/master
|
|
pushing to ssh://user@dummy/master
|
|
searching for changes
|
|
remote: sqllock: failed to get sql lock after * seconds (glob)
|
|
remote: (elapsed=*, repository=$TESTTMP/master, success=false, valuetype=lockwait) (glob)
|
|
remote: abort: fake timeout
|
|
abort: not a Mercurial bundle
|
|
[255]
|
|
$ cd ..
|
|
$ cp $TESTTMP/orighgrc master/.hg/hgrc
|
|
|
|
# Verify sqllock times are logged
|
|
$ cd client
|
|
$ hg push ssh://user@dummy/master
|
|
pushing to ssh://user@dummy/master
|
|
searching for changes
|
|
remote: sqllock: waited for sql lock for * seconds (glob)
|
|
remote: (elapsed=*, repository=$TESTTMP/master, success=true, valuetype=lockwait) (glob)
|
|
remote: adding changesets
|
|
remote: adding manifests
|
|
remote: adding file changes
|
|
remote: added 1 changesets with 1 changes to 1 files
|
|
remote: sqllock: held sql lock for * seconds (glob)
|
|
remote: (elapsed=*, repository=$TESTTMP/master, valuetype=lockheld) (glob)
|