Add unit testing of bitbucket trigger handler

buildtrigger/test/test_bitbuckethandler.py

@@ -1,24 +1,22 @@
 import pytest
 
-from mock import Mock
-from datetime import datetime
-
-from buildtrigger.bitbuckethandler import BitbucketBuildTrigger
-from buildtrigger.triggerutil import (InvalidPayloadException, SkipRequestException,
-                                      TriggerStartException, ValidationRequestException)
-from endpoints.building import PreparedBuild
-from util.morecollections import AttrDict
+from buildtrigger.test.bitbucketmock import get_bitbucket_trigger
 
 @pytest.fixture
 def bitbucket_trigger():
-  return _get_bitbucket_trigger()
+  return get_bitbucket_trigger()
 
-def get_mock_bitbucket():
-  client_mock = Mock()
-  return client_mock
 
-def _get_bitbucket_trigger(subdir=''):
-  trigger_obj = AttrDict(dict(auth_token='foobar', id='sometrigger'))
-  trigger = BitbucketBuildTrigger(trigger_obj, {'build_source': 'foo/bar', 'subdir': subdir})
-  trigger._get_client = get_mock_bitbucket
-  return trigger
+def test_list_build_subdirs(bitbucket_trigger):
+  assert bitbucket_trigger.list_build_subdirs() == ['']
+
+
+@pytest.mark.parametrize('subdir, contents', [
+  ('', 'hello world'),
+  ('somesubdir', 'hi universe'),
+  ('unknownpath', None),
+])
+def test_load_dockerfile_contents(subdir, contents):
+  trigger = get_bitbucket_trigger(subdir)
+  assert trigger.load_dockerfile_contents() == contents
+