1 files changed, 170 insertions, 0 deletions
diff --git a/vendor/github.com/goamz/goamz/exp/mturk/mturk_test.go b/vendor/github.com/goamz/goamz/exp/mturk/mturk_test.go
new file mode 100644
index 000000000..0c546571f
--- /dev/null
+++ b/vendor/github.com/goamz/goamz/exp/mturk/mturk_test.go
@@ -0,0 +1,170 @@
+package mturk_test
+
+import (
+	"net/url"
+	"testing"
+
+	"github.com/goamz/goamz/aws"
+	"github.com/goamz/goamz/exp/mturk"
+	"github.com/goamz/goamz/testutil"
+	. "gopkg.in/check.v1"
+)
+
+func Test(t *testing.T) {
+	TestingT(t)
+}
+
+var _ = Suite(&S{})
+
+type S struct {
+	mturk *mturk.MTurk
+}
+
+var testServer = testutil.NewHTTPServer()
+
+func (s *S) SetUpSuite(c *C) {
+	testServer.Start()
+	auth := aws.Auth{AccessKey: "abc", SecretKey: "123"}
+	u, err := url.Parse(testServer.URL)
+	if err != nil {
+		panic(err.Error())
+	}
+
+	s.mturk = &mturk.MTurk{
+		Auth: auth,
+		URL:  u,
+	}
+}
+
+func (s *S) TearDownTest(c *C) {
+	testServer.Flush()
+}
+
+func (s *S) TestCreateHITExternalQuestion(c *C) {
+	testServer.Response(200, nil, BasicHitResponse)
+
+	question := mturk.ExternalQuestion{
+		ExternalURL: "http://www.amazon.com",
+		FrameHeight: 200,
+	}
+	reward := mturk.Price{
+		Amount:       "0.01",
+		CurrencyCode: "USD",
+	}
+	hit, err := s.mturk.CreateHIT("title", "description", question, reward, 1, 2, "key1,key2", 3, nil, "annotation")
+
+	testServer.WaitRequest()
+
+	c.Assert(err, IsNil)
+	c.Assert(hit, NotNil)
+
+	c.Assert(hit.HITId, Equals, "28J4IXKO2L927XKJTHO34OCDNASCDW")
+	c.Assert(hit.HITTypeId, Equals, "2XZ7D1X3V0FKQVW7LU51S7PKKGFKDF")
+}
+
+func (s *S) TestCreateHITHTMLQuestion(c *C) {
+	testServer.Response(200, nil, BasicHitResponse)
+
+	question := mturk.HTMLQuestion{
+		HTMLContent: mturk.HTMLContent{`<![CDATA[
+<!DOCTYPE html>
+<html>
+ <head>
+  <meta http-equiv='Content-Type' content='text/html; charset=UTF-8'/>
+  <script type='text/javascript' src='https://s3.amazonaws.com/mturk-public/externalHIT_v1.js'></script>
+ </head>
+ <body>
+  <form name='mturk_form' method='post' id='mturk_form' action='https://www.mturk.com/mturk/externalSubmit'>
+  <input type='hidden' value='' name='assignmentId' id='assignmentId'/>
+  <h1>What's up?</h1>
+  <p><textarea name='comment' cols='80' rows='3'></textarea></p>
+  <p><input type='submit' id='submitButton' value='Submit' /></p></form>
+  <script language='Javascript'>turkSetAssignmentID();</script>
+ </body>
+</html>
+]]>`},
+		FrameHeight: 200,
+	}
+	reward := mturk.Price{
+		Amount:       "0.01",
+		CurrencyCode: "USD",
+	}
+	hit, err := s.mturk.CreateHIT("title", "description", question, reward, 1, 2, "key1,key2", 3, nil, "annotation")
+
+	testServer.WaitRequest()
+
+	c.Assert(err, IsNil)
+	c.Assert(hit, NotNil)
+
+	c.Assert(hit.HITId, Equals, "28J4IXKO2L927XKJTHO34OCDNASCDW")
+	c.Assert(hit.HITTypeId, Equals, "2XZ7D1X3V0FKQVW7LU51S7PKKGFKDF")
+}
+
+func (s *S) TestSearchHITs(c *C) {
+	testServer.Response(200, nil, SearchHITResponse)
+
+	hitResult, err := s.mturk.SearchHITs()
+
+	c.Assert(err, IsNil)
+	c.Assert(hitResult, NotNil)
+
+	c.Assert(hitResult.NumResults, Equals, uint(1))
+	c.Assert(hitResult.PageNumber, Equals, uint(1))
+	c.Assert(hitResult.TotalNumResults, Equals, uint(1))
+
+	c.Assert(len(hitResult.HITs), Equals, 1)
+	c.Assert(hitResult.HITs[0].HITId, Equals, "2BU26DG67D1XTE823B3OQ2JF2XWF83")
+	c.Assert(hitResult.HITs[0].HITTypeId, Equals, "22OWJ5OPB0YV6IGL5727KP9U38P5XR")
+	c.Assert(hitResult.HITs[0].CreationTime, Equals, "2011-12-28T19:56:20Z")
+	c.Assert(hitResult.HITs[0].Title, Equals, "test hit")
+	c.Assert(hitResult.HITs[0].Description, Equals, "please disregard, testing only")
+	c.Assert(hitResult.HITs[0].HITStatus, Equals, "Reviewable")
+	c.Assert(hitResult.HITs[0].MaxAssignments, Equals, uint(1))
+	c.Assert(hitResult.HITs[0].Reward.Amount, Equals, "0.01")
+	c.Assert(hitResult.HITs[0].Reward.CurrencyCode, Equals, "USD")
+	c.Assert(hitResult.HITs[0].AutoApprovalDelayInSeconds, Equals, uint(2592000))
+	c.Assert(hitResult.HITs[0].AssignmentDurationInSeconds, Equals, uint(30))
+	c.Assert(hitResult.HITs[0].NumberOfAssignmentsPending, Equals, uint(0))
+	c.Assert(hitResult.HITs[0].NumberOfAssignmentsAvailable, Equals, uint(1))
+	c.Assert(hitResult.HITs[0].NumberOfAssignmentsCompleted, Equals, uint(0))
+}
+
+func (s *S) TestGetAssignmentsForHIT_NoAnswer(c *C) {
+	testServer.Response(200, nil, GetAssignmentsForHITNoAnswerResponse)
+
+	assignment, err := s.mturk.GetAssignmentsForHIT("emptyassignment")
+
+	testServer.WaitRequest()
+
+	c.Assert(err, IsNil)
+	c.Assert(assignment, NotNil)
+
+	c.Assert(assignment.HITId, Equals, "")
+}
+
+func (s *S) TestGetAssignmentsForHIT_Answer(c *C) {
+	testServer.Response(200, nil, GetAssignmentsForHITAnswerResponse)
+
+	assignment, err := s.mturk.GetAssignmentsForHIT("emptyassignment")
+
+	testServer.WaitRequest()
+
+	c.Assert(err, IsNil)
+	c.Assert(assignment, NotNil)
+
+	c.Assert(assignment.AssignmentId, Equals, "2QKNTL0XULRGFAQWUWDD05FP94V2O3")
+	c.Assert(assignment.WorkerId, Equals, "A1ZUQ2YDM61713")
+	c.Assert(assignment.HITId, Equals, "2W36VCPWZ9RN5DX1MBJ7VN3D6WEPAM")
+	c.Assert(assignment.AssignmentStatus, Equals, "Submitted")
+	c.Assert(assignment.AutoApprovalTime, Equals, "2014-02-26T09:39:48Z")
+	c.Assert(assignment.AcceptTime, Equals, "2014-01-27T09:39:38Z")
+	c.Assert(assignment.SubmitTime, Equals, "2014-01-27T09:39:48Z")
+	c.Assert(assignment.ApprovalTime, Equals, "")
+
+	answers := assignment.Answers()
+	c.Assert(len(answers), Equals, 4)
+	c.Assert(answers["tags"], Equals, "asd")
+	c.Assert(answers["text_in_image"], Equals, "asd")
+	c.Assert(answers["is_pattern"], Equals, "yes")
+	c.Assert(answers["is_map"], Equals, "yes")
+}